add workspaces to user

go.mod

@@ -3,7 +3,7 @@ module gitea.tecamino.com/paadi/access-handler
 go 1.24.5
 
 require (
-	gitea.tecamino.com/paadi/dbHandler v1.1.9
+	gitea.tecamino.com/paadi/dbHandler v1.1.10
 	gitea.tecamino.com/paadi/tecamino-logger v0.2.1
 	github.com/gin-gonic/gin v1.11.0
 	github.com/go-playground/assert/v2 v2.2.0

go.sum

@@ -1,5 +1,5 @@
-gitea.tecamino.com/paadi/dbHandler v1.1.9 h1:B0sXVMYfjI1CQB2tX/aF+NoMeC0vlGhKs9MEYWf3oe8=
-gitea.tecamino.com/paadi/dbHandler v1.1.9/go.mod h1:y/xn/POJg1DO++67uKvnO23lJQgh+XFQq7HZCS9Getw=
+gitea.tecamino.com/paadi/dbHandler v1.1.10 h1:zZQbDTJ0bu6CIW90Zms8yYIzTLHtWPNhVKRxLUXEDuE=
+gitea.tecamino.com/paadi/dbHandler v1.1.10/go.mod h1:y/xn/POJg1DO++67uKvnO23lJQgh+XFQq7HZCS9Getw=
 gitea.tecamino.com/paadi/tecamino-logger v0.2.1 h1:sQTBKYPdzn9mmWX2JXZBtGBvNQH7cuXIwsl4TD0aMgE=
 gitea.tecamino.com/paadi/tecamino-logger v0.2.1/go.mod h1:FkzRTldUBBOd/iy2upycArDftSZ5trbsX5Ira5OzJgM=
 github.com/bytedance/sonic v1.14.0 h1:/OfKt8HFw0kh2rj8N0F6C/qPGRESq0BbaNZgcNXXzQQ=

handlers/access.go

@@ -82,6 +82,14 @@ func NewAccessHandler(path string, logger *logging.Logger) (aH *AccessHandler, e
 		return
 	}
 
+	logger.Debug("NewAccessHandler", "add workspace table")
+	// Add the role table to the database
+	err = aH.AddWorkspaceTable()
+	if err != nil {
+		aH.logger.Error("NewAccessHandler", err)
+		return
+	}
+
 	//indicator that user database was created
 	aH.NewDBCreated = aH.dbHandler.NewCreatedDB
 

handlers/login.go

@@ -215,7 +215,13 @@ func (aH *AccessHandler) Me(c *gin.Context) {
 		return
 	}
 
-	claims := token.Claims.(jwt.MapClaims)
+	claims, ok := token.Claims.(jwt.MapClaims)
+	if !ok {
+		aH.logger.Error("Me", "error: token.Claims.(jwt.MapClaims)")
+		c.JSON(http.StatusInternalServerError, nil)
+		return
+	}
+
 	c.JSON(http.StatusOK, gin.H{
 		"id":   claims["id"],
 		"user": claims["username"],

handlers/user.go

@@ -112,6 +112,10 @@ func (aH *AccessHandler) AddUser(c *gin.Context) {
 		return
 	}
 
+	if len(user.Workspaces) > 0 {
+		aH.dbHandler.AddRelation(&user, user.Workspaces, "Workspaces")
+	}
+
 	c.JSON(http.StatusOK, gin.H{
 		"message": fmt.Sprintf("user '%s' successfully added", user.Name),
 	})
@@ -200,6 +204,10 @@ func (aH *AccessHandler) UpdateUser(c *gin.Context) {
 		c.JSON(http.StatusInternalServerError, nil)
 		return
 	}
+	if len(user.Workspaces) > 0 {
+		aH.dbHandler.AddRelation(&user, user.Workspaces, "Workspaces")
+	}
+
 	c.JSON(http.StatusOK, models.NewJsonMessageResponse("successfully updated user '"+user.Email+"'"))
 }
 

handlers/workspace.go

@@ -0,0 +1,150 @@
+package handlers
+
+import (
+	"fmt"
+	"net/http"
+	"slices"
+	"strconv"
+
+	"gitea.tecamino.com/paadi/access-handler/models"
+	"github.com/gin-gonic/gin"
+)
+
+func (aH *AccessHandler) AddWorkspaceTable() error {
+	return aH.dbHandler.AddNewTable(models.Workspace{})
+}
+
+func (aH *AccessHandler) AddWorkspace(c *gin.Context) {
+	var workspace models.Workspace
+	err := c.BindJSON(&workspace)
+	if err != nil {
+		aH.logger.Error("AddWorkspace", err)
+		c.JSON(http.StatusBadRequest, models.NewJsonErrorResponse(err))
+		return
+	}
+
+	// Check if a workspace with this name already exists
+	if aH.dbHandler.Exists(&models.Workspace{}, "workspace", workspace.Name, false) {
+		aH.logger.Error("AddWorkspace", fmt.Sprintf("workspace with name %s already exists", workspace.Name))
+		c.JSON(http.StatusBadRequest, models.NewJsonMessageResponse(fmt.Sprintf("workspace with name %s already exists", workspace.Name)))
+		return
+	}
+
+	// Insert new workspace with provided permissions
+	aH.dbHandler.AddNewColum(&models.Workspace{
+		Name:        workspace.Name,
+		Description: workspace.Description,
+	})
+
+	c.JSON(http.StatusOK, gin.H{
+		"message": fmt.Sprintf("workspace '%s' successfully added", workspace.Name),
+	})
+}
+
+func (aH *AccessHandler) GetWorkspace(c *gin.Context) {
+	var i int
+	var err error
+	var workspaces []models.Workspace
+
+	workspace := c.Query("workspace")
+	id := c.Query("id")
+
+	if workspace != "" {
+		err = aH.dbHandler.GetByKey(&workspaces, "workspace", workspace, false)
+	} else if id != "" {
+		i, err = strconv.Atoi(id)
+		if err != nil {
+			c.JSON(http.StatusBadRequest, gin.H{
+				"message": err.Error(),
+			})
+			return
+		}
+		err = aH.dbHandler.GetById(&workspaces, uint(i))
+	} else {
+		err = aH.dbHandler.GetById(&workspaces, 0)
+
+	}
+
+	if err != nil {
+		aH.logger.Error("GetWorkspace", err)
+		c.JSON(http.StatusInternalServerError, nil)
+		return
+	}
+	c.JSON(http.StatusOK, workspaces)
+}
+
+func (aH *AccessHandler) UpdateWorkspace(c *gin.Context) {
+	var workspace models.Workspace
+	if err := c.BindJSON(&workspace); err != nil {
+		aH.logger.Error("UpdateWorkspace", err)
+		c.JSON(http.StatusInternalServerError, nil)
+		return
+	}
+
+	if workspace.Id != 0 {
+		err := aH.dbHandler.UpdateValuesById(&workspace, workspace.Id)
+		if err != nil {
+			aH.logger.Error("UpdateWorkspace", err)
+			c.JSON(http.StatusInternalServerError, nil)
+			return
+		}
+	} else {
+		err := aH.dbHandler.UpdateValuesByKey(&workspace, "", "workspace", workspace.Name)
+		if err != nil {
+			aH.logger.Error("UpdateWorkspace", err)
+			c.JSON(http.StatusInternalServerError, nil)
+			return
+		}
+	}
+
+	c.JSON(http.StatusOK, models.NewJsonMessageResponse("successfully updated workspace '"+workspace.Name+"'"))
+}
+
+func (aH *AccessHandler) DeleteWorkspace(c *gin.Context) {
+
+	var request struct {
+		Workspaces []models.Workspace `json:"workspaces"`
+	}
+
+	err := c.BindJSON(&request)
+	if err != nil {
+		aH.logger.Error("DeleteWorkspace", err)
+		c.JSON(http.StatusBadRequest, nil)
+		return
+	}
+
+	if len(request.Workspaces) == 0 {
+		aH.logger.Error("DeleteWorkspace", "no ids given to be deleted")
+		c.JSON(http.StatusBadRequest, models.NewJsonMessageResponse("no workspaces given to be deleted"))
+		return
+	}
+
+	for _, workspace := range request.Workspaces {
+		err = aH.dbHandler.DeleteByKey(&models.Workspace{}, "id", workspace, false)
+		if err != nil {
+			aH.logger.Error("DeleteWorkspace", err)
+			c.JSON(http.StatusInternalServerError, nil)
+			return
+		}
+
+		var users []models.User
+		if err := aH.dbHandler.GetById(&users, 0); err != nil {
+			aH.logger.Error("DeleteWorkspace", err)
+			c.JSON(http.StatusInternalServerError, nil)
+			return
+		}
+		for _, u := range users {
+			if u.Settings.Workspace == workspace.Name {
+				u.Settings.Workspace = ""
+			}
+
+			u.Workspaces = slices.DeleteFunc(u.Workspaces, func(w *models.Workspace) bool {
+				return w != nil && w.Name == workspace.Name
+			})
+		}
+	}
+
+	c.JSON(http.StatusOK, gin.H{
+		"message": "workspace(s) deleted",
+	})
+}

models/settings.go

@@ -15,6 +15,7 @@ type Settings struct {
 	AppName            string `json:"appName,omitempty"`
 	DatabaseName       string `json:"databaseName,omitempty"`
 	DatabaseToken      string `json:"databaseToken,omitempty"`
+	Workspace          string `json:"workspace,omitempty"`
 }
 
 func (s *Settings) DefaultQuasarSettings() {

models/user.go

@@ -5,15 +5,16 @@ import (
 )
 
 type User struct {
-	Id          uint     `gorm:"primaryKey" json:"id"`
-	Name        string   `gorm:"column:user_name" json:"user"`
-	Email       string   `gorm:"column:email" json:"email"`
-	RoleID      *uint    `gorm:"column:roleId" json:"roleId,omitempty"`
-	Role        *Role    `gorm:"foreignKey:RoleID" json:"role,omitempty"`
-	Password    string   `gorm:"column:password" json:"password"`
-	NewPassword string   `gorm:"-" json:"newPassword,omitempty"`
-	Expiration  string   `gorm:"column:expiration" json:"expiration,omitempty"`
-	Settings    Settings `gorm:"type:json" json:"settings"`
+	Id          uint         `gorm:"primaryKey" json:"id"`
+	Name        string       `gorm:"column:user_name" json:"user"`
+	Email       string       `gorm:"column:email" json:"email"`
+	RoleID      *uint        `gorm:"column:roleId" json:"roleId,omitempty"`
+	Role        *Role        `gorm:"foreignKey:RoleID" json:"role,omitempty"`
+	Password    string       `gorm:"column:password" json:"password"`
+	NewPassword string       `gorm:"-" json:"newPassword,omitempty"`
+	Expiration  string       `gorm:"column:expiration" json:"expiration,omitempty"`
+	Settings    Settings     `gorm:"type:json" json:"settings"`
+	Workspaces  []*Workspace `gorm:"many2many:users_workspaces;" `
 }
 
 func (u *User) IsValid() bool {

models/workspace.go

@@ -0,0 +1,8 @@
+package models
+
+type Workspace struct {
+	Id          uint    `gorm:"primaryKey" json:"id"`
+	Name        string  `gorm:"column:name" json:"name"`
+	Description string  `gorm:"type:json" json:"description"`
+	Users       []*User `gorm:"many2many:users_workspaces;" `
+}