AiAssistantMcp.js 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713
  1. const db = require('../../plugin/DataBase/db')
  2. const { axiosWithQgOutbound } = require('../Lepao/qgOutboundAxios')
  3. const Logger = require('../Logger')
  4. const path = require('path')
  5. const QxsBookList = require('../../apis/QuXuanShu/GetBookList').GetBookList
  6. const EmailTemplate = require('../../plugin/Email/emailTemplate')
  7. const ASSISTANT_UUID = 'e4fe0277-0b1a-41a1-b25f-8b6e4cec3281'
  8. function required(values) {
  9. return values.some(v => v === '' || v === null || v === undefined)
  10. }
  11. function isProvided(value) {
  12. return value !== '' && value !== null && value !== undefined
  13. }
  14. function pageArgs({ current = 1, pagesize = 10 } = {}, max = 20) {
  15. const c = Math.max(1, Number.parseInt(current, 10) || 1)
  16. const p = Math.min(Math.max(1, Number.parseInt(pagesize, 10) || 10), max)
  17. return { current: c, pagesize: p, offset: (c - 1) * p }
  18. }
  19. function pagination(page, totalRows) {
  20. const total = Number(totalRows?.[0]?.total || 0)
  21. return {
  22. current: page.current,
  23. pagesize: page.pagesize,
  24. total,
  25. total_pages: Math.ceil(total / page.pagesize)
  26. }
  27. }
  28. function asJson(data) {
  29. return JSON.stringify(data)
  30. }
  31. function pad2(value) {
  32. return String(value).padStart(2, '0')
  33. }
  34. function formatBeijingTime(value) {
  35. if (value === '' || value === null || value === undefined) return ''
  36. if (typeof value === 'string' && !/^\d+$/.test(value.trim())) {
  37. const match = value.trim().match(/^(\d{4})[-/](\d{1,2})[-/](\d{1,2})(?:[ T](\d{1,2}):(\d{1,2})(?::(\d{1,2}))?)?/)
  38. if (match && !/[zZ]|[+-]\d{2}:?\d{2}$/.test(value.trim())) {
  39. const [, y, m, d, h = '0', min = '0', s = '0'] = match
  40. return `${y}-${pad2(m)}-${pad2(d)} ${pad2(h)}:${pad2(min)}:${pad2(s)}`
  41. }
  42. }
  43. const numberValue = typeof value === 'number' || /^\d+$/.test(String(value).trim())
  44. ? Number(value)
  45. : null
  46. const millis = numberValue !== null
  47. ? (numberValue < 1000000000000 ? numberValue * 1000 : numberValue)
  48. : new Date(value).getTime()
  49. if (!Number.isFinite(millis)) return String(value)
  50. const date = new Date(millis + 8 * 60 * 60 * 1000)
  51. return `${date.getUTCFullYear()}-${pad2(date.getUTCMonth() + 1)}-${pad2(date.getUTCDate())} ${pad2(date.getUTCHours())}:${pad2(date.getUTCMinutes())}:${pad2(date.getUTCSeconds())}`
  52. }
  53. function parseJson(value, fallback = []) {
  54. if (!value) return fallback
  55. if (Array.isArray(value)) return value
  56. try {
  57. const parsed = JSON.parse(value)
  58. return parsed == null ? fallback : parsed
  59. } catch (_) {
  60. return fallback
  61. }
  62. }
  63. function stateText(value, map) {
  64. return map[value] || String(value)
  65. }
  66. function normalizeText(value, max = 2000) {
  67. return String(value || '').trim().slice(0, max)
  68. }
  69. function normalizeNumber(value) {
  70. if (!isProvided(value)) return null
  71. const numberValue = Number(value)
  72. return Number.isFinite(numberValue) ? numberValue : null
  73. }
  74. function normalizeRunArea(area) {
  75. if (!isProvided(area)) return ''
  76. const value = String(area).trim()
  77. if (value === '随机分配') return ''
  78. const allowed = new Set([
  79. '兰花湖校区跑区',
  80. '主校区北跑区',
  81. '主校区南跑区',
  82. '重庆工商大学茶园校区',
  83. ''
  84. ])
  85. return allowed.has(value) ? value : null
  86. }
  87. function mapLepaoAccount(row) {
  88. return {
  89. student_num: row.student_num,
  90. name: row.name || '',
  91. state: stateText(row.state, { 0: '需登录', 1: '正常', 2: '异常' }),
  92. area: row.area || '随机分配',
  93. auto_run: row.auto_run === 1,
  94. auto_time: row.auto_time,
  95. today_auto_time: row.today_auto_time,
  96. target_count: row.target_count,
  97. total_num: row.total_num,
  98. term_num: row.term_num,
  99. auto_day: parseJson(row.auto_day, []),
  100. notice_type: row.notice_type,
  101. email: row.email || '',
  102. notes: row.notes || ''
  103. }
  104. }
  105. function mapLepaoRecord(row) {
  106. return {
  107. ...row,
  108. time: formatBeijingTime(row.time)
  109. }
  110. }
  111. function mapCountLedger(row) {
  112. return {
  113. ...row,
  114. created_at: formatBeijingTime(row.created_at)
  115. }
  116. }
  117. function mapOrder(row) {
  118. const orderStateMap = {
  119. 0: '待支付',
  120. 1: '已支付',
  121. 2: '已取消',
  122. 3: '异常',
  123. 4: '已退款'
  124. }
  125. return {
  126. order_no: row.orderId,
  127. state: stateText(row.state, orderStateMap),
  128. create_time: formatBeijingTime(row.create_time),
  129. price: row.price,
  130. pay_type: row.pay_type,
  131. goods_name: row.goods_name
  132. }
  133. }
  134. function mapPowerTask(row) {
  135. const { id, ...task } = row
  136. return {
  137. ...task,
  138. koufei_date: formatBeijingTime(task.koufei_date),
  139. update_time: formatBeijingTime(task.update_time)
  140. }
  141. }
  142. function mapPowerRecord(row) {
  143. return {
  144. ...row,
  145. time: formatBeijingTime(row.time),
  146. change_time: formatBeijingTime(row.change_time)
  147. }
  148. }
  149. class AiAssistantMcp {
  150. constructor() {
  151. this.logger = new Logger(path.join(__dirname, '../../logs/AiAssistantMcp.log'), 'INFO')
  152. this.emailRegex = /^[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}$/
  153. this.banEmailList = ['icloud.com']
  154. }
  155. async list_lepao_accounts(args) {
  156. const { sender, student_num, name, area, state, auto_run, current, pagesize } = args
  157. if (required([sender])) return '缺少 sender'
  158. const page = pageArgs({ current, pagesize })
  159. const where = ['create_user = ?']
  160. const params = [sender]
  161. if (student_num) {
  162. where.push('student_num LIKE ?')
  163. params.push(`%${student_num}%`)
  164. }
  165. if (name) {
  166. where.push('name LIKE ?')
  167. params.push(`%${name}%`)
  168. }
  169. if (area !== undefined && area !== null && area !== '') {
  170. const normalizedArea = normalizeRunArea(area)
  171. if (normalizedArea === null) return '跑区不合法'
  172. where.push('area = ?')
  173. params.push(normalizedArea)
  174. }
  175. if (state !== undefined && state !== null && state !== '' && Number(state) !== -1) {
  176. where.push('state = ?')
  177. params.push(state)
  178. }
  179. if (auto_run !== undefined && auto_run !== null && auto_run !== '') {
  180. where.push('auto_run = ?')
  181. params.push(Number(auto_run))
  182. }
  183. const totalRows = await db.query(
  184. `SELECT COUNT(*) AS total
  185. FROM lepao_account
  186. WHERE ${where.join(' AND ')}`,
  187. params
  188. )
  189. const rows = await db.query(
  190. `SELECT student_num, name, state, area, auto_run, auto_time, today_auto_time,
  191. target_count, total_num, term_num, auto_day, notice_type, email, notes
  192. FROM lepao_account
  193. WHERE ${where.join(' AND ')}
  194. ORDER BY update_time DESC
  195. LIMIT ? OFFSET ?`,
  196. [...params, String(page.pagesize), String(page.offset)]
  197. )
  198. if (!rows || !totalRows) return '查询失败'
  199. return asJson({ list: rows.map(mapLepaoAccount), pagination: pagination(page, totalRows) })
  200. }
  201. async list_lepao_records(args) {
  202. const { sender, lepao_account, name, start_time, end_time, result, current, pagesize } = args
  203. if (required([sender])) return '缺少 sender'
  204. const page = pageArgs({ current, pagesize })
  205. const where = ['(r.uuid = ? OR a.create_user = ?)']
  206. const params = [sender, sender]
  207. if (lepao_account) {
  208. where.push('r.lepao_account LIKE ?')
  209. params.push(`%${lepao_account}%`)
  210. }
  211. if (name) {
  212. where.push('a.name LIKE ?')
  213. params.push(`%${name}%`)
  214. }
  215. if (result) {
  216. where.push('r.result LIKE ?')
  217. params.push(`%${result}%`)
  218. }
  219. if (start_time) {
  220. where.push('r.time >= ?')
  221. params.push(start_time)
  222. }
  223. if (end_time) {
  224. where.push('r.time < ?')
  225. params.push(end_time)
  226. }
  227. const totalRows = await db.query(
  228. `SELECT COUNT(*) AS total
  229. FROM lepao_record r
  230. LEFT JOIN lepao_account a ON r.lepao_account = a.student_num
  231. WHERE ${where.join(' AND ')}`,
  232. params
  233. )
  234. const rows = await db.query(
  235. `SELECT r.public_id, r.time, r.result, r.run_mode, r.lepao_account, a.name
  236. FROM lepao_record r
  237. LEFT JOIN lepao_account a ON r.lepao_account = a.student_num
  238. WHERE ${where.join(' AND ')}
  239. ORDER BY r.id DESC
  240. LIMIT ? OFFSET ?`,
  241. [...params, String(page.pagesize), String(page.offset)]
  242. )
  243. if (!rows || !totalRows) return '查询失败'
  244. return asJson({ list: rows.map(mapLepaoRecord), pagination: pagination(page, totalRows) })
  245. }
  246. async list_lepao_count_ledger(args) {
  247. const { sender, biz_type, remark, start_time, end_time, current, pagesize } = args
  248. if (required([sender])) return '缺少 sender'
  249. const page = pageArgs({ current, pagesize })
  250. const where = ['user_uuid = ?']
  251. const params = [sender]
  252. if (biz_type) {
  253. where.push('biz_type = ?')
  254. params.push(biz_type)
  255. }
  256. if (remark) {
  257. where.push('remark LIKE ?')
  258. params.push(`%${remark}%`)
  259. }
  260. if (start_time) {
  261. where.push('created_at >= FROM_UNIXTIME(? / 1000)')
  262. params.push(start_time)
  263. }
  264. if (end_time) {
  265. where.push('created_at <= FROM_UNIXTIME(? / 1000)')
  266. params.push(end_time)
  267. }
  268. const totalRows = await db.query(
  269. `SELECT COUNT(*) AS total
  270. FROM lepao_count_ledger
  271. WHERE ${where.join(' AND ')}`,
  272. params
  273. )
  274. const rows = await db.query(
  275. `SELECT delta, balance_before, balance_after, biz_type, remark,
  276. UNIX_TIMESTAMP(created_at) * 1000 AS created_at
  277. FROM lepao_count_ledger
  278. WHERE ${where.join(' AND ')}
  279. ORDER BY id DESC
  280. LIMIT ? OFFSET ?`,
  281. [...params, String(page.pagesize), String(page.offset)]
  282. )
  283. if (!rows || !totalRows) return '查询失败'
  284. return asJson({ list: rows.map(mapCountLedger), pagination: pagination(page, totalRows) })
  285. }
  286. async list_goods(args) {
  287. const { keyword, category, current, pagesize } = args
  288. const page = pageArgs({ current, pagesize })
  289. const where = ['state = 1']
  290. const params = []
  291. if (keyword) {
  292. where.push('name LIKE ?')
  293. params.push(`%${keyword}%`)
  294. }
  295. if (category) {
  296. where.push('category = ?')
  297. params.push(category)
  298. }
  299. const totalRows = await db.query(
  300. `SELECT COUNT(*) AS total
  301. FROM goods
  302. WHERE ${where.join(' AND ')}`,
  303. params
  304. )
  305. const rows = await db.query(
  306. `SELECT name, price, num, \`limit\`, description, category, features
  307. FROM goods
  308. WHERE ${where.join(' AND ')}
  309. ORDER BY id DESC
  310. LIMIT ? OFFSET ?`,
  311. [...params, String(page.pagesize), String(page.offset)]
  312. )
  313. if (!rows || !totalRows) return '查询失败'
  314. return asJson({
  315. list: rows.map(r => ({ ...r, features: parseJson(r.features, []) })),
  316. pagination: pagination(page, totalRows)
  317. })
  318. }
  319. async list_my_orders(args) {
  320. const { sender, state, order_id, current, pagesize } = args
  321. if (required([sender])) return '缺少 sender'
  322. const page = pageArgs({ current, pagesize })
  323. const where = ['o.create_user = ?']
  324. const params = [sender]
  325. if (state !== undefined && state !== null && state !== '' && Number(state) !== -1) {
  326. where.push('o.state = ?')
  327. params.push(state)
  328. }
  329. if (order_id) {
  330. where.push('o.orderId = ?')
  331. params.push(order_id)
  332. }
  333. const totalRows = await db.query(
  334. `SELECT COUNT(*) AS total
  335. FROM orders o
  336. JOIN goods g ON o.goods_id = g.id
  337. WHERE ${where.join(' AND ')}`,
  338. params
  339. )
  340. const rows = await db.query(
  341. `SELECT o.orderId, o.state, o.create_time, o.price, o.pay_type, g.name AS goods_name
  342. FROM orders o
  343. JOIN goods g ON o.goods_id = g.id
  344. WHERE ${where.join(' AND ')}
  345. ORDER BY o.id DESC
  346. LIMIT ? OFFSET ?`,
  347. [...params, String(page.pagesize), String(page.offset)]
  348. )
  349. if (!rows || !totalRows) return '查询失败'
  350. return asJson({
  351. list: rows.map(mapOrder),
  352. pagination: pagination(page, totalRows)
  353. })
  354. }
  355. async save_lepao_account(args) {
  356. const { sender, student_num } = args
  357. if (required([sender, student_num])) return '缺少 sender 或 student_num'
  358. const existRows = await db.query(
  359. `SELECT id, create_user, student_num, email, area, auto_time, auto_run, target_count, total_num, auto_day, notice_type, notes
  360. FROM lepao_account
  361. WHERE student_num = ?
  362. LIMIT 1`,
  363. [student_num]
  364. )
  365. if (!existRows) return '查询账号失败'
  366. if (existRows.length > 0 && existRows[0].create_user && existRows[0].create_user !== sender) {
  367. return '该乐跑账号已被其他用户绑定'
  368. }
  369. const existing = existRows[0] || {}
  370. const targetCount = isProvided(args.target_count)
  371. ? normalizeNumber(args.target_count)
  372. : (existRows.length > 0 ? normalizeNumber(existing.target_count) : 30)
  373. if (targetCount === null || targetCount < 0 || targetCount > 99) return '乐跑目标次数不在合法范围内'
  374. args.target_count = targetCount
  375. const noticeType = isProvided(args.notice_type) ? args.notice_type : (existing.notice_type || 'email')
  376. const email = isProvided(args.email) ? String(args.email).trim() : (existing.email || '')
  377. if (noticeType === 'email') {
  378. if (!this.emailRegex.test(email)) return '邮箱格式不正确'
  379. const emailDomain = email.split('@')[1].toLowerCase()
  380. if (this.banEmailList.includes(emailDomain)) return `暂不支持使用 ${emailDomain} 域名的邮箱,请更换其他邮箱后重试`
  381. } else if (email && !this.emailRegex.test(email)) {
  382. return '邮箱格式不正确'
  383. }
  384. const autoRun = isProvided(args.auto_run) ? Number(args.auto_run) : (existRows.length > 0 ? Number(existing.auto_run) : 1)
  385. if (![0, 1].includes(autoRun)) return 'auto_run 只能为 0 或 1'
  386. args.auto_run = autoRun
  387. const autoDay = isProvided(args.auto_day)
  388. ? args.auto_day
  389. : (existRows.length > 0 ? parseJson(existing.auto_day, []) : [0, 1, 2, 3, 4, 5, 6])
  390. if (autoRun === 1 && (!Array.isArray(autoDay) || !autoDay.every(v => Number.isInteger(v) && v >= 0 && v <= 6))) return '自动乐跑星期不合法'
  391. if (isProvided(args.auto_day)) args.auto_day = autoDay
  392. const totalNum = normalizeNumber(existing.total_num) || 0
  393. if (autoRun === 1 && totalNum >= targetCount && targetCount !== 0) return '该账号累计跑步次数已达到预设目标次数,请尝试增大目标次数后再试'
  394. const fields = []
  395. const params = []
  396. const mutable = [
  397. ['email', 'email'],
  398. ['auto_time', 'auto_time'],
  399. ['auto_run', 'auto_run'],
  400. ['target_count', 'target_count'],
  401. ['notice_type', 'notice_type'],
  402. ['notes', 'notes']
  403. ]
  404. if (isProvided(args.area)) {
  405. const area = normalizeRunArea(args.area)
  406. if (area === null) return '跑区不合法'
  407. fields.push('area = ?')
  408. params.push(area)
  409. }
  410. if (isProvided(args.notice_type) && !['email', 'none', 'bot'].includes(args.notice_type)) return '通知方式不合法'
  411. if (isProvided(args.auto_time)) {
  412. const autoTime = Number(args.auto_time)
  413. if (!Number.isInteger(autoTime) || autoTime < -1 || autoTime > 23) return '自动乐跑时间不合法'
  414. args.auto_time = autoTime
  415. }
  416. if (isProvided(args.auto_day)) {
  417. fields.push('auto_day = ?')
  418. params.push(JSON.stringify(args.auto_day))
  419. }
  420. for (const [argName, column] of mutable) {
  421. if (isProvided(args[argName])) {
  422. fields.push(`${column} = ?`)
  423. params.push(args[argName])
  424. }
  425. }
  426. const now = Date.now()
  427. let result
  428. if (existRows.length > 0) {
  429. if (fields.length === 0) return '没有需要修改的字段'
  430. fields.push('update_time = ?', 'create_user = ?')
  431. params.push(now, sender)
  432. result = await db.query(
  433. `UPDATE lepao_account SET ${fields.join(', ')} WHERE student_num = ? AND (create_user = ? OR create_user IS NULL)`,
  434. [...params, student_num, sender]
  435. )
  436. } else {
  437. const area = isProvided(args.area) ? normalizeRunArea(args.area) : ''
  438. if (area === null) return '跑区不合法'
  439. const autoDay = isProvided(args.auto_day) ? args.auto_day : [0, 1, 2, 3, 4, 5, 6]
  440. if (!Array.isArray(autoDay) || !autoDay.every(v => Number.isInteger(v) && v >= 0 && v <= 6)) return '自动乐跑星期不合法'
  441. const email = args.email || ''
  442. if (email && !this.emailRegex.test(email)) return '邮箱格式不正确'
  443. result = await db.query(
  444. `INSERT INTO lepao_account
  445. (student_num, email, area, auto_time, auto_run, target_count, create_user, create_time, update_time, notes, auto_day, notice_type)
  446. VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)`,
  447. [
  448. student_num,
  449. email,
  450. area,
  451. isProvided(args.auto_time) ? Number(args.auto_time) : -1,
  452. isProvided(args.auto_run) ? Number(args.auto_run) : 1,
  453. isProvided(args.target_count) ? args.target_count : 30,
  454. sender,
  455. now,
  456. now,
  457. args.notes || '',
  458. JSON.stringify(autoDay),
  459. args.notice_type || 'email'
  460. ]
  461. )
  462. const bindCode = Math.random().toString(36).slice(2, 8).toUpperCase()
  463. await db.query('INSERT IGNORE INTO lepao_extra (student_num, bind_code) VALUES (?, ?)', [student_num, bindCode])
  464. }
  465. if (!result || result.affectedRows < 1) return '保存失败或无权修改'
  466. return asJson({ success: true, student_num })
  467. }
  468. async save_power_task(args) {
  469. const { sender, task_id, area, building, room } = args
  470. if (required([sender])) return '缺少 sender'
  471. const now = Date.now()
  472. let existing = []
  473. if (task_id) {
  474. existing = await db.query('SELECT id, area, building, room FROM power_task WHERE id = ? AND create_user = ? LIMIT 1', [task_id, sender])
  475. } else if ([area, building, room].every(isProvided)) {
  476. existing = await db.query(
  477. 'SELECT id, area, building, room FROM power_task WHERE create_user = ? AND area = ? AND building = ? AND room = ? LIMIT 1',
  478. [sender, area, building, room]
  479. )
  480. }
  481. if (!existing) return '查询电费任务失败'
  482. const fields = []
  483. const params = []
  484. for (const key of ['area', 'building', 'room', 'email', 'lowest', 'notes']) {
  485. if (isProvided(args[key])) {
  486. fields.push(`${key} = ?`)
  487. params.push(args[key])
  488. }
  489. }
  490. let result
  491. if (existing.length > 0) {
  492. if (fields.length === 0) return '没有需要修改的字段'
  493. fields.push('is_notice = 0', 'update_time = ?')
  494. params.push(now)
  495. result = await db.query(
  496. `UPDATE power_task SET ${fields.join(', ')} WHERE id = ? AND create_user = ?`,
  497. [...params, existing[0].id, sender]
  498. )
  499. } else {
  500. if (required([area, building, room, args.lowest])) return '新增电费任务需提供 area、building、room、lowest'
  501. let balance = ''
  502. let koufeiDate = ''
  503. try {
  504. const endpoint = `https://hqpay.ctbu.edu.cn/weixin/ashx/frmuser.ashx?test=lastlist&pid=${room}&dyid=${building}`
  505. const apiRes = await axiosWithQgOutbound({ method: 'get', url: endpoint, timeout: 15000, logger: this.logger, scene: 'AiAssistantPowerAccount' })
  506. if (!apiRes || !apiRes.data || !apiRes.data[0]) return '获取电费信息失败,请稍后再试'
  507. balance = apiRes.data[0][1]
  508. koufeiDate = apiRes.data[0][2]
  509. } catch (_) {
  510. return '获取电费信息失败,请稍后再试'
  511. }
  512. result = await db.query(
  513. 'INSERT INTO power_task (create_user, create_time, update_time, area, building, room, email, lowest, notes, balance, koufei_date) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)',
  514. [sender, now, now, area, building, room, args.email || '', args.lowest, args.notes || '', balance, koufeiDate]
  515. )
  516. }
  517. if (!result || result.affectedRows < 1) return '保存失败或无权修改'
  518. return asJson({ success: true, area, building, room })
  519. }
  520. async list_power_bills(args) {
  521. const { sender, area, building, room, current, pagesize } = args
  522. if (required([sender])) return 'missing sender'
  523. const page = pageArgs({ current, pagesize })
  524. const taskWhere = ['create_user = ?']
  525. const taskParams = [sender]
  526. if (area) {
  527. taskWhere.push('area = ?')
  528. taskParams.push(area)
  529. }
  530. if (building) {
  531. taskWhere.push('building = ?')
  532. taskParams.push(building)
  533. }
  534. if (room) {
  535. taskWhere.push('room = ?')
  536. taskParams.push(room)
  537. }
  538. const totalRows = await db.query(
  539. `SELECT COUNT(*) AS total
  540. FROM power_task
  541. WHERE ${taskWhere.join(' AND ')}`,
  542. taskParams
  543. )
  544. const tasks = await db.query(
  545. `SELECT id, area, building, room, balance, lowest, koufei_date, update_time, notes
  546. FROM power_task
  547. WHERE ${taskWhere.join(' AND ')}
  548. ORDER BY id DESC
  549. LIMIT ? OFFSET ?`,
  550. [...taskParams, String(page.pagesize), String(page.offset)]
  551. )
  552. if (!tasks || !totalRows) return 'query failed'
  553. return asJson({
  554. tasks: tasks.map(mapPowerTask),
  555. pagination: pagination(page, totalRows)
  556. })
  557. }
  558. async list_power_bill_records(args) {
  559. const { sender, area, building, room, current, pagesize } = args
  560. if (required([sender, area, building, room])) return 'missing sender, area, building or room'
  561. const page = pageArgs({ current, pagesize })
  562. const tasks = await db.query(
  563. `SELECT id, area, building, room, balance, lowest, koufei_date, update_time, notes
  564. FROM power_task
  565. WHERE create_user = ? AND area = ? AND building = ? AND room = ?
  566. ORDER BY id DESC
  567. LIMIT 1`,
  568. [sender, area, building, room]
  569. )
  570. if (!tasks) return 'query failed'
  571. if (tasks.length === 0) return asJson({ task: null, records: [], pagination: pagination(page, [{ total: 0 }]) })
  572. const totalRows = await db.query(
  573. `SELECT COUNT(*) AS total
  574. FROM power_change_record
  575. WHERE task_id = ?`,
  576. [tasks[0].id]
  577. )
  578. const records = await db.query(
  579. `SELECT time, balance, old_balance, change_time
  580. FROM power_change_record
  581. WHERE task_id = ?
  582. ORDER BY id DESC
  583. LIMIT ? OFFSET ?`,
  584. [tasks[0].id, String(page.pagesize), String(page.offset)]
  585. )
  586. if (!records || !totalRows) return 'query failed'
  587. return asJson({
  588. task: mapPowerTask(tasks[0]),
  589. records: records.map(mapPowerRecord),
  590. pagination: pagination(page, totalRows)
  591. })
  592. }
  593. async query_qxs_book_list(args) {
  594. const { sender, username, password, current, pagesize } = args
  595. if (required([sender, username, password])) return '缺少 sender、username 或 password'
  596. try {
  597. const page = pageArgs({ current, pagesize })
  598. const qxs = new QxsBookList()
  599. const { accessToken } = await qxs.qxsLogin(username, password)
  600. const { userInfo } = await qxs.qsxUserInfo(accessToken)
  601. const { bookList } = await qxs.qsxGetList(accessToken, userInfo.termCode)
  602. const list = bookList || []
  603. return asJson({
  604. user: {
  605. realname: userInfo.userName,
  606. termName: userInfo.termName
  607. },
  608. list: list.slice(page.offset, page.offset + page.pagesize).map(book => ({
  609. title: book.title || book.bookName || '',
  610. isbn: book.isbn || '',
  611. author: book.author || '',
  612. publisher: book.publisher || '',
  613. className: book.className || ''
  614. })),
  615. pagination: pagination(page, [{ total: list.length }])
  616. })
  617. } catch (err) {
  618. return err.message || '查询趣选书书单失败'
  619. }
  620. }
  621. async submit_work_order(args) {
  622. const { sender, title, content, email = '', files = [] } = args
  623. if (required([sender, title, content])) return '缺少 sender、title 或 content'
  624. if (email && !this.emailRegex.test(email)) return '邮箱格式不正确'
  625. if (email && this.banEmailList.includes(email.split('@')[1].toLowerCase())) return '暂不支持该邮箱域名'
  626. const safeTitle = normalizeText(title, 80)
  627. const safeContent = normalizeText(content, 4000)
  628. const safeFiles = Array.isArray(files) ? files.map(item => String(item || '').trim()).filter(Boolean).slice(0, 6) : []
  629. if (!safeTitle || !safeContent) return '标题或内容不能为空'
  630. const time = Date.now()
  631. const msg = [
  632. {
  633. time,
  634. content: safeContent,
  635. files: safeFiles,
  636. uuid: sender,
  637. type: 'user'
  638. },
  639. {
  640. time,
  641. content: '您的问题已收到,我们将尽快处理,请耐心等待。',
  642. uuid: ASSISTANT_UUID,
  643. type: 'system'
  644. }
  645. ]
  646. const result = await db.query(
  647. 'INSERT INTO work_order (title, email, msg, create_user, create_time, update_time) VALUES (?, ?, ?, ?, ?, ?)',
  648. [safeTitle, email, msg, sender, time, time]
  649. )
  650. if (!result || result.affectedRows !== 1) return '提交工单失败'
  651. const kefuRows = await db.query(`
  652. SELECT email FROM users
  653. WHERE JSON_CONTAINS(permission, '"admin"') OR JSON_CONTAINS(permission, '"service"')
  654. `)
  655. const emails = [...new Set((kefuRows || []).map(row => row.email).filter(Boolean))]
  656. for (const targetEmail of emails) {
  657. await EmailTemplate.orderNewReply(targetEmail, { id: result.insertId, content: safeContent, files: safeFiles })
  658. }
  659. return asJson({ success: true, message: '工单提交成功,可在工单列表查看处理进度' })
  660. }
  661. }
  662. module.exports.AI_ASSISTANT_MCP = new AiAssistantMcp()