package controllers import ( "strconv" "strings" "server/middleware" "server/models" "server/pkg/identity" "server/pkg/jwtutil" "github.com/beego/beego/v2/client/orm" "github.com/beego/beego/v2/core/logs" beego "github.com/beego/beego/v2/server/web" ) // ==================== 移动端 PasswordStore ==================== // // 移动端「账号管理」(密码库)接口。与 Web 端 /backend/passwordStore 同一套数据 // (yz_backend_password_store,一个平台下挂多个账号),按 JWT 中的 tid + user_id // 隔离,小程序「账号管理」页面调用。 type AppPasswordStoreController struct { beego.Controller } // 只放行移动端与租户端令牌:/api/** 仅被 uniapp 调用,平台令牌不该能打移动端接口。 func (c *AppPasswordStoreController) appPwdClaims() (*jwtutil.Claims, error) { claims, err := identity.Claims(c.Ctx.Request.Header.Get("Authorization"), identity.EndApp, identity.EndBackend) if err != nil { return nil, err } if claims.UserID <= 0 || claims.TenantId <= 0 { return nil, orm.ErrNoRows } return claims, nil } // GetList GET /api/passwordStore/list?keyword= func (c *AppPasswordStoreController) GetList() { cl, e := c.appPwdClaims() if e != nil { passwordReply(&c.Controller, 401, "未登录或无权限", nil) return } qs := models.Orm.QueryTable(new(models.BackendPasswordStore)). Filter("is_deleted", 0).Filter("tid", cl.TenantId).Filter("user_id", cl.UserID) if k := strings.TrimSpace(c.GetString("keyword")); k != "" { cond := orm.NewCondition(). Or("platform__icontains", k). Or("url__icontains", k). Or("accounts__icontains", k). Or("remark__icontains", k) qs = qs.SetCond(orm.NewCondition().And("is_deleted", 0).And("tid", cl.TenantId).And("user_id", cl.UserID).AndCond(cond)) } var rows []models.BackendPasswordStore if _, e = qs.OrderBy("-id").All(&rows); e != nil && e != orm.ErrNoRows { logs.Error("[passwordStore] app list failed: %v", e) passwordReply(&c.Controller, 500, "查询失败: "+e.Error(), nil) return } res := make([]PasswordStoreItemDTO, 0, len(rows)) for _, row := range rows { res = append(res, PasswordStoreItemDTO{ ID: row.ID, Tid: row.Tid, Platform: row.Platform, URL: row.URL, Accounts: accountsFromJSON(row.Accounts), Remark: row.Remark, UserID: row.UserID, CreateTime: row.CreateTime, UpdateTime: row.UpdateTime, }) } passwordReply(&c.Controller, 200, "success", map[string]interface{}{"list": res, "total": len(res)}) } // GetDetail GET /api/passwordStore/detail/:id func (c *AppPasswordStoreController) GetDetail() { cl, e := c.appPwdClaims() if e != nil { passwordReply(&c.Controller, 401, "未登录或无权限", nil) return } id, e := strconv.Atoi(c.Ctx.Input.Param(":id")) if e != nil { passwordReply(&c.Controller, 400, "参数错误", nil) return } var m models.BackendPasswordStore e = models.Orm.QueryTable(new(models.BackendPasswordStore)). Filter("id", id).Filter("is_deleted", 0).Filter("tid", cl.TenantId).Filter("user_id", cl.UserID).One(&m) if e == orm.ErrNoRows { passwordReply(&c.Controller, 404, "记录不存在", nil) return } if e != nil { logs.Error("[passwordStore] app detail failed: %v", e) passwordReply(&c.Controller, 500, "查询失败: "+e.Error(), nil) return } passwordReply(&c.Controller, 200, "success", PasswordStoreItemDTO{ ID: m.ID, Tid: m.Tid, Platform: m.Platform, URL: m.URL, Accounts: accountsFromJSON(m.Accounts), Remark: m.Remark, UserID: m.UserID, CreateTime: m.CreateTime, UpdateTime: m.UpdateTime, }) } // Create POST /api/passwordStore/create func (c *AppPasswordStoreController) Create() { cl, e := c.appPwdClaims() if e != nil { passwordReply(&c.Controller, 401, "未登录或无权限", nil) return } p, e := parsePassword(&c.Controller) if e != nil { passwordReply(&c.Controller, 400, e.Error(), nil) return } if !validPassword(p) { passwordReply(&c.Controller, 400, "请输入平台名称", nil) return } if e := validateAccountsUnique(p.Accounts); e != nil { passwordReply(&c.Controller, 400, e.Error(), nil) return } qs := models.Orm.QueryTable(new(models.BackendPasswordStore)). Filter("is_deleted", 0).Filter("tid", cl.TenantId).Filter("user_id", cl.UserID) if existing, e := findSamePlatformRecord(qs, 0, p.Platform, p.URL); e != nil { passwordReply(&c.Controller, 500, "查询失败: "+e.Error(), nil) return } else if existing > 0 { passwordReply(&c.Controller, 400, "同平台同网址记录已存在,请勿重复添加", nil) return } m := models.BackendPasswordStore{ Tid: cl.TenantId, UserID: uint64(cl.UserID), Platform: p.Platform, URL: p.URL, Accounts: accountsToJSON(p.Accounts), Remark: p.Remark, } if _, e = models.Orm.Insert(&m); e != nil { logs.Error("[passwordStore] app create failed: %v", e) passwordReply(&c.Controller, 500, "创建失败: "+e.Error(), nil) return } passwordReply(&c.Controller, 200, "success", PasswordStoreItemDTO{ ID: m.ID, Tid: m.Tid, Platform: m.Platform, URL: m.URL, Accounts: p.Accounts, Remark: m.Remark, UserID: m.UserID, CreateTime: m.CreateTime, UpdateTime: m.UpdateTime, }) } // Update POST /api/passwordStore/update/:id func (c *AppPasswordStoreController) Update() { cl, e := c.appPwdClaims() if e != nil { passwordReply(&c.Controller, 401, "未登录或无权限", nil) return } id, e := strconv.Atoi(c.Ctx.Input.Param(":id")) if e != nil { passwordReply(&c.Controller, 400, "参数错误", nil) return } p, e := parsePassword(&c.Controller) if e != nil { passwordReply(&c.Controller, 400, e.Error(), nil) return } if !validPassword(p) { passwordReply(&c.Controller, 400, "请输入平台名称", nil) return } if e := validateAccountsUnique(p.Accounts); e != nil { passwordReply(&c.Controller, 400, e.Error(), nil) return } var m models.BackendPasswordStore e = models.Orm.QueryTable(new(models.BackendPasswordStore)). Filter("id", id).Filter("is_deleted", 0).Filter("tid", cl.TenantId).Filter("user_id", cl.UserID).One(&m) if e == orm.ErrNoRows { passwordReply(&c.Controller, 404, "记录不存在", nil) return } if e != nil { logs.Error("[passwordStore] app update query failed: %v", e) passwordReply(&c.Controller, 500, "查询失败: "+e.Error(), nil) return } qs := models.Orm.QueryTable(new(models.BackendPasswordStore)). Filter("is_deleted", 0).Filter("tid", cl.TenantId).Filter("user_id", cl.UserID) if existing, e := findSamePlatformRecord(qs, m.ID, p.Platform, p.URL); e != nil { passwordReply(&c.Controller, 500, "查询失败: "+e.Error(), nil) return } else if existing > 0 { passwordReply(&c.Controller, 400, "同平台同网址记录已存在,请勿重复添加", nil) return } m.Platform = p.Platform m.URL = p.URL m.Remark = p.Remark m.Accounts = accountsToJSON(p.Accounts) if _, e = models.Orm.Update(&m); e != nil { logs.Error("[passwordStore] app update failed: %v", e) passwordReply(&c.Controller, 500, "更新失败: "+e.Error(), nil) return } passwordReply(&c.Controller, 200, "success", PasswordStoreItemDTO{ ID: m.ID, Tid: m.Tid, Platform: m.Platform, URL: m.URL, Accounts: p.Accounts, Remark: m.Remark, UserID: m.UserID, CreateTime: m.CreateTime, UpdateTime: m.UpdateTime, }) } // Delete DELETE /api/passwordStore/delete/:id func (c *AppPasswordStoreController) Delete() { cl, e := c.appPwdClaims() if e != nil { passwordReply(&c.Controller, 401, "未登录或无权限", nil) return } id, e := strconv.Atoi(c.Ctx.Input.Param(":id")) if e != nil { passwordReply(&c.Controller, 400, "参数错误", nil) return } var m models.BackendPasswordStore e = models.Orm.QueryTable(new(models.BackendPasswordStore)). Filter("id", id).Filter("is_deleted", 0).Filter("tid", cl.TenantId).Filter("user_id", cl.UserID).One(&m) if e == orm.ErrNoRows { passwordReply(&c.Controller, 404, "记录不存在", nil) return } if e != nil { logs.Error("[passwordStore] app delete query failed: %v", e) passwordReply(&c.Controller, 500, "查询失败: "+e.Error(), nil) return } if _, e = models.Orm.Delete(&m); e != nil { logs.Error("[passwordStore] app delete failed: %v", e) passwordReply(&c.Controller, 500, "删除失败: "+e.Error(), nil) return } middleware.WriteDeleteLog(uint64(cl.UserID), cl.TenantId, "passwordStore", m.Platform) passwordReply(&c.Controller, 200, "success", nil) }