You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
79 lines
1.9 KiB
79 lines
1.9 KiB
package server
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"github.com/halverneus/static-file-server/config"
|
|
"github.com/halverneus/static-file-server/handle"
|
|
)
|
|
|
|
var (
|
|
// Values to be overridden to simplify unit testing.
|
|
selectHandler = handlerSelector
|
|
selectListener = listenerSelector
|
|
)
|
|
|
|
// Run server.
|
|
func Run() error {
|
|
if config.Get.Debug {
|
|
config.Log()
|
|
}
|
|
// Choose and set the appropriate, optimized static file serving function.
|
|
handler := selectHandler()
|
|
|
|
// Serve files over HTTP or HTTPS based on paths to TLS files being
|
|
// provided.
|
|
listener := selectListener()
|
|
|
|
binding := fmt.Sprintf("%s:%d", config.Get.Host, config.Get.Port)
|
|
return listener(binding, handler)
|
|
}
|
|
|
|
// handlerSelector returns the appropriate request handler based on
|
|
// configuration.
|
|
func handlerSelector() (handler http.HandlerFunc) {
|
|
var serveFileHandler handle.FileServerFunc
|
|
|
|
serveFileHandler = http.ServeFile
|
|
if config.Get.Debug {
|
|
serveFileHandler = handle.WithLogging(serveFileHandler)
|
|
}
|
|
|
|
if config.Get.Referrers != nil {
|
|
serveFileHandler = handle.WithReferrers(serveFileHandler, config.Get.Referrers)
|
|
}
|
|
|
|
// Choose and set the appropriate, optimized static file serving function.
|
|
if 0 == len(config.Get.URLPrefix) {
|
|
handler = handle.Basic(serveFileHandler, config.Get.Folder)
|
|
} else {
|
|
handler = handle.Prefix(
|
|
serveFileHandler,
|
|
config.Get.Folder,
|
|
config.Get.URLPrefix,
|
|
)
|
|
}
|
|
|
|
// Determine whether index files should hidden.
|
|
if !config.Get.ShowListing {
|
|
handler = handle.IgnoreIndex(handler)
|
|
}
|
|
return
|
|
}
|
|
|
|
// listenerSelector returns the appropriate listener handler based on
|
|
// configuration.
|
|
func listenerSelector() (listener handle.ListenerFunc) {
|
|
// Serve files over HTTP or HTTPS based on paths to TLS files being
|
|
// provided.
|
|
if 0 < len(config.Get.TLSCert) {
|
|
listener = handle.TLSListening(
|
|
config.Get.TLSCert,
|
|
config.Get.TLSKey,
|
|
)
|
|
} else {
|
|
listener = handle.Listening()
|
|
}
|
|
return
|
|
}
|
|
|