123 lines
3.2 KiB
Go
123 lines
3.2 KiB
Go
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
|
|
}
|