mirror of
https://github.com/usememos/memos.git
synced 2024-09-21 23:05:59 +08:00
95 lines
2.9 KiB
Go
95 lines
2.9 KiB
Go
package v1
|
|
|
|
import (
|
|
"net/http"
|
|
"time"
|
|
|
|
"github.com/labstack/echo/v4"
|
|
"github.com/labstack/echo/v4/middleware"
|
|
|
|
"github.com/usememos/memos/api/resource"
|
|
"github.com/usememos/memos/plugin/telegram"
|
|
"github.com/usememos/memos/server/profile"
|
|
"github.com/usememos/memos/store"
|
|
)
|
|
|
|
type APIV1Service struct {
|
|
Secret string
|
|
Profile *profile.Profile
|
|
Store *store.Store
|
|
telegramBot *telegram.Bot
|
|
}
|
|
|
|
// @title memos API
|
|
// @version 1.0
|
|
// @description A privacy-first, lightweight note-taking service.
|
|
//
|
|
// @contact.name API Support
|
|
// @contact.url https://github.com/orgs/usememos/discussions
|
|
//
|
|
// @license.name MIT License
|
|
// @license.url https://github.com/usememos/memos/blob/main/LICENSE
|
|
//
|
|
// @BasePath /
|
|
//
|
|
// @externalDocs.url https://usememos.com/
|
|
// @externalDocs.description Find out more about Memos.
|
|
func NewAPIV1Service(secret string, profile *profile.Profile, store *store.Store, telegramBot *telegram.Bot) *APIV1Service {
|
|
return &APIV1Service{
|
|
Secret: secret,
|
|
Profile: profile,
|
|
Store: store,
|
|
telegramBot: telegramBot,
|
|
}
|
|
}
|
|
|
|
func (s *APIV1Service) Register(rootGroup *echo.Group) {
|
|
// Register RSS routes.
|
|
s.registerRSSRoutes(rootGroup)
|
|
|
|
// Register API v1 routes.
|
|
apiV1Group := rootGroup.Group("/api/v1")
|
|
apiV1Group.Use(middleware.RateLimiterWithConfig(middleware.RateLimiterConfig{
|
|
Store: middleware.NewRateLimiterMemoryStoreWithConfig(
|
|
middleware.RateLimiterMemoryStoreConfig{Rate: 30, Burst: 100, ExpiresIn: 3 * time.Minute},
|
|
),
|
|
IdentifierExtractor: func(ctx echo.Context) (string, error) {
|
|
id := ctx.RealIP()
|
|
return id, nil
|
|
},
|
|
ErrorHandler: func(context echo.Context, err error) error {
|
|
return context.JSON(http.StatusForbidden, nil)
|
|
},
|
|
DenyHandler: func(context echo.Context, identifier string, err error) error {
|
|
return context.JSON(http.StatusTooManyRequests, nil)
|
|
},
|
|
}))
|
|
apiV1Group.Use(func(next echo.HandlerFunc) echo.HandlerFunc {
|
|
return JWTMiddleware(s, next, s.Secret)
|
|
})
|
|
s.registerSystemRoutes(apiV1Group)
|
|
s.registerSystemSettingRoutes(apiV1Group)
|
|
s.registerAuthRoutes(apiV1Group)
|
|
s.registerIdentityProviderRoutes(apiV1Group)
|
|
s.registerUserRoutes(apiV1Group)
|
|
s.registerTagRoutes(apiV1Group)
|
|
s.registerStorageRoutes(apiV1Group)
|
|
s.registerResourceRoutes(apiV1Group)
|
|
s.registerMemoRoutes(apiV1Group)
|
|
s.registerMemoOrganizerRoutes(apiV1Group)
|
|
s.registerMemoRelationRoutes(apiV1Group)
|
|
|
|
// Register public routes.
|
|
publicGroup := rootGroup.Group("/o")
|
|
publicGroup.Use(func(next echo.HandlerFunc) echo.HandlerFunc {
|
|
return JWTMiddleware(s, next, s.Secret)
|
|
})
|
|
s.registerGetterPublicRoutes(publicGroup)
|
|
// Create and register resource public routes.
|
|
resourceService := resource.NewService(s.Profile, s.Store)
|
|
resourceService.RegisterResourcePublicRoutes(publicGroup)
|
|
|
|
// programmatically set API version same as the server version
|
|
SwaggerInfo.Version = s.Profile.Version
|
|
}
|