feat: add trade data method (#6)

v1.4.1

Author: chengzuopeng

README.md

@@ -165,6 +165,7 @@ console.log(`共获取 ${allQuotes.length} 只股票`);
 |------|------|
 | `getFundFlow` | 资金流向 |
 | `getPanelLargeOrder` | 盘口大单占比 |
+| `getTradingCalendar` | A 股交易日历 |
 
 ### 批量查询
 

README_EN.md

@@ -165,6 +165,7 @@ console.log(`Fetched ${allQuotes.length} stocks`);
 |--------|-------------|
 | `getFundFlow` | Fund flow |
 | `getPanelLargeOrder` | Large order ratio |
+| `getTradingCalendar` | A-share trading calendar |
 
 ### Batch Query
 

package.json

@@ -1,6 +1,6 @@
 {
   "name": "stock-sdk",
-  "version": "1.4.0",
+  "version": "1.4.1",
   "type": "module",
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",

src/core/constants.ts

@@ -11,6 +11,9 @@ export const A_SHARE_LIST_URL = 'https://assets.linkdiary.cn/shares/zh_a_list.js
 export const US_LIST_URL = 'https://assets.linkdiary.cn/shares/us_list.json';
 export const HK_LIST_URL = 'https://assets.linkdiary.cn/shares/hk_list.json';
 
+// A 股交易日历
+export const TRADE_CALENDAR_URL = 'https://assets.linkdiary.cn/shares/trade-data-list.txt';
+
 /** @deprecated 使用 A_SHARE_LIST_URL 代替 */
 export const CODE_LIST_URL = A_SHARE_LIST_URL;
 

src/index.test.ts

@@ -1902,3 +1902,30 @@ describe('概念板块', () => {
     });
   });
 });
+
+// ==================== 扩展数据 ====================
+
+describe('扩展数据', () => {
+  describe('getTradingCalendar', () => {
+    it('should return A股交易日历', async () => {
+      const res = await sdk.getTradingCalendar();
+      expect(Array.isArray(res)).toBe(true);
+      expect(res.length).toBeGreaterThan(0);
+      // 验证日期格式 YYYY-MM-DD
+      expect(res[0]).toMatch(/^\d{4}-\d{2}-\d{2}$/);
+      // 验证包含已知的历史交易日
+      expect(res).toContain('1990-12-19'); // 上证所开业日
+      expect(res).toContain('2024-01-02'); // 2024年第一个交易日
+    });
+
+    it('should return dates in chronological order', async () => {
+      const res = await sdk.getTradingCalendar();
+      // 检查前几个日期是按时间顺序排列的
+      for (let i = 1; i < Math.min(10, res.length); i++) {
+        expect(new Date(res[i]).getTime()).toBeGreaterThan(
+          new Date(res[i - 1]).getTime()
+        );
+      }
+    });
+  });
+});

src/providers/tencent/index.ts

@@ -31,6 +31,9 @@ export {
   type GetAllAShareQuotesOptions,
 } from './batch';
 
+// 交易日历
+export { getTradingCalendar } from './tradeCalendar';
+
 // 解析器（供内部使用）
 export * from './parsers';
 

src/providers/tencent/tradeCalendar.ts

@@ -0,0 +1,26 @@
+/**
+ * A 股交易日历
+ */
+import type { RequestClient } from '../../core';
+import { TRADE_CALENDAR_URL } from '../../core/constants';
+
+/**
+ * 获取 A 股交易日历
+ * @param client 请求客户端
+ * @returns 交易日期字符串数组，格式如 ['1990-12-19', '1990-12-20', ...]
+ */
+export async function getTradingCalendar(client: RequestClient): Promise<string[]> {
+  const text = await client.get<string>(TRADE_CALENDAR_URL);
+  
+  if (!text || text.trim() === '') {
+    return [];
+  }
+  
+  // 按逗号分割并过滤空字符串
+  return text
+    .trim()
+    .split(',')
+    .map((date) => date.trim())
+    .filter((date) => date.length > 0);
+}
+

src/sdk.ts

@@ -345,6 +345,16 @@ export class StockSDK {
     return this.client.getTencentQuote(params);
   }
 
+  // ==================== 扩展数据 ====================
+
+  /**
+   * 获取 A 股交易日历
+   * @returns 交易日期字符串数组，格式如 ['1990-12-19', '1990-12-20', ...]
+   */
+  getTradingCalendar(): Promise<string[]> {
+    return tencent.getTradingCalendar(this.client);
+  }
+
   // ==================== 技术指标 ====================
 
   /**

website/.vitepress/theme/components/Playground.vue

@@ -406,6 +406,16 @@ console.log(flows[0].mainNetRatio);  // 主力净流入占比`
 console.log(orders[0].buyLargeRatio);   // 买盘大单占比
 console.log(orders[0].sellLargeRatio);  // 卖盘大单占比`
   },
+  getTradingCalendar: {
+    name: 'getTradingCalendar',
+    desc: '获取 A 股交易日历',
+    category: 'extended',
+    params: [],
+    code: `const calendar = await sdk.getTradingCalendar();
+console.log(calendar.length);        // 交易日总数
+console.log(calendar[0]);            // '1990-12-19' (第一个交易日)
+console.log(calendar.slice(-5));     // 最近 5 个交易日`
+  },
 }
 
 // 按分类分组方法
@@ -515,6 +525,10 @@ async function fetchData() {
         data = await sdk.value.getPanelLargeOrder(codes)
         break
       }
+      case 'getTradingCalendar': {
+        data = await sdk.value.getTradingCalendar()
+        break
+      }
       case 'getHKQuotes': {
         const codes = params.codes.split(',').map(c => c.trim()).filter(Boolean)
         data = await sdk.value.getHKQuotes(codes)

website/api/fund-flow.md

@@ -1,4 +1,61 @@
-# 资金流向
+# 扩展数据
+
+## getTradingCalendar
+
+获取 A 股交易日历，返回从 1990 年至未来的所有交易日列表。
+
+### 签名
+
+```typescript
+getTradingCalendar(): Promise<string[]>
+```
+
+### 返回类型
+
+```typescript
+string[]  // 交易日期数组，格式如 ['1990-12-19', '1990-12-20', ...]
+```
+
+### 示例
+
+```typescript
+const calendar = await sdk.getTradingCalendar();
+
+console.log(`共有 ${calendar.length} 个交易日`);
+console.log(`第一个交易日: ${calendar[0]}`);  // 1990-12-19
+console.log(`最后一个交易日: ${calendar[calendar.length - 1]}`);
+
+// 判断某天是否为交易日
+function isTradingDay(date: string): boolean {
+  return calendar.includes(date);
+}
+
+console.log(isTradingDay('2024-01-02'));  // true
+console.log(isTradingDay('2024-01-01'));  // false (元旦)
+```
+
+### 应用场景
+
+```typescript
+// 获取最近 N 个交易日
+function getRecentTradingDays(n: number): string[] {
+  const today = new Date().toISOString().slice(0, 10);
+  const idx = calendar.findIndex(d => d >= today);
+  return calendar.slice(Math.max(0, idx - n), idx);
+}
+
+// 计算两个日期之间的交易日数量
+function countTradingDays(start: string, end: string): number {
+  return calendar.filter(d => d >= start && d <= end).length;
+}
+
+// 获取下一个交易日
+function getNextTradingDay(date: string): string | undefined {
+  return calendar.find(d => d > date);
+}
+```
+
+---
 
 ## getFundFlow