@watsonserve/stock-base 0.0.4 → 0.0.6

package/dao.js

@@ -34,6 +34,22 @@ export function toStockPoint(market, st, timestamp) {
         .floatField('ma_250', st.ma_250 || 0)
         .timestamp(timestamp);
 }
+export function toDivPoints(market, nc, data) {
+    const { curr, amount, ex, paid } = data;
+    return [
+        new Point('Dividend').tag('nc', `${nc}.${market}`).tag('type', 'ex').tag('currency', curr).floatField('amount', amount).timestamp(ex),
+        new Point('Dividend').tag('nc', `${nc}.${market}`).tag('type', 'paid').tag('currency', curr).floatField('amount', amount).timestamp(paid)
+    ];
+}
+function groupBy(key, list) {
+    return list.reduce((pre, item) => {
+        const foo = item[key];
+        const fooList = pre[foo] || [];
+        pre[foo] = fooList;
+        fooList.push(item);
+        return pre;
+    }, {});
+}
 export class InfluxDAO {
     origin = '';
     org = '';
@@ -100,4 +116,31 @@ export class InfluxDAO {
         const start = getStartTime();
         return this.__query(['c', 'v'], start, [EnMarket.SGX, EnMarket.USA, EnMarket.HKEX], ncs);
     }
+    async readDividend(ncs) {
+        const { origin: url, token, org, bucket } = this;
+        const client = new InfluxDB({ url, token });
+        const queryClient = client.getQueryApi(org);
+        const ncQuery = ncs.map(nc => `r["nc"] == "${nc}"`).join(' or ');
+        const fluxQuery = `from(bucket: "${bucket}")
+    |> range(start: -1y, stop: 1y)
+    |> filter(fn: (r) => r["_measurement"] == "Dividend")
+    |> filter(fn: (r) => r["_field"] == "amount")
+    |> filter(fn: (r) => ${ncQuery})
+    |> filter(fn: (r) => r["type"] == "ex")
+    `;
+        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(Object.entries(groupBy('nc', list)));
+                },
+            });
+        });
+    }
 }

package/log.js

@@ -7,7 +7,7 @@ export var EnLogLevel;
     EnLogLevel[EnLogLevel["ERROR"] = 3] = "ERROR";
 })(EnLogLevel || (EnLogLevel = {}));
 function findAppName() {
-    const __dirPath = import.meta.dirname.split('/').reverse();
+    const __dirPath = process.argv.find(arg => arg.endsWith('.js'))?.split('/').reverse().slice(1) || [];
     if (['release', 'src', 'dist'].includes(__dirPath[0])) {
         __dirPath.shift();
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@watsonserve/stock-base",
-  "version": "0.0.4",
+  "version": "0.0.6",
   "description": "",
   "type": "module",
   "keywords": [],

package/types/dao.d.ts

@@ -1,5 +1,5 @@
 import { Point } from '@influxdata/influxdb-client';
-import { EnMarket, type IStock } from './stock.js';
+import { EnMarket, IDividend, type IStock } from './stock.js';
 export interface IDBConf {
     origin: string;
     org: string;
@@ -8,6 +8,7 @@ export interface IDBConf {
 }
 export declare function toFxPoint(sgd: number, hkd: number, cny: number, timestamp: number): Point;
 export declare function toStockPoint(market: EnMarket, st: IStock, timestamp: number): Point;
+export declare function toDivPoints(market: string, nc: string, data: IDividend): Point[];
 export declare class InfluxDAO {
     private origin;
     private org;
@@ -19,4 +20,5 @@ export declare class InfluxDAO {
     private __query;
     readFxLatest(): Promise<any[]>;
     readStockLatest(ncs: string[]): Promise<any[]>;
+    readDividend(ncs: string[]): Promise<any[]>;
 }

package/types/stock.d.ts

@@ -37,6 +37,15 @@ export interface StockPoint {
     timestamp: number;
     st: IStock;
 }
+export interface IDividend {
+    curr: string;
+    amount?: number;
+    ratio?: number;
+    annc: Date;
+    ex: Date;
+    paid: Date;
+    announcement_URL?: string;
+}
 export declare class Stock {
     nc: string;
     name: string;