Moved server packages to a separate folder

server/middleware/Recover.go

@@ -0,0 +1,59 @@
+package middleware
+
+import (
+	"fmt"
+	"net/http"
+	"os"
+	"runtime"
+	"strings"
+
+	"github.com/aerogo/aero"
+	"github.com/animenotifier/notify.moe/arn"
+)
+
+// Recover recovers from panics and shows them as the response body.
+func Recover(next aero.Handler) aero.Handler {
+	return func(ctx aero.Context) error {
+		defer func() {
+			r := recover()
+
+			if r == nil {
+				return
+			}
+
+			err, ok := r.(error)
+
+			if !ok {
+				err = fmt.Errorf("%v", r)
+			}
+
+			stack := make([]byte, 4096)
+			length := runtime.Stack(stack, false)
+			stackString := string(stack[:length])
+			fmt.Fprint(os.Stderr, stackString)
+
+			// Save crash in database
+			crash := &arn.Crash{
+				Error: err.Error(),
+				Stack: stackString,
+				Path:  ctx.Path(),
+			}
+
+			crash.ID = arn.GenerateID("Crash")
+			crash.Created = arn.DateTimeUTC()
+			user := arn.GetUserFromContext(ctx)
+
+			if user != nil {
+				crash.CreatedBy = user.ID
+			}
+
+			crash.Save()
+
+			// Send HTML
+			message := "<div class='crash'>" + err.Error() + "<br><br>" + strings.ReplaceAll(stackString, "\n", "<br>") + "</div>"
+			_ = ctx.Error(http.StatusInternalServerError, message)
+		}()
+
+		return next(ctx)
+	}
+}