quote-observer 2.0.1 → 2.0.3

package/core.js

@@ -5,10 +5,14 @@
  * @Last Modified time: 2021-11-09 10:43:03
  */
 
+const fs = require('fs');
 const logger = require('./logger')();
 const Futu = require('./services/futu');
 const KLine = require('./services/kline');
 
+// Update KLine every 10 minutes
+const KLINE_UPDATE_PERIOD = 10 * 60 * 1000;
+
 // https://github.com/Marak/colors.js/blob/56de9f0983f68cd0a08c5b76d10a783e4b881716/lib/styles.js
 const styleCodes = {
   black: [30, 39],
@@ -23,13 +27,13 @@ const styleCodes = {
 };
 
 function colors(string, name) {
-  let [open, close] = styleCodes[name];
+  const [open, close] = styleCodes[name];
   return `\u001b[${open}m${string}\u001b[${close}m`;
 }
 
 function moving_average(datas, days = 3) {
   if (!Array.isArray(datas)) return 0;
-  let total = datas
+  const total = datas
     .slice(0, days)
     .map(v => parseFloat(v[2]))
     .reduce((sum, v) => sum += v, 0);
@@ -44,7 +48,7 @@ function moving_average(datas, days = 3) {
 // => 2tm0 = 3tm4 + 3tm3 - 2tm2 - 2tm1
 // => tm0 = 1.5tm4 + 1.5tm3 - tm2 - tm1
 function marginal_value(datas) {
-  let [tm4, tm3, tm2, tm1, tm0] = datas.slice(0, 5).map(v => parseFloat(v[2])).reverse();
+  const [tm4, tm3, tm2, tm1, tm0] = datas.slice(0, 5).map(v => parseFloat(v[2])).reverse();
   return {
     now: tm0,
     curr: 1.5 * tm4 + 1.5 * tm3 - tm2 - tm1,
@@ -65,9 +69,9 @@ const padding = [
 ];
 const pad = (items) => {
   return items.map((item, index) => {
-    let conf = padding[index];
-    let size = conf.size || conf;
-    let text = item[index ? 'padStart' : 'padEnd'](size - item.split(/[^\x00-\xff]/).length - 1);
+    const conf = padding[index];
+    const size = conf.size || conf;
+    const text = item[index ? 'padStart' : 'padEnd'](size - item.split(/[^\x00-\xff]/).length - 1);
     return conf.color ? colors(text, conf.color(item)) : text;
   }).join('');
 };
@@ -76,13 +80,13 @@ const LINE = new Array(128).fill('=').join('');
 
 const STOCKS_MAP = new Map();
 function holding(code) {
-  let stock = STOCKS_MAP.get(code);
+  const stock = STOCKS_MAP.get(code);
   return (stock && stock.market_val) ? '*' : '';
 }
 
 function roc(MV) {
-  let { now, next } = MV;
-  let v = 100 * (next - now) / now;
+  const { now, next } = MV;
+  const v = 100 * (next - now) / now;
   return (v > 0 ? '+' : '') + v.toFixed(2) + '%';
 }
 
@@ -91,33 +95,37 @@ function ceil(v, d = 2) {
 }
 
 function sale_signal(code, now) {
-  let stock = STOCKS_MAP.get(code);
-  let { cost_price } = stock;
+  const stock = STOCKS_MAP.get(code);
+  const { cost_price } = stock;
   if (!cost_price) return '[=====][00]';
-  let irate = 100 * (now - cost_price) / cost_price;
-  let srate = [-25, -15, -8, -5, -2].map(signal => irate < signal ? '-' : ' ').join('');
+  const irate = 100 * (now - cost_price) / cost_price;
+  const srate = [-25, -15, -8, -5, -2].map(signal => irate < signal ? '-' : ' ').join('');
   return `[${srate}][${ceil(irate)}]`;
 }
 
+function random_delay() {
+  return new Promise(r => setTimeout(r, Math.random() * 60 * 1000));
+}
+
 function start() {
-  let quotes = [...STOCKS_MAP.keys()].map(code => KLine(code, 10));
+  const quotes = [...STOCKS_MAP.keys()].map(code => random_delay().then(() => KLine(code, 10, STOCKS_MAP.get(code)?.tid)));
   Promise.all(quotes).then(resps => {
     console.reset();
     console.log((new Date()).toTimeString().split(' ')[0]);
     console.log(LINE);
-    console.log(pad(['ID', '名称', 'NOW', 'MA3', 'MA5', 'MVC', 'MVN', 'MVN-ROC', 'S-SIG']));
+    console.log(pad(['ID', 'NAME', 'NOW', 'MA3', 'MA5', 'MVC', 'MVN', 'MVN-ROC', 'S-SIG']));
     console.log(LINE);
-    let rows = [[], LINE.replace(/=/g, '-'), []];
-    let errors = resps.filter((data, index) => {
-      let { fid, id, info, data: resp } = data;
+    const rows = [[], LINE.replace(/=/g, '-'), []];
+    const errors = resps.filter((data, index) => {
+      const { fid, id, info, data: resp } = data;
       if (!id || !info || !resp) return false;
-      let name = String(info[1] || '').replace(/(ETF|\s).+$/i, '');
-      let MA3 = moving_average(resp, 3);
-      let MA5 = moving_average(resp, 5);
-      let MV = marginal_value(resp);
+      const name = String(info[1] || '').replace(/(ETF|\s).+$/i, '');
+      const MA3 = moving_average(resp, 3);
+      const MA5 = moving_average(resp, 5);
+      const MV = marginal_value(resp);
       if (!MV.now) return global.MARKET[id] = `us${info[2]}`;
-      let hold_star = holding(fid);
-      let row = pad([
+      const hold_star = holding(fid);
+      const row = pad([
         fid + hold_star, String(name).replace(/-\w+$/, ''),
         MV.now.toFixed(3), MA3.toFixed(3), MA5.toFixed(3),
         MV.curr.toFixed(3), MV.next.toFixed(3), roc(MV),
@@ -128,10 +136,9 @@ function start() {
     console.log(rows.flat().join('\n'));
     console.log(LINE);
     if (errors.length) throw new Error('Errors occur.');
-    setTimeout(() => start(), 10 * 1000);
+    setTimeout(() => start(), KLINE_UPDATE_PERIOD);
   }).catch((error) => {
     logger.error(error.message);
-    setTimeout(() => start(), 1 * 1000);
   });
 }
 
@@ -143,6 +150,14 @@ function update_stocks() {
   // SH,SZ,HK,US => [ 'SH', 'SZ', 'HK', 'US' ]
   const markets = String(global.config?.opts?.market || '').trim()
     .split(',').map(v => v.trim().toUpperCase()).filter(v => v);
+  const codes = global.config?.opts?.codes;
+  if (codes && fs.existsSync(codes)) {
+    fs.readFileSync(codes, 'utf8').toString().trim().split('\n').map(v => {
+      const stock = JSON.parse(v.trim());
+      STOCKS_MAP.set(stock.code, stock);
+    });
+    return Promise.resolve();
+  }
   return Promise.all([Futu.position(), Futu.following()]).then(([holders, follows]) => {
     STOCKS_MAP.clear();
     [

package/index.js

@@ -15,6 +15,7 @@ const pkg = require('./package.json');
 program
   .version(pkg.version)
   .option('-S, --save <root>', `save response into root`, '')
+  .option('-C, --codes <root>', `load codes from file`, '')
   .option('-H, --holder', `show stock holder`, true)
   .option('    --no-holder', `hide stock holder`)
   .option('-F, --follow', `show stock follow`, false)

package/logger.js

@@ -19,9 +19,9 @@ const Logger = new Console(
 fs.writeFileSync(logfile, '');
 
 module.exports = () => {
-  let error = (new Error()).stack.toString().split('\n')[2] || '';
-  let filename = (error.match(/[\\\/\(]([-\w\.]+.\w+):\d+:\d+\)$/) || [])[1] || 'unknown';
-  let logger = {};
+  const error = (new Error()).stack.toString().split('\n')[2] || '';
+  const filename = (error.match(/[\\\/\(]([-\w\.]+.\w+):\d+:\d+\)$/) || [])[1] || 'unknown';
+  const logger = {};
   'trace,debug,info,warn,error,fatal,mark'.split(',').forEach(name => {
     logger[name] = (...args) => {
       Logger.info(`[${(new Date()).toISOString()}] [${filename}] [${name.toUpperCase()}]`, ...args);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "quote-observer",
-  "version": "2.0.1",
+  "version": "2.0.3",
   "description": "quote observer",
   "main": "index.js",
   "scripts": {

package/services/futu.js

@@ -10,8 +10,8 @@ const conf = global.config.futu;
 const logger = require('../logger')();
 
 function apply(name) {
-  let headers = { Authorization: conf.auth };
-  let url = `${conf.base}${conf[name]}`;
+  const headers = { Authorization: conf.auth };
+  const url = `${conf.base}${conf[name]}`;
   logger.info(`Query ${name}: ${url}`);
   return axios
     .get(url, { headers })

package/services/kline.js

@@ -39,27 +39,33 @@ const save_resp = (data) => {
   }
 };
 
-function kline(id, days = 10) {
-  let [mkt, code] = id.split('.');
-  if (!code) {
+const convert2tid = (mkt, code, tid) => {
+  if (tid) return tid;
+  if (!code) return '';
+  return fix_id(`${mkt.toLowerCase()}${code}${'US' === mkt ? '.OQ' : ''}`);
+};
+
+function kline(id, days = 10, tid = '') {
+  const [mkt, code] = id.split('.');
+  const _id = convert2tid(mkt, code, tid);
+  if (!_id) {
     logger.info(`[${id}] INDEX SKIP`);
     return {};
   }
-  let _id = fix_id(`${mkt.toLowerCase()}${code}${'US' === mkt ? '.OQ' : ''}`);
-  let _var = 'kline_dayqfq';
-  let params = {
+  const _var = 'kline_dayqfq';
+  const params = {
     _var,
     _r: Math.random(),
     param: `${_id},day,,,${days},qfq`,
   };
-  let url = `${base_url}${path_mkt[mkt]}?${querystring.stringify(params)}`;
+  const url = `${base_url}${path_mkt[mkt]}?${querystring.stringify(params)}`;
   logger.info(`[${id}] -> ${url}`);
   return axios.get(url)
     .then((raw = '') => {
       try {
-        let resp = JSON.parse(raw.data.toString().replace(`${_var}=`, ''));
-        let data = (resp || {}).data[_id] || [];
-        let info = (data.qt || {})[_id] || [];
+        const resp = JSON.parse(raw.data.toString().replace(`${_var}=`, ''));
+        const data = (resp || {}).data[_id] || [];
+        const info = (data.qt || {})[_id] || [];
         return save_resp({ fid: id, id: _id, info, data: (data.day || data.qfqday || []).reverse() });
       } catch (error) {
         logger.error(`[${id}] PARSE RESPONSE ERROR:`, error.message);