Move server initialization to new server package

server/Main.go

@@ -0,0 +1,6 @@
+package server
+
+func Main() {
+	app := New()
+	app.Run()
+}

server/New.go

@@ -0,0 +1,106 @@
+package server
+
+import (
+	"net/http"
+	"strings"
+
+	"github.com/aerogo/aero"
+	nanostore "github.com/aerogo/session-store-nano"
+	"github.com/akyoto/color"
+	"github.com/animenotifier/notify.moe/arn"
+	"github.com/animenotifier/notify.moe/assets"
+	"github.com/animenotifier/notify.moe/auth"
+	"github.com/animenotifier/notify.moe/graphql"
+	"github.com/animenotifier/notify.moe/middleware"
+	"github.com/animenotifier/notify.moe/pages"
+	"github.com/animenotifier/notify.moe/utils/htmlemail"
+	"github.com/animenotifier/notify.moe/utils/https"
+	"github.com/animenotifier/notify.moe/utils/routetests"
+)
+
+func New() *aero.Application {
+	app := aero.New()
+
+	// Sessions
+	app.Sessions.Duration = 3600 * 24 * 30 * 6
+	app.Sessions.Store = nanostore.New(arn.DB.Collection("Session"))
+	app.Sessions.SameSite = http.SameSiteNoneMode
+
+	// Content security policy
+	app.ContentSecurityPolicy.Set("img-src", "https: data:")
+	app.ContentSecurityPolicy.Set("connect-src", "https: wss: data:")
+	app.ContentSecurityPolicy.Set("font-src", "https: data:")
+
+	// Security
+	https.Configure(app)
+
+	// Assets
+	assets.Configure(app)
+
+	// Pages
+	pages.Configure(app)
+
+	// Rewrite
+	app.Rewrite(pages.Rewrite)
+
+	// Middleware
+	app.Use(
+		middleware.Recover,
+		middleware.HTTPSRedirect,
+		middleware.OpenGraph,
+		middleware.Log,
+		middleware.Session,
+		middleware.UserInfo,
+	)
+
+	// API
+	arn.API.Install(app)
+
+	// Development server configuration
+	if arn.IsDevelopment() {
+		assets.Domain = "beta.notify.moe"
+		assets.Manifest.Name += " - Beta"
+	}
+
+	// Authentication
+	auth.Install(app)
+
+	// GraphQL
+	graphql.Install(app)
+
+	// Close the database node on shutdown
+	app.OnEnd(arn.Node.Close)
+
+	// Don't push when an underscore URL has been requested
+	app.AddPushCondition(func(ctx aero.Context) bool {
+		return !strings.HasPrefix(ctx.Path(), "/_")
+	})
+
+	// Show errors in the console
+	app.OnError(func(ctx aero.Context, err error) {
+		color.Red(err.Error())
+	})
+
+	// Emails
+	arn.HTMLEmailRenderer = &htmlemail.Renderer{}
+
+	// Check that this is the server
+	if !arn.Node.IsServer() && !arn.IsTest() {
+		panic("Another program is currently running as the database server")
+	}
+
+	// Prefetch all collections
+	arn.DB.Prefetch()
+
+	// Do not use HTTP/2 push on service worker requests
+	app.AddPushCondition(func(ctx aero.Context) bool {
+		return !strings.Contains(ctx.Request().Header("Referer"), "/service-worker")
+	})
+
+	// Specify test routes
+	for route, examples := range routetests.All() {
+		app.Test(route, examples...)
+	}
+
+	return app
+}