package main import ( "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/routetests" ) func main() { // Configure and start app := aero.New() configure(app).Run() } func configure(app *aero.Application) *aero.Application { // Sessions app.Sessions.Duration = 3600 * 24 * 30 * 6 app.Sessions.Store = nanostore.New(arn.DB.Collection("Session")) // 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 configureHTTPS(app) // Assets assets.Configure(app) // Pages pages.Configure(app) // Rewrite app.Rewrite(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()) }) // 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 }