serve-cms/app/service/tool.js

/* eslint-disable strict */
const Service = require('egg').Service;
const svgCaptcha = require('svg-captcha');
// const cheerio = require('cheerio');
// const puppeteer = require('puppeteer');
const charset = require('superagent-charset');
const superagent = charset(require('superagent'));
const cheerio = require('cheerio');
const fs = require('fs');
const { CrudService } = require('naf-framework-mongoose/lib/service');
class ToolsService extends CrudService {

  // 产生验证码
  async captcha() {
    const captcha = svgCaptcha.create({
      size: 4, // 大小
      fontSize: 50, // 字体大小
      width: 100, // 宽
      height: 40, // 高
      bacground: '#cc9966', // 背景色
    });
    return captcha;
  }

  // 发送短信验证码
  async sendmessage(ctx, app, randomstr) {
    const message = '【吉林省就业中心】您的验证码为:' + randomstr + '，请在30分钟内完成输入，欢迎使用吉林省智慧就业企业服务平台。';
    const data = '?Id=300&Name=wwqcgh&Psw=jljyzx-wwqcgh&Message=' + message + '&Phone=' + ctx.query.mobile + '&Timestamp=0';
    // const data = '?Id=300&Name=wwqcgh&Psw=jljyzx-wwqcgh&Message=' + message + '&Phone=' + phone + '&Timestamp=0';
    const path = ctx.app.config.messageDir + data;
    const result = await app.curl(path, {
      method: 'GET',
      dataType: 'text/xml',
    });
    return result;
  }


  async creeper() {
    // 目标链接 吉林省人事考试网第一页
    const targetUrl = 'http://www.jlzkb.com/cms/root/ksbmList.vm?dir=L-iAg-ivleaKpeWQjS_kuovkuJrljZXkvY3mi5vogZjogIPor5U&page=1&rows=10';
    const columnTitle = '吉林省事业单位招聘';
    // 查看是否有这个栏目，没有则创建一个
    let column = await this.service.column.model.find({ news_type: '0', title: columnTitle });
    if (column.length === 0) {
      column = await this.service.column.model.create({ site: '99991', news_type: '0', title: columnTitle, type: '', parent_id: '', parent: '', is_use: '' });
    } else {
      column = column[0];
    }
    await this.creeperCreate(targetUrl, column);
  }


  // 输入路径返回文本
  async creeperCreate(targetUrl, column) {
    // 目标链接 吉林省人事考试网第一页
    // 用来暂时保存解析到的内容和图片地址数据
    const hrefOld = targetUrl;
    const hrefAdd = 'http://www.jlzkb.com/cms/root/';
    const uri = 'http://www.jlzkb.com';
    // const imgs = [];
    // 创建附件文件夹(暂定直接跳到该网站下载)
    // this.mkdir('./attachment');

    // 发起请求
    superagent.get(targetUrl).charset('utf-8').buffer(true)
      .end((error, res) => {
        if (error) { // 请求出错，打印错误，返回
          console.log(error);
          return;
        }
        // cheerio需要先load html
        const $ = cheerio.load(res.text);
        // 循环列表，获取标题、a标签路径、日期，然后根据a标签路径再次进行爬出内容、保存即可
        $('#DivInfoList tr').each((index, element) => {
        // 这些数据都是存放在news中的
          const title = $(element).find('td a').attr('title');
          const thisHref = $(element).find('td a').attr('href');
          const time = $(element).find('td[width="12%"]').text();
          const publishTime = time.substring(1, 11);
          // 这里可以给时间做判断当前日期（如果需要的话可以做为判断条件）
          const nowDate = new Date().toLocaleDateString();
          if (publishTime !== nowDate) {
            // 为undefined时，不需要进行下一步了
            if (thisHref !== undefined) {
              const thisAllHref = hrefAdd + thisHref;
              // 请求内容
              superagent.get(thisAllHref).charset('utf-8').buffer(true)
                .end((error, res) => {
                  if (error) { // 请求出错，打印错误，返回
                    console.log(error);
                    return;
                  }
                  const $ = cheerio.load(res.text);
                  // 获取内容保存
                  const content = $('#fontzoom').children('p').clone();
                  content.find(':nth-child(n)').remove();
                  const attachment = [];
                  // 每页都查询是否有附件存在，如果有，下载到本地，保存即可
                  $('#fontzoom p a').each((index, element) => {
                    const thisHref = $(element).attr('href');
                    if (thisHref.substring(0, 4) !== 'http') {
                      const url = uri + thisHref;
                      const fileName = $(element).text();
                      // const filepath = this.downloadAttachment(url, fileName);
                      const file = {
                        name: fileName,
                        uri: url,
                      };
                      attachment.push(file);
                    }
                  });
                  const news = this.service.news.model.create({
                    site: column.site,
                    title,
                    pic: '',
                    content,
                    type: '',
                    parent_id: column.id,
                    parent: column.title,
                    publish_time: publishTime,
                    attachment,
                    is_use: '0'
                  });
                });
            }
          }
        });
        // 点击下一页
        const href = $('#DivPageControl a').eq(2).attr('href');
        const hrefNew = hrefAdd + href;
        // 第一次路径与第二次路径比较，相同，就不需要调自己了
        console.log(hrefNew);
        console.log(hrefOld);
        if (hrefNew !== hrefOld) {
          // this.creeperCreate(hrefNew, column);
        // over
        }
      });
  }
  // 创建目录

  async mkdir(_path) {
    if (fs.existsSync(_path)) {
      console.log(`${_path}目录已存在`);
    } else {
      fs.mkdir(_path, error => {
        if (error) {
          return console.log(`创建${_path}目录失败`);
        }
        console.log(`创建${_path}目录成功`);
      });
    }
  }
  // -------------------------------------------------------这里应返回路径供给前台下载（图片存放问题一并解决）-----------------------------------------------------
  // 下载爬到的附件
  async downloadAttachment(thisHref, fileName) {
    // 下载图片存放到指定目录
    const stream = fs.createWriteStream('./attachment/' + fileName);
    const req = superagent.get(thisHref); // 响应流
    req.pipe(stream);
    console.log(thisHref);
    console.log(fileName);
    return '这里返回路径保存即可';
    // console.log('开始下载>>>>>>>>>>>>>>');
  }

}
module.exports = ToolsService;