Move App to server package

server/app.go

@@ -0,0 +1,100 @@
+package server
+
+import (
+	"net/http"
+	"os"
+	"path/filepath"
+	"strings"
+	"time"
+
+	"github.com/cloudsonic/sonic-server/log"
+	"github.com/cloudsonic/sonic-server/scanner"
+	"github.com/go-chi/chi"
+	"github.com/go-chi/chi/middleware"
+	"github.com/go-chi/cors"
+)
+
+type Server struct {
+	Importer *scanner.Importer
+	router   *chi.Mux
+}
+
+func New(importer *scanner.Importer) *Server {
+	a := &Server{Importer: importer}
+	initMimeTypes()
+	a.initRoutes()
+	a.initImporter()
+	return a
+}
+
+func (a *Server) MountRouter(path string, subRouter http.Handler) {
+	a.router.Group(func(r chi.Router) {
+		r.Use(middleware.Logger)
+		r.Mount(path, subRouter)
+	})
+}
+
+func (a *Server) Run(addr string) {
+	log.Info("Started CloudSonic server", "address", addr)
+	log.Error(http.ListenAndServe(addr, a.router))
+}
+
+func (a *Server) initRoutes() {
+	r := chi.NewRouter()
+
+	r.Use(cors.Default().Handler)
+	r.Use(middleware.RequestID)
+	r.Use(middleware.RealIP)
+	r.Use(middleware.Recoverer)
+	r.Use(middleware.Compress(5, "application/xml", "application/json", "application/javascript"))
+	r.Use(middleware.Heartbeat("/ping"))
+	r.Use(InjectLogger)
+
+	r.Get("/", func(w http.ResponseWriter, r *http.Request) {
+		http.Redirect(w, r, "/Jamstash", 302)
+	})
+	workDir, _ := os.Getwd()
+	filesDir := filepath.Join(workDir, "Jamstash-master/dist")
+	FileServer(r, "/Jamstash", http.Dir(filesDir))
+
+	a.router = r
+}
+
+func (a *Server) initImporter() {
+	go a.startPeriodicScans()
+}
+
+func (a *Server) startPeriodicScans() {
+	for {
+		select {
+		case <-time.After(5 * time.Second):
+			a.Importer.CheckForUpdates(false)
+		}
+	}
+}
+
+func FileServer(r chi.Router, path string, root http.FileSystem) {
+	if strings.ContainsAny(path, "{}*") {
+		panic("FileServer does not permit URL parameters.")
+	}
+
+	fs := http.StripPrefix(path, http.FileServer(root))
+
+	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) {
+		fs.ServeHTTP(w, r)
+	})
+}
+
+func InjectLogger(next http.Handler) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		ctx := r.Context()
+		ctx = log.NewContext(r.Context(), "requestId", ctx.Value(middleware.RequestIDKey))
+		next.ServeHTTP(w, r.WithContext(ctx))
+	})
+}

server/mime_types.go

@@ -0,0 +1,44 @@
+package server
+
+import "mime"
+
+func initMimeTypes() {
+	mt := map[string]string{
+		".mp3":  "audio/mpeg",
+		".ogg":  "audio/ogg",
+		".oga":  "audio/ogg",
+		".opus": "audio/ogg",
+		".ogx":  "application/ogg",
+		".aac":  "audio/mp4",
+		".m4a":  "audio/mp4",
+		".flac": "audio/flac",
+		".wav":  "audio/x-wav",
+		".wma":  "audio/x-ms-wma",
+		".ape":  "audio/x-monkeys-audio",
+		".mpc":  "audio/x-musepack",
+		".shn":  "audio/x-shn",
+		".flv":  "video/x-flv",
+		".avi":  "video/avi",
+		".mpg":  "video/mpeg",
+		".mpeg": "video/mpeg",
+		".mp4":  "video/mp4",
+		".m4v":  "video/x-m4v",
+		".mkv":  "video/x-matroska",
+		".mov":  "video/quicktime",
+		".wmv":  "video/x-ms-wmv",
+		".ogv":  "video/ogg",
+		".divx": "video/divx",
+		".m2ts": "video/MP2T",
+		".ts":   "video/MP2T",
+		".webm": "video/webm",
+		".gif":  "image/gif",
+		".jpg":  "image/jpeg",
+		".jpeg": "image/jpeg",
+		".png":  "image/png",
+		".bmp":  "image/bmp",
+	}
+
+	for ext, typ := range mt {
+		_ = mime.AddExtensionType(ext, typ)
+	}
+}