iframe-proxy | Sunbelt Computer Software

97 lines (85 loc) · 2.14 KB

package httpserver
	"net/http"
	"path/filepath"
	preTag      = "<pre>"
	preTagClose = "</pre>"
	aTag        = "<a"
	htmlHeader  = `<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01//EN" "http://www.w3.org/TR/html4/strict.dtd">
<meta http-equiv="Content-Type" content="text/html; charset=utf-8">
<meta name="viewport" content="width=device-width, initial-scale=1">
<title>Directory listing for %s</title>
	htmlFooter = `<hr>
type pythonStyleHandler struct {
	origWriter http.ResponseWriter
	root       http.Dir
func (h *pythonStyleHandler) Header() http.Header {
	return h.origWriter.Header()
func (h *pythonStyleHandler) writeListItem(b []byte, written *int) {
	i, _ = fmt.Fprint(h.origWriter, "<li>")
	*written += i
	i, _ = h.origWriter.Write(bytes.Trim(b, "\r\n"))
	*written += i
	i, _ = fmt.Fprint(h.origWriter, "</li>\n")
	*written += i
func (h *pythonStyleHandler) Write(b []byte) (int, error) {
	written := 0
	if bytes.HasPrefix(b, []byte(preTag)) {
		_, _ = io.Discard.Write(b)
		i, _ = fmt.Fprintln(h.origWriter, "<ul>")
		written += i
		return written, nil
	if bytes.HasPrefix(b, []byte(preTagClose)) {
		_, _ = io.Discard.Write(b)
		i, _ = fmt.Fprintln(h.origWriter, "</ul>")
		written += i
		return written, nil
	if bytes.HasPrefix(b, []byte(aTag)) {
		h.writeListItem(b, &written)
	return i, nil
func (h *pythonStyleHandler) WriteHeader(statusCode int) {
	h.origWriter.WriteHeader(statusCode)
func (h *pythonStyleHandler) ServeHTTP(writer http.ResponseWriter, request *http.Request) {
	target := filepath.Join(string(h.root), filepath.Clean(request.URL.Path))
	file, err := os.Stat(target)
	if err != nil || !file.IsDir() {
		http.ServeFile(writer, request, target)
		_, _ = fmt.Fprintf(writer, htmlHeader, request.URL.Path)
		_, _ = fmt.Fprintf(writer, "<h1>Directory listing for %s</h1>\n<hr>\n", request.URL.Path)
		h.origWriter = writer
		http.ServeFile(h, request, target)
		_, _ = fmt.Fprint(writer, htmlFooter)
func PythonStyle(root http.Dir) http.Handler {
	return &pythonStyleHandler{
		root: root,

Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Sunbelt Computer Software

PL/B Language Development and Support

FilesExpand file tree

pythonliststyle.go

Latest commit

History

pythonliststyle.go

File metadata and controls