jarviscrawlercore/src/service/plugins/alimama.js

jarvis crawler core

const messages = require('../../../pbjs/result_pb');
const {alimamaKeepalive} = require('../../alimama/keepalive');
const {alimamaSearch} = require('../../alimama/search');
const {alimamaGetTop} = require('../../alimama/gettop');
const {alimamaGetShop} = require('../../alimama/shop');
const {replyError, replyMsg, setReplyCrawler} = require('../utils');
const {newReplyAlimama} = require('../../proto.alimama.utils');

/**
 * callAlimama - alimama
 * @param {object} browser - browser
 * @param {object} cfg - cfg
 * @param {object} call - call
 * @param {object} param - RequestAlimama
 * @param {object} request - RequestCrawler
 */
function callAlimama(browser, cfg, call, param, request) {
  let timeout = 3 * 60 * 1000;
  if (request.getTimeout()) {
    timeout = request.getTimeout();
  }

  if (param.getMode() == messages.AlimamaMode.ALIMMM_KEEPALIVE) {
    alimamaKeepalive(browser, cfg.alimamacfg, timeout)
        .then((ret) => {
          if (ret.error) {
            replyError(call, ret.error, true);

            return;
          }

          const reply = new messages.ReplyCrawler();

          const val = newReplyAlimama(
              messages.AlimamaMode.ALIMMM_KEEPALIVE,
              undefined,
          );

          setReplyCrawler(reply, messages.CrawlerType.CT_ALIMAMA, val);

          replyMsg(call, reply, true);
        })
        .catch((err) => {
          replyError(call, err.toString(), true);
        });
  } else if (param.getMode() == messages.AlimamaMode.ALIMMM_SEARCH) {
    alimamaSearch(browser, param.getText(), cfg.alimamacfg, timeout)
        .then((ret) => {
          if (ret.error) {
            replyError(call, ret.error, true);

            return;
          }

          const reply = new messages.ReplyCrawler();

          const val = newReplyAlimama(
              messages.AlimamaMode.ALIMMM_SEARCH,
              ret.ret,
          );

          setReplyCrawler(reply, messages.CrawlerType.CT_ALIMAMA, val);

          replyMsg(call, reply, true);
        })
        .catch((err) => {
          replyError(call, err.toString(), true);
        });
  } else if (param.getMode() == messages.AlimamaMode.ALIMMM_GETTOP) {
    alimamaGetTop(browser, cfg.alimamacfg, timeout)
        .then((ret) => {
          if (ret.error) {
            replyError(call, ret.error, true);

            return;
          }

          const reply = new messages.ReplyCrawler();

          const val = newReplyAlimama(
              messages.AlimamaMode.ALIMMM_GETTOP,
              ret.ret,
          );

          setReplyCrawler(reply, messages.CrawlerType.CT_ALIMAMA, val);

          replyMsg(call, reply, true);
        })
        .catch((err) => {
          replyError(call, err.toString(), true);
        });
  } else if (param.getMode() == messages.AlimamaMode.ALIMMM_GETSHOP) {
    alimamaGetShop(browser, param.getUrl(), cfg.alimamacfg, timeout)
        .then((ret) => {
          if (ret.error) {
            replyError(call, ret.error, true);

            return;
          }

          const reply = new messages.ReplyCrawler();

          const val = newReplyAlimama(messages.AlimamaMode.ALIMMM_GETSHOP, ret.ret);

          setReplyCrawler(reply, messages.CrawlerType.CT_ALIMAMA, val);

          replyMsg(call, reply, true);
        })
        .catch((err) => {
          replyError(call, err.toString(), true);
        });
  } else {
    replyError(call, 'invalid mode', true);
  }
}

exports.callAlimama = callAlimama;