index.js

// 本地调试用
// 参考 https://github.com/motdotla/dotenv
if (process.env.LOCAL_TEST) {
  require('dotenv').config()
}

process.on('unhandledRejection', err => {
  throw err
})

const pLimit = require('p-limit')
const Notifier = require('./lib/Notifier')
const { version } = require('./package.json')
const parseToken = require('./lib/parse-token')
const updateNotifier = require('./lib/update-notifier')
const { getCoupons, getRule } = require('./lib/coupons')

const TOKEN = process.env.TOKEN
const notifier = new Notifier({
  barkKey: process.env.BARK_KEY,
  workWechat: process.env.QYWX_SEND_CONF,
  serverChanToken: process.env.SC_SEND_KEY,
  telegram: {
    botToken: process.env.TG_BOT_TOKEN,
    userId: process.env.TG_USER_ID
  }
})

const NOTIFY_TITLE = '外卖神券天天领😋'
const MAX_RETRY_COUNT = 2

console.log(`
───────────────────────────────────────
 actions-mtwm-coupons
 外卖神券天天领
────────────────────────

 Ver. ${version}

 Github @vv314`)

async function printRule() {
  const rule = await getRule()

  if (rule.length) {
    console.log('—————————— 活动规则 ——————————\n')
    rule.forEach((item, i) => {
      console.log(`${i + 1}. ${item}`)
    })
  }
}

function stringifyCoupons(coupons) {
  return coupons
    .map(item => `- ￥${item.amount}（${item.amountLimit}）`)
    .join('\n')
}

function sendUserNotify(msg, account) {
  const result = []
  const user = account.alias

  if (account.barkKey) {
    const qywxRes = notifier
      .sendBark(NOTIFY_TITLE, msg, { key: account.barkKey })
      .then(res => `@${user} ${res.msg}`)

    result.push(qywxRes)
  }

  if (account.qywxUid) {
    const qywxRes = notifier
      .sendWorkWechat(NOTIFY_TITLE, msg, {
        uid: account.qywxUid
      })
      .then(res => `@${user} ${res.msg}`)

    result.push(qywxRes)
  }

  if (account.tgUid) {
    const tgRes = notifier
      .sendTelegram(NOTIFY_TITLE, msg, { uid: account.tgUid })
      .then(res => `@${user} ${res.msg}`)

    result.push(tgRes)
  }

  return result.map(p => p.then(r => `[用户通知] ${r}`))
}

function sendGlobalNotify(tasks) {
  const message = tasks.map(t => `账号 ${t.user}:\n${t.data}`).join('\n\n')

  return notifier
    .notify(NOTIFY_TITLE, message)
    .map(p => p.then(res => `[全局通知] ${res.msg}`))
}

async function doJob(account, progress) {
  const res = await getCoupons(account.token, MAX_RETRY_COUNT)

  console.log(
    `\n────────── [${progress.mark()}] 账号: ${account.alias} ──────────\n`
  )

  if (res.code != 0) {
    console.log(res.msg, res.error)
    res.retryTimes && console.log(`重试: ${res.retryTimes} 次`)
    console.log('\n😦 领取失败')

    return {
      user: account.alias,
      data: `领取失败: ${res.msg}`,
      pushInfo: []
    }
  }

  const { coupons, phone } = res.data

  console.log(...coupons)
  console.log(`\n红包已放入账号：${phone}`)
  console.log(`\n🎉 领取成功！`)

  const message = stringifyCoupons(coupons)
  const pushInfo = sendUserNotify(message, account)

  return { user: account.alias, data: message, pushInfo }
}

async function runTaskQueue(tokenList) {
  const asyncPool = pLimit(5)
  const progress = {
    count: 0,
    mark() {
      return `${++this.count}/${tokenList.length}`
    }
  }

  return Promise.all(
    tokenList.map(account => asyncPool(doJob, account, progress))
  )
}

async function printNotifyResult(pushInfo) {
  if (pushInfo.length) {
    console.log(`\n────────── 推送通知 ──────────\n`)

    // 异步打印结果
    pushInfo.forEach(p => p.then(res => console.log(res)))
  }

  return Promise.all(pushInfo)
}

async function checkUpdate() {
  let message

  try {
    message = await updateNotifier()
  } catch (e) {
    console.log('\n', e)
  }

  if (!message) return

  console.log(`\n────────── 更新提醒 ──────────\n`)
  console.log(message)
}

async function main() {
  await printRule()

  const tokens = parseToken(TOKEN)
  const tasks = await runTaskQueue(tokens)

  const userPushInfo = tasks.map(info => info.pushInfo).flat()
  const globalPushInfo = sendGlobalNotify(tasks)

  // 打印通知结果，用户通知优先
  await printNotifyResult(userPushInfo.concat(globalPushInfo))

  checkUpdate()
}

main()