284 lines
8.5 KiB
Go
284 lines
8.5 KiB
Go
package main
|
||
|
||
import (
|
||
"embed"
|
||
"log"
|
||
"net/http"
|
||
"os"
|
||
"os/signal"
|
||
"strings"
|
||
"sync"
|
||
"syscall"
|
||
"text/template"
|
||
)
|
||
|
||
func main() {
|
||
stop()
|
||
start()
|
||
}
|
||
|
||
//go:embed docs
|
||
var webFiles embed.FS
|
||
|
||
//go:embed docs/index.html
|
||
var indexTemplateContent string
|
||
|
||
//go:embed docs/404.html
|
||
var notFoundTemplateContent string
|
||
|
||
var (
|
||
indexTemplate *template.Template
|
||
notFoundTemplate *template.Template
|
||
indexOnce sync.Once
|
||
notFoundOnce sync.Once
|
||
)
|
||
|
||
func start() {
|
||
log.Printf("[Info] Starting MeowBox Docs...\r\n")
|
||
log.Printf("[Info] Web service started on 0.0.0.0:2230")
|
||
route()
|
||
err := http.ListenAndServe("0.0.0.0:2230", nil)
|
||
if err != nil {
|
||
log.Fatalf("[Error] Failed to start server: %v", err)
|
||
}
|
||
}
|
||
|
||
func stop() {
|
||
sigCh := make(chan os.Signal, 1)
|
||
signal.Notify(sigCh, syscall.SIGINT, syscall.SIGTERM)
|
||
|
||
go func() {
|
||
sig := <-sigCh
|
||
log.Printf("[Info] Received signal: %v，stopping...\r\n", sig)
|
||
os.Exit(0)
|
||
}()
|
||
}
|
||
|
||
// StaticFileHandler Embedded static file service
|
||
func staticFileHandler(w http.ResponseWriter, r *http.Request) {
|
||
path := r.URL.Path
|
||
|
||
// Clean path, make sure it starts with "docs/"
|
||
path = strings.TrimPrefix(path, "/")
|
||
if path == "" {
|
||
if err := notFoundTemplate.Execute(w, ""); err != nil {
|
||
log.Printf("[Error] Failed to render error page: %v", err)
|
||
}
|
||
return
|
||
}
|
||
|
||
// Concatenate embed.FS path
|
||
filePath := "docs/" + path
|
||
data, err := webFiles.ReadFile(filePath)
|
||
if err != nil {
|
||
log.Printf("[Error] Static file not found: %s", filePath)
|
||
if err := notFoundTemplate.Execute(w, ""); err != nil {
|
||
log.Printf("[Error] Failed to render error page: %v", err)
|
||
}
|
||
return
|
||
}
|
||
|
||
// Set Content-Type
|
||
contentType := getContentType(path)
|
||
w.Header().Set("Content-Type", contentType)
|
||
|
||
_, _ = w.Write(data)
|
||
}
|
||
|
||
func indexHandler(w http.ResponseWriter, r *http.Request) {
|
||
if r.URL.Path != "/" {
|
||
notFoundHandler(w, r)
|
||
return
|
||
}
|
||
loadIndexTemplate()
|
||
log.Printf("[Info] [Web Access] Handler index page, on path: %s", r.URL.Path)
|
||
w.Header().Set("Content-Type", "text/html; charset=utf-8")
|
||
if err := indexTemplate.Execute(w, nil); err != nil {
|
||
log.Printf("[Error] Failed to render index page: %v", err)
|
||
}
|
||
}
|
||
|
||
func notFoundHandler(w http.ResponseWriter, r *http.Request) {
|
||
loadNotFoundTemplate()
|
||
log.Printf("[Info] [Web Access] Handler http error page: 404, on path: %s", r.URL.Path)
|
||
w.WriteHeader(404)
|
||
w.Header().Set("Content-Type", "text/html; charset=utf-8")
|
||
if err := notFoundTemplate.Execute(w, nil); err != nil {
|
||
log.Printf("[Error] Failed to render index page: %v", err)
|
||
}
|
||
}
|
||
|
||
func loadIndexTemplate() {
|
||
indexOnce.Do(func() {
|
||
var err error
|
||
indexTemplate, err = template.New("index").Parse(indexTemplateContent)
|
||
if err != nil {
|
||
log.Fatalf("[Error] Failed to parse index template: %v", err)
|
||
}
|
||
})
|
||
}
|
||
|
||
func loadNotFoundTemplate() {
|
||
notFoundOnce.Do(func() {
|
||
var err error
|
||
notFoundTemplate, err = template.New("error").Parse(notFoundTemplateContent)
|
||
if err != nil {
|
||
log.Fatalf("[Error] Failed to parse 404 template: %v", err)
|
||
}
|
||
})
|
||
}
|
||
|
||
func route() {
|
||
http.HandleFunc("/", indexHandler)
|
||
http.HandleFunc("/.well-known/", func(w http.ResponseWriter, r *http.Request) {
|
||
w.WriteHeader(http.StatusNoContent)
|
||
})
|
||
http.HandleFunc("/favicon.ico", staticFileHandler)
|
||
http.HandleFunc("/css/", staticFileHandler)
|
||
http.HandleFunc("/font-awesome/", staticFileHandler)
|
||
http.HandleFunc("/js/", staticFileHandler)
|
||
http.HandleFunc("/img/", staticFileHandler)
|
||
}
|
||
|
||
func getContentType(filename string) string {
|
||
// Convert file names to lowercase for case insensitive comparison
|
||
lowerFilename := strings.ToLower(filename)
|
||
|
||
switch {
|
||
// Text files
|
||
case strings.HasSuffix(lowerFilename, ".css"):
|
||
return "text/css; charset=utf-8"
|
||
case strings.HasSuffix(lowerFilename, ".js"):
|
||
return "application/javascript; charset=utf-8"
|
||
case strings.HasSuffix(lowerFilename, ".json"):
|
||
return "application/json; charset=utf-8"
|
||
case strings.HasSuffix(lowerFilename, ".xml"):
|
||
return "application/xml; charset=utf-8"
|
||
case strings.HasSuffix(lowerFilename, ".html"), strings.HasSuffix(lowerFilename, ".htm"):
|
||
return "text/html; charset=utf-8"
|
||
case strings.HasSuffix(lowerFilename, ".txt"):
|
||
return "text/plain; charset=utf-8"
|
||
case strings.HasSuffix(lowerFilename, ".md"):
|
||
return "text/markdown; charset=utf-8"
|
||
case strings.HasSuffix(lowerFilename, ".csv"):
|
||
return "text/csv; charset=utf-8"
|
||
|
||
// Image files
|
||
case strings.HasSuffix(lowerFilename, ".webp"):
|
||
return "image/webp"
|
||
case strings.HasSuffix(lowerFilename, ".png"):
|
||
return "image/png"
|
||
case strings.HasSuffix(lowerFilename, ".jpg"), strings.HasSuffix(lowerFilename, ".jpeg"):
|
||
return "image/jpeg"
|
||
case strings.HasSuffix(lowerFilename, ".gif"):
|
||
return "image/gif"
|
||
case strings.HasSuffix(lowerFilename, ".bmp"):
|
||
return "image/bmp"
|
||
case strings.HasSuffix(lowerFilename, ".ico"):
|
||
return "image/x-icon"
|
||
case strings.HasSuffix(lowerFilename, ".svg"), strings.HasSuffix(lowerFilename, ".svgz"):
|
||
return "image/svg+xml"
|
||
case strings.HasSuffix(lowerFilename, ".tiff"), strings.HasSuffix(lowerFilename, ".tif"):
|
||
return "image/tiff"
|
||
case strings.HasSuffix(lowerFilename, ".avif"):
|
||
return "image/avif"
|
||
|
||
// Audio files
|
||
case strings.HasSuffix(lowerFilename, ".mp3"):
|
||
return "audio/mpeg"
|
||
case strings.HasSuffix(lowerFilename, ".wav"):
|
||
return "audio/wav"
|
||
case strings.HasSuffix(lowerFilename, ".ogg"):
|
||
return "audio/ogg"
|
||
case strings.HasSuffix(lowerFilename, ".flac"):
|
||
return "audio/flac"
|
||
case strings.HasSuffix(lowerFilename, ".aac"):
|
||
return "audio/aac"
|
||
case strings.HasSuffix(lowerFilename, ".m4a"):
|
||
return "audio/mp4"
|
||
|
||
// Video files
|
||
case strings.HasSuffix(lowerFilename, ".mp4"):
|
||
return "video/mp4"
|
||
case strings.HasSuffix(lowerFilename, ".webm"):
|
||
return "video/webm"
|
||
case strings.HasSuffix(lowerFilename, ".ogg"), strings.HasSuffix(lowerFilename, ".ogv"):
|
||
return "video/ogg"
|
||
case strings.HasSuffix(lowerFilename, ".mov"):
|
||
return "video/quicktime"
|
||
case strings.HasSuffix(lowerFilename, ".avi"):
|
||
return "video/x-msvideo"
|
||
case strings.HasSuffix(lowerFilename, ".wmv"):
|
||
return "video/x-ms-wmv"
|
||
case strings.HasSuffix(lowerFilename, ".flv"):
|
||
return "video/x-flv"
|
||
case strings.HasSuffix(lowerFilename, ".mkv"):
|
||
return "video/x-matroska"
|
||
|
||
// Font files
|
||
case strings.HasSuffix(lowerFilename, ".woff"):
|
||
return "font/woff"
|
||
case strings.HasSuffix(lowerFilename, ".woff2"):
|
||
return "font/woff2"
|
||
case strings.HasSuffix(lowerFilename, ".ttf"):
|
||
return "font/ttf"
|
||
case strings.HasSuffix(lowerFilename, ".otf"):
|
||
return "font/otf"
|
||
|
||
// Archive files
|
||
case strings.HasSuffix(lowerFilename, ".zip"):
|
||
return "application/zip"
|
||
case strings.HasSuffix(lowerFilename, ".rar"):
|
||
return "application/x-rar-compressed"
|
||
case strings.HasSuffix(lowerFilename, ".gz"):
|
||
return "application/gzip"
|
||
case strings.HasSuffix(lowerFilename, ".tar"):
|
||
return "application/x-tar"
|
||
case strings.HasSuffix(lowerFilename, ".7z"):
|
||
return "application/x-7z-compressed"
|
||
case strings.HasSuffix(lowerFilename, ".bz2"):
|
||
return "application/x-bzip2"
|
||
case strings.HasSuffix(lowerFilename, ".xz"):
|
||
return "application/x-xz"
|
||
|
||
// Document files
|
||
case strings.HasSuffix(lowerFilename, ".pdf"):
|
||
return "application/pdf"
|
||
case strings.HasSuffix(lowerFilename, ".doc"):
|
||
return "application/msword"
|
||
case strings.HasSuffix(lowerFilename, ".docx"):
|
||
return "application/vnd.openxmlformats-officedocument.wordprocessingml.document"
|
||
case strings.HasSuffix(lowerFilename, ".xls"):
|
||
return "application/vnd.ms-excel"
|
||
case strings.HasSuffix(lowerFilename, ".xlsx"):
|
||
return "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
|
||
case strings.HasSuffix(lowerFilename, ".ppt"):
|
||
return "application/vnd.ms-powerpoint"
|
||
case strings.HasSuffix(lowerFilename, ".pptx"):
|
||
return "application/vnd.openxmlformats-officedocument.presentationml.presentation"
|
||
case strings.HasSuffix(lowerFilename, ".odt"):
|
||
return "application/vnd.oasis.opendocument.text"
|
||
case strings.HasSuffix(lowerFilename, ".ods"):
|
||
return "application/vnd.oasis.opendocument.spreadsheet"
|
||
case strings.HasSuffix(lowerFilename, ".odp"):
|
||
return "application/vnd.oasis.opendocument.presentation"
|
||
|
||
// Other files
|
||
case strings.HasSuffix(lowerFilename, ".rtf"):
|
||
return "application/rtf"
|
||
case strings.HasSuffix(lowerFilename, ".epub"):
|
||
return "application/epub+zip"
|
||
case strings.HasSuffix(lowerFilename, ".apk"):
|
||
return "application/vnd.android.package-archive"
|
||
case strings.HasSuffix(lowerFilename, ".exe"):
|
||
return "application/x-msdownload"
|
||
case strings.HasSuffix(lowerFilename, ".dmg"):
|
||
return "application/x-apple-diskimage"
|
||
case strings.HasSuffix(lowerFilename, ".iso"):
|
||
return "application/x-iso9660-image"
|
||
|
||
default:
|
||
return "application/octet-stream"
|
||
}
|
||
}
|