refactor: backend

2025-06-05 22:09:59 +02:00 · 2022-02-03 15:32:03 +08:00
parent 3d8997a43a
commit d661134b03
42 changed files with 2601 additions and 1289 deletions
--- a/server/shortcut.go
+++ b/server/shortcut.go
@ -0,0 +1,122 @@
+package server
+
+import (
+	"fmt"
+	"memos/api"
+	"net/http"
+	"strconv"
+	"time"
+
+	"github.com/google/jsonapi"
+	"github.com/labstack/echo/v4"
+)
+
+func (s *Server) registerShortcutRoutes(g *echo.Group) {
+	g.POST("/shortcut", func(c echo.Context) error {
+		userId := c.Get(getUserIdContextKey()).(int)
+		shortcutCreate := &api.ShortcutCreate{
+			CreatorId: userId,
+		}
+		if err := jsonapi.UnmarshalPayload(c.Request().Body, shortcutCreate); err != nil {
+			return echo.NewHTTPError(http.StatusBadRequest, "Malformatted post shortcut request").SetInternal(err)
+		}
+
+		shortcut, err := s.ShortcutService.CreateShortcut(shortcutCreate)
+		if err != nil {
+			return echo.NewHTTPError(http.StatusInternalServerError, "Failed to create shortcut").SetInternal(err)
+		}
+
+		c.Response().Header().Set(echo.HeaderContentType, echo.MIMEApplicationJSONCharsetUTF8)
+		if err := jsonapi.MarshalPayload(c.Response().Writer, shortcut); err != nil {
+			return echo.NewHTTPError(http.StatusInternalServerError, "Failed to marshal shortcut response").SetInternal(err)
+		}
+
+		return nil
+	})
+	g.PATCH("/shortcut/:shortcutId", func(c echo.Context) error {
+		shortcutId, err := strconv.Atoi(c.Param("shortcutId"))
+		if err != nil {
+			return echo.NewHTTPError(http.StatusBadRequest, fmt.Sprintf("ID is not a number: %s", c.Param("shortcutId"))).SetInternal(err)
+		}
+
+		shortcutPatch := &api.ShortcutPatch{
+			Id: shortcutId,
+		}
+		if err := jsonapi.UnmarshalPayload(c.Request().Body, shortcutPatch); err != nil {
+			return echo.NewHTTPError(http.StatusBadRequest, "Malformatted patch shortcut request").SetInternal(err)
+		}
+
+		if shortcutPatch.Pinned != nil {
+			pinnedTs := int64(0)
+			if *shortcutPatch.Pinned {
+				pinnedTs = time.Now().Unix()
+			}
+			shortcutPatch.PinnedTs = &pinnedTs
+		}
+
+		shortcut, err := s.ShortcutService.PatchShortcut(shortcutPatch)
+		if err != nil {
+			return echo.NewHTTPError(http.StatusInternalServerError, "Failed to patch shortcut").SetInternal(err)
+		}
+
+		c.Response().Header().Set(echo.HeaderContentType, echo.MIMEApplicationJSONCharsetUTF8)
+		if err := jsonapi.MarshalPayload(c.Response().Writer, shortcut); err != nil {
+			return echo.NewHTTPError(http.StatusInternalServerError, "Failed to marshal shortcut response").SetInternal(err)
+		}
+
+		return nil
+	})
+	g.GET("/shortcut", func(c echo.Context) error {
+		userId := c.Get(getUserIdContextKey()).(int)
+		shortcutFind := &api.ShortcutFind{
+			CreatorId: &userId,
+		}
+		list, err := s.ShortcutService.FindShortcut(shortcutFind)
+		if err != nil {
+			return echo.NewHTTPError(http.StatusInternalServerError, "Failed to fetch shortcut list").SetInternal(err)
+		}
+
+		c.Response().Header().Set(echo.HeaderContentType, echo.MIMEApplicationJSONCharsetUTF8)
+		if err := jsonapi.MarshalPayload(c.Response().Writer, list); err != nil {
+			return echo.NewHTTPError(http.StatusInternalServerError, "Failed to marshal shortcut list response").SetInternal(err)
+		}
+
+		return nil
+	})
+	g.GET("/shortcut/:shortcutId", func(c echo.Context) error {
+		shortcutId, err := strconv.Atoi(c.Param("shortcutId"))
+		if err != nil {
+			return echo.NewHTTPError(http.StatusBadRequest, fmt.Sprintf("ID is not a number: %s", c.Param("shortcutId"))).SetInternal(err)
+		}
+
+		shortcutFind := &api.ShortcutFind{
+			Id: &shortcutId,
+		}
+		shortcut, err := s.ShortcutService.FindShortcut(shortcutFind)
+		if err != nil {
+			return echo.NewHTTPError(http.StatusInternalServerError, "Failed to fetch shortcut").SetInternal(err)
+		}
+
+		c.Response().Header().Set(echo.HeaderContentType, echo.MIMEApplicationJSONCharsetUTF8)
+		if err := jsonapi.MarshalPayload(c.Response().Writer, shortcut); err != nil {
+			return echo.NewHTTPError(http.StatusInternalServerError, "Failed to marshal shortcut response").SetInternal(err)
+		}
+
+		return nil
+	})
+	g.DELETE("/shortcut/:shortcutId", func(c echo.Context) error {
+		shortcutId, err := strconv.Atoi(c.Param("shortcutId"))
+		if err != nil {
+			return echo.NewHTTPError(http.StatusBadRequest, fmt.Sprintf("ID is not a number: %s", c.Param("shortcutId"))).SetInternal(err)
+		}
+
+		shortcutDelete := &api.ShortcutDelete{
+			Id: shortcutId,
+		}
+		if err := s.ShortcutService.DeleteShortcut(shortcutDelete); err != nil {
+			return echo.NewHTTPError(http.StatusInternalServerError, "Failed to delete shortcut").SetInternal(err)
+		}
+
+		return nil
+	})
+}