Node.js 实现抢票小工具 & 短信通知提醒功能

<td class="b">
<span>这里为日期</span>
<span>如果有余票则显示余票数量</span>
</td>

const axios = require('axios')
const querystring = require("querystring"); //序列化对象，用qs也行，都一样
var QcloudSms = require("qcloudsms_js");
var cheerio = require('cheerio');
var schedule = require('node-schedule');

var obj = {
 data: {
  lineId: 111130, //路线id
  vehTime: 0722, //发车时间，
  startTime: 0751, //预计上车时间
  onStationId: 564492, //预定的站点id
  offStationId: 17990,//到站id
  onStationName: '宝安交通运输局③', //预定的站点名称
  offStationName: "深港产学研基地",//预定到站名称
  tradePrice: 0,//总金额
  saleDates: '17',//车票日期
  beginDate: '',//订票时间，滞空，用于抓取到余票后填入数据
 },
 phoneNumber: 123123123, //用户手机号，接收短信的手机号
 cookie: 'JSESSIONID=TESTCOOKIE', // 抓取到的cookie
 day: "17" //定17号的票，这个主要是用于抢指定日期的票，滞空则为抢当月所有余票
}

class QueryTicket{
 /**
  *Creates an instance of QueryTicket.
  * @param {Object} { data, phoneNumber, cookie, day }
  * @param data {Object} 请求余票接口的requery参数
  * @param phoneNumber {Number} 用户手机号，短信需要用到
  * @param cookie {String} cookie信息
  * @params day {String} 某日的票，如'18'
  * @memberof QueryTicket 请求余票接口
  */
 constructor({ data, phoneNumber, cookie, day }) {
  this.data = data 
  this.cookie = cookie
  this.day = day
  this.phoneNumber = phoneNumber
  this.postData = querystring.stringify(data)
  this.times = 0;  //记录次数
  var stop = false //通过特定接口才能修改stop值，防止外部随意串改
  this.getStop = function () { //获取是否停止
   return stop 
  }
  this.setStop = function (ifStop) { //设置是否停止
   stop = ifStop
  }
 }
}

class QueryTicket{
 constructor({ data, phoneNumber, cookie, day }) {
 //constructor代码... 
 }
  init(){}//初始化
  handleQueryTicket(){}//查询余票的逻辑
  requestTicket(){} //调用查询余票接口
  handleBuyTicket(){} //购票相关逻辑
  requestOrder(){}//调用购票接口
  handleInfoUser(){}//通知用户的逻辑
  sendMSg(){} //发短信接口
}

class QueryTicket{
 constructor({ data, phoneNumber, cookie, day }) {
 //constructor代码... 
 }
 //初始化,因为涉及到异步请求，所以我们使用`async await`
  async init(){
     let ticketList = await this.handleQueryTicket() //返回查询到的余票数组
  }
  //查询余票的逻辑
  handleQueryTicket(){ 
  let ticketList = [] //余票数组
  let res = await this.requestTicket()
  this.times++ //计数器，记录请求查询多少次
  var str = res.data.replace(/\\/g, "") //格式化返回值
  var $ = cheerio.load(`<div class="main">${str}</div>`) // cheerio载入查询接口response的html节点数据
  let list = $(".main").find(".b") //查找是否有余票的dom节点
  // 如果没有余票，打印出请求多少次,然后返回，不执行下面的代码
  if (!list.length) {
   console.log(`用户${this.phoneNumber}：无票，已进行${this.times}次`)
   return
  }

  // 如果有余票
  list.each((idx, item) => {
   var str = $(item).html() //str这时格式是<span>21</span><span>&$x4F59;0</span>
   //最后一个span 的内容其实"余0"，也就是无票，只不过是被转码了而已
   //因此要在下一步对其进行格式化
   var arr = str.split(/<span>|<\/span>|\&\#x4F59\;/).filter(item => !!item === true) 
   let data = {
    day: arr[0],
    ticketLeft: arr[1]
   }
   
   //如果是要抢指定日期的票
   if (this.day) {
   //如果有指定日期的余票
    if (parseInt(data.day) === parseInt(data.day)) {
     ticketList.push(data)
    }
   } else {
   //如果不是，则返回查询到的所有余票
    ticketList.push(data)
   }
  })
  return ticketList
  }
   //调用查询余票接口
  requestTicket(){
  return axios.post('http://weixin.xxxx.net/ebus/front/wxQueryController.do?BcTicketCalendar', this.postData, {
   headers: {
    'Content-Type': 'application/x-www-form-urlencoded',
    'User-Agent': "Mozilla/5.0 (iPhone; CPU iPhone OS 8_0 like Mac OS X) AppleWebKit/600.1.4 (KHTML, like Gecko) Mobile/12A365 MicroMessenger/5.4.1 NetType/WIFI",
    "Cookie": this.cookie
   }
  })  
  }
  handleBuyTicket(){} //购票相关逻辑
  requestOrder(){}//调用购票接口
  handleInfoUser(){}//通知用户的逻辑
  sendMSg(){} //发短信接口
}

class QueryTicket{
 constructor({ data, phoneNumber, cookie, day }) {
 //constructor代码... 
 }
 //初始化
  async init(){
  let ticketList = await this.handleQueryTicket()
  //如果有余票
  if (ticketList.length) {
  //把余票传入购票逻辑方法，返回短信通知所需要的数据
   let resParse = await this.handleBuyTicket(ticketList)
  }
  }
  //查询余票的逻辑
  async handleQueryTicket(){
  // 查询余票代码...
  }
  //调用查询余票接口
  requestTicket(){
  //调用查询余票接口代码...  
  } 
  //购票相关逻辑
  async handleBuyTicket(ticketList){
  let year = new Date().getFullYear() //年份，
  let month = new Date().getMonth() + 1 //月份，拼接购票日期用得上，因为余票接口只返回几号
  let {
   onStationName,//起始站点名
   offStationName,//结束站点名
   lineId,//线路id
   vehTime,//发车时间
   startTime,//预计上车时间
   onStationId,//上车的站台id
   offStationId //到站的站台id
   } = this.data // 初始化的数据
  let station = `${onStationName}-${offStationName}` //站点，发短信时候用到:"宝安交通局-深港产学研基地"
  let dateStr = ""; //车票日期
  let tickAmount = "" //总张数
  ticketList.forEach(item => {
   dateStr = dateStr + `${year}-${month}-${item.day},`
   tickAmount = tickAmount + `${item.ticketLeft}张,`
  })
  var buyTicket = {
   lineId,//线路id
   vehTime,//发车时间
   startTime,//预计上车时间
   onStationId,//上车的站点id
   offStationId,//目标站点id
   tradePrice: '5', //金额
   saleDates: dateStr.slice(0, -1),
   payType: '2' //支付方式，微信支付
  }
  // 调用购票接口
   let data = querystring.stringify(buyTicket)
   let res = await this.requestOrder(data) //返回json数据，是否购票成功等等
   //把发短信所需要数据都要传入
  return Object.assign({}, JSON.parse(res.data), { queryParam: { dateStr, tickAmount, startTime, station } })
  }//购票相关逻辑
  //调用购票接口
  requestOrder(obj){
  return axios.post('http://weixin.xxxx.net/ebus/front/wxQueryController.do?BcTicketBuy', obj, {
   headers: {
    'Content-Type': 'application/x-www-form-urlencoded',
    'User-Agent': "Mozilla/5.0 (iPhone; CPU iPhone OS 8_0 like Mac OS X) AppleWebKit/600.1.4 (KHTML, like Gecko) Mobile/12A365 MicroMessenger/5.4.1 NetType/WIFI",
    "Cookie": this.cookie
   }
  })
  }
  handleInfoUser(){}//通知用户的逻辑
  sendMSg(){} //发短信接口
}

class QueryTicket{
 constructor({ data, phoneNumber, cookie, day }) {
 //constructor代码... 
 }
 //初始化
  async init(){
  let ticketList = await this.handleQueryTicket()
  //如果有余票
  if (ticketList.length) {
  //把余票传入购票逻辑方法，返回短信通知所需要的数据
   let resParse = await this.handleBuyTicket(ticketList)
  //执行通知逻辑
   this.handleInfoUser(resParse)
  }
  }
  
  //查询余票的逻辑
  async handleQueryTicket(){
  // 查询余票代码...
  }
  //调用查询余票接口
  requestTicket(){
  //调用查询余票接口代码...  
  } 
  //购票相关逻辑
  async handleBuyTicket(ticketList){
  //购票代码...
  }
  //调用购票接口
  requestOrder(obj){
  //购票接口请求代码...
  }
  //通知用户的逻辑
  async handleInfoUser(parseData){
  //获取上一步购票的response数据和我们拼接的数据
  let { returnCode, returnData: { main: { lineName, tradePrice } }, queryParam: { dateStr, tickAmount, startTime, station } } = parseData
  //如果购票成功，则返回500
  if (returnCode === "500") {
   let res = await this.sendMsg({
    dateStr, //日期
    tickAmount: tickAmount.slice(0, -1), //总张数
    station, //站点
    lineName, //巴士名称/路线名称
    tradePrice,//总价
    startTime,//出发时间
    phoneNumber: this.phoneNumber,//手机号
   })
   //如果发信成功，则不再进行抢票操作
   if (res.result === 0 && res.errmsg === "OK") {
    this.setStop(true)
   } else {
   //失败不做任何操作
    console.log(res.errmsg)
   }
  } else {
   //失败不做任何操作
   console.log(resParse['returnInfo'])
  }    
  }
  //发短信接口
  sendMSg(){
  let { dateStr, tickAmount, station, lineName, phoneNumber, startTime, tradePrice } = obj
  var appid = 140034324; // SDK AppID 以1400开头
  // 短信应用 SDK AppKey
  var appkey = "asdfdsvajwienin23493nadsnzxc";
  // 短信模板 ID，需要在短信控制台中申请
  var templateId = 7839; // NOTE: 这里的模板ID`7839`只是示例，真实的模板 ID 需要在短信控制台中申请
  // 签名
  var smsSign = "测试短信"; // NOTE: 签名参数使用的是`签名内容`，而不是`签名ID`。这里的签名"腾讯云"只是示例，真实的签名需要在短信控制台申请
  // 实例化 QcloudSms
  var qcloudsms = QcloudSms(appid, appkey);
  var ssender = qcloudsms.SmsSingleSender();
  // 这里的params就是短信里面可以自定义的内容，也就是填入{1}{2}..的内容
  var params = [dateStr, station, lineName, startTime, tickAmount, tradePrice];
  //用promise来封装下异步操作
  return new Promise((resolve, reject) => {
   ssender.sendWithParam(86, phoneNumber, templateId, params, smsSign, "", "", function (err, res, resData) {
    if (err) {
     reject(err)
    } else {
     resolve(resData)
    }
   });
  })
  } 
}

// 定时任务
class SetInter {
 constructor({ timer, fn }) {
  this.timer = timer // 每几秒执行
  this.fn = fn //执行的回调
  this.rule = new schedule.RecurrenceRule(); //实例化一个对象
  this.rule.second = this.setRule() // 调用原型方法，schedule的语法而已
  this.init()
 }
 setRule() {
  let rule = [];
  let i = 1;
  while (i < 60) {
   rule.push(i)
   i += this.timer
  }
  return rule //假设传入的timer为5，则表示定时任务每5秒执行一次
  // [1, 6, 11, 16, 21, 26, 31, 36, 41, 46, 51, 56] 
 }
 init() {
  schedule.scheduleJob(this.rule, () => {
   this.fn() // 定时调用传入的回调方法
  });
 }
}

data: { //用户1
  lineId: 111130,
  vehTime: 0722,
  startTime: 0751,
  onStationId: 564492,
  offStationId: 17990,
  onStationName: '宝安交通运输局③',
  offStationName: "深港产学研基地",
  tradePrice: 0,
  saleDates: '',
  beginDate: '',
 },
 phoneNumber: 123123123,
 cookie: 'JSESSIONID=TESTCOOKIE',
 day: "17"
}
var obj2 = { //用户2
 data: {
  lineId: 134423,
  vehTime: 1820,
  startTime: 1855,
  onStationId: 4322,
  offStationId: 53231,
  onStationName: '百度国际大厦',
  offStationName: "裕安路口",
  tradePrice: 0,
  saleDates: '',
  beginDate: '',
 },
 phoneNumber: 175932123124,
 cookie: 'JSESSIONID=TESTCOOKIE',
 day: "" 
}
var ticket = new QueryTicket(obj) //用户1
var ticket2 = new QueryTicket(obj2) //用户2

new SetInter({
 timer: 1, //每秒执行一次，建议5秒，不然怕被ip拉黑，我这里只是为了方便下面截图
 fn: function () {
  [ticket,ticket2].map(item => { //同时进行两个用户的抢票
   if (!item.getStop()) { //调用实例的原型方法，判断是否停止抢票，如果没有则继续抢
    item.init()
   } else { // 如果抢到票了，则不继续抢票
    console.log('stop')
   }
  })
 }
})