103 lines
2.3 KiB
Go

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.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
}