Moved server packages to a separate folder

server/middleware/Log.go

@@ -0,0 +1,90 @@
+package middleware
+
+import (
+	"net/http"
+	"os"
+	"strconv"
+	"strings"
+	"time"
+
+	"github.com/aerogo/aero"
+	"github.com/aerogo/log"
+	"github.com/animenotifier/notify.moe/arn"
+)
+
+var (
+	requestLog = log.New()
+	errorLog   = log.New()
+	ipLog      = log.New()
+)
+
+// Initialize log files
+func init() {
+	// The request log contains every single request to the server
+	requestLog.AddWriter(log.File("logs/request.log"))
+
+	// The IP log contains the IPs accessing the server
+	ipLog.AddWriter(log.File("logs/ip.log"))
+
+	// The error log contains all failed requests
+	errorLog.AddWriter(log.File("logs/error.log"))
+	errorLog.AddWriter(os.Stderr)
+}
+
+// Log middleware logs every request into logs/request.log and errors into logs/error.log.
+func Log(next aero.Handler) aero.Handler {
+	return func(ctx aero.Context) error {
+		start := time.Now()
+		err := next(ctx)
+		responseTime := time.Since(start)
+
+		logRequest(ctx, responseTime)
+		return err
+	}
+}
+
+// Logs a single request
+func logRequest(ctx aero.Context, responseTime time.Duration) {
+	responseTimeString := strconv.Itoa(int(responseTime.Nanoseconds()/1000000)) + " ms"
+	repeatSpaceCount := 8 - len(responseTimeString)
+
+	if repeatSpaceCount < 0 {
+		repeatSpaceCount = 0
+	}
+
+	responseTimeString = strings.Repeat(" ", repeatSpaceCount) + responseTimeString
+
+	user := arn.GetUserFromContext(ctx)
+	ip := ctx.IP()
+	hostNames, cached := GetHostsForIP(ip)
+
+	if !cached && len(hostNames) > 0 {
+		ipLog.Info("%s = %s", ip, strings.Join(hostNames, ", "))
+	}
+
+	// Log every request
+	id := "id"
+	nick := "guest"
+
+	if user != nil {
+		id = user.ID
+		nick = user.Nick
+	}
+
+	requestLog.Info("%s | %s | %s | %s | %d | %s", nick, id, ip, responseTimeString, ctx.Status(), ctx.Path())
+
+	// Log all requests that failed
+	switch ctx.Status() {
+	case http.StatusOK, http.StatusTemporaryRedirect, http.StatusPermanentRedirect:
+		// Ok.
+
+	default:
+		errorLog.Error("%s | %s | %s | %s | %d | %s", nick, id, ip, responseTimeString, ctx.Status(), ctx.Path())
+	}
+
+	// Notify us about long requests.
+	// However ignore requests under /auth/ because those depend on 3rd party servers.
+	if responseTime >= 500*time.Millisecond && !strings.HasPrefix(ctx.Path(), "/auth/") && !strings.HasPrefix(ctx.Path(), "/sitemap/") && !strings.HasPrefix(ctx.Path(), "/api/sse/") {
+		errorLog.Error("%s | %s | %s | %s | %d | %s (long response time)", nick, id, ip, responseTimeString, ctx.Status(), ctx.Path())
+	}
+}