memos/server/server.go

165 lines
3.9 KiB
Go
Raw Normal View History

2022-02-03 15:32:03 +08:00
package server
import (
2023-01-02 23:18:12 +08:00
"context"
2023-02-03 10:30:18 +08:00
"database/sql"
2023-01-02 23:18:12 +08:00
"encoding/json"
2022-02-03 15:32:03 +08:00
"fmt"
2022-02-04 18:54:24 +08:00
"time"
2022-02-03 15:32:03 +08:00
2023-01-02 23:18:12 +08:00
"github.com/pkg/errors"
"github.com/usememos/memos/api"
2022-06-27 22:09:06 +08:00
"github.com/usememos/memos/server/profile"
"github.com/usememos/memos/store"
"github.com/usememos/memos/store/db"
2022-06-27 22:09:06 +08:00
2022-02-03 15:32:03 +08:00
"github.com/labstack/echo/v4"
"github.com/labstack/echo/v4/middleware"
)
type Server struct {
2023-02-03 10:30:18 +08:00
e *echo.Echo
db *sql.DB
2022-02-03 15:32:03 +08:00
ID string
Profile *profile.Profile
Store *store.Store
2022-02-03 15:32:03 +08:00
}
func NewServer(ctx context.Context, profile *profile.Profile) (*Server, error) {
2022-02-03 15:32:03 +08:00
e := echo.New()
e.Debug = true
e.HideBanner = true
2022-05-20 22:48:36 +08:00
e.HidePort = true
2022-02-03 15:32:03 +08:00
db := db.NewDB(profile)
if err := db.Open(ctx); err != nil {
return nil, errors.Wrap(err, "cannot open db")
}
2023-02-03 10:30:18 +08:00
s := &Server{
e: e,
db: db.DBInstance,
Profile: profile,
}
storeInstance := store.New(db.DBInstance, profile)
s.Store = storeInstance
2022-02-04 18:54:24 +08:00
e.Use(middleware.LoggerWithConfig(middleware.LoggerConfig{
2022-08-19 00:45:02 +08:00
Format: `{"time":"${time_rfc3339}",` +
`"method":"${method}","uri":"${uri}",` +
`"status":${status},"error":"${error}"}` + "\n",
2022-02-04 18:54:24 +08:00
}))
e.Use(middleware.Gzip())
2022-07-30 14:52:37 +08:00
e.Use(middleware.CORS())
2023-01-07 10:51:34 +08:00
e.Use(middleware.SecureWithConfig(middleware.SecureConfig{
2023-02-12 17:29:23 +08:00
Skipper: defaultGetRequestSkipper,
2023-01-07 10:51:34 +08:00
XSSProtection: "1; mode=block",
ContentTypeNosniff: "nosniff",
XFrameOptions: "SAMEORIGIN",
HSTSPreloadEnabled: false,
}))
2022-02-04 18:54:24 +08:00
e.Use(middleware.TimeoutWithConfig(middleware.TimeoutConfig{
ErrorMessage: "Request timeout",
Timeout: 30 * time.Second,
}))
serverID, err := s.getSystemServerID(ctx)
if err != nil {
return nil, err
}
s.ID = serverID
2022-02-03 15:32:03 +08:00
embedFrontend(e)
secret := "usememos"
2022-05-02 09:40:25 +08:00
if profile.Mode == "prod" {
secret, err = s.getSystemSecretSessionName(ctx)
if err != nil {
return nil, err
}
2022-03-29 07:30:29 +08:00
}
rootGroup := e.Group("")
s.registerRSSRoutes(rootGroup)
2022-09-09 07:40:21 +08:00
publicGroup := e.Group("/o")
publicGroup.Use(func(next echo.HandlerFunc) echo.HandlerFunc {
return JWTMiddleware(s, next, secret)
})
2023-01-02 23:18:12 +08:00
registerGetterPublicRoutes(publicGroup)
s.registerResourcePublicRoutes(publicGroup)
2022-02-03 15:32:03 +08:00
apiGroup := e.Group("/api")
apiGroup.Use(func(next echo.HandlerFunc) echo.HandlerFunc {
return JWTMiddleware(s, next, secret)
2022-02-03 15:32:03 +08:00
})
2022-03-29 20:53:43 +08:00
s.registerSystemRoutes(apiGroup)
s.registerAuthRoutes(apiGroup, secret)
2022-02-03 15:32:03 +08:00
s.registerUserRoutes(apiGroup)
s.registerMemoRoutes(apiGroup)
2023-04-22 10:42:24 +08:00
s.registerMemoResourceRoutes(apiGroup)
2022-02-03 15:32:03 +08:00
s.registerShortcutRoutes(apiGroup)
s.registerResourceRoutes(apiGroup)
2022-06-21 21:58:33 +08:00
s.registerTagRoutes(apiGroup)
s.registerStorageRoutes(apiGroup)
s.registerIdentityProviderRoutes(apiGroup)
s.registerOpenAIRoutes(apiGroup)
s.registerMemoRelationRoutes(apiGroup)
2022-02-03 15:32:03 +08:00
return s, nil
2022-02-03 15:32:03 +08:00
}
func (s *Server) Start(ctx context.Context) error {
2023-01-02 23:18:12 +08:00
if err := s.createServerStartActivity(ctx); err != nil {
return errors.Wrap(err, "failed to create activity")
}
return s.e.Start(fmt.Sprintf(":%d", s.Profile.Port))
2023-01-01 21:32:17 +08:00
}
2023-01-02 23:18:12 +08:00
2023-02-03 10:30:18 +08:00
func (s *Server) Shutdown(ctx context.Context) {
ctx, cancel := context.WithTimeout(ctx, 10*time.Second)
defer cancel()
// Shutdown echo server
if err := s.e.Shutdown(ctx); err != nil {
fmt.Printf("failed to shutdown server, error: %v\n", err)
}
// Close database connection
if err := s.db.Close(); err != nil {
fmt.Printf("failed to close database, error: %v\n", err)
}
fmt.Printf("memos stopped properly\n")
}
func (s *Server) GetEcho() *echo.Echo {
return s.e
}
2023-01-02 23:18:12 +08:00
func (s *Server) createServerStartActivity(ctx context.Context) error {
payload := api.ActivityServerStartPayload{
2023-01-03 20:05:37 +08:00
ServerID: s.ID,
Profile: s.Profile,
2023-01-02 23:18:12 +08:00
}
payloadBytes, err := json.Marshal(payload)
2023-01-02 23:18:12 +08:00
if err != nil {
return errors.Wrap(err, "failed to marshal activity payload")
}
2023-01-05 20:56:50 +08:00
activity, err := s.Store.CreateActivity(ctx, &api.ActivityCreate{
2023-01-02 23:18:12 +08:00
CreatorID: api.UnknownID,
Type: api.ActivityServerStart,
Level: api.ActivityInfo,
Payload: string(payloadBytes),
2023-01-02 23:18:12 +08:00
})
2023-01-07 11:49:58 +08:00
if err != nil || activity == nil {
return errors.Wrap(err, "failed to create activity")
}
2023-01-02 23:18:12 +08:00
return err
}