main.js

import schedule from 'node-schedule'
import dayjs from 'dayjs'
import {
  getAccessToken,
  sendMessageReply,
  getAggregatedData,
  getCallbackTemplateParams,
} from './src/services/index.js'
import config from './config/exp-config.js'
import cornTime from './config/server-config.js'
import mainForTest from './main-for-test.js'

export default async function mainForProd() {
  // 获取accessToken
  console.log('\n\n')
  console.log(dayjs().format('YYYY-MM-DD HH:mm:ss'))
  console.log('---')
  console.log('【连接微信公众平台】开始')
  const accessToken = await getAccessToken()
  if (!accessToken) {
    console.log('遇到错误，执行终止！')
    return
  }
  console.log('【连接微信公众平台】结束')
  console.log('---')

  // 处理好的用户数据
  console.log('---')
  console.log('【数据获取】开始')
  const aggregatedData = await getAggregatedData()
  console.log('【数据获取】结束')
  console.log('---')

  // 公众号推送消息
  console.log('---')
  console.log('【常规模板】推送开始')
  const {
    needPostNum,
    successPostNum,
    failPostNum,
    successPostIds,
    failPostIds,
  } = await sendMessageReply(aggregatedData, accessToken)
  console.log('【常规模板】推送结束')
  console.log('---')

  // 获取回执信息
  const callbackTemplateParams = getCallbackTemplateParams({
    needPostNum,
    successPostNum,
    failPostNum,
    successPostIds,
    failPostIds,
  })

  // 发送回执
  if (config.CALLBACK_TEMPLATE_ID) {
    console.log('---')
    console.log('【推送完成提醒】推送开始')
    await sendMessageReply(config.CALLBACK_USERS, accessToken, config.CALLBACK_TEMPLATE_ID, callbackTemplateParams)
    console.log('【推送完成提醒】推送结束')
    console.log('---')
  }
}

const main = () => {
  if (process.env.APP_MODE === 'params-log') {
    mainForTest()
  } else if (process.env.APP_MODE === 'server') {
    console.log('======【定时推送服务已启动, enjoying it】======')
    console.log(`目前定时推送的配置为：【${cornTime}】`)
    schedule.scheduleJob(cornTime, () => {
      mainForProd()
    })
  } else if (process.env.APP_MODE === 'prod') {
    mainForProd()
  }
}

main()