star.js

#!/usr/bin/env node --harmony

/**
 * @author      hustcer
 * @license     MIT
 * @copyright   TraceInvest.com
 * @create      05/18/2015
 */
"use strict";

let _       = require('lodash'),
    colors  = require('colors'),
    Promise = require('bluebird'),
    cmd     = require('commander');

const pkg   = require('./package.json');
let   conf  = require('./lib/conf.js').conf;
let action  = null;   // cmd action

/**
 * Available colors are:
 * bold, italic, underline, inverse, yellow, cyan, white,
 * green, red, grey, blue, rainbow, zebra, random, magenta,
 */
const COLOR_THEME = {
    error  : 'red',
    info   : 'blue',
    data   : 'grey',
    header : 'blue',
    warn   : 'yellow',
    em     : 'magenta'
};

colors.setTheme(COLOR_THEME);

cmd
  .version(pkg.version)
  .usage('[options]' + ' OR'.em + ' star code1,code2，code3...，codeN')
  .description('Star is a command line tool for STock Analysis and Research.')
  .option('-a, --all'          , 'display all stocks.')
  .option('-o, --hold'         , 'display all held stocks.')
  .option('-C, --cal'          , 'display finance calendar of the future month.')
  .option('-I, --ignore'       , 'display all ignored stocks.')
  .option('-i, --insider [c]'  , 'display insider trading records of specified stocks.')
  .option('--latest-sz'        , 'query latest insider tradings of ShenZhen market, should be used with "-i" or "--insider".')
  .option('--latest-sh'        , 'query latest insider tradings of ShangHai market, should be used with "-i" or "--insider".')
  .option('--show-detail'      , 'show detail latest insider trading records, should be used with "-i" or "--insider".')
  .option('-w, --watch [c1...]', 'watch specified stocks or watch all the stocks in watch list.')
  .option('-r, --reverse'      , 'sort stocks in ascending order according to designated field.')
  .option('-l, --limit <n>'    , 'set total display limit of current page.', parseInt)
  .option('-p, --page  <n>'    , 'specify the page index to display.', parseInt)
  .option('-d, --data  <data>' , 'specify data provider, should be one of "sina" or "tencent".')
  .option('-f, --file  <file>' , 'specify the symbol file path.')
  .option('--from <2014/06/01>', 'specify the beginning date of insider tradings.')
  .option('--to <2015/07/09>'  , 'specify the ending date of insider tradings.')
  .option('--span <3m>'        , 'specify the month span of insider tradings ahead of today, should be 1m~24m.')
  .option('-L, --lte   <pct> ' , 'filter the symbols whose upside potential is lower than or equal to the specified percentage.', parseInt)
  .option('-G, --gte   <pct> ' , 'filter the symbols whose upside potential is greater than or equal to the specified percentage.', parseInt)
  .option('-U, --under <star>' , 'filter the symbols whose star is under or equal to the specified star value.', parseInt)
  .option('-A, --above <star>' , 'filter the symbols whose star is above or equal to the specified star value.', parseInt)
  .option('--lteb [pct]'       , "filter the symbols whose current price is lower than or equal to it's buy/cheap price", parseInt)
  .option('--gtes [pct]'       , "filter the symbols whose current price is greater than or equal to it's sell/expensive price", parseInt)
  .option('-g, --grep  <kw>  ' , 'specify the keyword to grep in name or comment, multiple keywords should be separated by ",".')
  .option('--remove  <kw>    ' , 'remove the symbols from result with the specified keywords in name or comment, multiple keywords should be separated by ",".')
  .option('-e, --exclude <pre>', 'exclude stocks whose code number begin with: 300,600,002 or 000, etc. multiple prefixs can be\n' + ' '.repeat(20) +
                                 ' used and should be separated by "," or "，". ')
  .option('-c, --contain <pre>', 'display stocks whose code number begin with: 300,600,002 or 000, etc. multiple prefixs can be\n' + ' '.repeat(20) +
                                 ' used and should be separated by "," or "，". ')
  .option('-s, --sort <sort>'  , 'specify sorting field, could be sorted by: code/star/price/targetp/incp/bdiff/sdiff/capacity/\n' + ' '.repeat(20) +
                                 ' pe/pb to sort stocks by code, star, price, price to target percent, price increase percent, \n' + ' '.repeat(20) +
                                 ' (s.price - s.cheap)/s.price, (s.price - s.expensive)/s.price, capacity, PE and PB separately.\n' + ' '.repeat(20) +
                                 ' and sort by capacity/pe/pb only works while using tencent data source.')
  .parse(process.argv);

let actions = {
    'WATCH'  : function(){

        let Watch = require('./lib/watch.js').Watch;
        Watch.doWatch(cmd.watch);

    },
    'CAL' : function(){

        let Cal = require('./lib/cal.js').Cal;
        Cal.showCal();

    },
    'INSIDER': function(){

        let async   = require('async');
        let Insider = require('./lib/insider.js').Insider;

        if(cmd.latestSz){ Insider.querySZLatest(); return false; }
        if(cmd.latestSh){ Insider.querySHLatest(); return false; }

        if(cmd.insider === true){
          console.error('Input Error: "--insider" or "-i" should be used with symbol code or "--latest-sh"/"--latest-sz"'.error);
          return false;
        }

        let query   = cmd.insider.replace(/，/g, ',');
        let symbols = _.trimRight(query, ',').split(',');
        if(symbols.length > conf.chunkSize){
            console.error(('You can query at most ' + conf.chunkSize + ' symbols once.').error);
            return false;
        }

        async.eachSeries(symbols, function(c, callback){
            Insider.queryInsider(c, callback);
        }, function(){
            console.log('ALL DONE!');
        });

    },
    'QUERY'  : function(){

        let Query = require('./lib/query.js').Query;
        Query.doQuery(cmd.args[0]);

    },
    'TRACE'  : function(){

        let Trace   = require('./lib/trace.js').Trace;
        let symbols = Trace.getFilteredSymbols();

        if(!symbols){ return false; }

        let symList = _.chunk(symbols, conf.chunkSize);

        Promise
            .resolve(symList)
            .each(syms => Trace.querySymbols(syms))
            .then(()   => Trace.printResults())
            .then(()   => Trace.printSummary());
    }
};

let doCmd = function() {

    // 'TRACE' is the default action
    action  = 'TRACE';

    if(cmd.watch)  {  action = 'WATCH';    }
    if(cmd.insider){  action = 'INSIDER';  }
    if(cmd.cal)    {  action = 'CAL';      }

    if(cmd.args.length === 1 ){

      action = 'QUERY';

    }else if (cmd.args.length > 1) {

      console.error('Input error, please try again, or run "star -h" for more help.'.error);
      return false;
    }

    actions[action]();
};

// Get the Job Done!
doCmd();