Moved everything out of internal

api/server.go

@@ -0,0 +1,139 @@
+package api
+
+import (
+	"context"
+	"crypto/tls"
+	"errors"
+	"fmt"
+	"github.com/99designs/gqlgen/handler"
+	"github.com/go-chi/chi"
+	"github.com/go-chi/chi/middleware"
+	"github.com/gobuffalo/packr/v2"
+	"github.com/rs/cors"
+	"github.com/stashapp/stash/logger"
+	"github.com/stashapp/stash/models"
+	"net/http"
+	"path"
+	"runtime/debug"
+	"strings"
+)
+
+const httpPort = "9998"
+const httpsPort = "9999"
+
+var certsBox *packr.Box
+var uiBox *packr.Box
+
+func Start() {
+	//port := os.Getenv("PORT")
+	//if port == "" {
+	//	port = defaultPort
+	//}
+
+	certsBox = packr.New("Cert Box", "../../certs")
+	uiBox = packr.New("UI Box", "../../ui/v1/dist/stash-frontend")
+
+	r := chi.NewRouter()
+
+	r.Use(middleware.Recoverer)
+	r.Use(middleware.Logger)
+	r.Use(middleware.DefaultCompress)
+	r.Use(middleware.StripSlashes)
+	r.Use(cors.AllowAll().Handler)
+	r.Use(BaseURLMiddleware)
+
+	recoverFunc := handler.RecoverFunc(func(ctx context.Context, err interface{}) error {
+		logger.Error(err)
+		debug.PrintStack()
+
+		message := fmt.Sprintf("Internal system error. Error <%v>", err)
+		return errors.New(message)
+	})
+	requestMiddleware := handler.RequestMiddleware(func(ctx context.Context, next func(ctx context.Context) []byte) []byte {
+		//api.GetRequestContext(ctx).Variables[]
+		return next(ctx)
+	})
+	gqlHandler := handler.GraphQL(models.NewExecutableSchema(models.Config{Resolvers: &Resolver{}}), recoverFunc, requestMiddleware)
+
+	// https://stash.server:9999/certs/server.crt
+	r.Handle("/certs/*", http.FileServer(certsBox))
+
+	r.Handle("/graphql", gqlHandler)
+	r.Handle("/playground", handler.Playground("GraphQL playground", "/graphql"))
+
+	r.Mount("/gallery", galleryRoutes{}.Routes())
+	r.Mount("/performer", performerRoutes{}.Routes())
+	r.Mount("/scene", sceneRoutes{}.Routes())
+	r.Mount("/studio", studioRoutes{}.Routes())
+
+	// Serve the angular app
+	r.HandleFunc("/*", func(w http.ResponseWriter, r *http.Request) {
+		ext := path.Ext(r.URL.Path)
+		if ext == ".html" || ext == ""  {
+			data := uiBox.Bytes("index.html")
+			_, _ = w.Write(data)
+		} else {
+			http.FileServer(uiBox).ServeHTTP(w, r)
+		}
+	})
+
+	httpsServer := &http.Server{
+		Addr: ":"+httpsPort,
+		Handler: r,
+		TLSConfig: makeTLSConfig(),
+	}
+	server := &http.Server{
+		Addr: ":"+httpPort,
+		Handler: r,
+	}
+
+	go func() {
+		logger.Infof("stash is running on HTTP at http://localhost:9998/")
+		logger.Fatal(server.ListenAndServe())
+	}()
+
+	logger.Infof("stash is running on HTTPS at https://localhost:9999/")
+	logger.Fatal(httpsServer.ListenAndServeTLS("", ""))
+}
+
+func makeTLSConfig() *tls.Config {
+	cert, err := certsBox.Find("server.crt")
+	key, err := certsBox.Find("server.key")
+
+	certs := make([]tls.Certificate, 1)
+	certs[0], err = tls.X509KeyPair(cert, key)
+	if err != nil {
+		panic(err)
+	}
+	tlsConfig := &tls.Config{
+		Certificates: certs,
+	}
+
+	return tlsConfig
+}
+
+
+type contextKey struct {
+	name string
+}
+var (
+	BaseURLCtxKey = &contextKey{"BaseURL"}
+)
+func BaseURLMiddleware(next http.Handler) http.Handler {
+	fn := func(w http.ResponseWriter, r *http.Request) {
+		ctx := r.Context()
+
+		var scheme string
+		if strings.Compare("https", r.URL.Scheme) == 0 || r.Proto == "HTTP/2.0" {
+			scheme = "https"
+		} else {
+			scheme = "http"
+		}
+		baseURL := scheme + "://" + r.Host
+
+		r = r.WithContext(context.WithValue(ctx, BaseURLCtxKey, baseURL))
+
+		next.ServeHTTP(w, r)
+	}
+	return http.HandlerFunc(fn)
+}