SetUserBan.js 2.8 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071
  1. const API = require("../../../lib/API")
  2. const db = require("../../../plugin/DataBase/db")
  3. const AccessControl = require("../../../lib/AccessControl")
  4. const { BaseStdResponse } = require("../../../BaseStdResponse")
  5. class SetUserBan extends API {
  6. constructor() {
  7. super()
  8. this.setPath("/Admin/User/SetUserBan")
  9. this.setMethod("POST")
  10. }
  11. async onRequest(req, res) {
  12. let { uuid, session, userid, is_banned } = req.body
  13. const flag = Number(is_banned)
  14. if ([uuid, session, userid].some(value => value === "" || value === null || value === undefined))
  15. return res.json({ ...BaseStdResponse.MISSING_PARAMETER })
  16. if (![0, 1].includes(flag))
  17. return res.json({ ...BaseStdResponse.ERR, msg: "参数错误" })
  18. if (!await AccessControl.checkSession(uuid, session))
  19. return res.status(401).json({ ...BaseStdResponse.ACCESS_DENIED })
  20. const permission = await AccessControl.getPermission(uuid)
  21. if (!permission.includes("admin") && !permission.includes("service"))
  22. return res.json({ ...BaseStdResponse.PERMISSION_DENIED })
  23. if (userid === uuid)
  24. return res.json({ ...BaseStdResponse.ERR, msg: "不能封禁自己的账号" })
  25. const targetRows = await db.query(
  26. "SELECT permission FROM users WHERE uuid = ? LIMIT 1",
  27. [userid]
  28. )
  29. if (!targetRows || targetRows.length === 0)
  30. return res.json({ ...BaseStdResponse.MISSING_FILE, msg: "未找到用户" })
  31. const targetPermission = targetRows[0].permission || []
  32. if (targetPermission.includes("admin") || targetPermission.includes("service"))
  33. return res.json({ ...BaseStdResponse.ERR, msg: "不能封禁管理员或客服账号" })
  34. const conn = await db.connect()
  35. try {
  36. const [r] = await conn.execute(
  37. "UPDATE users SET is_banned = ? WHERE uuid = ?",
  38. [flag, userid]
  39. )
  40. if (!r || r.affectedRows !== 1)
  41. return res.json({ ...BaseStdResponse.MISSING_FILE, msg: "未找到用户或更新失败" })
  42. if (flag === 1)
  43. await AccessControl.invalidateSession(userid)
  44. return res.json({
  45. ...BaseStdResponse.OK,
  46. msg: flag === 1 ? "已封禁该用户" : "已解除封禁"
  47. })
  48. } catch (err) {
  49. this.logger.error(`设置用户封禁状态失败: ${err.message || err}`)
  50. return res.json({ ...BaseStdResponse.ERR, msg: "操作失败,请稍后再试" })
  51. } finally {
  52. if (conn?.connection && typeof conn.connection.release === "function" && typeof conn?.release === "function") {
  53. conn.release()
  54. }
  55. }
  56. }
  57. }
  58. module.exports.SetUserBan = SetUserBan