123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218 |
- package v1
- import (
- "encoding/json"
- "fmt"
- "net/http"
- "regexp"
- "sort"
- "github.com/labstack/echo/v4"
- "golang.org/x/exp/slices"
- "github.com/usememos/memos/store"
- )
- 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.GET("/tag/suggestion", s.GetTagSuggestion)
- g.POST("/tag/delete", s.DeleteTag)
- }
- // 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"
- // @Router /api/v1/tag [GET]
- func (s *APIV1Service) GetTagList(c echo.Context) error {
- ctx := c.Request().Context()
- userID, ok := c.Get(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"
- // @Router /api/v1/tag [POST]
- func (s *APIV1Service) CreateTag(c echo.Context) error {
- ctx := c.Request().Context()
- userID, ok := c.Get(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)
- 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"
- // @Router /api/v1/tag/delete [POST]
- func (s *APIV1Service) DeleteTag(c echo.Context) error {
- ctx := c.Request().Context()
- userID, ok := c.Get(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"
- // @Router /api/v1/tag/suggestion [GET]
- func (s *APIV1Service) GetTagSuggestion(c echo.Context) error {
- ctx := c.Request().Context()
- userID, ok := c.Get(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 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
- }
|