101 lines
3.3 KiB
Go
101 lines
3.3 KiB
Go
package restricted
|
|
|
|
import (
|
|
"strconv"
|
|
|
|
"git.ma-al.com/goc_daniel/b2b/app/config"
|
|
"git.ma-al.com/goc_daniel/b2b/app/model"
|
|
"git.ma-al.com/goc_daniel/b2b/app/service/storageService"
|
|
"git.ma-al.com/goc_daniel/b2b/app/utils/i18n"
|
|
"git.ma-al.com/goc_daniel/b2b/app/utils/localeExtractor"
|
|
"git.ma-al.com/goc_daniel/b2b/app/utils/nullable"
|
|
"git.ma-al.com/goc_daniel/b2b/app/utils/response"
|
|
"git.ma-al.com/goc_daniel/b2b/app/utils/responseErrors"
|
|
"github.com/gofiber/fiber/v3"
|
|
)
|
|
|
|
type StorageHandler struct {
|
|
storageService *storageService.StorageService
|
|
config *config.Config
|
|
}
|
|
|
|
func NewStorageHandler() *StorageHandler {
|
|
return &StorageHandler{
|
|
storageService: storageService.New(),
|
|
config: config.Get(),
|
|
}
|
|
}
|
|
|
|
func StorageHandlerRoutes(r fiber.Router) fiber.Router {
|
|
handler := NewStorageHandler()
|
|
|
|
// for all users
|
|
r.Get("/list-content/*", handler.ListContent)
|
|
r.Get("/download-file/*", handler.DownloadFile)
|
|
|
|
// for admins only
|
|
r.Get("/create-new-webdav-token", handler.CreateNewWebdavToken)
|
|
|
|
return r
|
|
}
|
|
|
|
// accepted path looks like e.g. "/folder1/" or "folder1"
|
|
func (h *StorageHandler) ListContent(c fiber.Ctx) error {
|
|
// relative path defaults to root directory
|
|
abs_path, err := h.storageService.AbsPath(h.config.Storage.RootFolder, c.Params("*"))
|
|
if err != nil {
|
|
return c.Status(responseErrors.GetErrorStatus(err)).
|
|
JSON(response.Make(nullable.GetNil(""), 0, responseErrors.GetErrorCode(c, err)))
|
|
}
|
|
|
|
entries_in_list, err := h.storageService.ListContent(abs_path)
|
|
|
|
if err != nil {
|
|
return c.Status(responseErrors.GetErrorStatus(err)).
|
|
JSON(response.Make(nullable.GetNil(""), 0, responseErrors.GetErrorCode(c, err)))
|
|
}
|
|
|
|
return c.JSON(response.Make(entries_in_list, 0, i18n.T_(c, response.Message_OK)))
|
|
}
|
|
|
|
func (h *StorageHandler) DownloadFile(c fiber.Ctx) error {
|
|
abs_path, err := h.storageService.AbsPath(h.config.Storage.RootFolder, c.Params("*"))
|
|
if err != nil {
|
|
return c.Status(responseErrors.GetErrorStatus(err)).
|
|
JSON(response.Make(nullable.GetNil(""), 0, responseErrors.GetErrorCode(c, err)))
|
|
}
|
|
|
|
f, filename, filesize, err := h.storageService.DownloadFilePrep(abs_path)
|
|
if err != nil {
|
|
return c.Status(responseErrors.GetErrorStatus(err)).
|
|
JSON(response.Make(nullable.GetNil(""), 0, responseErrors.GetErrorCode(c, err)))
|
|
}
|
|
|
|
c.Attachment(filename)
|
|
c.Set("Content-Length", strconv.FormatInt(filesize, 10))
|
|
c.Set("Content-Type", "application/octet-stream")
|
|
return c.SendStream(f, int(filesize))
|
|
}
|
|
|
|
func (h *StorageHandler) CreateNewWebdavToken(c fiber.Ctx) error {
|
|
userID, ok := localeExtractor.GetUserID(c)
|
|
if !ok {
|
|
return c.Status(responseErrors.GetErrorStatus(responseErrors.ErrInvalidBody)).
|
|
JSON(response.Make(nullable.GetNil(""), 0, responseErrors.GetErrorCode(c, responseErrors.ErrInvalidBody)))
|
|
}
|
|
|
|
userRole, ok := localeExtractor.GetOriginalUserRole(c)
|
|
if !ok || model.CustomerRole(userRole.Name) != model.RoleAdmin {
|
|
return c.Status(responseErrors.GetErrorStatus(responseErrors.ErrAdminAccessRequired)).
|
|
JSON(response.Make(nullable.GetNil(""), 0, responseErrors.GetErrorCode(c, responseErrors.ErrAdminAccessRequired)))
|
|
}
|
|
|
|
new_token, err := h.storageService.NewWebdavToken(userID)
|
|
if err != nil {
|
|
return c.Status(responseErrors.GetErrorStatus(err)).
|
|
JSON(response.Make(nullable.GetNil(""), 0, responseErrors.GetErrorCode(c, err)))
|
|
}
|
|
|
|
return c.JSON(response.Make(&new_token, 0, i18n.T_(c, response.Message_OK)))
|
|
}
|