first commit

app/app.go

@@ -0,0 +1,122 @@
+package app
+
+import (
+	"crypto/tls"
+	_ "embed"
+	"fmt"
+	"io"
+	"log"
+	"net/http"
+	"os"
+
+	"git.0x0001f346.de/andreas/ablage/config"
+	"github.com/julienschmidt/httprouter"
+)
+
+//go:embed assets/index.html
+var assetIndexHTML []byte
+
+//go:embed assets/favicon.svg
+var assetFaviconSVG []byte
+
+//go:embed assets/script.js
+var assetScriptJS []byte
+
+//go:embed assets/style.css
+var assetStyleCSS []byte
+
+func Init() {
+	router := httprouter.New()
+
+	router.NotFound = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		http.Redirect(w, r, "/", http.StatusSeeOther)
+	})
+
+	router.MethodNotAllowed = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		http.Redirect(w, r, "/", http.StatusSeeOther)
+	})
+
+	router.GET(httpPathRoot, httpGetRoot)
+	router.GET(httpPathConfig, httpGetConfig)
+	router.GET(httpPathFaviconICO, httpGetFaviconICO)
+	router.GET(httpPathFaviconSVG, httpGetFaviconSVG)
+	router.GET(httpPathFiles, httpGetFiles)
+	router.GET(httpPathFilesDeleteFilename, httpGetFilesDeleteFilename)
+	router.GET(httpPathFilesGetFilename, httpGetFilesGetFilename)
+	router.GET(httpPathScriptJS, httpGetScriptJS)
+	router.GET(httpPathStyleCSS, httpGetStyleCSS)
+	router.POST(httpPathUpload, httpPostUpload)
+
+	var handler http.Handler = router
+
+	if config.GetBasicAuthMode() {
+		handler = basicAuthMiddleware(handler, config.GetBasicAuthUsername(), config.GetBasicAuthPassword())
+	}
+
+	if config.GetHttpMode() {
+		config.PrintStartupBanner()
+		err := http.ListenAndServe(fmt.Sprintf(":%d", config.GetPortToListenOn()), handler)
+		if err != nil {
+			fmt.Fprintf(os.Stderr, "Ablage exited with error:\n%v\n", err)
+			os.Exit(1)
+		}
+		return
+	}
+
+	tlsCert, err := tls.X509KeyPair(config.GetTLSCertificate(), config.GetTLSKey())
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "Faild to parse PEM encoded public/private key pair:\n%v\n", err)
+		os.Exit(1)
+	}
+
+	server := &http.Server{
+		Addr:     fmt.Sprintf(":%d", config.GetPortToListenOn()),
+		ErrorLog: log.New(io.Discard, "", 0),
+		Handler:  handler,
+		TLSConfig: &tls.Config{
+			Certificates: []tls.Certificate{tlsCert},
+		},
+		TLSNextProto: make(map[string]func(*http.Server, *tls.Conn, http.Handler)),
+	}
+
+	config.PrintStartupBanner()
+
+	err = server.ListenAndServeTLS("", "")
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "Ablage exited with error:\n%v\n", err)
+		os.Exit(1)
+	}
+}
+
+func getClientIP(r *http.Request) string {
+	if r.Header.Get("X-Forwarded-For") != "" {
+		return r.Header.Get("X-Forwarded-For")
+	}
+
+	return r.RemoteAddr
+}
+
+func isBrowserDisplayableFileType(extension string) bool {
+	browserDisplayableFileTypes := map[string]struct{}{
+		// audio
+		".mp3": {}, ".ogg": {}, ".wav": {},
+		// pictures
+		".bmp": {}, ".gif": {}, ".ico": {}, ".jpg": {}, ".jpeg": {},
+		".png": {}, ".svg": {}, ".webp": {},
+		// programming
+		".bat": {}, ".cmd": {}, ".c": {}, ".cpp": {}, ".go": {},
+		".h": {}, ".hpp": {}, ".java": {}, ".kt": {}, ".lua": {},
+		".php": {}, ".pl": {}, ".ps1": {}, ".py": {}, ".rb": {},
+		".rs": {}, ".sh": {}, ".swift": {}, ".ts": {}, ".tsx": {},
+		// text
+		".csv": {}, ".log": {}, ".md": {}, ".pdf": {}, ".txt": {},
+		// video
+		".mp4": {}, ".webm": {},
+		// web
+		".css": {}, ".js": {}, ".html": {},
+	}
+
+	_, isBrowserDisplayableFileType := browserDisplayableFileTypes[extension]
+
+	return isBrowserDisplayableFileType
+}