mirror of
https://github.com/aykhans/slash-e.git
synced 2025-04-18 21:19:44 +00:00
feat: migrate workspace user to api v1
This commit is contained in:
parent
24d42694c8
commit
44ef82fb4a
75
api/user.go
75
api/user.go
@ -1,10 +1,25 @@
|
|||||||
package api
|
package api
|
||||||
|
|
||||||
import (
|
// Role is the type of a role.
|
||||||
"fmt"
|
type Role string
|
||||||
"net/mail"
|
|
||||||
|
const (
|
||||||
|
// RoleAdmin is the ADMIN role.
|
||||||
|
RoleAdmin Role = "ADMIN"
|
||||||
|
// RoleUser is the USER role.
|
||||||
|
RoleUser Role = "USER"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
func (e Role) String() string {
|
||||||
|
switch e {
|
||||||
|
case RoleAdmin:
|
||||||
|
return "ADMIN"
|
||||||
|
case RoleUser:
|
||||||
|
return "USER"
|
||||||
|
}
|
||||||
|
return "USER"
|
||||||
|
}
|
||||||
|
|
||||||
type User struct {
|
type User struct {
|
||||||
ID int `json:"id"`
|
ID int `json:"id"`
|
||||||
|
|
||||||
@ -29,57 +44,3 @@ type UserCreate struct {
|
|||||||
OpenID string `json:"-"`
|
OpenID string `json:"-"`
|
||||||
Role Role `json:"-"`
|
Role Role `json:"-"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func (create UserCreate) Validate() error {
|
|
||||||
if len(create.Email) < 3 {
|
|
||||||
return fmt.Errorf("email is too short, minimum length is 6")
|
|
||||||
}
|
|
||||||
if !validateEmail(create.Email) {
|
|
||||||
return fmt.Errorf("invalid email format")
|
|
||||||
}
|
|
||||||
if len(create.Password) < 3 {
|
|
||||||
return fmt.Errorf("password is too short, minimum length is 6")
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
type UserPatch struct {
|
|
||||||
ID int
|
|
||||||
|
|
||||||
// Standard fields
|
|
||||||
RowStatus *RowStatus `json:"rowStatus"`
|
|
||||||
|
|
||||||
// Domain specific fields
|
|
||||||
Email *string `json:"email"`
|
|
||||||
DisplayName *string `json:"displayName"`
|
|
||||||
Password *string `json:"password"`
|
|
||||||
ResetOpenID *bool `json:"resetOpenId"`
|
|
||||||
PasswordHash *string `json:"-"`
|
|
||||||
OpenID *string `json:"-"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type UserFind struct {
|
|
||||||
ID *int `json:"id"`
|
|
||||||
|
|
||||||
// Standard fields
|
|
||||||
RowStatus *RowStatus `json:"rowStatus"`
|
|
||||||
|
|
||||||
// Domain specific fields
|
|
||||||
Email *string `json:"email"`
|
|
||||||
DisplayName *string `json:"displayName"`
|
|
||||||
OpenID *string `json:"openId"`
|
|
||||||
Role *Role `json:"-"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type UserDelete struct {
|
|
||||||
ID int
|
|
||||||
}
|
|
||||||
|
|
||||||
// validateEmail validates the email.
|
|
||||||
func validateEmail(email string) bool {
|
|
||||||
if _, err := mail.ParseAddress(email); err != nil {
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
@ -24,4 +24,5 @@ func (s *APIV1Service) Start(apiV1Group *echo.Group, secret string) {
|
|||||||
s.registerAuthRoutes(apiV1Group, secret)
|
s.registerAuthRoutes(apiV1Group, secret)
|
||||||
s.registerUserRoutes(apiV1Group)
|
s.registerUserRoutes(apiV1Group)
|
||||||
s.registerWorkspaceRoutes(apiV1Group)
|
s.registerWorkspaceRoutes(apiV1Group)
|
||||||
|
s.registerWorkspaceUserRoutes(apiV1Group)
|
||||||
}
|
}
|
||||||
|
@ -1,18 +1,34 @@
|
|||||||
package server
|
package v1
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"context"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net/http"
|
"net/http"
|
||||||
"strconv"
|
"strconv"
|
||||||
|
|
||||||
"github.com/boojack/shortify/api"
|
"github.com/boojack/shortify/store"
|
||||||
"github.com/boojack/shortify/internal/errorutil"
|
|
||||||
|
|
||||||
"github.com/labstack/echo/v4"
|
"github.com/labstack/echo/v4"
|
||||||
)
|
)
|
||||||
|
|
||||||
func (s *Server) registerWorkspaceUserRoutes(g *echo.Group) {
|
type WorkspaceUser struct {
|
||||||
|
WorkspaceID int `json:"workspaceId"`
|
||||||
|
UserID int `json:"userId"`
|
||||||
|
Role Role `json:"role"`
|
||||||
|
|
||||||
|
// Related fields
|
||||||
|
Username string `json:"username"`
|
||||||
|
Nickname string `json:"nickname"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type UpsertWorkspaceUserRequest struct {
|
||||||
|
WorkspaceID int `json:"workspaceId"`
|
||||||
|
UserID int `json:"userId"`
|
||||||
|
Role Role `json:"role"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *APIV1Service) registerWorkspaceUserRoutes(g *echo.Group) {
|
||||||
g.POST("/workspace/:id/user", func(c echo.Context) error {
|
g.POST("/workspace/:id/user", func(c echo.Context) error {
|
||||||
ctx := c.Request().Context()
|
ctx := c.Request().Context()
|
||||||
userID, ok := c.Get(getUserIDContextKey()).(int)
|
userID, ok := c.Get(getUserIDContextKey()).(int)
|
||||||
@ -25,33 +41,39 @@ func (s *Server) registerWorkspaceUserRoutes(g *echo.Group) {
|
|||||||
return echo.NewHTTPError(http.StatusBadRequest, "Malformatted workspace id").SetInternal(err)
|
return echo.NewHTTPError(http.StatusBadRequest, "Malformatted workspace id").SetInternal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
currentWorkspaceUser, err := s.Store.FindWordspaceUser(ctx, &api.WorkspaceUserFind{
|
currentWorkspaceUser, err := s.Store.GetWorkspaceUser(ctx, &store.FindWorkspaceUser{
|
||||||
WorkspaceID: &workspaceID,
|
WorkspaceID: &workspaceID,
|
||||||
UserID: &userID,
|
UserID: &userID,
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to find workspace user").SetInternal(err)
|
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to find workspace user").SetInternal(err)
|
||||||
}
|
}
|
||||||
if currentWorkspaceUser.Role != api.RoleAdmin {
|
if currentWorkspaceUser.Role != store.RoleAdmin {
|
||||||
return echo.NewHTTPError(http.StatusForbidden, "Access forbidden to add workspace user").SetInternal(err)
|
return echo.NewHTTPError(http.StatusForbidden, "Access forbidden to add workspace user").SetInternal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
workspaceUserUpsert := &api.WorkspaceUserUpsert{
|
upsert := &UpsertWorkspaceUserRequest{
|
||||||
WorkspaceID: workspaceID,
|
WorkspaceID: workspaceID,
|
||||||
}
|
}
|
||||||
if err := json.NewDecoder(c.Request().Body).Decode(workspaceUserUpsert); err != nil {
|
if err := json.NewDecoder(c.Request().Body).Decode(upsert); err != nil {
|
||||||
return echo.NewHTTPError(http.StatusBadRequest, "Malformatted post workspace user request").SetInternal(err)
|
return echo.NewHTTPError(http.StatusBadRequest, "Malformatted post workspace user request").SetInternal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
workspaceUser, err := s.Store.UpsertWorkspaceUser(ctx, workspaceUserUpsert)
|
workspaceUser, err := s.Store.UpsertWorkspaceUserV1(ctx, &store.WorkspaceUser{
|
||||||
|
WorkspaceID: upsert.WorkspaceID,
|
||||||
|
UserID: upsert.UserID,
|
||||||
|
Role: convertRoleToStore(upsert.Role),
|
||||||
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to upsert workspace user").SetInternal(err)
|
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to upsert workspace user").SetInternal(err)
|
||||||
}
|
}
|
||||||
if err := s.Store.ComposeWorkspaceUser(ctx, workspaceUser); err != nil {
|
|
||||||
|
composedWorkspaceUser, err := s.composeWorkspaceUser(ctx, workspaceUser)
|
||||||
|
if err != nil {
|
||||||
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to compose workspace user").SetInternal(err)
|
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to compose workspace user").SetInternal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return c.JSON(http.StatusOK, composeResponse(workspaceUser))
|
return c.JSON(http.StatusOK, composedWorkspaceUser)
|
||||||
})
|
})
|
||||||
|
|
||||||
g.GET("/workspace/:id/user", func(c echo.Context) error {
|
g.GET("/workspace/:id/user", func(c echo.Context) error {
|
||||||
@ -61,20 +83,23 @@ func (s *Server) registerWorkspaceUserRoutes(g *echo.Group) {
|
|||||||
return echo.NewHTTPError(http.StatusBadRequest, "Malformatted workspace id").SetInternal(err)
|
return echo.NewHTTPError(http.StatusBadRequest, "Malformatted workspace id").SetInternal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
workspaceUserList, err := s.Store.FindWordspaceUserList(ctx, &api.WorkspaceUserFind{
|
workspaceUserList, err := s.Store.ListWorkspaceUsers(ctx, &store.FindWorkspaceUser{
|
||||||
WorkspaceID: &workspaceID,
|
WorkspaceID: &workspaceID,
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to find workspace user list").SetInternal(err)
|
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to find workspace user list").SetInternal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
composedList := make([]*WorkspaceUser, 0, len(workspaceUserList))
|
||||||
for _, workspaceUser := range workspaceUserList {
|
for _, workspaceUser := range workspaceUserList {
|
||||||
if err := s.Store.ComposeWorkspaceUser(ctx, workspaceUser); err != nil {
|
composedWorkspaceUser, err := s.composeWorkspaceUser(ctx, workspaceUser)
|
||||||
|
if err != nil {
|
||||||
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to compose workspace user").SetInternal(err)
|
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to compose workspace user").SetInternal(err)
|
||||||
}
|
}
|
||||||
|
composedList = append(composedList, composedWorkspaceUser)
|
||||||
}
|
}
|
||||||
|
|
||||||
return c.JSON(http.StatusOK, composeResponse(workspaceUserList))
|
return c.JSON(http.StatusOK, composedList)
|
||||||
})
|
})
|
||||||
|
|
||||||
g.GET("/workspace/:workspaceId/user/:userId", func(c echo.Context) error {
|
g.GET("/workspace/:workspaceId/user/:userId", func(c echo.Context) error {
|
||||||
@ -88,7 +113,7 @@ func (s *Server) registerWorkspaceUserRoutes(g *echo.Group) {
|
|||||||
return echo.NewHTTPError(http.StatusBadRequest, "Malformatted user id").SetInternal(err)
|
return echo.NewHTTPError(http.StatusBadRequest, "Malformatted user id").SetInternal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
workspaceUser, err := s.Store.FindWordspaceUser(ctx, &api.WorkspaceUserFind{
|
workspaceUser, err := s.Store.GetWorkspaceUser(ctx, &store.FindWorkspaceUser{
|
||||||
WorkspaceID: &workspaceID,
|
WorkspaceID: &workspaceID,
|
||||||
UserID: &userID,
|
UserID: &userID,
|
||||||
})
|
})
|
||||||
@ -96,11 +121,12 @@ func (s *Server) registerWorkspaceUserRoutes(g *echo.Group) {
|
|||||||
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to find workspace user").SetInternal(err)
|
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to find workspace user").SetInternal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := s.Store.ComposeWorkspaceUser(ctx, workspaceUser); err != nil {
|
composedWorkspaceUser, err := s.composeWorkspaceUser(ctx, workspaceUser)
|
||||||
|
if err != nil {
|
||||||
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to compose workspace user").SetInternal(err)
|
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to compose workspace user").SetInternal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return c.JSON(http.StatusOK, composeResponse(workspaceUser))
|
return c.JSON(http.StatusOK, composedWorkspaceUser)
|
||||||
})
|
})
|
||||||
|
|
||||||
g.DELETE("/workspace/:workspaceId/user/:userId", func(c echo.Context) error {
|
g.DELETE("/workspace/:workspaceId/user/:userId", func(c echo.Context) error {
|
||||||
@ -120,36 +146,68 @@ func (s *Server) registerWorkspaceUserRoutes(g *echo.Group) {
|
|||||||
return echo.NewHTTPError(http.StatusBadRequest, "Malformatted user id").SetInternal(err)
|
return echo.NewHTTPError(http.StatusBadRequest, "Malformatted user id").SetInternal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
currentWorkspaceUser, err := s.Store.FindWordspaceUser(ctx, &api.WorkspaceUserFind{
|
currentWorkspaceUser, err := s.Store.GetWorkspaceUser(ctx, &store.FindWorkspaceUser{
|
||||||
WorkspaceID: &workspaceID,
|
WorkspaceID: &workspaceID,
|
||||||
UserID: ¤tUserID,
|
UserID: ¤tUserID,
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to find workspace user").SetInternal(err)
|
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to find workspace user").SetInternal(err)
|
||||||
}
|
}
|
||||||
if currentUserID != userID && currentWorkspaceUser.Role != api.RoleAdmin {
|
if currentUserID != userID && currentWorkspaceUser.Role != store.RoleAdmin {
|
||||||
return echo.NewHTTPError(http.StatusForbidden, "Access forbidden to delete workspace user").SetInternal(err)
|
return echo.NewHTTPError(http.StatusForbidden, "Access forbidden to delete workspace user").SetInternal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
workspaceUserDelete := &api.WorkspaceUserDelete{
|
if err := s.Store.DeleteWorkspaceUserV1(ctx, &store.DeleteWorkspaceUser{
|
||||||
WorkspaceID: workspaceID,
|
WorkspaceID: workspaceID,
|
||||||
UserID: userID,
|
UserID: userID,
|
||||||
}
|
}); err != nil {
|
||||||
if err := s.Store.DeleteWorkspaceUser(ctx, workspaceUserDelete); err != nil {
|
|
||||||
if errorutil.ErrorCode(err) == errorutil.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 echo.NewHTTPError(http.StatusInternalServerError, "Failed to delete workspace user").SetInternal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
shortcutDelete := &api.ShortcutDelete{
|
|
||||||
CreatorID: &userID,
|
|
||||||
WorkspaceID: &workspaceID,
|
|
||||||
}
|
|
||||||
if err := s.Store.DeleteShortcut(ctx, shortcutDelete); err != nil {
|
|
||||||
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to delete shortcut").SetInternal(err)
|
|
||||||
}
|
|
||||||
|
|
||||||
return c.JSON(http.StatusOK, true)
|
return c.JSON(http.StatusOK, true)
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func convertRoleToStore(role Role) store.Role {
|
||||||
|
switch role {
|
||||||
|
case RoleAdmin:
|
||||||
|
return store.RoleAdmin
|
||||||
|
case RoleUser:
|
||||||
|
return store.RoleUser
|
||||||
|
default:
|
||||||
|
return store.RoleUser
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func convertRoleFromStore(role store.Role) Role {
|
||||||
|
switch role {
|
||||||
|
case store.RoleAdmin:
|
||||||
|
return RoleAdmin
|
||||||
|
case store.RoleUser:
|
||||||
|
return RoleUser
|
||||||
|
default:
|
||||||
|
return RoleUser
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *APIV1Service) composeWorkspaceUser(ctx context.Context, workspaceUser *store.WorkspaceUser) (*WorkspaceUser, error) {
|
||||||
|
user, err := s.Store.GetUser(ctx, &store.FindUser{
|
||||||
|
ID: &workspaceUser.UserID,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if user == nil {
|
||||||
|
return nil, fmt.Errorf("Failed to find user %d", workspaceUser.UserID)
|
||||||
|
}
|
||||||
|
|
||||||
|
composedWorkspaceUser := &WorkspaceUser{
|
||||||
|
WorkspaceID: workspaceUser.WorkspaceID,
|
||||||
|
UserID: workspaceUser.UserID,
|
||||||
|
Role: convertRoleFromStore(workspaceUser.Role),
|
||||||
|
Username: user.Username,
|
||||||
|
Nickname: user.Nickname,
|
||||||
|
}
|
||||||
|
|
||||||
|
return composedWorkspaceUser, nil
|
||||||
|
}
|
@ -1,56 +0,0 @@
|
|||||||
package api
|
|
||||||
|
|
||||||
type Workspace struct {
|
|
||||||
ID int `json:"id"`
|
|
||||||
|
|
||||||
// Standard fields
|
|
||||||
CreatorID int `json:"creatorId"`
|
|
||||||
CreatedTs int64 `json:"createdTs"`
|
|
||||||
UpdatedTs int64 `json:"updatedTs"`
|
|
||||||
RowStatus RowStatus `json:"rowStatus"`
|
|
||||||
|
|
||||||
// Domain specific fields
|
|
||||||
Name string `json:"name"`
|
|
||||||
Title string `json:"title"`
|
|
||||||
Description string `json:"description"`
|
|
||||||
|
|
||||||
// Related fields
|
|
||||||
WorkspaceUserList []*WorkspaceUser `json:"workspaceUserList"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type WorkspaceCreate struct {
|
|
||||||
CreatorID int
|
|
||||||
|
|
||||||
Name string `json:"name"`
|
|
||||||
Title string `json:"title"`
|
|
||||||
Description string `json:"description"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type WorkspacePatch struct {
|
|
||||||
ID int
|
|
||||||
|
|
||||||
// Standard fields
|
|
||||||
RowStatus *RowStatus `json:"rowStatus"`
|
|
||||||
|
|
||||||
// Domain specific fields
|
|
||||||
Name *string `json:"name"`
|
|
||||||
Title *string `json:"title"`
|
|
||||||
Description *string `json:"description"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type WorkspaceFind struct {
|
|
||||||
ID *int `json:"id"`
|
|
||||||
|
|
||||||
// Standard fields
|
|
||||||
RowStatus *RowStatus `json:"rowStatus"`
|
|
||||||
|
|
||||||
// Domain specific fields
|
|
||||||
Name *string `json:"name"`
|
|
||||||
|
|
||||||
// Related fields
|
|
||||||
MemberID *int
|
|
||||||
}
|
|
||||||
|
|
||||||
type WorkspaceDelete struct {
|
|
||||||
ID int
|
|
||||||
}
|
|
@ -1,48 +0,0 @@
|
|||||||
package api
|
|
||||||
|
|
||||||
// Role is the type of a role.
|
|
||||||
type Role string
|
|
||||||
|
|
||||||
const (
|
|
||||||
// RoleAdmin is the ADMIN role.
|
|
||||||
RoleAdmin Role = "ADMIN"
|
|
||||||
// RoleUser is the USER role.
|
|
||||||
RoleUser Role = "USER"
|
|
||||||
)
|
|
||||||
|
|
||||||
func (e Role) String() string {
|
|
||||||
switch e {
|
|
||||||
case RoleAdmin:
|
|
||||||
return "ADMIN"
|
|
||||||
case RoleUser:
|
|
||||||
return "USER"
|
|
||||||
}
|
|
||||||
return "USER"
|
|
||||||
}
|
|
||||||
|
|
||||||
type WorkspaceUser struct {
|
|
||||||
WorkspaceID int `json:"workspaceId"`
|
|
||||||
UserID int `json:"userId"`
|
|
||||||
Role Role `json:"role"`
|
|
||||||
|
|
||||||
// Related fields
|
|
||||||
Email string `json:"email"`
|
|
||||||
DisplayName string `json:"displayName"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type WorkspaceUserUpsert struct {
|
|
||||||
WorkspaceID int `json:"workspaceId"`
|
|
||||||
UserID int `json:"userId"`
|
|
||||||
Role Role `json:"role"`
|
|
||||||
UpdatedTs *int64 `json:"updatedTs"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type WorkspaceUserFind struct {
|
|
||||||
WorkspaceID *int
|
|
||||||
UserID *int
|
|
||||||
}
|
|
||||||
|
|
||||||
type WorkspaceUserDelete struct {
|
|
||||||
WorkspaceID int
|
|
||||||
UserID int
|
|
||||||
}
|
|
@ -64,7 +64,6 @@ func NewServer(profile *profile.Profile, store *store.Store) (*Server, error) {
|
|||||||
apiGroup.Use(func(next echo.HandlerFunc) echo.HandlerFunc {
|
apiGroup.Use(func(next echo.HandlerFunc) echo.HandlerFunc {
|
||||||
return JWTMiddleware(s, next, string(secret))
|
return JWTMiddleware(s, next, string(secret))
|
||||||
})
|
})
|
||||||
s.registerWorkspaceUserRoutes(apiGroup)
|
|
||||||
s.registerShortcutRoutes(apiGroup)
|
s.registerShortcutRoutes(apiGroup)
|
||||||
|
|
||||||
// Register API v1 routes.
|
// Register API v1 routes.
|
||||||
|
@ -7,6 +7,7 @@ import (
|
|||||||
"strconv"
|
"strconv"
|
||||||
|
|
||||||
"github.com/boojack/shortify/api"
|
"github.com/boojack/shortify/api"
|
||||||
|
"github.com/boojack/shortify/store"
|
||||||
|
|
||||||
"github.com/labstack/echo/v4"
|
"github.com/labstack/echo/v4"
|
||||||
)
|
)
|
||||||
@ -66,7 +67,7 @@ func (s *Server) registerShortcutRoutes(g *echo.Group) {
|
|||||||
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to find shortcut").SetInternal(err)
|
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to find shortcut").SetInternal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
workspaceUser, err := s.Store.FindWordspaceUser(ctx, &api.WorkspaceUserFind{
|
workspaceUser, err := s.Store.GetWorkspaceUser(ctx, &store.FindWorkspaceUser{
|
||||||
UserID: &userID,
|
UserID: &userID,
|
||||||
WorkspaceID: &shortcut.WorkspaceID,
|
WorkspaceID: &shortcut.WorkspaceID,
|
||||||
})
|
})
|
||||||
@ -74,7 +75,7 @@ func (s *Server) registerShortcutRoutes(g *echo.Group) {
|
|||||||
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to find workspace user").SetInternal(err)
|
return echo.NewHTTPError(http.StatusInternalServerError, "Failed to find workspace user").SetInternal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if shortcut.CreatorID != userID && workspaceUser.Role != api.RoleAdmin {
|
if shortcut.CreatorID != userID && workspaceUser.Role != store.RoleAdmin {
|
||||||
return echo.NewHTTPError(http.StatusForbidden, "Forbidden to patch shortcut")
|
return echo.NewHTTPError(http.StatusForbidden, "Forbidden to patch shortcut")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -3,11 +3,7 @@ package store
|
|||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"database/sql"
|
"database/sql"
|
||||||
"fmt"
|
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
"github.com/boojack/shortify/api"
|
|
||||||
"github.com/boojack/shortify/internal/errorutil"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// Role is the type of a role.
|
// Role is the type of a role.
|
||||||
@ -174,220 +170,3 @@ func listWorkspaceUsers(ctx context.Context, tx *sql.Tx, find *FindWorkspaceUser
|
|||||||
|
|
||||||
return list, nil
|
return list, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// workspaceUserRaw is the store model for WorkspaceUser.
|
|
||||||
type workspaceUserRaw struct {
|
|
||||||
WorkspaceID int
|
|
||||||
UserID int
|
|
||||||
Role api.Role
|
|
||||||
}
|
|
||||||
|
|
||||||
func (raw *workspaceUserRaw) toWorkspaceUser() *api.WorkspaceUser {
|
|
||||||
return &api.WorkspaceUser{
|
|
||||||
WorkspaceID: raw.WorkspaceID,
|
|
||||||
UserID: raw.UserID,
|
|
||||||
Role: raw.Role,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Store) ComposeWorkspaceUser(ctx context.Context, workspaceUser *api.WorkspaceUser) error {
|
|
||||||
user, err := s.GetUser(ctx, &FindUser{
|
|
||||||
ID: &workspaceUser.UserID,
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
workspaceUser.Email = user.Email
|
|
||||||
workspaceUser.DisplayName = user.Nickname
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Store) ComposeWorkspaceUserListForWorkspace(ctx context.Context, workspace *api.Workspace) error {
|
|
||||||
workspaceUserList, err := s.FindWordspaceUserList(ctx, &api.WorkspaceUserFind{
|
|
||||||
WorkspaceID: &workspace.ID,
|
|
||||||
})
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, workspaceUser := range workspaceUserList {
|
|
||||||
if err := s.ComposeWorkspaceUser(ctx, workspaceUser); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
workspace.WorkspaceUserList = workspaceUserList
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Store) UpsertWorkspaceUser(ctx context.Context, upsert *api.WorkspaceUserUpsert) (*api.WorkspaceUser, error) {
|
|
||||||
tx, err := s.db.BeginTx(ctx, nil)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
defer tx.Rollback()
|
|
||||||
|
|
||||||
workspaceUserRaw, err := upsertWorkspaceUser(ctx, tx, upsert)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := tx.Commit(); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
workspaceUser := workspaceUserRaw.toWorkspaceUser()
|
|
||||||
|
|
||||||
return workspaceUser, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Store) FindWordspaceUserList(ctx context.Context, find *api.WorkspaceUserFind) ([]*api.WorkspaceUser, error) {
|
|
||||||
tx, err := s.db.BeginTx(ctx, nil)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
defer tx.Rollback()
|
|
||||||
|
|
||||||
workspaceUserRawList, err := findWorkspaceUserList(ctx, tx, find)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
list := []*api.WorkspaceUser{}
|
|
||||||
for _, raw := range workspaceUserRawList {
|
|
||||||
list = append(list, raw.toWorkspaceUser())
|
|
||||||
}
|
|
||||||
|
|
||||||
return list, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Store) FindWordspaceUser(ctx context.Context, find *api.WorkspaceUserFind) (*api.WorkspaceUser, error) {
|
|
||||||
tx, err := s.db.BeginTx(ctx, nil)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
defer tx.Rollback()
|
|
||||||
|
|
||||||
list, err := findWorkspaceUserList(ctx, tx, find)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(list) == 0 {
|
|
||||||
return nil, &errorutil.Error{Code: errorutil.NotFound, Err: fmt.Errorf("not found workspace user with filter %+v", find)}
|
|
||||||
} else if len(list) > 1 {
|
|
||||||
return nil, &errorutil.Error{Code: errorutil.Conflict, Err: fmt.Errorf("found %d workspaces user with filter %+v, expect 1", len(list), find)}
|
|
||||||
}
|
|
||||||
|
|
||||||
workspaceUser := list[0].toWorkspaceUser()
|
|
||||||
return workspaceUser, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (s *Store) DeleteWorkspaceUser(ctx context.Context, delete *api.WorkspaceUserDelete) error {
|
|
||||||
tx, err := s.db.BeginTx(ctx, nil)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
defer tx.Rollback()
|
|
||||||
|
|
||||||
err = deleteWorkspaceUser(ctx, tx, delete)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
err = tx.Commit()
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
func upsertWorkspaceUser(ctx context.Context, tx *sql.Tx, upsert *api.WorkspaceUserUpsert) (*workspaceUserRaw, error) {
|
|
||||||
set := []string{"workspace_id", "user_id", "role"}
|
|
||||||
args := []any{upsert.WorkspaceID, upsert.UserID, upsert.Role}
|
|
||||||
placeholder := []string{"?", "?", "?"}
|
|
||||||
|
|
||||||
if v := upsert.UpdatedTs; v != nil {
|
|
||||||
set, args, placeholder = append(set, "updated_ts"), append(args, *v), append(placeholder, "?")
|
|
||||||
}
|
|
||||||
|
|
||||||
query := `
|
|
||||||
INSERT INTO workspace_user (
|
|
||||||
` + strings.Join(set, ", ") + `
|
|
||||||
)
|
|
||||||
VALUES (` + strings.Join(placeholder, ",") + `)
|
|
||||||
ON CONFLICT(workspace_id, user_id) DO UPDATE
|
|
||||||
SET
|
|
||||||
role = EXCLUDED.role
|
|
||||||
RETURNING workspace_id, user_id, role
|
|
||||||
`
|
|
||||||
var workspaceUserRaw workspaceUserRaw
|
|
||||||
if err := tx.QueryRowContext(ctx, query, args...).Scan(
|
|
||||||
&workspaceUserRaw.WorkspaceID,
|
|
||||||
&workspaceUserRaw.UserID,
|
|
||||||
&workspaceUserRaw.Role,
|
|
||||||
); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return &workspaceUserRaw, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func findWorkspaceUserList(ctx context.Context, tx *sql.Tx, find *api.WorkspaceUserFind) ([]*workspaceUserRaw, error) {
|
|
||||||
where, args := []string{"1 = 1"}, []any{}
|
|
||||||
|
|
||||||
if v := find.WorkspaceID; v != nil {
|
|
||||||
where, args = append(where, "workspace_id = ?"), append(args, *v)
|
|
||||||
}
|
|
||||||
if v := find.UserID; v != nil {
|
|
||||||
where, args = append(where, "user_id = ?"), append(args, *v)
|
|
||||||
}
|
|
||||||
|
|
||||||
query := `
|
|
||||||
SELECT
|
|
||||||
workspace_id,
|
|
||||||
user_id,
|
|
||||||
role
|
|
||||||
FROM workspace_user
|
|
||||||
WHERE ` + strings.Join(where, " AND ")
|
|
||||||
rows, err := tx.QueryContext(ctx, query, args...)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
defer rows.Close()
|
|
||||||
|
|
||||||
workspaceUserRawList := make([]*workspaceUserRaw, 0)
|
|
||||||
for rows.Next() {
|
|
||||||
var workspaceUserRaw workspaceUserRaw
|
|
||||||
if err := rows.Scan(
|
|
||||||
&workspaceUserRaw.WorkspaceID,
|
|
||||||
&workspaceUserRaw.UserID,
|
|
||||||
&workspaceUserRaw.Role,
|
|
||||||
); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
workspaceUserRawList = append(workspaceUserRawList, &workspaceUserRaw)
|
|
||||||
}
|
|
||||||
|
|
||||||
if err := rows.Err(); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return workspaceUserRawList, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func deleteWorkspaceUser(ctx context.Context, tx *sql.Tx, delete *api.WorkspaceUserDelete) error {
|
|
||||||
result, err := tx.ExecContext(ctx, `
|
|
||||||
DELETE FROM workspace_user WHERE workspace_id = ? AND user_id = ?
|
|
||||||
`, delete.WorkspaceID, delete.UserID)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
rows, _ := result.RowsAffected()
|
|
||||||
if rows == 0 {
|
|
||||||
return &errorutil.Error{Code: errorutil.NotFound, Err: fmt.Errorf("workspace user not found")}
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user