@watsonserve/stock-base 0.0.1

package/dao.js

@@ -0,0 +1,103 @@
+import { InfluxDB, Point } from '@influxdata/influxdb-client';
+import { request } from 'node:http';
+import { EnCurrency, EnMarket } from './stock.js';
+function getStartTime() {
+    switch (new Date().getUTCDay()) {
+        case 0: // Sunday
+            return '-2d';
+        case 1:
+            return '-3d';
+        default:
+    }
+    return '-1d';
+}
+export function toFxPoint(sgd, hkd, cny, timestamp) {
+    return new Point(EnMarket.FX)
+        .floatField('SGD', sgd)
+        .floatField('HKD', hkd)
+        .floatField('CNY', cny)
+        .timestamp(timestamp);
+}
+export function toStockPoint(market, st, timestamp) {
+    return new Point(market)
+        .tag('nc', st.nc)
+        .floatField('o', st.o || 0)
+        .floatField('h', st.h || 0)
+        .floatField('l', st.l || 0)
+        .floatField('c', st.c || 0)
+        .intField('v', st.v || 0)
+        .floatField('boll_u', st.boll_u || 0)
+        .floatField('boll_l', st.boll_l || 0)
+        .floatField('ma_5', st.ma_5 || 0)
+        .floatField('ma_20', st.ma_20 || 0)
+        .floatField('ma_60', st.ma_60 || 0)
+        .floatField('ma_250', st.ma_250 || 0)
+        .timestamp(timestamp);
+}
+export class InfluxDAO {
+    origin = '';
+    org = '';
+    bucket = '';
+    token = '';
+    constructor(dbConf) {
+        const { origin, org, bucket, token } = dbConf;
+        this.origin = origin;
+        this.org = org;
+        this.bucket = bucket;
+        this.token = token;
+    }
+    async rm(measurement, start, stop) {
+        const conf = {
+            method: 'POST',
+            hostname: `${this.origin}/api/v2/delete?org=${this.org}&bucket=${this.bucket}`,
+            headers: {
+                'Authorization': `Token ${this.token}`,
+                'Content-Type': 'application/json',
+            },
+        };
+        const body = JSON.stringify({ start, stop, predicate: `_measurement="${measurement}"` });
+        return new Promise((s, j) => request(conf, res => res.on('error', j).on('end', s)).end(body).on('error', j));
+    }
+    async writeToInfluxDB(points) {
+        const { origin: url, token, org, bucket } = this;
+        const client = new InfluxDB({ url, token });
+        const writeClient = client.getWriteApi(org, bucket, 's');
+        writeClient.writePoints(points);
+        await writeClient.flush();
+        await writeClient.close();
+    }
+    async __query(fileds, start, markets, ncs) {
+        const { origin: url, token, org, bucket } = this;
+        const client = new InfluxDB({ url, token });
+        const queryClient = client.getQueryApi(org);
+        const mkQuery = markets.map(mk => `r["_measurement"] == "${mk}"`).join(' or ');
+        const ncQuery = ncs.map(nc => `r["nc"] == "${nc}"`).join(' or ');
+        const fdQuery = fileds.map(fd => `r["_field"] == "${fd}"`).join(' or ');
+        const filters = [mkQuery, ncQuery, fdQuery].filter(Boolean).map(str => `|> filter(fn: (r) => ${str})`).join('\n');
+        const fluxQuery = `from(bucket: "${bucket}")
+  |> range(start: ${start})
+  ${filters}
+  |> pivot(rowKey:["_time"], columnKey: ["_field"], valueColumn: "_value")`;
+        return new Promise((resolve, reject) => {
+            const list = [];
+            queryClient.queryRows(fluxQuery, {
+                next: (row, tableMeta) => {
+                    const item = tableMeta.toObject(row);
+                    const { result, table, _start, _stop, _measurement, _time, nc, ...content } = item;
+                    list.push({ time: new Date(_time), market: _measurement, nc, ...content });
+                },
+                error: reject,
+                complete: () => resolve(list),
+            });
+        });
+    }
+    ;
+    async readFxLatest() {
+        const start = getStartTime();
+        return this.__query([EnCurrency.SGD, EnCurrency.HKD, EnCurrency.CNY], start, [EnMarket.FX], []);
+    }
+    async readStockLatest(ncs) {
+        const start = getStartTime();
+        return this.__query(['c', 'v'], start, [EnMarket.SGX, EnMarket.USA, EnMarket.HKEX], ncs);
+    }
+}

package/index.js

@@ -0,0 +1,3 @@
+export * from './stock.js';
+export * from './dao.js';
+export * from './log.js';

package/log.js

@@ -0,0 +1,42 @@
+import { appendFile } from 'fs/promises';
+export var EnLogLevel;
+(function (EnLogLevel) {
+    EnLogLevel[EnLogLevel["DEBUG"] = 7] = "DEBUG";
+    EnLogLevel[EnLogLevel["INFO"] = 6] = "INFO";
+    EnLogLevel[EnLogLevel["WARN"] = 4] = "WARN";
+    EnLogLevel[EnLogLevel["ERROR"] = 3] = "ERROR";
+})(EnLogLevel || (EnLogLevel = {}));
+function findAppName() {
+    const __dirPath = import.meta.dirname.split('/').reverse();
+    if (['release', 'src', 'dist'].includes(__dirPath[0])) {
+        __dirPath.shift();
+    }
+    return __dirPath[0];
+}
+export class Log {
+    _logDir = `/var/log/${process.env.APP_NAME || findAppName()}`;
+    level = EnLogLevel.INFO;
+    constructor(logDir = '') {
+        if (!logDir)
+            return;
+        this._logDir = logDir;
+    }
+    log(lev, msg) {
+        if (lev > this.level)
+            return Promise.resolve();
+        return appendFile(`${this._logDir}/${EnLogLevel[lev].toLowerCase()}.log`, `[${new Date().toISOString()}] ${msg}\n`);
+    }
+    debug(msg) {
+        return this.log(EnLogLevel.DEBUG, msg);
+    }
+    info(msg) {
+        return this.log(EnLogLevel.INFO, msg);
+    }
+    warn(msg) {
+        return this.log(EnLogLevel.WARN, msg);
+    }
+    error(msg) {
+        return this.log(EnLogLevel.ERROR, msg.toString());
+    }
+}
+export const log = new Log();

package/package.json

@@ -0,0 +1,18 @@
+{
+  "name": "@watsonserve/stock-base",
+  "version": "0.0.1",
+  "description": "",
+  "type": "module",
+  "keywords": [],
+  "author": "",
+  "license": "ISC",
+  "packageManager": "pnpm@10.28.2",
+  "dependencies": {
+    "@influxdata/influxdb-client": "^1.35.0",
+    "influx": "^5.12.0"
+  },
+  "main": "index.js",
+  "publishConfig": {
+    "access": "public"
+  }
+}

package/stock.js

@@ -0,0 +1,69 @@
+export var EnMarket;
+(function (EnMarket) {
+    EnMarket["FX"] = "FX";
+    EnMarket["JPX"] = "JPX";
+    EnMarket["CHN"] = "CHN";
+    EnMarket["HKEX"] = "HKEX";
+    EnMarket["SGX"] = "SGX";
+    EnMarket["LSE"] = "LSE";
+    EnMarket["USA"] = "USA";
+})(EnMarket || (EnMarket = {}));
+export var EnCurrency;
+(function (EnCurrency) {
+    EnCurrency["SGD"] = "SGD";
+    EnCurrency["USD"] = "USD";
+    EnCurrency["HKD"] = "HKD";
+    EnCurrency["CNY"] = "CNY";
+})(EnCurrency || (EnCurrency = {}));
+export class Stock {
+    nc;
+    name;
+    currency;
+    o = 0;
+    c = 0;
+    h = 0;
+    l = 0;
+    v = 0;
+    constructor(code, name = '', currency = EnCurrency.USD) {
+        this.nc = code;
+        this.name = name;
+        this.currency = currency;
+    }
+    setInfo(name, close, open = 0, height = 0, low = 0, vol = 0) {
+        this.name = name;
+        this.o = open;
+        this.c = close;
+        this.h = height;
+        this.l = low;
+        this.v = vol;
+    }
+}
+export class HandleStock extends Stock {
+    count = 0;
+    cost = 0;
+    category = '';
+    usd_cost = 0;
+    proportion = 0;
+    _usdValue = 0;
+    constructor(nc, count = 0, cost = 0.0, currency = EnCurrency.USD, category = '') {
+        super(nc, '', currency);
+        this.count = count;
+        this.cost = cost;
+        this.category = category;
+        this.usd_cost = cost;
+    }
+    setFx(fx) {
+        this._usdValue = this.c * this.count / fx;
+        this.usd_cost = this.cost / fx;
+    }
+    set asset(asset) {
+        this.proportion = this.usdValue / asset * 100.0;
+        this.proportion;
+    }
+    get usdValue() {
+        return this._usdValue || (this.c * this.count);
+    }
+    toJSON() {
+        return `{"category": "${this.category}", "title": "${this.name}", "amount": ${this.usdValue.toFixed(2)}, "proportion": "${this.proportion.toFixed(2)}%"}`;
+    }
+}