71 lines
1.6 KiB
Go
71 lines
1.6 KiB
Go
//
|
|
// FileServer
|
|
// ===========
|
|
// This example demonstrates how to serve static files from your filesystem.
|
|
//
|
|
//
|
|
// Boot the server:
|
|
// ----------------
|
|
// $ go run main.go
|
|
//
|
|
// Client requests:
|
|
// ----------------
|
|
// $ curl http://localhost:3333/files/
|
|
// <pre>
|
|
// <a href="notes.txt">notes.txt</a>
|
|
// </pre>
|
|
//
|
|
// $ curl http://localhost:3333/files/notes.txt
|
|
// Notessszzz
|
|
//
|
|
package main
|
|
|
|
import (
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/go-chi/chi"
|
|
"github.com/go-chi/chi/middleware"
|
|
)
|
|
|
|
func main() {
|
|
r := chi.NewRouter()
|
|
r.Use(middleware.Logger)
|
|
|
|
// Index handler
|
|
r.Get("/", func(w http.ResponseWriter, r *http.Request) {
|
|
w.Write([]byte("hi"))
|
|
})
|
|
|
|
// Create a route along /files that will serve contents from
|
|
// the ./data/ folder.
|
|
workDir, _ := os.Getwd()
|
|
filesDir := http.Dir(filepath.Join(workDir, "data"))
|
|
FileServer(r, "/files", filesDir)
|
|
|
|
http.ListenAndServe("localhost:3333", r)
|
|
}
|
|
|
|
// FileServer conveniently sets up a http.FileServer handler to serve
|
|
// static files from a http.FileSystem.
|
|
func FileServer(r chi.Router, path string, root http.FileSystem) {
|
|
if strings.ContainsAny(path, "{}*") {
|
|
panic("FileServer does not permit any URL parameters.")
|
|
}
|
|
|
|
if path != "/" && path[len(path)-1] != '/' {
|
|
r.Get(path, http.RedirectHandler(path+"/", 301).ServeHTTP)
|
|
path += "/"
|
|
}
|
|
path += "*"
|
|
|
|
r.Get(path, func(w http.ResponseWriter, r *http.Request) {
|
|
rctx := chi.RouteContext(r.Context())
|
|
pathPrefix := strings.TrimSuffix(rctx.RoutePattern(), "/*")
|
|
fs := http.StripPrefix(pathPrefix, http.FileServer(root))
|
|
fs.ServeHTTP(w, r)
|
|
})
|
|
}
|