123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198 |
- package server
- import (
- "context"
- "encoding/json"
- "fmt"
- "net/http"
- "time"
- "github.com/google/uuid"
- "github.com/pkg/errors"
- apiv1 "github.com/usememos/memos/api/v1"
- "github.com/usememos/memos/common/util"
- "github.com/usememos/memos/plugin/telegram"
- "github.com/usememos/memos/server/profile"
- "github.com/usememos/memos/store"
- "github.com/labstack/echo/v4"
- "github.com/labstack/echo/v4/middleware"
- )
- type Server struct {
- e *echo.Echo
- ID string
- Secret string
- Profile *profile.Profile
- Store *store.Store
- telegramBot *telegram.Bot
- }
- func NewServer(ctx context.Context, profile *profile.Profile, store *store.Store) (*Server, error) {
- e := echo.New()
- e.Debug = true
- e.HideBanner = true
- e.HidePort = true
- s := &Server{
- e: e,
- Store: store,
- Profile: profile,
- }
- telegramBotHandler := newTelegramHandler(store)
- s.telegramBot = telegram.NewBotWithHandler(telegramBotHandler)
- e.Use(middleware.LoggerWithConfig(middleware.LoggerConfig{
- Format: `{"time":"${time_rfc3339}",` +
- `"method":"${method}","uri":"${uri}",` +
- `"status":${status},"error":"${error}"}` + "\n",
- }))
- e.Use(middleware.Gzip())
- e.Use(middleware.CORS())
- e.Use(middleware.SecureWithConfig(middleware.SecureConfig{
- Skipper: defaultGetRequestSkipper,
- XSSProtection: "1; mode=block",
- ContentTypeNosniff: "nosniff",
- XFrameOptions: "SAMEORIGIN",
- HSTSPreloadEnabled: false,
- }))
- e.Use(middleware.TimeoutWithConfig(middleware.TimeoutConfig{
- Skipper: func(c echo.Context) bool {
- // this is a hack to skip timeout for openai chat streaming
- // because streaming require to flush response. But the timeout middleware will break it.
- return c.Request().URL.Path == "/api/v1/openai/chat-streaming"
- },
- ErrorMessage: "Request timeout",
- Timeout: 30 * time.Second,
- }))
- serverID, err := s.getSystemServerID(ctx)
- if err != nil {
- return nil, fmt.Errorf("failed to retrieve system server ID: %w", err)
- }
- s.ID = serverID
- embedFrontend(e)
- secret := "usememos"
- if profile.Mode == "prod" {
- secret, err = s.getSystemSecretSessionName(ctx)
- if err != nil {
- return nil, fmt.Errorf("failed to retrieve system secret session name: %w", err)
- }
- }
- s.Secret = secret
- rootGroup := e.Group("")
- apiV1Service := apiv1.NewAPIV1Service(s.Secret, profile, store)
- apiV1Service.Register(rootGroup)
- return s, nil
- }
- func (s *Server) Start(ctx context.Context) error {
- if err := s.createServerStartActivity(ctx); err != nil {
- return errors.Wrap(err, "failed to create activity")
- }
- go s.telegramBot.Start(ctx)
- go autoBackup(ctx, s.Store)
- return s.e.Start(fmt.Sprintf(":%d", s.Profile.Port))
- }
- 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.Store.GetDB().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
- }
- func (s *Server) getSystemServerID(ctx context.Context) (string, error) {
- serverIDSetting, err := s.Store.GetSystemSetting(ctx, &store.FindSystemSetting{
- Name: apiv1.SystemSettingServerIDName.String(),
- })
- if err != nil {
- return "", err
- }
- if serverIDSetting == nil || serverIDSetting.Value == "" {
- serverIDSetting, err = s.Store.UpsertSystemSetting(ctx, &store.SystemSetting{
- Name: apiv1.SystemSettingServerIDName.String(),
- Value: uuid.NewString(),
- })
- if err != nil {
- return "", err
- }
- }
- return serverIDSetting.Value, nil
- }
- func (s *Server) getSystemSecretSessionName(ctx context.Context) (string, error) {
- secretSessionNameValue, err := s.Store.GetSystemSetting(ctx, &store.FindSystemSetting{
- Name: apiv1.SystemSettingSecretSessionName.String(),
- })
- if err != nil {
- return "", err
- }
- if secretSessionNameValue == nil || secretSessionNameValue.Value == "" {
- secretSessionNameValue, err = s.Store.UpsertSystemSetting(ctx, &store.SystemSetting{
- Name: apiv1.SystemSettingSecretSessionName.String(),
- Value: uuid.NewString(),
- })
- if err != nil {
- return "", err
- }
- }
- return secretSessionNameValue.Value, nil
- }
- func (s *Server) createServerStartActivity(ctx context.Context) error {
- payload := apiv1.ActivityServerStartPayload{
- ServerID: s.ID,
- Profile: s.Profile,
- }
- payloadBytes, err := json.Marshal(payload)
- if err != nil {
- return errors.Wrap(err, "failed to marshal activity payload")
- }
- activity, err := s.Store.CreateActivity(ctx, &store.Activity{
- CreatorID: apiv1.UnknownID,
- Type: apiv1.ActivityServerStart.String(),
- Level: apiv1.ActivityInfo.String(),
- Payload: string(payloadBytes),
- })
- if err != nil || activity == nil {
- return errors.Wrap(err, "failed to create activity")
- }
- return err
- }
- func defaultGetRequestSkipper(c echo.Context) bool {
- return c.Request().Method == http.MethodGet
- }
- func defaultAPIRequestSkipper(c echo.Context) bool {
- path := c.Path()
- return util.HasPrefixes(path, "/api", "/api/v1")
- }
|