yoake/server/webroot/server.go

95 lines
3.3 KiB
Go

package webroot
import (
"encoding/json"
"log"
"os"
"github.com/eternal-flame-AD/yoake/config"
"github.com/eternal-flame-AD/yoake/internal/auth"
"github.com/eternal-flame-AD/yoake/internal/canvaslms"
"github.com/eternal-flame-AD/yoake/internal/comm"
"github.com/eternal-flame-AD/yoake/internal/db"
"github.com/eternal-flame-AD/yoake/internal/echoerror"
"github.com/eternal-flame-AD/yoake/internal/entertainment"
"github.com/eternal-flame-AD/yoake/internal/health"
"github.com/eternal-flame-AD/yoake/internal/servetpl"
"github.com/eternal-flame-AD/yoake/internal/session"
"github.com/eternal-flame-AD/yoake/internal/twilio"
"github.com/eternal-flame-AD/yoake/internal/utilapi"
"github.com/eternal-flame-AD/yoake/server"
"github.com/gorilla/context"
"github.com/gorilla/sessions"
"github.com/labstack/echo/v4"
"github.com/labstack/echo/v4/middleware"
)
func Init(hostname string, comm *comm.Communicator, database db.DB) {
e := echo.New()
webroot := config.Config().WebRoot
if webroot.Root == "" {
log.Panicf("webroot not set, use . to override")
}
if webroot.SessionKey == "" {
log.Panicf("webroot session key not set")
}
sessionCookie := sessions.NewCookieStore([]byte(webroot.SessionKey))
fsCookie := sessions.NewFilesystemStore(webroot.SessionDir, []byte(webroot.SessionKey))
if webroot.Secure != nil {
e.Use(middleware.SecureWithConfig(*webroot.Secure))
}
if webroot.Log != nil {
filters := logCompileFilters(webroot.Log.Filter)
logOut := log.New(os.Stdout, "webroot: ", log.Ldate|log.Ltime)
lc := loggerConfig
lc.LogValuesFunc = func(c echo.Context, values middleware.RequestLoggerValues) (err error) {
entry := processLoggerValues(c, values)
if logFilterCategories(c, filters) {
var j []byte
if webroot.Log.Indent {
j, err = json.MarshalIndent(entry, "", " ")
} else {
j, err = json.Marshal(entry)
}
if err != nil {
return err
}
logOut.Println(string(j))
}
return nil
}
e.Use(middleware.RequestLoggerWithConfig(lc))
}
api := e.Group("/api", echoerror.Middleware(echoerror.JSONWriter))
{
canvaslms.Register(api.Group("/canvas", logMiddleware("api_canvas", nil)), comm)
utilapi.Register(api.Group("/util", logMiddleware("api_util", nil)))
comm.RegisterAPIRoute(api.Group("/comm", logMiddleware("api_comm", nil)))
auth.Register(api.Group("/auth", logMiddleware("api_auth", nil)))
entertainment.Register(api.Group("/entertainment", logMiddleware("api_entertainment", nil)), database)
health.Register(api.Group("/health", logMiddleware("api_health", nil)), database, comm)
}
e.Use(
echoerror.Middleware(echoerror.HTMLWriter),
func(next echo.HandlerFunc) echo.HandlerFunc {
return func(c echo.Context) error {
defer context.Clear(c.Request())
c.Set(session.SessionStoreKeyPrefix+"cookie", (sessions.Store)(sessionCookie))
c.Set(session.SessionStoreKeyPrefix+"fs", (sessions.Store)(fsCookie))
return next(c)
}
},
middleware.Gzip(),
auth.Middleware(sessionCookie),
logMiddleware("twilio", twilio.VerifyMiddleware("/twilio", config.Config().Twilio.BaseURL)),
middleware.Rewrite(map[string]string{"*/": "$1/index.html"}),
logMiddleware("template", servetpl.ServeTemplateDir(webroot.Root)),
logMiddleware("static", middleware.Static(webroot.Root)))
server.RegisterHostname(hostname, &server.Host{Echo: e})
}