mtfosbot/libs/line-message/commands/index.js

const DB = require('../../database')
const actions = require('./actions')

const parseCMD = async (text = '', source = {}) => {
  if (typeof text !== 'string' || text.trim().length === 0) return null
  if (!source || typeof source !== 'object' || !('type' in source)) return null
  let txt = text.trim()
  let arr = txt.split(' ').map(t => t.trim())
  if (arr.length === 0) return null
  if (arr[0][0] !== '!') return null
  let cmd = arr[0].replace(/^!/, '')
  cmd = cmd.toLowerCase()

  let db = await DB.connect()
  let reply = null

  try {
    // query normal command
    let result = await db.query({
      text: `select "message", "group" from "public"."commands" where "cmd" = $1 and ("group" = '' or "group" = $2)`,
      values: [cmd, source.groupId]
    })
    if (result.rowCount > 0) {
      let obj = result.rows.filter(t => t.group === source.groupId)
      if (obj.length === 0) obj = result.rows[0]
      else obj = obj[0]
      let content = obj.message
      let m = content.match(/{{(.+?)}}/g)
      if (m !== null && m.length > 0) {
        for (let i = 0; i < m.length; i++) {
          let c = m[i].replace(/^{{/, '').replace(/}}$/, '')
          let carr = c.split('=')
          if (carr.length > 1) c = carr
          let res = await actions(c, arr.slice(1).join(' '), source)
          content = content.replace(m[i], res || '')
        }
      }
      if (content.trim().length > 0) {
        reply = {
          reply: content
        }
      }
    }

    if (reply === null) {
      // query keyword commands
      let keyCMD = await db.query({
        text: `select "message", "group", "cmd" from "public"."key_commands" where ("group" = '' or "group" = $1)`,
        values: [source.groupId]
      })
      if (keyCMD.rowCount > 0) {
        let obj = keyCMD.rows.filter(t => t.group === '')
        let obj2 = keyCMD.rows.filter(t => t.group === source.groupId)
        obj = obj.map(t => {
          for (let i of obj2) {
            if (i.cmd === t.cmd && i.group !== '') return i
          }
          return t
        })
        let regex = null
        let txt = ''
        for (let i of obj) {
          txt += (txt.length > 0 ? '|' : '') + i.cmd
        }
        regex = new RegExp(`(${txt})`)
        let m = arr.slice(1).join(' ').match(regex)
        if (m !== null && m.length > 0) {
          let key = obj.filter(t => t.cmd === m[0])
          if (key.length > 0) {
            let content = key[0].message
            let m = content.match(/{{(.+?)}}/g)
            if (m !== null && m.length > 0) {
              for (let i = 0; i < m.length; i++) {
                let c = m[i].replace(/^{{/, '').replace(/}}$/, '')
                let carr = c.split('=')
                if (carr.length > 1) c = carr
                let res = await actions(c, arr.slice(1).join(' '), source)
                content = content.replace(m[i], res || '')
              }
            }
            if (content.trim().length > 0) {
              reply = {
                reply: content
              }
            }
          }
        }
      }
    }
  } catch (err) {
    console.log(err)
  }

  db.release()
  return reply
}

module.exports = parseCMD