basicly implement acl crud ops with roles and resources

This commit is contained in:
2025-12-20 17:38:15 +02:00
parent c188b46519
commit 904f446447
18 changed files with 1607 additions and 324 deletions

View File

@@ -0,0 +1,11 @@
package api_acladmin
type errorInvalidRequestBody struct {
Error string `json:"error" example:"INVALID_REQUEST_BODY"`
Details string `json:"details" example:"Request body is not valid JSON"`
}
type errorInternalServerError struct {
Error string `json:"error"`
Details string `json:"details"`
}

28
api/acl_admin/errors.go Normal file
View File

@@ -0,0 +1,28 @@
package api_acladmin
const (
ErrorInvalidRequestBody = "INVALID_REQUEST_BODY"
ErrorInternalServerError = "INTERNAL_SERVER_ERROR"
// Roles
ErrorFailedToCreateRole = "FAILED_TO_CREATE_ROLE"
ErrorFailedToGetRole = "FAILED_TO_GET_ROLE"
ErrorFailedToUpdateRole = "FAILED_TO_UPDATE_ROLE"
ErrorFailedToDeleteRole = "FAILED_TO_DELETE_ROLE"
ErrorInvalidRoleID = "INVALID_ROLE_ID"
ErrorRoleNotFound = "ROLE_NOT_FOUND"
// Resources
ErrorFailedToCreateResource = "FAILED_TO_CREATE_RESOURCE"
ErrorFailedToGetResource = "FAILED_TO_GET_RESOURCE"
ErrorFailedToUpdateResource = "FAILED_TO_UPDATE_RESOURCE"
ErrorFailedToDeleteResource = "FAILED_TO_DELETE_RESOURCE"
ErrorInvalidResourceID = "INVALID_RESOURCE_ID"
ErrorResourceNotFound = "RESOURCE_NOT_FOUND"
)
const (
ErrorACLServiceNotInitialized = "ACL service is not initialized"
)

View File

@@ -1,13 +1,11 @@
package api_acladmin package api_acladmin
import ( import (
"encoding/json"
"net/http"
"git.oblat.lv/alex/triggerssmith/internal/acl" "git.oblat.lv/alex/triggerssmith/internal/acl"
"git.oblat.lv/alex/triggerssmith/internal/auth" "git.oblat.lv/alex/triggerssmith/internal/auth"
"git.oblat.lv/alex/triggerssmith/internal/config" "git.oblat.lv/alex/triggerssmith/internal/config"
"git.oblat.lv/alex/triggerssmith/internal/server"
//"git.oblat.lv/alex/triggerssmith/internal/server"
"github.com/go-chi/chi/v5" "github.com/go-chi/chi/v5"
) )
@@ -32,91 +30,221 @@ func MustRoute(config *config.Config, aclService *acl.Service, authService *auth
a: aclService, a: aclService,
auth: authService, auth: authService,
} }
// GET /roles — список ролей
// POST /roles — создать роль
// GET /roles/{roleId} — получить роль
// PATCH /roles/{roleId} — обновить роль (если нужно)
// DELETE /roles/{roleId} — удалить роль
// GET /resources — список ресурсов
// POST /resources — создать ресурс
// GET /resources/{resId} — получить ресурс
// PATCH /resources/{resId} — обновить ресурс
// DELETE /resources/{resId} — удалить ресурс
// GET /users/{userId}/roles — роли пользователя
// POST /users/{userId}/roles — назначить роль пользователю
// DELETE /users/{userId}/roles/{roleId} — снять роль
// GET /roles/{roleId}/resources — ресурсы роли
// POST /roles/{roleId}/resources — назначить ресурс роли
// DELETE /roles/{roleId}/resources/{resId} — убрать ресурс
return func(r chi.Router) { return func(r chi.Router) {
r.Get("/roles", h.getRoles) // Roles
r.Post("/create-role", h.createRole) r.Get("/roles", h.getRoles) // list all roles
r.Post("/assign-role", h.assignRoleToUser) r.Post("/roles", h.createRole) // create a new role
r.Get("/user-roles", h.getUserRoles) r.Get("/roles/{roleId}", h.getRole) // get a role by ID
r.Post("/remove-role", h.removeRoleFromUser) r.Patch("/roles/{roleId}", h.updateRole) // update a role by ID
r.Delete("/roles/{roleId}", h.deleteRole) // delete a role by ID
r.Get("/resources", h.getResources) // // Resources
r.Post("/create-resource", h.createResource) r.Get("/resources", h.getResources) // list all resources
r.Post("/assign-resource", h.assignResourceToRole) r.Post("/resources", h.createResource) // create a new resource
r.Get("/role-resources", h.getRoleResources) r.Get("/resources/{resourceId}", h.getResource) // get a resource by ID
r.Post("/remove-resource", h.removeResourceFromRole) r.Patch("/resources/{resourceId}", h.updateResource) // update a resource by ID
r.Delete("/resources/{resourceId}", h.deleteResource) // delete a resource by ID
r.Get("/permissions", h.getResources) // legacy support // Users
r.Post("/create-permissions", h.createResource) // legacy support // r.Get("/users/{userId}/roles", h.getUserRoles) // get all roles for a user
r.Post("/assign-permissions", h.assignResourceToRole) // legacy support // r.Post("/users/{userId}/roles", h.assignRoleToUser) // assign a role to a user
r.Get("/role-permissions", h.getRoleResources) // legacy support // r.Delete("/users/{userId}/roles/{roleId}", h.removeRoleFromUser) // remove a role from a user
r.Post("/remove-permissions", h.removeResourceFromRole) // legacy support
// r.Get("/roles", h.getRoles)
// r.Post("/create-role", h.createRole)
// r.Post("/assign-role", h.assignRoleToUser)
// r.Get("/user-roles", h.getUserRoles)
// r.Post("/remove-role", h.removeRoleFromUser)
// r.Get("/resources", h.getResources)
// r.Post("/create-resource", h.createResource)
// r.Post("/assign-resource", h.assignResourceToRole)
// r.Get("/role-resources", h.getRoleResources)
// r.Post("/remove-resource", h.removeResourceFromRole)
// r.Get("/permissions", h.getResources) // legacy support
// r.Post("/create-permissions", h.createResource) // legacy support
// r.Post("/assign-permissions", h.assignResourceToRole) // legacy support
// r.Get("/role-permissions", h.getRoleResources) // legacy support
// r.Post("/remove-permissions", h.removeResourceFromRole) // legacy support
} }
} }
type rolesResponse []struct { // type assignRoleRequest struct {
ID uint `json:"id"` // UserID int `json:"userId"`
Name string `json:"name"` // RoleID int `json:"roleId"`
} // }
func (h *aclAdminHandler) getRoles(w http.ResponseWriter, r *http.Request) { // func (h *aclAdminHandler) assignRoleToUser(w http.ResponseWriter, r *http.Request) {
roles, err := h.a.GetRoles() // var req assignRoleRequest
if err != nil { // if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
http.Error(w, "Internal server error", http.StatusInternalServerError) // http.Error(w, "Invalid request body", http.StatusBadRequest)
return // return
} // }
w.Header().Set("Content-Type", "application/json") // if req.UserID < 0 || req.RoleID < 0 {
err = json.NewEncoder(w).Encode(func() rolesResponse { // http.Error(w, "Invalid user or role ID", http.StatusBadRequest)
// Transform acl.Role to rolesResponse // return
resp := make(rolesResponse, 0, len(roles)) // }
for _, role := range roles { // if err := h.a.AssignRoleToUser(uint(req.RoleID), uint(req.UserID)); err != nil {
resp = append(resp, struct { // http.Error(w, "Failed to assign role to user", http.StatusConflict)
ID uint `json:"id"` // return
Name string `json:"name"` // }
}{ // w.WriteHeader(http.StatusCreated)
ID: role.ID, // }
Name: role.Name,
})
}
return resp
}())
if err != nil {
http.Error(w, "Failed to encode response", http.StatusInternalServerError)
return
}
}
func (h *aclAdminHandler) createRole(w http.ResponseWriter, r *http.Request) { // type getUserRolesResponse getRolesResponse
server.NotImplemented(w)
}
func (h *aclAdminHandler) assignRoleToUser(w http.ResponseWriter, r *http.Request) { // func (h *aclAdminHandler) getUserRoles(w http.ResponseWriter, r *http.Request) {
server.NotImplemented(w) // uidStr := r.URL.Query().Get("userId")
} // if uidStr == "" {
// http.Error(w, "Missing userId parameter", http.StatusBadRequest)
// return
// }
// userID, err := strconv.Atoi(uidStr)
// if err != nil || userID < 0 {
// http.Error(w, "Invalid userId parameter", http.StatusBadRequest)
// return
// }
// roles, err := h.a.GetUserRoles(uint(userID))
// if err != nil {
// http.Error(w, "Internal server error", http.StatusInternalServerError)
// return
// }
// w.Header().Set("Content-Type", "application/json")
// err = json.NewEncoder(w).Encode(func() getUserRolesResponse {
// // Transform acl.Role to getUserRolesResponse
// resp := make(getUserRolesResponse, 0, len(roles))
// for _, role := range roles {
// resp = append(resp, struct {
// ID uint `json:"id"`
// Name string `json:"name"`
// }{
// ID: role.ID,
// Name: role.Name,
// })
// }
// return resp
// }())
// if err != nil {
// http.Error(w, "Failed to encode response", http.StatusInternalServerError)
// return
// }
// }
func (h *aclAdminHandler) getUserRoles(w http.ResponseWriter, r *http.Request) { // type removeRoleRequest struct {
server.NotImplemented(w) // UserID int `json:"userId"`
} // RoleID int `json:"roleId"`
// }
func (h *aclAdminHandler) removeRoleFromUser(w http.ResponseWriter, r *http.Request) { // func (h *aclAdminHandler) removeRoleFromUser(w http.ResponseWriter, r *http.Request) {
server.NotImplemented(w) // var req removeRoleRequest
} // if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
// http.Error(w, "Invalid request body", http.StatusBadRequest)
// return
// }
// if req.UserID < 0 || req.RoleID < 0 {
// http.Error(w, "Invalid user or role ID", http.StatusBadRequest)
// return
// }
// if err := h.a.RemoveRoleFromUser(uint(req.RoleID), uint(req.UserID)); err != nil {
// http.Error(w, "Failed to remove role from user", http.StatusConflict)
// return
// }
// w.WriteHeader(http.StatusNoContent)
// }
func (h *aclAdminHandler) getResources(w http.ResponseWriter, r *http.Request) { // type getResourcesResponse getRolesResponse
server.NotImplemented(w)
}
func (h *aclAdminHandler) createResource(w http.ResponseWriter, r *http.Request) { // func (h *aclAdminHandler) getResources(w http.ResponseWriter, r *http.Request) {
server.NotImplemented(w) // resources, err := h.a.GetResources()
} // if err != nil {
// http.Error(w, "Internal server error", http.StatusInternalServerError)
// return
// }
// w.Header().Set("Content-Type", "application/json")
// err = json.NewEncoder(w).Encode(func() getResourcesResponse {
// // Transform acl.Resource to getResourcesResponse
// resp := make(getResourcesResponse, 0, len(resources))
// for _, res := range resources {
// resp = append(resp, struct {
// ID uint `json:"id"`
// Name string `json:"name"`
// }{
// ID: res.ID,
// Name: res.Key,
// })
// }
// return resp
// }())
// if err != nil {
// http.Error(w, "Failed to encode response", http.StatusInternalServerError)
// return
// }
// }
func (h *aclAdminHandler) assignResourceToRole(w http.ResponseWriter, r *http.Request) { // type createResourceRequest struct {
server.NotImplemented(w) // Name string `json:"name"`
} // }
func (h *aclAdminHandler) getRoleResources(w http.ResponseWriter, r *http.Request) { // type createResourceResponse struct {
server.NotImplemented(w) // ID uint `json:"id"`
} // Name string `json:"name"`
// }
func (h *aclAdminHandler) removeResourceFromRole(w http.ResponseWriter, r *http.Request) { // func (h *aclAdminHandler) createResource(w http.ResponseWriter, r *http.Request) {
server.NotImplemented(w) // var req createResourceRequest
} // if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
// http.Error(w, "Invalid request body", http.StatusBadRequest)
// return
// }
// if req.Name == "" {
// http.Error(w, "Name is required", http.StatusBadRequest)
// return
// }
// id, err := h.a.CreateResource(req.Name)
// if err != nil {
// http.Error(w, "Failed to create resource", http.StatusConflict)
// return
// }
// w.WriteHeader(http.StatusCreated)
// w.Header().Set("Content-Type", "application/json")
// err = json.NewEncoder(w).Encode(createResourceResponse{
// ID: id,
// Name: req.Name,
// })
// if err != nil {
// http.Error(w, "Failed to encode response", http.StatusInternalServerError)
// return
// }
// }
// func (h *aclAdminHandler) assignResourceToRole(w http.ResponseWriter, r *http.Request) {
// server.NotImplemented(w)
// }
// func (h *aclAdminHandler) getRoleResources(w http.ResponseWriter, r *http.Request) {
// server.NotImplemented(w)
// }
// func (h *aclAdminHandler) removeResourceFromRole(w http.ResponseWriter, r *http.Request) {
// server.NotImplemented(w)
// }

321
api/acl_admin/resources.go Normal file
View File

@@ -0,0 +1,321 @@
package api_acladmin
import (
"encoding/json"
"log/slog"
"net/http"
"strconv"
"git.oblat.lv/alex/triggerssmith/internal/acl"
"github.com/go-chi/chi/v5"
)
// @Summary Get all resources
// @Tags resources
// @Produce json
// @Success 200 {object} getResourcesResponse
// @Failure 500 {object} errorInternalServerError
// @Router /api/acl/resources [get]
func (h *aclAdminHandler) getResources(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "application/json")
resources, err := h.a.GetResources()
if err != nil {
switch err {
case acl.ErrNotInitialized:
w.WriteHeader(http.StatusInternalServerError)
_ = json.NewEncoder(w).Encode(errorInternalServerError{
Error: ErrorInternalServerError,
Details: ErrorACLServiceNotInitialized,
})
return
default:
w.WriteHeader(http.StatusInternalServerError)
_ = json.NewEncoder(w).Encode(errorInternalServerError{
Error: ErrorInternalServerError,
Details: "Failed to get resources",
})
return
}
}
_ = json.NewEncoder(w).Encode(func() getResourcesResponse {
resp := make(getResourcesResponse, 0, len(resources))
for _, res := range resources {
resp = append(resp, struct {
ID uint `json:"id" example:"1"`
Key string `json:"key" example:"html.view"`
}{
ID: res.ID,
Key: res.Key,
})
}
return resp
}())
}
// @Summary Get resource by ID
// @Tags resources
// @Produce json
// @Param resourceId path int true "Resource ID" example(1)
// @Success 200 {object} getResourceResponse
// @Failure 400 {object} getResourceErrorInvalidResourceID
// @Failure 404 {object} getResourceErrorResourceNotFound
// @Failure 500 {object} errorInternalServerError
// @Router /api/acl/resources/{resourceId} [get]
func (h *aclAdminHandler) getResource(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "application/json")
resourceIDStr := chi.URLParam(r, "resourceId")
resourceID, err := strconv.Atoi(resourceIDStr)
if err != nil || resourceID < 0 {
w.WriteHeader(http.StatusBadRequest)
_ = json.NewEncoder(w).Encode(getResourceErrorInvalidResourceID{
Error: ErrorInvalidResourceID,
Details: "Resource ID must be positive integer",
})
return
}
resource, err := h.a.GetResourceByID(uint(resourceID))
if err != nil {
switch err {
case acl.ErrNotInitialized:
w.WriteHeader(http.StatusInternalServerError)
_ = json.NewEncoder(w).Encode(errorInternalServerError{
Error: ErrorInternalServerError,
Details: ErrorACLServiceNotInitialized,
})
return
case acl.ErrResourceNotFound:
w.WriteHeader(http.StatusNotFound)
_ = json.NewEncoder(w).Encode(getResourceErrorResourceNotFound{
Error: ErrorResourceNotFound,
Details: "No resource with ID " + resourceIDStr,
})
return
default:
w.WriteHeader(http.StatusInternalServerError)
_ = json.NewEncoder(w).Encode(errorInternalServerError{
Error: ErrorInternalServerError,
Details: "Failed to get resource with ID " + resourceIDStr,
})
return
}
}
_ = json.NewEncoder(w).Encode(getResourceResponse{
ID: resource.ID,
Key: resource.Key,
})
}
// @Summary Create resource
// @Tags resources
// @Accept json
// @Produce json
// @Param request body createResourceRequest true "Resource"
// @Success 201 {object} createResourceResponse
// @Failure 400 {object} errorInvalidRequestBody
// @Failure 400 {object} createResourceErrorInvalidResourceKey
// @Failure 409 {object} createResourceErrorResourceAlreadyExists
// @Failure 500 {object} errorInternalServerError
// @Router /api/acl/resources [post]
func (h *aclAdminHandler) createResource(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "application/json")
var req createResourceRequest
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
w.WriteHeader(http.StatusBadRequest)
_ = json.NewEncoder(w).Encode(errorInvalidRequestBody{
Error: ErrorInvalidRequestBody,
Details: "Request body is not valid JSON",
})
return
}
resourceID, err := h.a.CreateResource(req.Key)
if err != nil {
slog.Error("Failed to create resource", "error", err.Error())
switch err {
case acl.ErrNotInitialized:
w.WriteHeader(http.StatusInternalServerError)
_ = json.NewEncoder(w).Encode(errorInternalServerError{
Error: ErrorInternalServerError,
Details: ErrorACLServiceNotInitialized,
})
return
case acl.ErrInvalidResourceKey:
w.WriteHeader(http.StatusBadRequest)
_ = json.NewEncoder(w).Encode(createResourceErrorInvalidResourceKey{
Error: ErrorFailedToCreateResource,
Details: "Resource key must be non-empty",
})
return
case acl.ErrResourceAlreadyExists:
w.WriteHeader(http.StatusConflict)
_ = json.NewEncoder(w).Encode(createResourceErrorResourceAlreadyExists{
Error: ErrorFailedToCreateResource,
Details: "Resource with key '" + req.Key + "' already exists",
})
return
default:
w.WriteHeader(http.StatusInternalServerError)
_ = json.NewEncoder(w).Encode(errorInternalServerError{
Error: ErrorInternalServerError,
Details: "Failed to create resource with key '" + req.Key + "'",
})
return
}
}
w.WriteHeader(http.StatusCreated)
_ = json.NewEncoder(w).Encode(createResourceResponse{
ID: resourceID,
Key: req.Key,
})
}
// @Summary Update resource
// @Tags resources
// @Accept json
// @Produce json
// @Param resourceId path int true "Resource ID" example(1)
// @Param request body updateResourceRequest true "Resource"
// @Success 200 {object} updateResourceResponse
// @Failure 400 {object} errorInvalidRequestBody
// @Failure 400 {object} updateResourceErrorInvalidResourceID
// @Failure 400 {object} updateResourceErrorInvalidResourceKey
// @Failure 404 {object} updateResourceErrorResourceNotFound
// @Failure 409 {object} updateResourceErrorResourceKeyAlreadyExists
// @Failure 500 {object} errorInternalServerError
// @Router /api/acl/resources/{resourceId} [patch]
func (h *aclAdminHandler) updateResource(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "application/json")
var req updateResourceRequest
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
w.WriteHeader(http.StatusBadRequest)
_ = json.NewEncoder(w).Encode(errorInvalidRequestBody{
Error: ErrorInvalidRequestBody,
Details: "Request body is not valid JSON",
})
return
}
resourceIDStr := chi.URLParam(r, "resourceId")
resourceID, err := strconv.Atoi(resourceIDStr)
if err != nil || resourceID < 0 {
w.WriteHeader(http.StatusBadRequest)
_ = json.NewEncoder(w).Encode(updateResourceErrorInvalidResourceID{
Error: ErrorInvalidResourceID,
Details: "Resource ID must be positive integer",
})
return
}
err = h.a.UpdateResource(uint(resourceID), req.Key)
if err != nil {
slog.Error("Failed to update resource", "error", err.Error())
switch err {
case acl.ErrNotInitialized:
w.WriteHeader(http.StatusInternalServerError)
_ = json.NewEncoder(w).Encode(errorInternalServerError{
Error: ErrorInternalServerError,
Details: ErrorACLServiceNotInitialized,
})
return
case acl.ErrInvalidResourceKey:
w.WriteHeader(http.StatusBadRequest)
_ = json.NewEncoder(w).Encode(updateResourceErrorInvalidResourceKey{
Error: ErrorFailedToUpdateResource,
Details: "Invalid resource key",
})
return
case acl.ErrResourceNotFound:
w.WriteHeader(http.StatusNotFound)
_ = json.NewEncoder(w).Encode(updateResourceErrorResourceNotFound{
Error: ErrorFailedToUpdateResource,
Details: "No resource with ID " + resourceIDStr,
})
return
case acl.ErrSameResourceKey:
w.WriteHeader(http.StatusConflict)
_ = json.NewEncoder(w).Encode(updateResourceErrorResourceKeyAlreadyExists{
Error: ErrorFailedToUpdateResource,
Details: "Resource with key '" + req.Key + "' already exists",
})
return
default:
w.WriteHeader(http.StatusInternalServerError)
_ = json.NewEncoder(w).Encode(errorInternalServerError{
Error: ErrorInternalServerError,
Details: "Failed to update resource with key '" + req.Key + "'",
})
return
}
}
w.WriteHeader(http.StatusOK)
_ = json.NewEncoder(w).Encode(updateResourceResponse{
ID: uint(resourceID),
Key: req.Key,
})
}
// @Summary Delete resource
// @Tags resources
// @Produce json
// @Param resourceId path int true "Resource ID" example(1)
// @Success 200
// @Failure 400 {object} deleteResourceErrorInvalidResourceID
// @Failure 404 {object} deleteResourceErrorResourceNotFound
// @Failure 409 {object} deleteResourceErrorResourceInUse
// @Failure 500 {object} errorInternalServerError
// @Router /api/acl/resources/{resourceId} [delete]
func (h *aclAdminHandler) deleteResource(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "application/json")
resourceIDStr := chi.URLParam(r, "resourceId")
resourceID, err := strconv.Atoi(resourceIDStr)
if err != nil || resourceID < 0 {
w.WriteHeader(http.StatusBadRequest)
_ = json.NewEncoder(w).Encode(deleteResourceErrorInvalidResourceID{
Error: ErrorInvalidResourceID,
Details: "Resource ID must be positive integer",
})
return
}
err = h.a.DeleteResource(uint(resourceID))
if err != nil {
slog.Error("Failed to delete resource", "error", err.Error())
switch err {
case acl.ErrNotInitialized:
w.WriteHeader(http.StatusInternalServerError)
_ = json.NewEncoder(w).Encode(errorInternalServerError{
Error: ErrorInternalServerError,
Details: ErrorACLServiceNotInitialized,
})
return
case acl.ErrResourceNotFound:
w.WriteHeader(http.StatusNotFound)
_ = json.NewEncoder(w).Encode(deleteResourceErrorResourceNotFound{
Error: ErrorFailedToDeleteResource,
Details: "No resource with ID " + resourceIDStr,
})
return
case acl.ErrResourceInUse:
w.WriteHeader(http.StatusConflict)
_ = json.NewEncoder(w).Encode(deleteResourceErrorResourceInUse{
Error: ErrorFailedToDeleteResource,
Details: "Resource with ID " + resourceIDStr + " is used and cannot be deleted",
})
return
default:
w.WriteHeader(http.StatusInternalServerError)
_ = json.NewEncoder(w).Encode(errorInternalServerError{
Error: ErrorInternalServerError,
Details: "Failed to delete resource with ID '" + resourceIDStr + "'",
})
return
}
}
w.WriteHeader(http.StatusOK)
}

View File

@@ -0,0 +1,94 @@
package api_acladmin
/*******************************************************************/
// used in getResources()
type getResourcesResponse []struct {
ID uint `json:"id" example:"1"`
Key string `json:"key" example:"html.view"`
}
/*******************************************************************/
// used in getResource()
type getResourceResponse struct {
ID uint `json:"id" example:"1"`
Key string `json:"key" example:"html.view"`
}
type getResourceErrorInvalidResourceID struct {
Error string `json:"error" example:"INVALID_RESOURCE_ID"`
Details string `json:"details" example:"Resource ID must be positive integer"`
}
type getResourceErrorResourceNotFound struct {
Error string `json:"error" example:"RESOURCE_NOT_FOUND"`
Details string `json:"details" example:"No resource with ID 123"`
}
/*******************************************************************/
// used in createResource()
type createResourceRequest struct {
Key string `json:"key" example:"html.view"`
}
type createResourceResponse struct {
ID uint `json:"id" example:"1"`
Key string `json:"key" example:"html.view"`
}
type createResourceErrorResourceAlreadyExists struct {
Error string `json:"error" example:"FAILED_TO_CREATE_RESOURCE"`
Details string `json:"details" example:"Resource with key 'html.view' already exists"`
}
type createResourceErrorInvalidResourceKey struct {
Error string `json:"error" example:"FAILED_TO_CREATE_RESOURCE"`
Details string `json:"details" example:"Invalid resource key"`
}
/*******************************************************************/
// used in updateResource()
type updateResourceRequest struct {
Key string `json:"key" example:"html.view"`
}
type updateResourceResponse struct {
ID uint `json:"id" example:"1"`
Key string `json:"key" example:"html.view"`
}
type updateResourceErrorResourceNotFound struct {
Error string `json:"error" example:"RESOURCE_NOT_FOUND"`
Details string `json:"details" example:"No resource with ID 123"`
}
type updateResourceErrorInvalidResourceID struct {
Error string `json:"error" example:"INVALID_RESOURCE_ID"`
Details string `json:"details" example:"Resource ID must be positive integer"`
}
type updateResourceErrorInvalidResourceKey struct {
Error string `json:"error" example:"FAILED_TO_UPDATE_RESOURCE"`
Details string `json:"details" example:"Invalid resource key"`
}
type updateResourceErrorResourceKeyAlreadyExists struct {
Error string `json:"error" example:"FAILED_TO_UPDATE_RESOURCE"`
Details string `json:"details" example:"Resource with key 'html.view' already exists"`
}
/*******************************************************************/
// used in deleteResource()
type deleteResourceErrorResourceNotFound struct {
Error string `json:"error" example:"RESOURCE_NOT_FOUND"`
Details string `json:"details" example:"No resource with ID 123"`
}
type deleteResourceErrorInvalidResourceID struct {
Error string `json:"error" example:"INVALID_RESOURCE_ID"`
Details string `json:"details" example:"Resource ID must be positive integer"`
}
type deleteResourceErrorResourceInUse struct {
Error string `json:"error" example:"FAILED_TO_DELETE_RESOURCE"`
Details string `json:"details" example:"Resource with ID 123 is used and cannot be deleted"`
}

314
api/acl_admin/roles.go Normal file
View File

@@ -0,0 +1,314 @@
package api_acladmin
import (
"encoding/json"
"log/slog"
"net/http"
"strconv"
"git.oblat.lv/alex/triggerssmith/internal/acl"
"github.com/go-chi/chi/v5"
)
// @Summary Get all roles
// @Tags roles
// @Produce json
// @Success 200 {object} getRolesResponse
// @Failure 500 {object} errorInternalServerError
// @Router /api/acl/roles [get]
func (h *aclAdminHandler) getRoles(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "application/json")
roles, err := h.a.GetRoles()
if err != nil {
switch err {
case acl.ErrNotInitialized:
w.WriteHeader(http.StatusInternalServerError)
_ = json.NewEncoder(w).Encode(errorInternalServerError{
Error: ErrorInternalServerError,
Details: ErrorACLServiceNotInitialized,
})
return
default:
w.WriteHeader(http.StatusInternalServerError)
_ = json.NewEncoder(w).Encode(errorInternalServerError{
Error: ErrorInternalServerError,
Details: "Failed to get roles",
})
return
}
}
_ = json.NewEncoder(w).Encode(func() getRolesResponse {
// Transform acl.Role to getRolesResponse
resp := make(getRolesResponse, 0, len(roles))
for _, role := range roles {
resp = append(resp, struct {
ID uint `json:"id" example:"1"`
Name string `json:"name" example:"admin"`
}{
ID: role.ID,
Name: role.Name,
})
}
return resp
}())
}
// @Summary Get role by ID
// @Tags roles
// @Produce json
// @Param roleId path int true "Role ID" example(1)
// @Success 200 {object} getRoleResponse
// @Failure 400 {object} getRoleErrorInvalidRoleID
// @Failure 404 {object} getRoleErrorRoleNotFound
// @Failure 500 {object} errorInternalServerError
// @Router /api/acl/roles/{roleId} [get]
func (h *aclAdminHandler) getRole(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "application/json")
roleIDStr := chi.URLParam(r, "roleId")
roleID, err := strconv.Atoi(roleIDStr)
if err != nil || roleID < 0 {
w.WriteHeader(http.StatusBadRequest)
_ = json.NewEncoder(w).Encode(getRoleErrorInvalidRoleID{
Error: ErrorInvalidRoleID,
Details: "Role ID must be positive integer",
})
return
}
role, err := h.a.GetRoleByID(uint(roleID))
if err != nil {
switch err {
case acl.ErrNotInitialized:
w.WriteHeader(http.StatusInternalServerError)
_ = json.NewEncoder(w).Encode(errorInternalServerError{
Error: ErrorInternalServerError,
Details: ErrorACLServiceNotInitialized,
})
return
case acl.ErrRoleNotFound:
w.WriteHeader(http.StatusNotFound)
_ = json.NewEncoder(w).Encode(getRoleErrorRoleNotFound{
Error: ErrorRoleNotFound,
Details: "No role with ID " + roleIDStr,
})
return
default:
w.WriteHeader(http.StatusInternalServerError)
_ = json.NewEncoder(w).Encode(errorInternalServerError{
Error: ErrorInternalServerError,
Details: "Failed to get role with ID " + roleIDStr,
})
return
}
}
_ = json.NewEncoder(w).Encode(getRoleResponse{
ID: role.ID,
Name: role.Name,
})
}
// @Summary Create role
// @Tags roles
// @Accept json
// @Produce json
// @Param request body createRoleRequest true "Role"
// @Success 201 {object} createRoleResponse
// @Failure 400 {object} errorInvalidRequestBody
// @Failure 401 {object} createRoleErrorInvalidRoleName
// @Failure 409 {object} createRoleErrorRoleAlreadyExists
// @Failure 500 {object} errorInternalServerError
// @Router /api/acl/roles [post]
func (h *aclAdminHandler) createRole(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "application/json")
var req createRoleRequest
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
w.WriteHeader(http.StatusBadRequest)
_ = json.NewEncoder(w).Encode(errorInvalidRequestBody{
Error: ErrorInvalidRequestBody,
Details: "Request body is not valid JSON",
})
return
}
roleID, err := h.a.CreateRole(req.Name)
if err != nil {
slog.Error("Failed to create role", "error", err.Error())
switch err {
case acl.ErrNotInitialized:
w.WriteHeader(http.StatusInternalServerError)
_ = json.NewEncoder(w).Encode(errorInternalServerError{
Error: ErrorInternalServerError,
Details: ErrorACLServiceNotInitialized,
})
return
case acl.ErrRoleAlreadyExists:
w.WriteHeader(http.StatusConflict)
_ = json.NewEncoder(w).Encode(createRoleErrorRoleAlreadyExists{
Error: ErrorFailedToCreateRole,
Details: "Role with name '" + req.Name + "' already exists",
})
return
case acl.ErrInvalidRoleName:
w.WriteHeader(http.StatusBadRequest)
_ = json.NewEncoder(w).Encode(createRoleErrorInvalidRoleName{
Error: ErrorFailedToCreateRole,
Details: "Role name must be non-empty string",
})
return
default:
w.WriteHeader(http.StatusInternalServerError)
_ = json.NewEncoder(w).Encode(errorInternalServerError{
Error: ErrorInternalServerError,
Details: "Failed to create role with name '" + req.Name + "'",
})
return
}
}
w.WriteHeader(http.StatusCreated)
_ = json.NewEncoder(w).Encode(createRoleResponse{
ID: roleID,
Name: req.Name,
})
}
// @Summary Update role
// @Tags roles
// @Accept json
// @Produce json
// @Param roleId path int true "Role ID" example(1)
// @Param request body updateRoleRequest true "Role"
// @Success 200 {object} updateRoleResponse
// @Failure 400 {object} errorInvalidRequestBody
// @Failure 400 {object} updateRoleErrorInvalidRoleID
// @Failure 400 {object} updateRoleErrorInvalidRoleName
// @Failure 404 {object} updateRoleErrorRoleNotFound
// @Failure 409 {object} updateRoleErrorRoleNameAlreadyExists
// @Failure 500 {object} errorInternalServerError
// @Router /api/acl/roles/{roleId} [patch]
func (h *aclAdminHandler) updateRole(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "application/json")
var req updateRoleRequest
if err := json.NewDecoder(r.Body).Decode(&req); err != nil {
w.WriteHeader(http.StatusBadRequest)
_ = json.NewEncoder(w).Encode(errorInvalidRequestBody{
Error: ErrorInvalidRequestBody,
Details: "Request body is not valid JSON",
})
return
}
roleIDStr := chi.URLParam(r, "roleId")
roleID, err := strconv.Atoi(roleIDStr)
if err != nil || roleID < 0 {
w.WriteHeader(http.StatusBadRequest)
_ = json.NewEncoder(w).Encode(updateRoleErrorInvalidRoleID{
Error: ErrorInvalidRoleID,
Details: "Role ID must be positive integer",
})
return
}
err = h.a.UpdateRole(uint(roleID), req.Name)
// TODO: make error handling more specific in acl service
if err != nil {
slog.Error("Failed to update role", "error", err.Error())
switch err {
case acl.ErrNotInitialized:
w.WriteHeader(http.StatusInternalServerError)
_ = json.NewEncoder(w).Encode(errorInternalServerError{
Error: ErrorInternalServerError,
Details: ErrorACLServiceNotInitialized,
})
return
case acl.ErrInvalidRoleName:
w.WriteHeader(http.StatusBadRequest)
_ = json.NewEncoder(w).Encode(updateRoleErrorInvalidRoleName{
Error: ErrorFailedToUpdateRole,
Details: "Invalid role name",
})
return
case acl.ErrRoleNotFound:
w.WriteHeader(http.StatusNotFound)
_ = json.NewEncoder(w).Encode(updateRoleErrorRoleNotFound{
Error: ErrorFailedToUpdateRole,
Details: "No role with ID " + roleIDStr,
})
return
case acl.ErrSameRoleName:
w.WriteHeader(http.StatusConflict)
_ = json.NewEncoder(w).Encode(updateRoleErrorRoleNameAlreadyExists{
Error: ErrorFailedToUpdateRole,
Details: "Role with name '" + req.Name + "' already exists",
})
return
default:
w.WriteHeader(http.StatusInternalServerError)
_ = json.NewEncoder(w).Encode(errorInternalServerError{
Error: ErrorInternalServerError,
Details: "Failed to update role with name '" + req.Name + "'",
})
return
}
}
w.WriteHeader(http.StatusOK)
_ = json.NewEncoder(w).Encode(updateRoleResponse{
ID: uint(roleID),
Name: req.Name,
})
}
// @Summary Delete role
// @Tags roles
// @Produce json
// @Param roleId path int true "Role ID" example(1)
// @Success 200
// @Failure 400 {object} deleteRoleErrorInvalidRoleID
// @Failure 404 {object} deleteRoleErrorRoleNotFound
// @Failure 409 {object} deleteRoleErrorRoleInUse
// @Failure 500 {object} errorInternalServerError
// @Router /api/acl/roles/{roleId} [delete]
func (h *aclAdminHandler) deleteRole(w http.ResponseWriter, r *http.Request) {
w.Header().Set("Content-Type", "application/json")
roleIDStr := chi.URLParam(r, "roleId")
roleID, err := strconv.Atoi(roleIDStr)
if err != nil || roleID < 0 {
w.WriteHeader(http.StatusBadRequest)
_ = json.NewEncoder(w).Encode(deleteRoleErrorInvalidRoleID{
Error: ErrorInvalidRoleID,
Details: "Role ID must be positive integer",
})
return
}
err = h.a.DeleteRole(uint(roleID))
// TODO: make error handling more specific in acl service
if err != nil {
slog.Error("Failed to delete role", "error", err.Error())
switch err {
case acl.ErrNotInitialized:
w.WriteHeader(http.StatusInternalServerError)
_ = json.NewEncoder(w).Encode(errorInternalServerError{
Error: ErrorInternalServerError,
Details: ErrorACLServiceNotInitialized,
})
return
case acl.ErrRoleNotFound:
w.WriteHeader(http.StatusNotFound)
_ = json.NewEncoder(w).Encode(deleteRoleErrorRoleNotFound{
Error: ErrorFailedToDeleteRole,
Details: "No role with ID " + roleIDStr,
})
return
case acl.ErrRoleInUse:
w.WriteHeader(http.StatusConflict)
_ = json.NewEncoder(w).Encode(deleteRoleErrorRoleInUse{
Error: ErrorFailedToDeleteRole,
Details: "Role with ID " + roleIDStr + " is assigned to users and cannot be deleted",
})
return
default:
w.WriteHeader(http.StatusInternalServerError)
_ = json.NewEncoder(w).Encode(errorInternalServerError{
Error: ErrorInternalServerError,
Details: "Failed to delete role with ID '" + roleIDStr + "'",
})
return
}
}
w.WriteHeader(http.StatusOK)
}

View File

@@ -0,0 +1,94 @@
package api_acladmin
/*******************************************************************/
// used in getRoles()
type getRolesResponse []struct {
ID uint `json:"id" example:"1"`
Name string `json:"name" example:"admin"`
}
/*******************************************************************/
// used in getRole()
type getRoleResponse struct {
ID uint `json:"id" example:"1"`
Name string `json:"name" example:"admin"`
}
type getRoleErrorInvalidRoleID struct {
Error string `json:"error" example:"INVALID_ROLE_ID"`
Details string `json:"details" example:"Role ID must be positive integer"`
}
type getRoleErrorRoleNotFound struct {
Error string `json:"error" example:"ROLE_NOT_FOUND"`
Details string `json:"details" example:"No role with ID 123"`
}
/*******************************************************************/
// used in createRole()
type createRoleRequest struct {
Name string `json:"name" example:"admin"`
}
type createRoleResponse struct {
ID uint `json:"id" example:"1"`
Name string `json:"name" example:"admin"`
}
type createRoleErrorRoleAlreadyExists struct {
Error string `json:"error" example:"FAILED_TO_CREATE_ROLE"`
Details string `json:"details" example:"Role with name 'admin' already exists"`
}
type createRoleErrorInvalidRoleName struct {
Error string `json:"error" example:"FAILED_TO_CREATE_ROLE"`
Details string `json:"details" example:"Invalid role name"`
}
/*******************************************************************/
// used in updateRole()
type updateRoleRequest struct {
Name string `json:"name" example:"admin"`
}
type updateRoleResponse struct {
ID uint `json:"id" example:"1"`
Name string `json:"name" example:"admin"`
}
type updateRoleErrorRoleNotFound struct {
Error string `json:"error" example:"ROLE_NOT_FOUND"`
Details string `json:"details" example:"No role with ID 123"`
}
type updateRoleErrorInvalidRoleID struct {
Error string `json:"error" example:"INVALID_ROLE_ID"`
Details string `json:"details" example:"Role ID must be positive integer"`
}
type updateRoleErrorInvalidRoleName struct {
Error string `json:"error" example:"FAILED_TO_UPDATE_ROLE"`
Details string `json:"details" example:"Invalid role name"`
}
type updateRoleErrorRoleNameAlreadyExists struct {
Error string `json:"error" example:"FAILED_TO_UPDATE_ROLE"`
Details string `json:"details" example:"Role with name 'admin' already exists"`
}
/*******************************************************************/
// used in deleteRole()
type deleteRoleErrorRoleNotFound struct {
Error string `json:"error" example:"ROLE_NOT_FOUND"`
Details string `json:"details" example:"No role with ID 123"`
}
type deleteRoleErrorInvalidRoleID struct {
Error string `json:"error" example:"INVALID_ROLE_ID"`
Details string `json:"details" example:"Role ID must be positive integer"`
}
type deleteRoleErrorRoleInUse struct {
Error string `json:"error" example:"FAILED_TO_DELETE_ROLE"`
Details string `json:"details" example:"Role with ID 123 is assigned to users and cannot be deleted"`
}

View File

@@ -65,7 +65,7 @@ type registerRequest struct {
} }
type registerResponse struct { type registerResponse struct {
UserID int64 `json:"id"` UserID uint `json:"id"`
Username string `json:"username"` Username string `json:"username"`
} }
@@ -92,6 +92,7 @@ func (h *authHandler) handleRegister(w http.ResponseWriter, r *http.Request) {
http.Error(w, "Failed to encode response", http.StatusInternalServerError) http.Error(w, "Failed to encode response", http.StatusInternalServerError)
return return
} }
w.WriteHeader(http.StatusCreated)
} }
type loginRequest struct { type loginRequest struct {
@@ -152,7 +153,7 @@ func (h *authHandler) handleLogout(w http.ResponseWriter, r *http.Request) {
} }
type meResponse struct { type meResponse struct {
UserID int64 `json:"id"` UserID uint `json:"id"`
Username string `json:"username"` Username string `json:"username"`
Email string `json:"email"` Email string `json:"email"`
} }

View File

@@ -202,7 +202,7 @@ var serveCmd = &cobra.Command{
} }
// also acl !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!! // also acl !!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!
userData, err := gorm.Open(sqlite.Open(filepath.Join(cfg.Data.DataPath, "user_data.sqlite3")), &gorm.Config{}) userData, err := gorm.Open(sqlite.Open(filepath.Join(cfg.Data.DataPath, "user_data.sqlite3")+"?_foreign_keys=on"), &gorm.Config{})
if err != nil { if err != nil {
slog.Error("Failed to open user database", slog.String("error", err.Error())) slog.Error("Failed to open user database", slog.String("error", err.Error()))
return return

21
internal/acl/errors.go Normal file
View File

@@ -0,0 +1,21 @@
package acl
// TODO: add more specific errors
import "fmt"
var (
ErrNotInitialized = fmt.Errorf("acl service is not initialized")
ErrRoleNotFound = fmt.Errorf("role not found")
ErrRoleAlreadyExists = fmt.Errorf("role already exists")
ErrInvalidRoleName = fmt.Errorf("role name is invalid")
ErrSameRoleName = fmt.Errorf("role name is the same as another role")
ErrRoleInUse = fmt.Errorf("role is in use")
ErrResourceNotFound = fmt.Errorf("resource not found")
ErrResourceAlreadyExists = fmt.Errorf("resource already exists")
ErrInvalidResourceKey = fmt.Errorf("invalid resource key")
ErrResourceInUse = fmt.Errorf("resource is in use")
ErrSameResourceKey = fmt.Errorf("resource key is the same as another resource")
)

View File

@@ -1,8 +1,8 @@
package acl package acl
type UserRole struct { type UserRole struct {
UserID uint `gorm:"primaryKey" json:"userId"` UserID uint `gorm:"index;not null;uniqueIndex:ux_user_role"`
RoleID uint `gorm:"primaryKey" json:"roleId"` RoleID uint `gorm:"index;not null;uniqueIndex:ux_user_role"`
Role Role `gorm:"constraint:OnDelete:CASCADE;foreignKey:RoleID;references:ID" json:"role"` Role Role `gorm:"constraint:OnDelete:CASCADE;foreignKey:RoleID;references:ID" json:"role"`
//User user.User `gorm:"constraint:OnDelete:CASCADE;foreignKey:UserID;references:ID"` //User user.User `gorm:"constraint:OnDelete:CASCADE;foreignKey:UserID;references:ID"`

140
internal/acl/resources.go Normal file
View File

@@ -0,0 +1,140 @@
package acl
import (
"errors"
"fmt"
"strings"
"gorm.io/gorm"
)
// GetResources returns all resources.
// May return [ErrNotInitialized] or db error.
func (s *Service) GetResources() ([]Resource, error) {
if !s.isInitialized() {
return nil, ErrNotInitialized
}
var resources []Resource
if err := s.db.Order("id").Find(&resources).Error; err != nil {
return nil, fmt.Errorf("db error: %w", err)
}
return resources, nil
}
// CreateResource creates a new resource with the given key or returns existing one.
// Returns ID of created resource.
// May return [ErrNotInitialized], [ErrInvalidResourceKey], [ErrResourceAlreadyExists] or db error.
func (s *Service) CreateResource(key string) (uint, error) {
if !s.isInitialized() {
return 0, ErrNotInitialized
}
key = strings.TrimSpace(key)
if key == "" {
return 0, ErrInvalidResourceKey
}
var res Resource
if err := s.db.Where("key = ?", key).First(&res).Error; err == nil {
// already exists
return res.ID, ErrResourceAlreadyExists
} else if !errors.Is(err, gorm.ErrRecordNotFound) {
// other db error
return 0, fmt.Errorf("db error: %w", err)
}
res = Resource{Key: key}
if err := s.db.Create(&res).Error; err != nil {
return 0, fmt.Errorf("db error: %w", err)
}
return res.ID, nil
}
// GetResourceByID returns the resource with the given ID.
// May return [ErrNotInitialized], [ErrResourceNotFound] or db error.
func (s *Service) GetResourceByID(resourceID uint) (*Resource, error) {
if !s.isInitialized() {
return nil, ErrNotInitialized
}
var res Resource
if err := s.db.First(&res, resourceID).Error; err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return nil, ErrResourceNotFound
}
return nil, fmt.Errorf("db error: %w", err)
}
return &res, nil
}
// UpdateResource updates the key of a resource.
// May return [ErrNotInitialized], [ErrInvalidResourceKey], [ErrResourceNotFound], [ErrSameResourceKey] or db error.
func (s *Service) UpdateResource(resourceID uint, newKey string) error {
if !s.isInitialized() {
return ErrNotInitialized
}
newKey = strings.TrimSpace(newKey)
if newKey == "" {
return ErrInvalidResourceKey
}
var res Resource
if err := s.db.First(&res, resourceID).Error; err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return ErrResourceNotFound
}
return fmt.Errorf("db error: %w", err)
}
// same key?
if res.Key == newKey {
return ErrSameResourceKey
}
// check if key used by another resource
var count int64
if err := s.db.Model(&Resource{}).
Where("key = ? AND id != ?", newKey, resourceID).
Count(&count).Error; err != nil {
return fmt.Errorf("db error: %w", err)
}
if count > 0 {
return ErrSameResourceKey
}
res.Key = newKey
if err := s.db.Save(&res).Error; err != nil {
return fmt.Errorf("failed to update resource: %w", err)
}
return nil
}
// DeleteResource deletes a resource.
// May return [ErrNotInitialized], [ErrResourceNotFound], [ErrResourceInUse] or db error.
func (s *Service) DeleteResource(resourceID uint) error {
if !s.isInitialized() {
return ErrNotInitialized
}
result := s.db.Delete(&Resource{}, resourceID)
if err := result.Error; err != nil {
if strings.Contains(err.Error(), "FOREIGN KEY constraint failed") {
return ErrResourceInUse
}
return fmt.Errorf("db error: %w", err)
}
if result.RowsAffected == 0 {
return ErrResourceNotFound
}
return nil
}

136
internal/acl/roles.go Normal file
View File

@@ -0,0 +1,136 @@
package acl
import (
"errors"
"fmt"
"strings"
"gorm.io/gorm"
)
// GetRoles returns all roles.
// May return [ErrNotInitialized] or db error.
func (s *Service) GetRoles() ([]Role, error) {
if !s.isInitialized() {
return nil, ErrNotInitialized
}
var roles []Role
if err := s.db.Preload("Resources").Order("id").Find(&roles).Error; err != nil {
return nil, fmt.Errorf("db error: %w", err)
}
return roles, nil
}
// CreateRole creates a new role with the given name or returns existing one.
// Returns the ID of the created role.
// May return [ErrNotInitialized], [ErrInvalidRoleName], [ErrRoleAlreadyExists] or db error.
func (s *Service) CreateRole(name string) (uint, error) {
if !s.isInitialized() {
return 0, ErrNotInitialized
}
name = strings.TrimSpace(name)
if name == "" {
return 0, ErrInvalidRoleName
}
var role Role
if err := s.db.Where("name = ?", name).First(&role).Error; err == nil {
// already exists
return role.ID, ErrRoleAlreadyExists
} else if !errors.Is(err, gorm.ErrRecordNotFound) {
// other database error
return 0, fmt.Errorf("db error: %w", err)
}
role = Role{Name: name}
if err := s.db.Create(&role).Error; err != nil {
return 0, fmt.Errorf("db error: %w", err)
}
return role.ID, nil
}
// GetRoleByID returns the role with the given ID or an error.
// May return [ErrNotInitialized], [ErrRoleNotFound] or db error.
func (s *Service) GetRoleByID(roleID uint) (*Role, error) {
if !s.isInitialized() {
return nil, ErrNotInitialized
}
var role Role
err := s.db.Preload("Resources").First(&role, roleID).Error
if err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return nil, ErrRoleNotFound
}
return nil, fmt.Errorf("db error: %w", err)
}
return &role, nil
}
// UpdateRole updates the name of a role.
// May return [ErrNotInitialized], [ErrInvalidRoleName], [ErrRoleNotFound], [ErrSameRoleName], or db error.
func (s *Service) UpdateRole(roleID uint, newName string) error {
if !s.isInitialized() {
return ErrNotInitialized
}
newName = strings.TrimSpace(newName)
if newName == "" {
return ErrInvalidRoleName
}
var role Role
err := s.db.First(&role, roleID).Error
if err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return ErrRoleNotFound
}
return fmt.Errorf("db error: %w", err)
}
// check for name conflicts
if role.Name == newName {
return ErrSameRoleName
}
var count int64
err = s.db.Model(&Role{}).Where("name = ? AND id != ?", newName, roleID).Count(&count).Error
if err != nil {
return fmt.Errorf("db error: %w", err)
}
if count > 0 {
return ErrSameRoleName
}
role.Name = newName
if err := s.db.Save(&role).Error; err != nil {
return fmt.Errorf("failed to update role: %w", err)
}
return nil
}
// DeleteRole deletes a role.
// May return [ErrNotInitialized], [ErrRoleNotFound], [ErrRoleInUse] or db error.
func (s *Service) DeleteRole(roleID uint) error {
if !s.isInitialized() {
return ErrNotInitialized
}
result := s.db.Delete(&Role{}, roleID)
if err := result.Error; err != nil {
if strings.Contains(err.Error(), "FOREIGN KEY constraint failed") {
return ErrRoleInUse
}
return fmt.Errorf("db error: %w", err)
}
if result.RowsAffected == 0 {
return ErrRoleNotFound
}
return nil
}

View File

@@ -1,6 +1,7 @@
package acl package acl
import ( import (
"errors"
"fmt" "fmt"
"gorm.io/gorm" "gorm.io/gorm"
@@ -40,30 +41,14 @@ func (s *Service) Init() error {
return nil return nil
} }
// Admin crud functions // Admin crud functions //
// CreateRole creates a new role with the given name // Resources
func (s *Service) CreateRole(name string) error {
if !s.isInitialized() {
return fmt.Errorf("acl service is not initialized")
}
role := Role{Name: name}
return s.db.FirstOrCreate(&role, &Role{Name: name}).Error
}
// CreateResource creates a new resource with the given key
func (s *Service) CreateResource(key string) error {
if !s.isInitialized() {
return fmt.Errorf("acl service is not initialized")
}
res := Resource{Key: key}
return s.db.FirstOrCreate(&res, &Resource{Key: key}).Error
}
// AssignResourceToRole assigns a resource to a role // AssignResourceToRole assigns a resource to a role
func (s *Service) AssignResourceToRole(roleID, resourceID uint) error { func (s *Service) AssignResourceToRole(roleID, resourceID uint) error {
if !s.isInitialized() { if !s.isInitialized() {
return fmt.Errorf("acl service is not initialized") return ErrNotInitialized
} }
rr := RoleResource{ rr := RoleResource{
RoleID: roleID, RoleID: roleID,
@@ -75,19 +60,25 @@ func (s *Service) AssignResourceToRole(roleID, resourceID uint) error {
// AssignRoleToUser assigns a role to a user // AssignRoleToUser assigns a role to a user
func (s *Service) AssignRoleToUser(roleID, userID uint) error { func (s *Service) AssignRoleToUser(roleID, userID uint) error {
if !s.isInitialized() { if !s.isInitialized() {
return fmt.Errorf("acl service is not initialized") return ErrNotInitialized
} }
ur := UserRole{ ur := UserRole{
UserID: userID, UserID: userID,
RoleID: roleID, RoleID: roleID,
} }
return s.db.FirstOrCreate(&ur, UserRole{UserID: userID, RoleID: roleID}).Error if err := s.db.Create(&ur).Error; err != nil {
if errors.Is(err, gorm.ErrDuplicatedKey) {
return fmt.Errorf("role already assigned to user")
}
return err
}
return nil
} }
// RemoveResourceFromRole removes a resource from a role // RemoveResourceFromRole removes a resource from a role
func (s *Service) RemoveResourceFromRole(roleID, resourceID uint) error { func (s *Service) RemoveResourceFromRole(roleID, resourceID uint) error {
if !s.isInitialized() { if !s.isInitialized() {
return fmt.Errorf("acl service is not initialized") return ErrNotInitialized
} }
return s.db.Where("role_id = ? AND resource_id = ?", roleID, resourceID).Delete(&RoleResource{}).Error return s.db.Where("role_id = ? AND resource_id = ?", roleID, resourceID).Delete(&RoleResource{}).Error
} }
@@ -95,35 +86,15 @@ func (s *Service) RemoveResourceFromRole(roleID, resourceID uint) error {
// RemoveRoleFromUser removes a role from a user // RemoveRoleFromUser removes a role from a user
func (s *Service) RemoveRoleFromUser(roleID, userID uint) error { func (s *Service) RemoveRoleFromUser(roleID, userID uint) error {
if !s.isInitialized() { if !s.isInitialized() {
return fmt.Errorf("acl service is not initialized") return ErrNotInitialized
} }
return s.db.Where("role_id = ? AND user_id = ?", roleID, userID).Delete(&UserRole{}).Error return s.db.Where("role_id = ? AND user_id = ?", roleID, userID).Delete(&UserRole{}).Error
} }
// GetRoles returns all roles
func (s *Service) GetRoles() ([]Role, error) {
if !s.isInitialized() {
return nil, fmt.Errorf("acl service is not initialized")
}
var roles []Role
err := s.db.Preload("Resources").Order("id").Find(&roles).Error
return roles, err
}
// GetPermissions returns all permissions
func (s *Service) GetPermissions() ([]Resource, error) {
if !s.isInitialized() {
return nil, fmt.Errorf("acl service is not initialized")
}
var resources []Resource
err := s.db.Order("id").Find(&resources).Error
return resources, err
}
// GetRoleResources returns all resources for a given role // GetRoleResources returns all resources for a given role
func (s *Service) GetRoleResources(roleID uint) ([]Resource, error) { func (s *Service) GetRoleResources(roleID uint) ([]Resource, error) {
if !s.isInitialized() { if !s.isInitialized() {
return nil, fmt.Errorf("acl service is not initialized") return nil, ErrNotInitialized
} }
var resources []Resource var resources []Resource
err := s.db.Joins("JOIN role_resources rr ON rr.resource_id = resources.id"). err := s.db.Joins("JOIN role_resources rr ON rr.resource_id = resources.id").
@@ -134,7 +105,7 @@ func (s *Service) GetRoleResources(roleID uint) ([]Resource, error) {
// GetUserRoles returns all roles for a given user // GetUserRoles returns all roles for a given user
func (s *Service) GetUserRoles(userID uint) ([]Role, error) { func (s *Service) GetUserRoles(userID uint) ([]Role, error) {
if !s.isInitialized() { if !s.isInitialized() {
return nil, fmt.Errorf("acl service is not initialized") return nil, ErrNotInitialized
} }
var roles []Role var roles []Role
err := s.db.Joins("JOIN user_roles ur ON ur.role_id = roles.id"). err := s.db.Joins("JOIN user_roles ur ON ur.role_id = roles.id").

View File

@@ -1,156 +1,158 @@
package acl_test package acl_test
import ( // DEPRECATED TEST FILE
"os"
"path/filepath"
"testing"
"git.oblat.lv/alex/triggerssmith/internal/acl" // import (
"git.oblat.lv/alex/triggerssmith/internal/user" // "os"
"gorm.io/driver/sqlite" // "path/filepath"
"gorm.io/gorm" // "testing"
)
func openTestDB(t *testing.T) *gorm.DB { // "git.oblat.lv/alex/triggerssmith/internal/acl"
t.Helper() // "git.oblat.lv/alex/triggerssmith/internal/user"
// "gorm.io/driver/sqlite"
// "gorm.io/gorm"
// )
// Путь к файлу базы // func openTestDB(t *testing.T) *gorm.DB {
dbPath := filepath.Join("testdata", "test.db") // t.Helper()
// Удаляем старую базу, если есть // // Путь к файлу базы
os.Remove(dbPath) // dbPath := filepath.Join("testdata", "test.db")
db, err := gorm.Open(sqlite.Open(dbPath), &gorm.Config{}) // // Удаляем старую базу, если есть
if err != nil { // os.Remove(dbPath)
t.Fatalf("failed to open test db: %v", err)
}
// Миграция таблицы User для связи с ACL // db, err := gorm.Open(sqlite.Open(dbPath), &gorm.Config{})
if err := db.AutoMigrate(&user.User{}); err != nil { // if err != nil {
t.Fatalf("failed to migrate User: %v", err) // t.Fatalf("failed to open test db: %v", err)
} // }
return db // // Миграция таблицы User для связи с ACL
} // if err := db.AutoMigrate(&user.User{}); err != nil {
// t.Fatalf("failed to migrate User: %v", err)
// }
func TestACLService_CRUD(t *testing.T) { // return db
db := openTestDB(t) // }
// Создаём сервис ACL // func TestACLService_CRUD(t *testing.T) {
svc, err := acl.NewService(db) // db := openTestDB(t)
if err != nil {
t.Fatalf("failed to create ACL service: %v", err)
}
if err := svc.Init(); err != nil { // // Создаём сервис ACL
t.Fatalf("failed to init ACL service: %v", err) // svc, err := acl.NewService(db)
} // if err != nil {
// t.Fatalf("failed to create ACL service: %v", err)
// }
// Создаём роли // if err := svc.Init(); err != nil {
if err := svc.CreateRole("admin"); err != nil { // t.Fatalf("failed to init ACL service: %v", err)
t.Fatalf("CreateRole failed: %v", err) // }
}
if err := svc.CreateRole("guest"); err != nil {
t.Fatalf("CreateRole failed: %v", err)
}
roles, err := svc.GetRoles() // // Создаём роли
if err != nil { // if err := svc.CreateRole("admin"); err != nil {
t.Fatalf("GetRoles failed: %v", err) // t.Fatalf("CreateRole failed: %v", err)
} // }
if len(roles) != 2 { // if err := svc.CreateRole("guest"); err != nil {
t.Fatalf("expected 2 roles, got %d", len(roles)) // t.Fatalf("CreateRole failed: %v", err)
} // }
// Создаём ресурсы // roles, err := svc.GetRoles()
if err := svc.CreateResource("*"); err != nil { // if err != nil {
t.Fatalf("CreateResource failed: %v", err) // t.Fatalf("GetRoles failed: %v", err)
} // }
if err := svc.CreateResource("html.view.*"); err != nil { // if len(roles) != 2 {
t.Fatalf("CreateResource failed: %v", err) // t.Fatalf("expected 2 roles, got %d", len(roles))
} // }
resources, err := svc.GetPermissions() // // Создаём ресурсы
if err != nil { // if err := svc.CreateResource("*"); err != nil {
t.Fatalf("GetPermissions failed: %v", err) // t.Fatalf("CreateResource failed: %v", err)
} // }
if len(resources) != 2 { // if err := svc.CreateResource("html.view.*"); err != nil {
t.Fatalf("expected 2 resources, got %d", len(resources)) // t.Fatalf("CreateResource failed: %v", err)
} // }
// 1. Создаём сервис user // resources, err := svc.GetPermissions()
store, err := user.NewGormUserStore(db) // if err != nil {
if err != nil { // t.Fatalf("GetPermissions failed: %v", err)
t.Fatalf("failed to create user store: %v", err) // }
} // if len(resources) != 2 {
userSvc, err := user.NewService(store) // t.Fatalf("expected 2 resources, got %d", len(resources))
if err != nil { // }
t.Fatalf("failed to create user service: %v", err)
}
// 2. Инициализируем // // 1. Создаём сервис user
if err := userSvc.Init(); err != nil { // store, err := user.NewGormUserStore(db)
t.Fatalf("failed to init user service: %v", err) // if err != nil {
} // t.Fatalf("failed to create user store: %v", err)
// }
// userSvc, err := user.NewService(store)
// if err != nil {
// t.Fatalf("failed to create user service: %v", err)
// }
user := &user.User{ // // 2. Инициализируем
Username: "testuser", // if err := userSvc.Init(); err != nil {
Email: "testuser@example.com", // t.Fatalf("failed to init user service: %v", err)
Password: "secret", // }
}
u := user // user := &user.User{
// Username: "testuser",
// Email: "testuser@example.com",
// Password: "secret",
// }
// 3. Создаём пользователя через сервис // u := user
err = userSvc.Create(user)
if err != nil {
t.Fatalf("failed to create user: %v", err)
}
// Привязываем роль к пользователю // // 3. Создаём пользователя через сервис
adminRoleID := roles[0].ID // err = userSvc.Create(user)
if err := svc.AssignRoleToUser(adminRoleID, uint(u.ID)); err != nil { // if err != nil {
t.Fatalf("AssignRoleToUser failed: %v", err) // t.Fatalf("failed to create user: %v", err)
} // }
userRoles, err := svc.GetUserRoles(uint(u.ID)) // // Привязываем роль к пользователю
if err != nil { // adminRoleID := roles[0].ID
t.Fatalf("GetUserRoles failed: %v", err) // if err := svc.AssignRoleToUser(adminRoleID, uint(u.ID)); err != nil {
} // t.Fatalf("AssignRoleToUser failed: %v", err)
if len(userRoles) != 1 || userRoles[0].ID != adminRoleID { // }
t.Fatalf("expected user to have admin role")
}
// Привязываем ресурсы к роли // userRoles, err := svc.GetUserRoles(uint(u.ID))
for _, res := range resources { // if err != nil {
if err := svc.AssignResourceToRole(adminRoleID, res.ID); err != nil { // t.Fatalf("GetUserRoles failed: %v", err)
t.Fatalf("AssignResourceToRole failed: %v", err) // }
} // if len(userRoles) != 1 || userRoles[0].ID != adminRoleID {
} // t.Fatalf("expected user to have admin role")
// }
roleResources, err := svc.GetRoleResources(adminRoleID) // // Привязываем ресурсы к роли
if err != nil { // for _, res := range resources {
t.Fatalf("GetRoleResources failed: %v", err) // if err := svc.AssignResourceToRole(adminRoleID, res.ID); err != nil {
} // t.Fatalf("AssignResourceToRole failed: %v", err)
if len(roleResources) != 2 { // }
t.Fatalf("expected role to have 2 resources") // }
}
// Удаляем ресурс из роли // roleResources, err := svc.GetRoleResources(adminRoleID)
if err := svc.RemoveResourceFromRole(adminRoleID, resources[0].ID); err != nil { // if err != nil {
t.Fatalf("RemoveResourceFromRole failed: %v", err) // t.Fatalf("GetRoleResources failed: %v", err)
} // }
roleResources, _ = svc.GetRoleResources(adminRoleID) // if len(roleResources) != 2 {
if len(roleResources) != 1 { // t.Fatalf("expected role to have 2 resources")
t.Fatalf("expected 1 resource after removal") // }
}
// Удаляем роль у пользователя // // Удаляем ресурс из роли
if err := svc.RemoveRoleFromUser(adminRoleID, uint(u.ID)); err != nil { // if err := svc.RemoveResourceFromRole(adminRoleID, resources[0].ID); err != nil {
t.Fatalf("RemoveRoleFromUser failed: %v", err) // t.Fatalf("RemoveResourceFromRole failed: %v", err)
} // }
userRoles, _ = svc.GetUserRoles(uint(u.ID)) // roleResources, _ = svc.GetRoleResources(adminRoleID)
if len(userRoles) != 0 { // if len(roleResources) != 1 {
t.Fatalf("expected user to have 0 roles after removal") // t.Fatalf("expected 1 resource after removal")
} // }
}
// // Удаляем роль у пользователя
// if err := svc.RemoveRoleFromUser(adminRoleID, uint(u.ID)); err != nil {
// t.Fatalf("RemoveRoleFromUser failed: %v", err)
// }
// userRoles, _ = svc.GetUserRoles(uint(u.ID))
// if len(userRoles) != 0 {
// t.Fatalf("expected user to have 0 roles after removal")
// }
// }

20
internal/server/error.go Normal file
View File

@@ -0,0 +1,20 @@
package server
import (
"encoding/json"
"net/http"
)
type ErrorResponse struct {
Error string `json:"error"`
Details string `json:"details,omitempty"`
}
func WriteError(w http.ResponseWriter, error, details string, statusCode int) {
w.Header().Set("Content-Type", "application/json")
w.WriteHeader(statusCode)
json.NewEncoder(w).Encode(ErrorResponse{
Error: error,
Details: details,
})
}

View File

@@ -6,7 +6,7 @@ import (
) )
type User struct { type User struct {
ID int64 `gorm:"primaryKey"` ID uint `gorm:"primaryKey"`
Username string `gorm:"uniqueIndex;not null"` Username string `gorm:"uniqueIndex;not null"`
Email string `gorm:"uniqueIndex;not null"` Email string `gorm:"uniqueIndex;not null"`
Password string `gorm:"not null"` Password string `gorm:"not null"`

View File

@@ -1,84 +1,86 @@
package user package user
import ( // DEPRECATED TEST FILE
"os"
"path/filepath"
"testing"
"gorm.io/driver/sqlite" // import (
"gorm.io/gorm" // "os"
) // "path/filepath"
// "testing"
func setupTestDB(t *testing.T) *gorm.DB { // "gorm.io/driver/sqlite"
t.Helper() // "gorm.io/gorm"
// )
dbPath := filepath.Join("testdata", "users.db") // func setupTestDB(t *testing.T) *gorm.DB {
// t.Helper()
_ = os.Remove(dbPath) // dbPath := filepath.Join("testdata", "users.db")
db, err := gorm.Open(sqlite.Open(dbPath), &gorm.Config{}) // _ = os.Remove(dbPath)
if err != nil {
t.Fatalf("failed to open db: %v", err)
}
if err := db.AutoMigrate(&User{}); err != nil { // db, err := gorm.Open(sqlite.Open(dbPath), &gorm.Config{})
t.Fatalf("failed to migrate: %v", err)
}
return db
}
func TestUsersCRUD(t *testing.T) {
db := setupTestDB(t)
store, err := NewGormUserStore(db)
if err != nil {
t.Fatalf("failed to create store: %v", err)
}
service, err := NewService(store)
if err != nil {
t.Fatalf("failed to create service: %v", err)
}
user := &User{
Username: "testuser",
Email: "test@example.com",
Password: "password123",
}
if err := service.Create(user); err != nil {
t.Fatalf("failed to create user: %v", err)
}
// retrieved, err := service.GetByID(user.ID)
// if err != nil { // if err != nil {
// t.Fatalf("failed to get user by ID: %v", err) // t.Fatalf("failed to open db: %v", err)
// }
// if retrieved.Username != user.Username {
// t.Fatalf("expected username %s, got %s", user.Username, retrieved.Username)
// } // }
// retrievedByUsername, err := service.GetByUsername(user.Username) // if err := db.AutoMigrate(&User{}); err != nil {
// if err != nil { // t.Fatalf("failed to migrate: %v", err)
// t.Fatalf("failed to get user by username: %v", err)
// }
// if retrievedByUsername.Email != user.Email {
// t.Fatalf("expected email %s, got %s", user.Email, retrievedByUsername.Email)
// } // }
// user.Email = "newemail@example.com" // return db
// if err := service.Update(user); err != nil {
// t.Fatalf("failed to update user: %v", err)
// } // }
// retrieved, err = service.GetByID(user.ID)
// func TestUsersCRUD(t *testing.T) {
// db := setupTestDB(t)
// store, err := NewGormUserStore(db)
// if err != nil { // if err != nil {
// t.Fatalf("failed to get user by ID: %v", err) // t.Fatalf("failed to create store: %v", err)
// }
// service, err := NewService(store)
// if err != nil {
// t.Fatalf("failed to create service: %v", err)
// }
// user := &User{
// Username: "testuser",
// Email: "test@example.com",
// Password: "password123",
// }
// if err := service.Create(user); err != nil {
// t.Fatalf("failed to create user: %v", err)
// }
// // retrieved, err := service.GetByID(user.ID)
// // if err != nil {
// // t.Fatalf("failed to get user by ID: %v", err)
// // }
// // if retrieved.Username != user.Username {
// // t.Fatalf("expected username %s, got %s", user.Username, retrieved.Username)
// // }
// // retrievedByUsername, err := service.GetByUsername(user.Username)
// // if err != nil {
// // t.Fatalf("failed to get user by username: %v", err)
// // }
// // if retrievedByUsername.Email != user.Email {
// // t.Fatalf("expected email %s, got %s", user.Email, retrievedByUsername.Email)
// // }
// // user.Email = "newemail@example.com"
// // if err := service.Update(user); err != nil {
// // t.Fatalf("failed to update user: %v", err)
// // }
// // retrieved, err = service.GetByID(user.ID)
// // if err != nil {
// // t.Fatalf("failed to get user by ID: %v", err)
// // }
// // if retrieved.Email != user.Email {
// // t.Fatalf("expected email %s, got %s", user.Email, retrieved.Email)
// // }
// err = service.Delete(user.ID)
// if err != nil {
// t.Fatalf("failed to delete user: %v", err)
// } // }
// if retrieved.Email != user.Email {
// t.Fatalf("expected email %s, got %s", user.Email, retrieved.Email)
// } // }
err = service.Delete(user.ID)
if err != nil {
t.Fatalf("failed to delete user: %v", err)
}
}