mirror of
https://github.com/usememos/memos.git
synced 2024-12-24 11:44:44 +03:00
4491c75135
* - Refactor several API routes from anonymous functions to regular definitions. Required to add parseable documentation comments. - Add API documentation comments using Swag Declarative Comments Format - Add echo-swagger to serve Swagger-UI at /api/index.html - Fix error response from extraneous parameter resourceId to relatedMemoId in DELETE("/memo/:memoId/relation/:relatedMemoId/type/:relationType") - Add an auto-generated ./docs/api/v1.md for quick reference on repo (generated by swagger-markdown) - Add auxiliary scripts to generate docs.go and swagger.yaml * fix: golangci-lint errors * fix: go fmt flag in swag scripts
248 lines
7.0 KiB
Go
248 lines
7.0 KiB
Go
package v1
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net/http"
|
|
"regexp"
|
|
"sort"
|
|
|
|
"github.com/labstack/echo/v4"
|
|
"github.com/pkg/errors"
|
|
"github.com/usememos/memos/api/auth"
|
|
"github.com/usememos/memos/store"
|
|
"golang.org/x/exp/slices"
|
|
)
|
|
|
|
type Tag struct {
|
|
Name string
|
|
CreatorID int32
|
|
}
|
|
|
|
type UpsertTagRequest struct {
|
|
Name string `json:"name"`
|
|
}
|
|
|
|
type DeleteTagRequest struct {
|
|
Name string `json:"name"`
|
|
}
|
|
|
|
func (s *APIV1Service) registerTagRoutes(g *echo.Group) {
|
|
g.GET("/tag", s.getTagList)
|
|
g.POST("/tag", s.createTag)
|
|
g.POST("/tag/delete", s.deleteTag)
|
|
g.GET("/tag/suggestion", s.getTagSuggestion)
|
|
}
|
|
|
|
// getTagList godoc
|
|
//
|
|
// @Summary Get a list of tags
|
|
// @Tags tag
|
|
// @Produce json
|
|
// @Success 200 {object} []string "Tag list"
|
|
// @Failure 400 {object} nil "Missing user id to find tag"
|
|
// @Failure 500 {object} nil "Failed to find tag list"
|
|
// @Security ApiKeyAuth
|
|
// @Router /api/v1/tag [GET]
|
|
func (s *APIV1Service) getTagList(c echo.Context) error {
|
|
ctx := c.Request().Context()
|
|
userID, ok := c.Get(auth.UserIDContextKey).(int32)
|
|
if !ok {
|
|
return echo.NewHTTPError(http.StatusBadRequest, "Missing user id to find tag")
|
|
}
|
|
|
|
list, err := s.Store.ListTags(ctx, &store.FindTag{
|
|
CreatorID: userID,
|
|
})
|
|
if err != nil {
|
|
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to find tag list").SetInternal(err)
|
|
}
|
|
|
|
tagNameList := []string{}
|
|
for _, tag := range list {
|
|
tagNameList = append(tagNameList, tag.Name)
|
|
}
|
|
return c.JSON(http.StatusOK, tagNameList)
|
|
}
|
|
|
|
// createTag godoc
|
|
//
|
|
// @Summary Create a tag
|
|
// @Tags tag
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param body body UpsertTagRequest true "Request object."
|
|
// @Success 200 {object} string "Created tag name"
|
|
// @Failure 400 {object} nil "Malformatted post tag request | Tag name shouldn't be empty"
|
|
// @Failure 401 {object} nil "Missing user in session"
|
|
// @Failure 500 {object} nil "Failed to upsert tag | Failed to create activity"
|
|
// @Security ApiKeyAuth
|
|
// @Router /api/v1/tag [POST]
|
|
func (s *APIV1Service) createTag(c echo.Context) error {
|
|
ctx := c.Request().Context()
|
|
userID, ok := c.Get(auth.UserIDContextKey).(int32)
|
|
if !ok {
|
|
return echo.NewHTTPError(http.StatusUnauthorized, "Missing user in session")
|
|
}
|
|
|
|
tagUpsert := &UpsertTagRequest{}
|
|
if err := json.NewDecoder(c.Request().Body).Decode(tagUpsert); err != nil {
|
|
return echo.NewHTTPError(http.StatusBadRequest, "Malformatted post tag request").SetInternal(err)
|
|
}
|
|
if tagUpsert.Name == "" {
|
|
return echo.NewHTTPError(http.StatusBadRequest, "Tag name shouldn't be empty")
|
|
}
|
|
|
|
tag, err := s.Store.UpsertTag(ctx, &store.Tag{
|
|
Name: tagUpsert.Name,
|
|
CreatorID: userID,
|
|
})
|
|
if err != nil {
|
|
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to upsert tag").SetInternal(err)
|
|
}
|
|
tagMessage := convertTagFromStore(tag)
|
|
if err := s.createTagCreateActivity(c, tagMessage); err != nil {
|
|
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to create activity").SetInternal(err)
|
|
}
|
|
return c.JSON(http.StatusOK, tagMessage.Name)
|
|
}
|
|
|
|
// deleteTag godoc
|
|
//
|
|
// @Summary Delete a tag
|
|
// @Tags tag
|
|
// @Accept json
|
|
// @Produce json
|
|
// @Param body body DeleteTagRequest true "Request object."
|
|
// @Success 200 {boolean} true "Tag deleted"
|
|
// @Failure 400 {object} nil "Malformatted post tag request | Tag name shouldn't be empty"
|
|
// @Failure 401 {object} nil "Missing user in session"
|
|
// @Failure 500 {object} nil "Failed to delete tag name: %v"
|
|
// @Security ApiKeyAuth
|
|
// @Router /api/v1/tag/delete [POST]
|
|
func (s *APIV1Service) deleteTag(c echo.Context) error {
|
|
ctx := c.Request().Context()
|
|
userID, ok := c.Get(auth.UserIDContextKey).(int32)
|
|
if !ok {
|
|
return echo.NewHTTPError(http.StatusUnauthorized, "Missing user in session")
|
|
}
|
|
|
|
tagDelete := &DeleteTagRequest{}
|
|
if err := json.NewDecoder(c.Request().Body).Decode(tagDelete); err != nil {
|
|
return echo.NewHTTPError(http.StatusBadRequest, "Malformatted post tag request").SetInternal(err)
|
|
}
|
|
if tagDelete.Name == "" {
|
|
return echo.NewHTTPError(http.StatusBadRequest, "Tag name shouldn't be empty")
|
|
}
|
|
|
|
err := s.Store.DeleteTag(ctx, &store.DeleteTag{
|
|
Name: tagDelete.Name,
|
|
CreatorID: userID,
|
|
})
|
|
if err != nil {
|
|
return echo.NewHTTPError(http.StatusInternalServerError, fmt.Sprintf("Failed to delete tag name: %v", tagDelete.Name)).SetInternal(err)
|
|
}
|
|
return c.JSON(http.StatusOK, true)
|
|
}
|
|
|
|
// getTagSuggestion godoc
|
|
//
|
|
// @Summary Get a list of tags suggested from other memos contents
|
|
// @Tags tag
|
|
// @Produce json
|
|
// @Success 200 {object} []string "Tag list"
|
|
// @Failure 400 {object} nil "Missing user session"
|
|
// @Failure 500 {object} nil "Failed to find memo list | Failed to find tag list"
|
|
// @Security ApiKeyAuth
|
|
// @Router /api/v1/tag/suggestion [GET]
|
|
func (s *APIV1Service) getTagSuggestion(c echo.Context) error {
|
|
ctx := c.Request().Context()
|
|
userID, ok := c.Get(auth.UserIDContextKey).(int32)
|
|
if !ok {
|
|
return echo.NewHTTPError(http.StatusBadRequest, "Missing user session")
|
|
}
|
|
normalRowStatus := store.Normal
|
|
memoFind := &store.FindMemo{
|
|
CreatorID: &userID,
|
|
ContentSearch: []string{"#"},
|
|
RowStatus: &normalRowStatus,
|
|
}
|
|
|
|
memoMessageList, err := s.Store.ListMemos(ctx, memoFind)
|
|
if err != nil {
|
|
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to find memo list").SetInternal(err)
|
|
}
|
|
|
|
list, err := s.Store.ListTags(ctx, &store.FindTag{
|
|
CreatorID: userID,
|
|
})
|
|
if err != nil {
|
|
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to find tag list").SetInternal(err)
|
|
}
|
|
tagNameList := []string{}
|
|
for _, tag := range list {
|
|
tagNameList = append(tagNameList, tag.Name)
|
|
}
|
|
|
|
tagMapSet := make(map[string]bool)
|
|
for _, memo := range memoMessageList {
|
|
for _, tag := range findTagListFromMemoContent(memo.Content) {
|
|
if !slices.Contains(tagNameList, tag) {
|
|
tagMapSet[tag] = true
|
|
}
|
|
}
|
|
}
|
|
tagList := []string{}
|
|
for tag := range tagMapSet {
|
|
tagList = append(tagList, tag)
|
|
}
|
|
sort.Strings(tagList)
|
|
return c.JSON(http.StatusOK, tagList)
|
|
}
|
|
|
|
func (s *APIV1Service) createTagCreateActivity(c echo.Context, tag *Tag) error {
|
|
ctx := c.Request().Context()
|
|
payload := ActivityTagCreatePayload{
|
|
TagName: tag.Name,
|
|
}
|
|
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: tag.CreatorID,
|
|
Type: ActivityTagCreate.String(),
|
|
Level: ActivityInfo.String(),
|
|
Payload: string(payloadBytes),
|
|
})
|
|
if err != nil || activity == nil {
|
|
return errors.Wrap(err, "failed to create activity")
|
|
}
|
|
return err
|
|
}
|
|
|
|
func convertTagFromStore(tag *store.Tag) *Tag {
|
|
return &Tag{
|
|
Name: tag.Name,
|
|
CreatorID: tag.CreatorID,
|
|
}
|
|
}
|
|
|
|
var tagRegexp = regexp.MustCompile(`#([^\s#,]+)`)
|
|
|
|
func findTagListFromMemoContent(memoContent string) []string {
|
|
tagMapSet := make(map[string]bool)
|
|
matches := tagRegexp.FindAllStringSubmatch(memoContent, -1)
|
|
for _, v := range matches {
|
|
tagName := v[1]
|
|
tagMapSet[tagName] = true
|
|
}
|
|
|
|
tagList := []string{}
|
|
for tag := range tagMapSet {
|
|
tagList = append(tagList, tag)
|
|
}
|
|
sort.Strings(tagList)
|
|
return tagList
|
|
}
|