memos/server/server.go

81 lines
1.8 KiB
Go
Raw Normal View History

2022-02-03 10:32:03 +03:00
package server
import (
"fmt"
2022-05-22 04:29:34 +03:00
"memos/server/profile"
2022-05-16 02:37:23 +03:00
"memos/store"
2022-02-04 13:54:24 +03:00
"time"
2022-02-03 10:32:03 +03:00
2022-02-05 12:04:23 +03:00
"github.com/gorilla/securecookie"
2022-02-03 10:32:03 +03:00
"github.com/gorilla/sessions"
"github.com/labstack/echo-contrib/session"
"github.com/labstack/echo/v4"
"github.com/labstack/echo/v4/middleware"
)
type Server struct {
e *echo.Echo
2022-05-22 04:29:34 +03:00
Profile *profile.Profile
2022-03-29 15:53:43 +03:00
2022-05-16 02:37:23 +03:00
Store *store.Store
2022-02-03 10:32:03 +03:00
}
2022-05-22 04:29:34 +03:00
func NewServer(profile *profile.Profile) *Server {
2022-02-03 10:32:03 +03:00
e := echo.New()
e.Debug = true
e.HideBanner = true
2022-05-20 17:48:36 +03:00
e.HidePort = true
2022-02-03 10:32:03 +03:00
2022-02-04 13:54:24 +03:00
e.Use(middleware.LoggerWithConfig(middleware.LoggerConfig{
Format: "${method} ${uri} ${status}\n",
}))
e.Use(middleware.TimeoutWithConfig(middleware.TimeoutConfig{
Skipper: middleware.DefaultSkipper,
ErrorMessage: "Request timeout",
Timeout: 30 * time.Second,
}))
2022-02-03 10:32:03 +03:00
e.Use(middleware.StaticWithConfig(middleware.StaticConfig{
Skipper: middleware.DefaultSkipper,
Root: "web/dist",
Browse: false,
HTML5: true,
2022-02-03 10:32:03 +03:00
}))
2022-03-29 15:53:43 +03:00
// In dev mode, set the const secret key to make login session persistence.
2022-05-21 07:33:18 +03:00
secret := []byte("usememos")
2022-05-02 04:40:25 +03:00
if profile.Mode == "prod" {
2022-03-29 02:30:29 +03:00
secret = securecookie.GenerateRandomKey(16)
}
e.Use(session.Middleware(sessions.NewCookieStore(secret)))
2022-02-03 10:32:03 +03:00
s := &Server{
2022-03-29 15:53:43 +03:00
e: e,
Profile: profile,
2022-02-03 10:32:03 +03:00
}
2022-03-29 15:53:43 +03:00
// Webhooks api skips auth checker.
2022-02-03 10:32:03 +03:00
webhookGroup := e.Group("/h")
s.registerWebhookRoutes(webhookGroup)
apiGroup := e.Group("/api")
apiGroup.Use(func(next echo.HandlerFunc) echo.HandlerFunc {
2022-05-16 02:37:23 +03:00
return BasicAuthMiddleware(s, next)
2022-02-03 10:32:03 +03:00
})
2022-03-29 15:53:43 +03:00
s.registerSystemRoutes(apiGroup)
2022-02-03 10:32:03 +03:00
s.registerAuthRoutes(apiGroup)
s.registerUserRoutes(apiGroup)
s.registerMemoRoutes(apiGroup)
s.registerShortcutRoutes(apiGroup)
s.registerResourceRoutes(apiGroup)
2022-06-21 16:58:33 +03:00
s.registerTagRoutes(apiGroup)
2022-02-03 10:32:03 +03:00
return s
}
func (server *Server) Run() error {
2022-03-29 15:53:43 +03:00
return server.e.Start(fmt.Sprintf(":%d", server.Profile.Port))
2022-02-03 10:32:03 +03:00
}