feat: register workspace user api

2025-06-14 20:07:50 +00:00 · 2022-09-21 23:26:05 +08:00
parent 3364d1bf39
commit c0699f159e
6 changed files with 197 additions and 21 deletions
--- a/server/server.go
+++ b/server/server.go
@ -64,6 +64,7 @@ func NewServer(profile *profile.Profile) *Server {
 	s.registerAuthRoutes(apiGroup)
 	s.registerUserRoutes(apiGroup)
 	s.registerWorkspaceRoutes(apiGroup)
+	s.registerWorkspaceUserRoutes(apiGroup)
 	s.registerShortcutRoutes(apiGroup)

 	return s
--- a/server/workspace_user.go
+++ b/server/workspace_user.go
@ -0,0 +1,159 @@
+package server
+
+import (
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"strconv"
+
+	"github.com/boojack/corgi/api"
+	"github.com/boojack/corgi/common"
+
+	"github.com/labstack/echo/v4"
+)
+
+func (s *Server) registerWorkspaceUserRoutes(g *echo.Group) {
+	g.POST("/workspace/:id/user", func(c echo.Context) error {
+		ctx := c.Request().Context()
+		userID, ok := c.Get(getUserIDContextKey()).(int)
+		if !ok {
+			return echo.NewHTTPError(http.StatusUnauthorized, "Missing auth session")
+		}
+
+		workspaceID, err := strconv.Atoi(c.Param("id"))
+		if err != nil {
+			return echo.NewHTTPError(http.StatusBadRequest, "Malformatted workspace id").SetInternal(err)
+		}
+
+		currentWorkspaceUser, err := s.Store.FindWordspaceUser(ctx, &api.WorkspaceUserFind{
+			WorkspaceID: &workspaceID,
+			UserID:      &userID,
+		})
+		if err != nil {
+			return echo.NewHTTPError(http.StatusInternalServerError, "Failed to find workspace user").SetInternal(err)
+		}
+		if currentWorkspaceUser.Role != api.RoleAdmin {
+			return echo.NewHTTPError(http.StatusForbidden, "Access forbidden to add workspace user").SetInternal(err)
+		}
+
+		workspaceUserUpsert := &api.WorkspaceUserUpsert{
+			WorkspaceID: workspaceID,
+		}
+		if err := json.NewDecoder(c.Request().Body).Decode(workspaceUserUpsert); err != nil {
+			return echo.NewHTTPError(http.StatusBadRequest, "Malformatted post workspace user request").SetInternal(err)
+		}
+
+		workspaceUser, err := s.Store.UpsertWorkspaceUser(ctx, workspaceUserUpsert)
+		if err != nil {
+			return echo.NewHTTPError(http.StatusInternalServerError, "Failed to upsert workspace user").SetInternal(err)
+		}
+		if err := s.Store.ComposeWorkspaceUser(ctx, workspaceUser); err != nil {
+			return echo.NewHTTPError(http.StatusInternalServerError, "Failed to compose workspace user").SetInternal(err)
+		}
+
+		c.Response().Header().Set(echo.HeaderContentType, echo.MIMEApplicationJSONCharsetUTF8)
+		if err := json.NewEncoder(c.Response().Writer).Encode(composeResponse(workspaceUser)); err != nil {
+			return echo.NewHTTPError(http.StatusInternalServerError, "Failed to encode workspace user response").SetInternal(err)
+		}
+		return nil
+	})
+
+	g.GET("/workspace/:id/user", func(c echo.Context) error {
+		ctx := c.Request().Context()
+		workspaceID, err := strconv.Atoi(c.Param("id"))
+		if err != nil {
+			return echo.NewHTTPError(http.StatusBadRequest, "Malformatted workspace id").SetInternal(err)
+		}
+
+		workspaceUserList, err := s.Store.FindWordspaceUserList(ctx, &api.WorkspaceUserFind{
+			WorkspaceID: &workspaceID,
+		})
+		if err != nil {
+			return echo.NewHTTPError(http.StatusInternalServerError, "Failed to find workspace user list").SetInternal(err)
+		}
+
+		for _, workspaceUser := range workspaceUserList {
+			if err := s.Store.ComposeWorkspaceUser(ctx, workspaceUser); err != nil {
+				return echo.NewHTTPError(http.StatusInternalServerError, "Failed to compose workspace user").SetInternal(err)
+			}
+		}
+
+		c.Response().Header().Set(echo.HeaderContentType, echo.MIMEApplicationJSONCharsetUTF8)
+		if err := json.NewEncoder(c.Response().Writer).Encode(composeResponse(workspaceUserList)); err != nil {
+			return echo.NewHTTPError(http.StatusInternalServerError, "Failed to encode workspace user list response").SetInternal(err)
+		}
+		return nil
+	})
+
+	g.GET("/workspace/:workspaceId/user/:userId", func(c echo.Context) error {
+		ctx := c.Request().Context()
+		workspaceID, err := strconv.Atoi(c.Param("workspaceId"))
+		if err != nil {
+			return echo.NewHTTPError(http.StatusBadRequest, "Malformatted workspace id").SetInternal(err)
+		}
+		userID, err := strconv.Atoi(c.Param("userId"))
+		if err != nil {
+			return echo.NewHTTPError(http.StatusBadRequest, "Malformatted user id").SetInternal(err)
+		}
+
+		workspaceUser, err := s.Store.FindWordspaceUser(ctx, &api.WorkspaceUserFind{
+			WorkspaceID: &workspaceID,
+			UserID:      &userID,
+		})
+		if err != nil {
+			return echo.NewHTTPError(http.StatusInternalServerError, "Failed to find workspace user").SetInternal(err)
+		}
+
+		if err := s.Store.ComposeWorkspaceUser(ctx, workspaceUser); err != nil {
+			return echo.NewHTTPError(http.StatusInternalServerError, "Failed to compose workspace user").SetInternal(err)
+		}
+
+		c.Response().Header().Set(echo.HeaderContentType, echo.MIMEApplicationJSONCharsetUTF8)
+		if err := json.NewEncoder(c.Response().Writer).Encode(composeResponse(workspaceUser)); err != nil {
+			return echo.NewHTTPError(http.StatusInternalServerError, "Failed to encode workspace user response").SetInternal(err)
+		}
+		return nil
+	})
+
+	g.DELETE("/workspace/:workspaceId/user/:userId", func(c echo.Context) error {
+		ctx := c.Request().Context()
+		userID, ok := c.Get(getUserIDContextKey()).(int)
+		if !ok {
+			return echo.NewHTTPError(http.StatusUnauthorized, "Missing auth session")
+		}
+
+		workspaceID, err := strconv.Atoi(c.Param("workspaceId"))
+		if err != nil {
+			return echo.NewHTTPError(http.StatusBadRequest, "Malformatted workspace id").SetInternal(err)
+		}
+
+		currentWorkspaceUser, err := s.Store.FindWordspaceUser(ctx, &api.WorkspaceUserFind{
+			WorkspaceID: &workspaceID,
+			UserID:      &userID,
+		})
+		if err != nil {
+			return echo.NewHTTPError(http.StatusInternalServerError, "Failed to find workspace user").SetInternal(err)
+		}
+		if currentWorkspaceUser.Role != api.RoleAdmin {
+			return echo.NewHTTPError(http.StatusForbidden, "Access forbidden to add workspace user").SetInternal(err)
+		}
+
+		userID, err = strconv.Atoi(c.Param("userId"))
+		if err != nil {
+			return echo.NewHTTPError(http.StatusBadRequest, "Malformatted user id").SetInternal(err)
+		}
+
+		workspaceUserDelete := &api.WorkspaceUserDelete{
+			WorkspaceID: workspaceID,
+			UserID:      userID,
+		}
+		if err := s.Store.DeleteWorkspaceUser(ctx, workspaceUserDelete); err != nil {
+			if common.ErrorCode(err) == common.NotFound {
+				return echo.NewHTTPError(http.StatusNotFound, fmt.Sprintf("Workspace user not found with workspace id %d and user id %d", workspaceID, userID))
+			}
+			return echo.NewHTTPError(http.StatusInternalServerError, "Failed to delete workspace user").SetInternal(err)
+		}
+
+		return c.JSON(http.StatusOK, true)
+	})
+}