fi-pool-server 0.1.0

package/README.md

@@ -0,0 +1,26 @@
+# @fi-pool/server
+
+Fi-Pool-Manager 核心服务层。
+
+## 职责
+
+- **数据获取**：通过腾讯财经接口获取 A 股日行情数据
+- **本地分析**：计算技术指标（MA、MACD、RSI、KDJ、布林带）
+- **LLM 分析**：调用本地 LM Studio 进行多角色分析和报告生成
+- **向量检索**：通过 sqlite-vec 进行语义搜索
+- **舆情搜索**：获取股票相关市场信息
+- **数据访问**：通过 Drizzle ORM 操作 SQLite 数据库
+
+## 目录结构
+
+```
+src/
+├── data/         # 数据获取模块
+├── analysis/     # 本地分析引擎
+├── llm/          # LLM 调用封装
+├── vector/       # 向量检索
+├── search/       # 舆情搜索
+├── db/           # 数据库访问层
+├── session/      # Session 管理
+└── index.ts      # 入口
+```

package/dist/db/index.d.ts

@@ -0,0 +1,18 @@
+/**
+ * 数据库连接与初始化
+ *
+ * 职责：
+ * - 读取 .env 配置的 DB_PATH
+ * - 创建 better-sqlite3 数据库连接
+ * - 初始化时自动创建数据目录
+ * - 暴露 drizzle 实例供全局使用
+ */
+import Database from 'better-sqlite3';
+import { drizzle } from 'drizzle-orm/better-sqlite3';
+import * as schema from './schema.js';
+export declare function getDbPath(): string;
+export declare function initDatabase(): ReturnType<typeof drizzle<typeof schema>>;
+export declare function getDatabase(): ReturnType<typeof drizzle<typeof schema>>;
+export declare function getSqlite(): Database.Database;
+export declare function closeDatabase(): void;
+//# sourceMappingURL=index.d.ts.map

package/dist/db/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/db/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AACtC,OAAO,EAAE,OAAO,EAAE,MAAM,4BAA4B,CAAC;AACrD,OAAO,KAAK,MAAM,MAAM,aAAa,CAAC;AAOtC,wBAAgB,SAAS,IAAI,MAAM,CAElC;AAED,wBAAgB,YAAY,IAAI,UAAU,CAAC,OAAO,OAAO,CAAC,OAAO,MAAM,CAAC,CAAC,CAoBxE;AAED,wBAAgB,WAAW,IAAI,UAAU,CAAC,OAAO,OAAO,CAAC,OAAO,MAAM,CAAC,CAAC,CAKvE;AAED,wBAAgB,SAAS,IAAI,QAAQ,CAAC,QAAQ,CAK7C;AAED,wBAAgB,aAAa,IAAI,IAAI,CAMpC"}

package/dist/db/index.js

@@ -0,0 +1,58 @@
+/**
+ * 数据库连接与初始化
+ *
+ * 职责：
+ * - 读取 .env 配置的 DB_PATH
+ * - 创建 better-sqlite3 数据库连接
+ * - 初始化时自动创建数据目录
+ * - 暴露 drizzle 实例供全局使用
+ */
+import Database from 'better-sqlite3';
+import { drizzle } from 'drizzle-orm/better-sqlite3';
+import * as schema from './schema.js';
+import { existsSync, mkdirSync } from 'fs';
+import { dirname, resolve } from 'path';
+let dbInstance = null;
+let sqliteInstance = null;
+export function getDbPath() {
+    return process.env.DB_PATH || './data/fi-pool.db';
+}
+export function initDatabase() {
+    if (dbInstance)
+        return dbInstance;
+    const dbPath = resolve(getDbPath());
+    const dbDir = dirname(dbPath);
+    if (!existsSync(dbDir)) {
+        try {
+            mkdirSync(dbDir, { recursive: true });
+        }
+        catch (err) {
+            throw new Error(`数据库目录创建失败: ${dbDir} — ${err instanceof Error ? err.message : String(err)}`);
+        }
+    }
+    sqliteInstance = new Database(dbPath);
+    sqliteInstance.pragma('journal_mode = WAL');
+    sqliteInstance.pragma('foreign_keys = ON');
+    dbInstance = drizzle(sqliteInstance, { schema });
+    return dbInstance;
+}
+export function getDatabase() {
+    if (!dbInstance) {
+        throw new Error('数据库未初始化，请先调用 initDatabase()');
+    }
+    return dbInstance;
+}
+export function getSqlite() {
+    if (!sqliteInstance) {
+        throw new Error('数据库未初始化，请先调用 initDatabase()');
+    }
+    return sqliteInstance;
+}
+export function closeDatabase() {
+    if (sqliteInstance) {
+        sqliteInstance.close();
+        sqliteInstance = null;
+        dbInstance = null;
+    }
+}
+//# sourceMappingURL=index.js.map

package/dist/db/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/db/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,QAAQ,MAAM,gBAAgB,CAAC;AACtC,OAAO,EAAE,OAAO,EAAE,MAAM,4BAA4B,CAAC;AACrD,OAAO,KAAK,MAAM,MAAM,aAAa,CAAC;AACtC,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAC3C,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AAExC,IAAI,UAAU,GAAqD,IAAI,CAAC;AACxE,IAAI,cAAc,GAA6B,IAAI,CAAC;AAEpD,MAAM,UAAU,SAAS;IACvB,OAAO,OAAO,CAAC,GAAG,CAAC,OAAO,IAAI,mBAAmB,CAAC;AACpD,CAAC;AAED,MAAM,UAAU,YAAY;IAC1B,IAAI,UAAU;QAAE,OAAO,UAAU,CAAC;IAElC,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;IACpC,MAAM,KAAK,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IAE9B,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;QACvB,IAAI,CAAC;YACH,SAAS,CAAC,KAAK,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACxC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,IAAI,KAAK,CAAC,cAAc,KAAK,MAAM,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC/F,CAAC;IACH,CAAC;IAED,cAAc,GAAG,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;IACtC,cAAc,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;IAC5C,cAAc,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;IAE3C,UAAU,GAAG,OAAO,CAAC,cAAc,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;IACjD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,MAAM,UAAU,WAAW;IACzB,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;IACjD,CAAC;IACD,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,MAAM,UAAU,SAAS;IACvB,IAAI,CAAC,cAAc,EAAE,CAAC;QACpB,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;IACjD,CAAC;IACD,OAAO,cAAc,CAAC;AACxB,CAAC;AAED,MAAM,UAAU,aAAa;IAC3B,IAAI,cAAc,EAAE,CAAC;QACnB,cAAc,CAAC,KAAK,EAAE,CAAC;QACvB,cAAc,GAAG,IAAI,CAAC;QACtB,UAAU,GAAG,IAAI,CAAC;IACpB,CAAC;AACH,CAAC"}

package/dist/db/migrate.d.ts

@@ -0,0 +1,11 @@
+/**
+ * 数据库迁移运行器
+ *
+ * 启动时查找数据库文件：
+ * - 不存在 → 初始化数据库并执行迁移
+ * - 存在 → 直接连接
+ * - 版本更新 → 通过 Drizzle 迁移脚本变更表结构（不覆盖已有数据）
+ */
+import { initDatabase } from './index.js';
+export declare function ensureDatabase(): ReturnType<typeof initDatabase>;
+//# sourceMappingURL=migrate.d.ts.map

package/dist/db/migrate.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"migrate.d.ts","sourceRoot":"","sources":["../../src/db/migrate.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAGH,OAAO,EAAE,YAAY,EAA0B,MAAM,YAAY,CAAC;AAQlE,wBAAgB,cAAc,IAAI,UAAU,CAAC,OAAO,YAAY,CAAC,CA2BhE"}

package/dist/db/migrate.js

@@ -0,0 +1,42 @@
+/**
+ * 数据库迁移运行器
+ *
+ * 启动时查找数据库文件：
+ * - 不存在 → 初始化数据库并执行迁移
+ * - 存在 → 直接连接
+ * - 版本更新 → 通过 Drizzle 迁移脚本变更表结构（不覆盖已有数据）
+ */
+import { migrate } from 'drizzle-orm/better-sqlite3/migrator';
+import { initDatabase, getDbPath } from './index.js';
+import { existsSync } from 'fs';
+import { resolve, dirname } from 'path';
+import { fileURLToPath } from 'url';
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = dirname(__filename);
+export function ensureDatabase() {
+    const dbPath = resolve(getDbPath());
+    const isNew = !existsSync(dbPath);
+    const db = initDatabase();
+    if (isNew) {
+        console.log(`[db] 数据库文件不存在，初始化: ${dbPath}`);
+    }
+    else {
+        console.log(`[db] 连接已有数据库: ${dbPath}`);
+    }
+    // 执行迁移 — 迁移文件夹路径相对于本文件位置
+    // 编译后：packages/server/dist/db/migrate.js → ../../drizzle
+    // 源码时：packages/server/src/db/migrate.ts → ../../drizzle
+    const migrationsFolder = resolve(__dirname, '../../drizzle');
+    try {
+        migrate(db, { migrationsFolder });
+        console.log('[db] 迁移执行完成');
+    }
+    catch (err) {
+        const msg = err instanceof Error ? err.message : String(err);
+        console.error('[db] 迁移失败:', msg);
+        console.error('[db] 数据库处于不一致状态，终止进程');
+        process.exit(1);
+    }
+    return db;
+}
+//# sourceMappingURL=migrate.js.map

package/dist/db/migrate.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"migrate.js","sourceRoot":"","sources":["../../src/db/migrate.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,EAAE,OAAO,EAAE,MAAM,qCAAqC,CAAC;AAC9D,OAAO,EAAE,YAAY,EAAe,SAAS,EAAE,MAAM,YAAY,CAAC;AAClE,OAAO,EAAE,UAAU,EAAE,MAAM,IAAI,CAAC;AAChC,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACxC,OAAO,EAAE,aAAa,EAAE,MAAM,KAAK,CAAC;AAEpC,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAClD,MAAM,SAAS,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;AAEtC,MAAM,UAAU,cAAc;IAC5B,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC,CAAC;IACpC,MAAM,KAAK,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;IAElC,MAAM,EAAE,GAAG,YAAY,EAAE,CAAC;IAE1B,IAAI,KAAK,EAAE,CAAC;QACV,OAAO,CAAC,GAAG,CAAC,sBAAsB,MAAM,EAAE,CAAC,CAAC;IAC9C,CAAC;SAAM,CAAC;QACN,OAAO,CAAC,GAAG,CAAC,iBAAiB,MAAM,EAAE,CAAC,CAAC;IACzC,CAAC;IAED,yBAAyB;IACzB,yDAAyD;IACzD,wDAAwD;IACxD,MAAM,gBAAgB,GAAG,OAAO,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC;IAC7D,IAAI,CAAC;QACH,OAAO,CAAC,EAAE,EAAE,EAAE,gBAAgB,EAAE,CAAC,CAAC;QAClC,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IAC7B,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC7D,OAAO,CAAC,KAAK,CAAC,YAAY,EAAE,GAAG,CAAC,CAAC;QACjC,OAAO,CAAC,KAAK,CAAC,sBAAsB,CAAC,CAAC;QACtC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;IAED,OAAO,EAAE,CAAC;AACZ,CAAC"}