echo/handler/static.go

package handler

import (
	"fmt"
	"io"
	"net/http"
	"path"

	"github.com/labstack/echo"
)

type (
	StaticOption struct {
		Root   string `json:"root"`
		Index  string `json:"index"`
		Browse bool   `json:"browse"`
	}
)

func Static(root string, option ...*StaticOption) echo.HandlerFunc {
	// Default options
	opt := &StaticOption{Index: "index.html"}
	if len(option) > 0 {
		opt = option[0]
	}

	return func(c echo.Context) error {
		fs := http.Dir(root)
		file := c.P(0)
		f, err := fs.Open(file)
		if err != nil {
			return echo.ErrNotFound
		}
		defer f.Close()

		fi, err := f.Stat()
		if err != nil {
			return err
		}

		if fi.IsDir() {
			/* NOTE:
			Not checking the Last-Modified header as it caches the response `304` when
			changing differnt directories for the same path.
			*/
			d := f

			// Index file
			file = path.Join(file, opt.Index)
			f, err = fs.Open(file)
			if err != nil {
				if opt.Browse {
					dirs, err := d.Readdir(-1)
					if err != nil {
						return err
					}

					// Create a directory index
					res := c.Response()
					res.Header().Set(echo.ContentType, echo.TextHTMLCharsetUTF8)
					if _, err = fmt.Fprintf(res, "<pre>\n"); err != nil {
						return err
					}
					for _, d := range dirs {
						name := d.Name()
						color := "#212121"
						if d.IsDir() {
							color = "#e91e63"
							name += "/"
						}
						if _, err = fmt.Fprintf(res, "<a href=\"%s\" style=\"color: %s;\">%s</a>\n", name, color, name); err != nil {
							return err
						}
					}
					_, err = fmt.Fprintf(res, "</pre>\n")
					return err
				}
				return echo.ErrNotFound
			}
			fi, _ = f.Stat() // Index file stat
		}
		c.Response().WriteHeader(http.StatusOK)
		io.Copy(c.Response(), f)
		return nil
		// TODO:
		// http.ServeContent(c.Response(), c.Request(), fi.Name(), fi.ModTime(), f)
	}
}

// Favicon serves the default favicon - GET /favicon.ico.
func Favicon() echo.HandlerFunc {
	return func(c echo.Context) error {
		return nil
	}
}