khairul169/vaulterm
1
0
Fork 0
mirror of https://github.com/khairul169/vaulterm.git synced 2026-09-15 17:03:30 +07:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: add team member role change & removal, add uptime server stats, etc

Browse Source
This commit is contained in:
khairul169
2024-11-14 16:28:44 +00:00
parent b574f83e74
commit 33dda374c7
25 changed files with 566 additions and 110 deletions
Download Patch File Download Diff File Expand all files Collapse all files
server/app/teams/members/repository.go
+39
View File
@@ -0,0 +1,39 @@
package members
import (
"gorm.io/gorm"
"gorm.io/gorm/clause"
"rul.sh/vaulterm/db"
"rul.sh/vaulterm/models"
"rul.sh/vaulterm/utils"
)
type TeamMembers struct {
db *gorm.DB
User *utils.UserContext
}
func NewRepository(r *TeamMembers) *TeamMembers {
if r == nil {
r = &TeamMembers{}
}
r.db = db.Get()
return r
}
func (r *TeamMembers) Add(data *models.TeamMembers) error {
ret := r.db.Clauses(clause.OnConflict{DoNothing: true}).Create(data)
return ret.Error
}
func (r *TeamMembers) SetRole(data *models.TeamMembers) error {
ret := r.db.
Where("team_id = ? AND user_id = ?", data.TeamID, data.UserID).
Updates(&models.TeamMembers{Role: data.Role})
return ret.Error
}
func (r *TeamMembers) Remove(data *models.TeamMembers) error {
ret := r.db.Delete(&models.TeamMembers{TeamID: data.TeamID, UserID: data.UserID})
return ret.Error
}
server/app/teams/members/router.go
+140
View File
@@ -0,0 +1,140 @@
package members
import (
"errors"
"github.com/gofiber/fiber/v2"
"rul.sh/vaulterm/app/teams"
"rul.sh/vaulterm/app/users"
"rul.sh/vaulterm/models"
"rul.sh/vaulterm/utils"
)
func Router(app fiber.Router) {
router := app.Group("/:id/members")
// router.Get("/", getAll)
router.Post("/", invite)
router.Put("/:userId/role", setRole)
router.Delete("/:userId", remove)
}
// func getAll(c *fiber.Ctx) error {
// user := utils.GetUser(c)
// repo := NewRepository(&TeamMembers{User: user})
// rows, err := repo.GetAll()
// if err != nil {
// return utils.ResponseError(c, err, 500)
// }
// return c.JSON(fiber.Map{"rows": rows})
// }
func invite(c *fiber.Ctx) error {
var body InviteSchema
if err := c.BodyParser(&body); err != nil {
return utils.ResponseError(c, err, 500)
}
user := utils.GetUser(c)
teamRepo := teams.NewRepository(&teams.Teams{User: user})
repo := NewRepository(&TeamMembers{User: user})
id := c.Params("id")
exist, _ := teamRepo.Exists(id)
if !exist {
return utils.ResponseError(c, errors.New("team not found"), 404)
}
if !user.TeamCanWrite(&id) {
return utils.ResponseError(c, errors.New("no access"), 403)
}
userRepo := users.NewRepository(&users.Users{User: user})
userData, _ := userRepo.Find(body.Username)
if userData.ID == "" {
return utils.ResponseError(c, errors.New("user not found"), 404)
}
err := repo.Add(&models.TeamMembers{TeamID: id, UserID: userData.ID, Role: body.Role})
if err != nil {
return utils.ResponseError(c, err, 500)
}
return c.JSON(true)
}
func setRole(c *fiber.Ctx) error {
var body PutRoleSchema
if err := c.BodyParser(&body); err != nil {
return utils.ResponseError(c, err, 500)
}
user := utils.GetUser(c)
teamRepo := teams.NewRepository(&teams.Teams{User: user})
repo := NewRepository(&TeamMembers{User: user})
id := c.Params("id")
userId := c.Params("userId")
exist, _ := teamRepo.Exists(id)
if !exist {
return utils.ResponseError(c, errors.New("team not found"), 404)
}
if !user.TeamCanWrite(&id) {
return utils.ResponseError(c, errors.New("no access"), 403)
}
userRepo := users.NewRepository(nil)
userData, _ := userRepo.Get(userId)
if userData == nil || userData.ID == "" {
return utils.ResponseError(c, errors.New("user not found"), 404)
}
if !userData.IsInTeam(&id) {
return utils.ResponseError(c, errors.New("user not in team"), 400)
}
err := repo.SetRole(&models.TeamMembers{TeamID: id, UserID: userData.ID, Role: body.Role})
if err != nil {
return utils.ResponseError(c, err, 500)
}
return c.JSON(true)
}
func remove(c *fiber.Ctx) error {
user := utils.GetUser(c)
teamRepo := teams.NewRepository(&teams.Teams{User: user})
repo := NewRepository(&TeamMembers{User: user})
id := c.Params("id")
userId := c.Params("userId")
exist, _ := teamRepo.Exists(id)
if !exist {
return utils.ResponseError(c, errors.New("team not found"), 404)
}
if !user.TeamCanWrite(&id) {
return utils.ResponseError(c, errors.New("no access"), 403)
}
userRepo := users.NewRepository(&users.Users{User: user})
userData, _ := userRepo.Get(userId)
if userData.ID == "" {
return utils.ResponseError(c, errors.New("user not found"), 404)
}
userRole := userData.GetTeamRole(&id)
if userRole == "" {
return utils.ResponseError(c, errors.New("user not in team"), 400)
}
if userRole == models.TeamRoleOwner {
return utils.ResponseError(c, errors.New("cannot remove owner"), 400)
}
err := repo.Remove(&models.TeamMembers{TeamID: id, UserID: userData.ID})
if err != nil {
return utils.ResponseError(c, err, 500)
}
return c.JSON(true)
}
server/app/teams/members/schema.go
+11
View File
@@ -0,0 +1,11 @@
package members
type InviteSchema struct {
Username string `json:"username"`
Role string `json:"role"`
}
type PutRoleSchema struct {
UserID string `json:"username"`
Role string `json:"role"`
}
server/app/teams/repository.go
-12
View File
@@ -2,7 +2,6 @@ package teams
import (
"gorm.io/gorm"
"gorm.io/gorm/clause"
"rul.sh/vaulterm/db"
"rul.sh/vaulterm/models"
"rul.sh/vaulterm/utils"
@@ -94,14 +93,3 @@ func (r *Teams) Delete(id string) error {
return nil
})
}
func (r *Teams) Invite(teamId string, userId string, role string) error {
ret := r.db.
Clauses(clause.OnConflict{DoNothing: true}).
Create(&models.TeamMembers{
TeamID: teamId,
UserID: userId,
Role: role,
})
return ret.Error
}
server/app/teams/router.go
+3 -34
View File
@@ -5,12 +5,11 @@ import (
"net/http"
"github.com/gofiber/fiber/v2"
"rul.sh/vaulterm/app/users"
"rul.sh/vaulterm/models"
"rul.sh/vaulterm/utils"
)
func Router(app fiber.Router) {
func Router(app fiber.Router) fiber.Router {
router := app.Group("/teams")
router.Get("/", getAll)
@@ -18,7 +17,8 @@ func Router(app fiber.Router) {
router.Post("/", create)
router.Put("/:id", update)
router.Delete("/:id", delete)
router.Post("/:id/invite", invite)
return router
}
func getAll(c *fiber.Ctx) error {
@@ -116,34 +116,3 @@ func delete(c *fiber.Ctx) error {
return c.JSON(true)
}
func invite(c *fiber.Ctx) error {
var body InviteTeamSchema
if err := c.BodyParser(&body); err != nil {
return utils.ResponseError(c, err, 500)
}
user := utils.GetUser(c)
repo := NewRepository(&Teams{User: user})
id := c.Params("id")
exist, _ := repo.Exists(id)
if !exist {
return utils.ResponseError(c, errors.New("team not found"), 404)
}
if !user.TeamCanWrite(&id) {
return utils.ResponseError(c, errors.New("no access"), 403)
}
userRepo := users.NewRepository(&users.Users{User: user})
userData, _ := userRepo.Find(body.Username)
if userData.ID == "" {
return utils.ResponseError(c, errors.New("user not found"), 404)
}
if err := repo.Invite(id, userData.ID, body.Role); err != nil {
return utils.ResponseError(c, err, 500)
}
return c.JSON(true)
}
server/app/teams/schema.go
-5
View File
@@ -9,8 +9,3 @@ type GetOptions struct {
ID string
WithMembers bool
}
type InviteTeamSchema struct {
Username string `json:"username"`
Role string `json:"role"`
}