98 lines
2.2 KiB
Go
98 lines
2.2 KiB
Go
package server
|
|
|
|
import (
|
|
"flag"
|
|
"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/pages"
|
|
"github.com/animenotifier/notify.moe/server/auth"
|
|
"github.com/animenotifier/notify.moe/server/middleware"
|
|
)
|
|
|
|
// New creates a new web server.
|
|
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:")
|
|
|
|
// Assets
|
|
assets.Configure(app)
|
|
|
|
// Pages
|
|
pages.Configure(app)
|
|
|
|
// Rewrite
|
|
app.Rewrite(pages.Rewrite)
|
|
|
|
// Middleware
|
|
if IsTest() {
|
|
app.Use(middleware.OpenGraph)
|
|
} else {
|
|
app.Use(
|
|
middleware.Recover,
|
|
middleware.OpenGraph,
|
|
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)
|
|
|
|
// 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() && !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")
|
|
})
|
|
|
|
return app
|
|
}
|
|
|
|
// IsTest returns true if the program is currently running in the "go test" tool.
|
|
func IsTest() bool {
|
|
return flag.Lookup("test.v") != nil
|
|
}
|