2018-07-16 03:52:37 +00:00
|
|
|
package handle
|
|
|
|
|
|
|
|
import (
|
|
|
|
"net/http"
|
|
|
|
"strings"
|
|
|
|
)
|
|
|
|
|
2018-07-16 23:31:30 +00:00
|
|
|
var (
|
|
|
|
// These assignments are for unit testing.
|
|
|
|
listenAndServe = http.ListenAndServe
|
|
|
|
listenAndServeTLS = http.ListenAndServeTLS
|
|
|
|
setHandler = http.HandleFunc
|
|
|
|
)
|
|
|
|
|
|
|
|
var (
|
|
|
|
server http.Server
|
|
|
|
)
|
|
|
|
|
|
|
|
// ListenerFunc accepts the {hostname:port} binding string required by HTTP
|
|
|
|
// listeners and the handler (router) function and returns any errors that
|
|
|
|
// occur.
|
2018-07-16 03:52:37 +00:00
|
|
|
type ListenerFunc func(string, http.HandlerFunc) error
|
|
|
|
|
|
|
|
// Basic file handler servers files from the passed folder.
|
|
|
|
func Basic(folder string) http.HandlerFunc {
|
|
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
http.ServeFile(w, r, folder+r.URL.Path)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Prefix file handler is an alternative to Basic where a URL prefix is removed
|
|
|
|
// prior to serving a file (http://my.machine/prefix/file.txt will serve
|
|
|
|
// file.txt from the root of the folder being served (ignoring 'prefix')).
|
|
|
|
func Prefix(folder, urlPrefix string) http.HandlerFunc {
|
|
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
if !strings.HasPrefix(r.URL.Path, urlPrefix) {
|
|
|
|
http.NotFound(w, r)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
http.ServeFile(w, r, folder+strings.TrimPrefix(r.URL.Path, urlPrefix))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// IgnoreIndex wraps an HTTP request. In the event of a folder root request,
|
|
|
|
// this function will automatically return 'NOT FOUND' as opposed to default
|
|
|
|
// behavior where the index file for that directory is retrieved.
|
|
|
|
func IgnoreIndex(serve http.HandlerFunc) http.HandlerFunc {
|
|
|
|
return func(w http.ResponseWriter, r *http.Request) {
|
|
|
|
if strings.HasSuffix(r.URL.Path, "/") {
|
|
|
|
http.NotFound(w, r)
|
|
|
|
return
|
|
|
|
}
|
|
|
|
serve(w, r)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Listening function for serving the handler function.
|
|
|
|
func Listening() ListenerFunc {
|
|
|
|
return func(binding string, handler http.HandlerFunc) error {
|
2018-07-16 23:31:30 +00:00
|
|
|
setHandler("/", handler)
|
|
|
|
return listenAndServe(binding, nil)
|
2018-07-16 03:52:37 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// TLSListening function for serving the handler function with encryption.
|
|
|
|
func TLSListening(tlsCert, tlsKey string) ListenerFunc {
|
|
|
|
return func(binding string, handler http.HandlerFunc) error {
|
2018-07-16 23:31:30 +00:00
|
|
|
setHandler("/", handler)
|
|
|
|
return listenAndServeTLS(binding, tlsCert, tlsKey, nil)
|
2018-07-16 03:52:37 +00:00
|
|
|
}
|
|
|
|
}
|