star/lib/watch.js

A STock Analysis and Research tool for terminal users.

'use strict';

/**
 * @author      hustcer
 * @license     MIT
 * @copyright   uzfin.com
 * @create      06/24/2015
 */

let vm      = require('vm'),
    fs      = require('fs'),
    _       = require('lodash'),
    yaml    = require('js-yaml'),
    blessed = require('blessed'),
    Promise = require('bluebird'),
    cmd     = require('commander'),
    contrib = require('blessed-contrib');

let conf    = require('./conf').conf;
let Iconv   = require('iconv').Iconv;
let iconv   = new Iconv('GBK', 'UTF-8//TRANSLIT//IGNORE');
let request = Promise.promisifyAll(require('request'), {multiArgs:true});

let table   = null;
let screen  = blessed.screen({ fullUnicode:true });

const MSG = {
    INPUT_ERROR      : '输入错误，当前只支持通过证券代码看盘，请检查后重新输入！',
    TOO_MANY_SYMS    : '输入的股票太多，一次最多支持盯盘25只股票！',
    SYMBOL_NOT_EXIST : '该股票代码不存在:',
};

// A client error like 400 Bad Request happened
let ClientError = e => (e.code >= 400 && e.code < 500 || e.code === 'ENOTFOUND');

// Set current data source
let src    = (cmd.data || '').toUpperCase();
let source = (src === 'SINA' || src === 'TENCENT') ? src : conf.dataSource;
let ds     =  conf.provider[source];

/**
 * Check validity of input symbols, and return a valid symbols array if there is.
 * @param  {String} syms    Input symbols' string.
 * @return {Array}  Array of input symbols
 */
let getValidSymbols = function getValidSymbols(syms){
    // When no symbols were provided, use symbols from watchList or holded symbols.
    if(syms === true){
        let Common   = require('./common').Common;
        let yamlConf = yaml.safeLoad(fs.readFileSync(Common.getSymbolFilePath(), 'utf8'));
        if(!cmd.hold && yamlConf.watchList && yamlConf.watchList.length > 0){
            syms = _.map(yamlConf.watchList, 'code').join(',');
        } else {
            syms = _.filter(yamlConf.symbols, s => s.hold);
            syms = _.map(syms, 'code').join(',');
        }
    }
    let symbols = syms.replace(/，/g, ',');
    let valid   = /^[0-9,]+$/.test(symbols);
    if(!valid){
        console.error(MSG.INPUT_ERROR.error);
        return false;
    }
    symbols = _.trimEnd(symbols, ',').split(',');
    if(symbols.length>conf.chunkSize){
        console.error(MSG.TOO_MANY_SYMS.error);
        return false;
    }
    return symbols;
};

/**
 * Init table layout and screen keyboard event
 */
let initUI = function initUI(){
    table = contrib.table({
        keys          : true,
        fg            : 'white',
        selectedFg    : 'black',
        selectedBg    : 'cyan',
        interactive   : true,
        label         : '股票列表',
        width         : '55%',
        height        : '80%',
        border        : { type: 'line', fg: 'cyan' },
        columnSpacing : 5,
        columnWidth   : [10, 7, 7, 7, 9],
    });

    /* eslint no-process-exit:0 */
    screen.key(['escape', 'q', 'C-c'], () => process.exit(0) );
};

/**
 * Update table data
 * @param  {Array} queryResults    New table data to be updated
 */
let updateData = function updateData(queryResults){
    let queryData = [];
    _.each(queryResults, q => {
        // Fix the padding of company name
        q.name   = (q.name.length === 3) ? `  ${q.name}`:q.name;
        q.price  = q.price.toFixed(2);
        q.inc    = q.inc.toFixed(2);
        q.incPct = `${q.incPct.toFixed(2)} %`;

        queryData.push([q.name, q.code, q.price, q.inc, q.incPct]);
    });

    table.setData({
        headers : ['公司', '代码', '当前价', '涨跌', '涨跌%'],
        data    : queryData,
    });
    table.focus();
    screen.append(table);
    screen.render();
};

/**
 * Query and refresh price data
 * @param  {Array} syms   Symbols' code array
 * @param  {String} query Query string
 */
let refreshData = function refreshData(syms, query){
    // symbol query results
    let queryResults = [];
    return request.getAsync(ds.url + query, { encoding:null }).spread((resp, body) => {
        body = iconv.convert(body).toString();
        vm.runInThisContext(body);

        _.each(syms, s => {
            let localVar = ds.flag + conf.market[s.substr(0,3)] + s;
            if(body.indexOf(localVar) < 0 || body.indexOf(`${localVar}="";`) >= 0){
                console.error(MSG.SYMBOL_NOT_EXIST.error, s.em);
                return false;
            }

            let res    = {};
            let splits = vm.runInThisContext(ds.flag + conf.market[s.substr(0,3)] + s).split(ds.sep);
            res.name   = splits[ds.nameIdx];
            res.code   = s;
            // 对于停牌股票价格以上个交易日收盘价格为准
            res.close  = +splits[ds.closeIdx];
            res.price  = +splits[ds.priceIdx] || res.close;
            res.inc    = res.price - res.close;
            res.incPct = ((res.price - res.close)/res.close) * 100;
            queryResults.push(res);
        });

        updateData(queryResults);

    // FIXME: fix 'ENOTFOUND' error
    }).catch(ClientError, () => {
        process.exit(1);
    });
};

/**
 * Query symbols' price info
 * @param  {Array} syms  Symbols array to query
 * @return {Array}       Basic info array of queried symbols
 */
let querySymbols = function querySymbols(syms){
    let query = _.map(syms, x => conf.market[x.substr(0,3)] + x).join(',');
    refreshData(syms, query);
    setInterval(refreshData.bind(this, syms, query), 3600);
};

/**
 * Do the query action.
 * @param  {String} syms    Input symbols' string.
 */
let doWatch = function doWatch(syms){
    initUI();
    Promise
        .resolve(syms)
        .then(getValidSymbols)
        .then(querySymbols);
};

// star -w 002625,600588,601179,600415,600816,600880,600703,000861,600410,000768,000712,600118,600893,002456,000902,000725,000002,000100
exports.Watch = {
    doWatch,
};