doesnm/quickshare
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat(users) add async handler for reset used space

Browse source
This commit is contained in:
hexxa 2022-03-09 15:29:38 +08:00 • committed by Hexxa
parent e05ef9b5fd
commit 4dcd2c56ca
8 changed files with 142 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

51
src/handlers/multiusers/handlers.go
View file

@ -1,6 +1,7 @@
package multiusers
import (
"encoding/json"
"errors"
"fmt"
"path/filepath"
@ -16,6 +17,7 @@ import (
"github.com/ihexxa/quickshare/src/db/userstore"
"github.com/ihexxa/quickshare/src/depidx"
q "github.com/ihexxa/quickshare/src/handlers"
"github.com/ihexxa/quickshare/src/worker/localworker"
)
var (
@ -48,6 +50,7 @@ func NewMultiUsersSvc(cfg gocfg.ICfg, deps *depidx.Deps) (*MultiUsersSvc, error)
apiRuleCname(userstore.AdminRole, "GET", "/v1/users/list"): true,
apiRuleCname(userstore.AdminRole, "GET", "/v1/users/self"): true,
apiRuleCname(userstore.AdminRole, "PATCH", "/v1/users/preferences"): true,
apiRuleCname(userstore.AdminRole, "PUT", "/v1/users//used-space"): true,
apiRuleCname(userstore.AdminRole, "POST", "/v1/roles/"): true,
apiRuleCname(userstore.AdminRole, "DELETE", "/v1/roles/"): true,
apiRuleCname(userstore.AdminRole, "GET", "/v1/roles/list"): true,
@ -77,6 +80,7 @@ func NewMultiUsersSvc(cfg gocfg.ICfg, deps *depidx.Deps) (*MultiUsersSvc, error)
apiRuleCname(userstore.AdminRole, "GET", "/v1/fs/sharings/dirs"): true,
apiRuleCname(userstore.AdminRole, "GET", "/v1/fs/sharings/ids"): true,
apiRuleCname(userstore.AdminRole, "POST", "/v1/fs/hashes/sha1"): true,
// user rules
apiRuleCname(userstore.UserRole, "GET", "/"): true,
apiRuleCname(userstore.UserRole, "GET", publicPath): true,
@ -125,11 +129,14 @@ func NewMultiUsersSvc(cfg gocfg.ICfg, deps *depidx.Deps) (*MultiUsersSvc, error)
apiRuleCname(userstore.VisitorRole, "GET", "/v1/fs/sharings/dirs"): true,
}
return &MultiUsersSvc{
handlers := &MultiUsersSvc{
cfg: cfg,
deps: deps,
apiACRules: apiACRules,
}, nil
}
deps.Workers().AddHandler(MsgTypeResetUsedSpace, handlers.resetUsedSpace)
return handlers, nil
}
func (h *MultiUsersSvc) Init(adminName, adminPwd string) (string, error) {
@ -707,3 +714,43 @@ func (h *MultiUsersSvc) SetPreferences(c *gin.Context) {
}
c.JSON(q.Resp(200))
}
type ResetUsedSpaceReq struct {
UserID uint64 `json:"userID,string"`
}
func (h *MultiUsersSvc) ResetUsedSpace(c *gin.Context) {
req := &ResetUsedSpaceReq{}
if err := c.ShouldBindJSON(&req); err != nil {
c.JSON(q.ErrResp(c, 400, err))
return
}
userInfo, err := h.deps.Users().GetUser(req.UserID)
if err != nil {
c.JSON(q.ErrResp(c, 500, err))
return
}
msg, err := json.Marshal(UsedSpaceParams{
UserID: req.UserID,
UserHomePath: userInfo.Name,
})
if err != nil {
c.JSON(q.ErrResp(c, 500, err))
return
}
err = h.deps.Workers().TryPut(
localworker.NewMsg(
h.deps.ID().Gen(),
map[string]string{localworker.MsgTypeKey: MsgTypeResetUsedSpace},
string(msg),
),
)
if err != nil {
c.JSON(q.ErrResp(c, 500, err))
return
}
c.JSON(q.Resp(200))
}