Add content policy and add abuse categories

main.go

@@ -2,8 +2,10 @@ package main
 
 import (
 	"flag"
+	"net"
 	"net/http"
 
+	"fornaxian.tech/pixeldrain_server/util"
 	web "fornaxian.tech/pixeldrain_web/init"
 
 	"github.com/Fornaxian/log"
@@ -14,17 +16,32 @@ import (
 // be directly embedded by another Go project. And when deployed it will run
 // independently.
 func main() {
+	var err error
+	var sock = flag.Bool("systemd-socket", false, "Enable/disable systemd socket activation")
 	var listen = flag.String("listen", ":8081", "The address which the API server will listen on")
 	var prefix = flag.String("prefix", "", "Prefix that comes before the API URL")
 	flag.Parse()
 
-	r := httprouter.New()
+	var listener net.Listener
 
-	web.Init(r, *prefix, true)
+	// Serve the API on a socket. If systemd-socket is enabled we'll reuse
+	// systemd's socket, else we'll create our own to serve on
+	if *sock {
+		// Socket activation enabled. Get the provided sockets and serve on them
+		if listener, err = util.SystemdSocketByName("pd-web.socket"); err != nil {
+			panic("Socket pd-web.socket not found")
+		}
+	} else {
+		// Socket activation disabled, so we create our own listener to serve on
+		if listener, err = net.Listen("tcp", *listen); err != nil {
+			panic(err)
+		}
+	}
 
-	err := http.ListenAndServe(*listen, r)
+	var router = httprouter.New()
+	web.Init(router, *prefix, true)
 
-	if err != nil {
+	if err = http.Serve(listener, router); err != nil {
 		log.Error("Can't listen and serve Pixeldrain Web: %v", err)
 	}
 }