jarviscrawlercore/src/service/cfg.js

jarvis crawler core

const yaml = require('yaml-js');
const fs = require('fs');
const alimamacfg = require('../alimama/cfg');

/**
 * load config
 * @param {string} cfgfile - cfgfile
 * @return {object} cfg - config
 */
function loadConfig(cfgfile) {
  const fd = fs.readFileSync(cfgfile);
  if (fd) {
    const cfg = yaml.load(fd);

    if (cfg.alimamaconfig) {
      cfg.alimamacfg = alimamacfg.loadConfig(cfg.alimamaconfig);
    }

    return cfg;
  }

  return undefined;
}

/**
 * check config
 * @param {object} cfg - config
 * @return {Error} err - error
 */
function checkConfig(cfg) {
  if (!cfg) {
    return new Error('config undefined');
  }

  if (!cfg.servAddr) {
    return new Error('no config.servAddr');
  }

  if (cfg.headless === undefined) {
    cfg.headless = true;
  }

  if (cfg.alimamacfg != undefined) {
    const err = alimamacfg.checkConfig(cfg.alimamacfg);
    if (err) {
      return err;
    }
  }

  return undefined;
}

/**
 * is valid token
 * @param {object} cfg - config
 * @param {string} token - token
 * @return {bool} isvalid - is valid token
 */
function isValidToken(cfg, token) {
  if (!cfg) {
    return false;
  }

  if (!cfg.clientToken) {
    return true;
  }

  if (typeof cfg.clientToken === 'string') {
    return token === cfg.clientToken;
  }

  if (Array.isArray(cfg.clientToken)) {
    return cfg.clientToken.indexOf(token) >= 0;
  }

  return false;
}

exports.loadConfig = loadConfig;
exports.checkConfig = checkConfig;
exports.isValidToken = isValidToken;