88 lines
1.8 KiB
Go

package main
import (
"strings"
"github.com/aerogo/aero"
"github.com/aerogo/session-store-nano"
"github.com/animenotifier/arn"
"github.com/animenotifier/notify.moe/auth"
"github.com/animenotifier/notify.moe/middleware"
"github.com/animenotifier/notify.moe/pages"
"github.com/animenotifier/notify.moe/utils/routetests"
)
var app = aero.New()
func main() {
// Configure and start
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:")
// Security
configureHTTPS(app)
// Assets
configureAssets(app)
// Pages
pages.Configure(app)
// Rewrite
app.Rewrite(rewrite)
// Middleware
app.Use(
middleware.Log(),
middleware.Session(),
middleware.UserInfo(),
)
// API
arn.API.Install(app)
// Development server configuration
if arn.IsDevelopment() {
app.Config.Domain = "beta.notify.moe"
app.Config.Title += " - Beta"
app.Config.Manifest.Name = app.Config.Title
// Test connectivity
app.OnStart(testConnectivity)
}
// Authentication
auth.Install(app)
// Close the database node on shutdown
app.OnEnd(arn.Node.Close)
// 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().Get("Referer"), "/service-worker")
})
// Specify test routes
for route, examples := range routetests.All() {
app.Test(route, examples)
}
return app
}