@hedgehog-finance/hedgehog-plugin 1.0.21 → 1.0.23

package/src/features/watchlist/tools.ts

@@ -1,7 +1,7 @@
 import { randomUUID } from "node:crypto";
 // @ts-ignore
 import { DatabaseSync } from "node:sqlite";
-import { z } from "openclaw/plugin-sdk/zod";
+import { z } from "zod";
 import { PluginRuntime } from "openclaw/plugin-sdk";
 import { getDB } from "../../core/database.js";
 import { logger } from "../../core/logger.js";
@@ -18,11 +18,30 @@ import {
 	SyncCategoriesParams,
 	SyncCategoriesParamsSchema,
 	BatchUpdateSortOrdersParams,
-	BatchUpdateSortOrdersParamsSchema
+	BatchUpdateSortOrdersParamsSchema,
+	GetIndustryListParamsSchema,
+	GetIndustryListParams
 } from "./schema.js";
 
 let watchlistMutationQueue: Promise<void> = Promise.resolve();
 
+const GetWatchlistAgentToolSchema = {
+	type: "object",
+	additionalProperties: false,
+	properties: {
+		categoryId: { type: "string", description: "分类 ID，不传返回所有" },
+		categoryType: { type: "string", enum: ["industry", "theme"], description: "分类类型" }
+	}
+};
+
+const GetIndustryListAgentToolSchema = {
+	type: "object",
+	additionalProperties: false,
+	properties: {
+		type: { type: "string", enum: ["industry", "theme", ""], description: "分类类型：industry 行业，theme 主题，空字符串或不传表示全部" }
+	}
+};
+
 function enqueueWatchlistMutation<T>(task: () => Promise<T>): Promise<T> {
 	const previous = watchlistMutationQueue;
 	let release!: () => void;
@@ -47,12 +66,12 @@ function normalizeTags(input: { name: string; weight?: number } | { name: string
 function normalizeWatchlistStock(stock: AddToWatchlistParams): AddToWatchlistParams {
 	return {
 		...stock,
-		stockCode: watchlistLogic._normalizeStockCodeForCache(stock.stockCode, stock.exchange)
+		stock_code: watchlistLogic._normalizeStockCodeForCache(stock.stock_code, stock.exchange)
 	};
 }
 
 function watchlistStockKey(stock: AddToWatchlistParams): string {
-	return `${watchlistLogic._normalizeStockCodeForCache(stock.stockCode, stock.exchange)}:${stock.exchange}`;
+	return `${watchlistLogic._normalizeStockCodeForCache(stock.stock_code, stock.exchange)}:${stock.exchange}`;
 }
 
 function updateWatchlistTags(
@@ -96,25 +115,91 @@ function upsertStockClassificationCache(
 		theme?: { name: string; weight?: number }[];
 	}
 ) {
-	const cacheCode = watchlistLogic._normalizeStockCodeForCache(stock.stockCode, stock.exchange);
-	const legacyCode = stock.stockCode.trim().toUpperCase().replace(/\.(SH|SS|SZ|HK|US)$/i, "");
+	const cacheCode = watchlistLogic._normalizeStockCodeForCache(stock.stock_code, stock.exchange);
+	const legacyCode = stock.stock_code.trim().toUpperCase().replace(/\.(SH|SS|SZ|HK|US)$/i, "");
 	db.prepare(`
-		INSERT OR REPLACE INTO global_stock_metadata (stockCode, exchange, stockName, industryJson, themeJson)
+		INSERT OR REPLACE INTO stock_classification_cache (stock_code, exchange, stock_name, industry_classification, theme_classification)
 		VALUES (?, ?, ?, ?, ?)
 	`).run(
 		cacheCode,
 		stock.exchange,
-		stock.stockName,
+		stock.stock_name,
 		JSON.stringify(classification.industry),
 		JSON.stringify(classification.theme || [])
 	);
 	if (legacyCode && legacyCode !== cacheCode) {
 		db.prepare(`
-			DELETE FROM global_stock_metadata WHERE stockCode = ? AND exchange = ?
+			DELETE FROM stock_classification_cache WHERE stock_code = ? AND exchange = ?
 		`).run(legacyCode, stock.exchange);
 	}
 }
 
+function readWatchlistRows(db: DatabaseSync, args: GetWatchlistParams, userId?: string): WatchlistRow[] {
+	let query: string;
+	const params: any[] = [];
+
+	if (userId) {
+		params.push(userId);
+		if (args.categoryId && args.categoryType) {
+			const table = args.categoryType === "industry" ? "watchlist_industry_items" : "watchlist_theme_items";
+			query = `
+				SELECT w.*, ci.weight as relWeight
+				FROM watchlist w
+				JOIN ${table} ci ON w.id = ci.watchlistId
+				WHERE w.userId = ? AND w.isDeleted = 0 AND ci.categoryId = ?
+				ORDER BY w.sortOrder ASC
+			`;
+			params.push(args.categoryId);
+		} else {
+			query = `
+				SELECT w.*
+				FROM watchlist w
+				WHERE w.userId = ? AND w.isDeleted = 0
+				ORDER BY w.sortOrder ASC
+			`;
+		}
+	} else if (args.categoryId && args.categoryType) {
+		const table = args.categoryType === "industry" ? "watchlist_industry_items" : "watchlist_theme_items";
+		query = `
+			SELECT w.*, ci.weight as relWeight
+			FROM watchlist w
+			JOIN ${table} ci ON w.id = ci.watchlistId
+			WHERE w.isDeleted = 0 AND ci.categoryId = ?
+			ORDER BY w.userId ASC, w.sortOrder ASC
+		`;
+		params.push(args.categoryId);
+	} else {
+		query = `
+			SELECT w.*
+			FROM watchlist w
+			WHERE w.isDeleted = 0
+			ORDER BY w.userId ASC, w.sortOrder ASC
+		`;
+	}
+
+	return db.prepare(query).all(...params) as WatchlistRow[];
+}
+
+function attachWatchlistTags(db: DatabaseSync, stocks: WatchlistRow[]) {
+	return stocks.map(stock => {
+		const industries = db.prepare(`
+			SELECT c.name FROM industry_theme_categories c
+			JOIN watchlist_industry_items i ON c.id = i.categoryId
+			WHERE i.watchlistId = ? ORDER BY i.weight DESC
+		`).all(stock.id) as { name: string }[];
+		const themes = db.prepare(`
+			SELECT c.name FROM industry_theme_categories c
+			JOIN watchlist_theme_items t ON c.id = t.categoryId
+			WHERE t.watchlistId = ? ORDER BY t.weight DESC
+		`).all(stock.id) as { name: string }[];
+		return {
+			...stock,
+			industries: industries.map(i => i.name),
+			themes: themes.map(t => t.name)
+		};
+	});
+}
+
 export const watchlistTools = {
 	add_to_watchlist: {
 		name: "add_to_watchlist",
@@ -129,8 +214,8 @@ export const watchlistTools = {
 				}
 				const db = getDB();
 				const stock = normalizeWatchlistStock(args);
-				const existingBeforeClassify = db.prepare("SELECT id, isDeleted FROM watchlist WHERE userId = ? AND stockCode = ? AND exchange = ?")
-					.get(uId, stock.stockCode, stock.exchange) as WatchlistRow | undefined;
+				const existingBeforeClassify = db.prepare("SELECT id, isDeleted FROM watchlist WHERE userId = ? AND stock_code = ? AND exchange = ?")
+					.get(uId, stock.stock_code, stock.exchange) as WatchlistRow | undefined;
 				if (existingBeforeClassify?.isDeleted === 0) {
 					return JSON.stringify({ success: true, skipped: true, reason: "duplicate", id: existingBeforeClassify.id });
 				}
@@ -142,7 +227,7 @@ export const watchlistTools = {
 
 				let classification: Awaited<ReturnType<typeof watchlistLogic.getStockClassification>>;
 				try {
-					classification = await watchlistLogic.getStockClassification(ctx.runtime, stock.stockName, stock.stockCode, stock.exchange, uId);
+					classification = await watchlistLogic.getStockClassification(ctx.runtime, stock.stock_name, stock.stock_code, stock.exchange, uId);
 				} catch (e: any) {
 					return JSON.stringify({ success: false, error: e.message });
 				}
@@ -157,14 +242,14 @@ export const watchlistTools = {
 					const nextOrder = (sortRow?.max ?? 0) + 1024;
 
 					let watchlistId: string;
-					const existingItem = db.prepare("SELECT id, isDeleted FROM watchlist WHERE userId = ? AND stockCode = ? AND exchange = ?").get(uId, stock.stockCode, stock.exchange) as WatchlistRow | undefined;
+					const existingItem = db.prepare("SELECT id, isDeleted FROM watchlist WHERE userId = ? AND stock_code = ? AND exchange = ?").get(uId, stock.stock_code, stock.exchange) as WatchlistRow | undefined;
 
 					if (existingItem) {
 						watchlistId = existingItem.id;
 						if (existingItem.isDeleted === 1) {
 							db.prepare(`
-								UPDATE watchlist SET isDeleted = 0, stockName = ?, sortOrder = ?, updatedAt = STRFTIME('%Y-%m-%dT%H:%M:%fZ', 'NOW') WHERE id = ?
-							`).run(stock.stockName, nextOrder, watchlistId);
+								UPDATE watchlist SET isDeleted = 0, stock_name = ?, sortOrder = ?, updatedAt = STRFTIME('%Y-%m-%dT%H:%M:%fZ', 'NOW') WHERE id = ?
+							`).run(stock.stock_name, nextOrder, watchlistId);
 						} else {
 							db.exec("ROLLBACK");
 							return JSON.stringify({ success: true, skipped: true, reason: "duplicate", id: watchlistId });
@@ -172,9 +257,9 @@ export const watchlistTools = {
 					} else {
 						watchlistId = randomUUID();
 						db.prepare(`
-							INSERT INTO watchlist (id, userId, stockCode, stockName, exchange, market, sortOrder)
+							INSERT INTO watchlist (id, userId, stock_code, stock_name, exchange, market, sortOrder)
 							VALUES (?, ?, ?, ?, ?, ?, ?)
-						`).run(watchlistId, uId, stock.stockCode, stock.stockName, stock.exchange, stock.market, nextOrder);
+						`).run(watchlistId, uId, stock.stock_code, stock.stock_name, stock.exchange, stock.market, nextOrder);
 					}
 
 					upsertStockClassificationCache(db, stock, classification);
@@ -207,12 +292,12 @@ export const watchlistTools = {
 				const uId = String(ctx.userId);
 				const uniqueStocks: AddToWatchlistParams[] = [];
 				const inputSeen = new Set<string>();
-				const skipped: { stockCode: string; exchange: string; reason: "input_duplicate" | "duplicate"; id?: string }[] = [];
+				const skipped: { stock_code: string; exchange: string; reason: "input_duplicate" | "duplicate"; id?: string }[] = [];
 				for (const rawStock of args.stocks) {
 					const stock = normalizeWatchlistStock(rawStock);
 					const key = watchlistStockKey(stock);
 					if (inputSeen.has(key)) {
-						skipped.push({ stockCode: stock.stockCode, exchange: stock.exchange, reason: "input_duplicate" });
+						skipped.push({ stock_code: stock.stock_code, exchange: stock.exchange, reason: "input_duplicate" });
 						continue;
 					}
 					inputSeen.add(key);
@@ -220,10 +305,10 @@ export const watchlistTools = {
 				}
 				const stocksToAdd: AddToWatchlistParams[] = [];
 				for (const stock of uniqueStocks) {
-					const existing = db.prepare("SELECT id, isDeleted FROM watchlist WHERE userId = ? AND stockCode = ? AND exchange = ?")
-						.get(uId, stock.stockCode, stock.exchange) as WatchlistRow | undefined;
+					const existing = db.prepare("SELECT id, isDeleted FROM watchlist WHERE userId = ? AND stock_code = ? AND exchange = ?")
+						.get(uId, stock.stock_code, stock.exchange) as WatchlistRow | undefined;
 					if (existing?.isDeleted === 0) {
-						skipped.push({ stockCode: stock.stockCode, exchange: stock.exchange, reason: "duplicate", id: existing.id });
+						skipped.push({ stock_code: stock.stock_code, exchange: stock.exchange, reason: "duplicate", id: existing.id });
 					} else {
 						stocksToAdd.push(stock);
 					}
@@ -246,7 +331,7 @@ export const watchlistTools = {
 				} catch (e: any) {
 					logger.warn({
 						count: stocksToAdd.length,
-						codes: stocksToAdd.map(stock => stock.stockCode),
+						codes: stocksToAdd.map(stock => stock.stock_code),
 						error: e.message || String(e)
 					}, "[Watchlist] batch_add_to_watchlist classification failed");
 					return JSON.stringify({
@@ -269,28 +354,28 @@ export const watchlistTools = {
 						currentMaxOrder = nextOrder;
 						const classification = batchResults[i];
 						if (!classification) {
-							throw new Error(`行业/主题关系分析失败: ${stock.stockName}`);
+							throw new Error(`行业/主题关系分析失败: ${stock.stock_name}`);
 						}
 
 						let watchlistId: string;
-						const existingItem = db.prepare("SELECT id, isDeleted FROM watchlist WHERE userId = ? AND stockCode = ? AND exchange = ?").get(uId, stock.stockCode, stock.exchange) as WatchlistRow | undefined;
+						const existingItem = db.prepare("SELECT id, isDeleted FROM watchlist WHERE userId = ? AND stock_code = ? AND exchange = ?").get(uId, stock.stock_code, stock.exchange) as WatchlistRow | undefined;
 
 						if (existingItem) {
 							watchlistId = existingItem.id;
 							if (existingItem.isDeleted === 1) {
 								db.prepare(`
-									UPDATE watchlist SET isDeleted = 0, stockName = ?, sortOrder = ?, updatedAt = STRFTIME('%Y-%m-%dT%H:%M:%fZ', 'NOW') WHERE id = ?
-								`).run(stock.stockName, nextOrder, watchlistId);
+									UPDATE watchlist SET isDeleted = 0, stock_name = ?, sortOrder = ?, updatedAt = STRFTIME('%Y-%m-%dT%H:%M:%fZ', 'NOW') WHERE id = ?
+								`).run(stock.stock_name, nextOrder, watchlistId);
 							} else {
-								skipped.push({ stockCode: stock.stockCode, exchange: stock.exchange, reason: "duplicate", id: watchlistId });
+								skipped.push({ stock_code: stock.stock_code, exchange: stock.exchange, reason: "duplicate", id: watchlistId });
 								return;
 							}
 						} else {
 							watchlistId = randomUUID();
 							db.prepare(`
-								INSERT INTO watchlist (id, userId, stockCode, stockName, exchange, market, sortOrder)
+								INSERT INTO watchlist (id, userId, stock_code, stock_name, exchange, market, sortOrder)
 								VALUES (?, ?, ?, ?, ?, ?, ?)
-							`).run(watchlistId, uId, stock.stockCode, stock.stockName, stock.exchange, stock.market, nextOrder);
+							`).run(watchlistId, uId, stock.stock_code, stock.stock_name, stock.exchange, stock.market, nextOrder);
 						}
 						results.push(watchlistId);
 						writtenItems.push({ id: watchlistId, stock, classification });
@@ -327,12 +412,9 @@ export const watchlistTools = {
 				db.exec("BEGIN TRANSACTION");
 				try {
 					const info = db.prepare(`
-						UPDATE watchlist
-						SET isDeleted = 1,
-							updatedAt = STRFTIME('%Y-%m-%dT%H:%M:%fZ', 'NOW')
+						DELETE FROM watchlist
 						WHERE id = ?
 						  AND userId = ?
-						  AND isDeleted = 0
 					`).run(args.id, uId);
 
 					if (info.changes > 0) {
@@ -360,54 +442,49 @@ export const watchlistTools = {
 
 	get_watchlist: {
 		name: "get_watchlist",
-		description: "获取自选股列表",
-		parameters: GetWatchlistParamsSchema,
-		registerTool: false,
-		execute: async (args: GetWatchlistParams, ctx: { userId: string }) => {
+		label: "获取自选股",
+		description: "获取当前全部自选股列表及其行业、主题标签。",
+		parameters: GetWatchlistAgentToolSchema,
+		registerTool: true,
+		execute: async (rawArgs?: GetWatchlistParams, ctx?: { userId: string }) => {
 			try {
+				const args = GetWatchlistParamsSchema.parse(rawArgs ?? {});
 				const db = getDB();
-				const uId = String(ctx.userId);
-				let query: string;
-				let params: any[] = [uId];
-
-				if (args.categoryId && args.categoryType) {
-					const table = args.categoryType === "industry" ? "watchlist_industry_items" : "watchlist_theme_items";
-					query = `
-						SELECT w.*, ci.weight as relWeight
-						FROM watchlist w
-						JOIN ${table} ci ON w.id = ci.watchlistId
-						WHERE w.userId = ? AND w.isDeleted = 0 AND ci.categoryId = ?
-						ORDER BY w.sortOrder ASC
-					`;
-					params.push(args.categoryId);
-				} else {
-					query = `
-						SELECT w.*
-						FROM watchlist w
-						WHERE w.userId = ? AND w.isDeleted = 0
-						ORDER BY w.sortOrder ASC
-					`;
+				const stocks = readWatchlistRows(db, args, ctx?.userId ? String(ctx.userId) : undefined);
+				const fullList = attachWatchlistTags(db, stocks);
+				return JSON.stringify({ success: true, data: fullList });
+			} catch (e: any) {
+				return JSON.stringify({ success: false, error: e.message });
+			}
+		}
+	},
+
+	get_industry_list: {
+		name: "get_industry_list",
+		label: "拉取申万一级行业分类或主题板块列表",
+		description: "拉取申万一级行业分类或主题板块列表。",
+		parameters: GetIndustryListAgentToolSchema,
+		registerTool: true,
+		execute: async (rawArgs?: GetIndustryListParams) => {
+			try {
+				const args = GetIndustryListParamsSchema.parse(rawArgs ?? {});
+				const db = getDB();
+				let query = "SELECT DISTINCT name FROM industry_theme_categories";
+				const params: any[] = [];
+				const conditions: string[] = [];
+
+				if (args.type && args.type !== "") {
+					conditions.push("type = ?");
+					params.push(args.type);
 				}
 
-				const stocks = db.prepare(query).all(...params) as WatchlistRow[];
-				const fullList = stocks.map(stock => {
-					const industries = db.prepare(`
-						SELECT c.name FROM watchlist_categories c
-						JOIN watchlist_industry_items i ON c.id = i.categoryId
-						WHERE i.watchlistId = ? ORDER BY i.weight DESC
-					`).all(stock.id) as { name: string }[];
-					const themes = db.prepare(`
-						SELECT c.name FROM watchlist_categories c
-						JOIN watchlist_theme_items t ON c.id = t.categoryId
-						WHERE t.watchlistId = ? ORDER BY t.weight DESC
-					`).all(stock.id) as { name: string }[];
-					return {
-						...stock,
-						industries: industries.map(i => i.name),
-						themes: themes.map(t => t.name)
-					};
-				});
-				return JSON.stringify({ success: true, data: fullList });
+				if (conditions.length > 0) {
+					query += ` WHERE ${conditions.join(" AND ")}`;
+				}
+				query += " ORDER BY name ASC";
+				const rows = db.prepare(query).all(...params) as { name: string }[];
+				const data = rows.map(r => r.name);
+				return JSON.stringify({ success: true, data });
 			} catch (e: any) {
 				return JSON.stringify({ success: false, error: e.message });
 			}
@@ -424,15 +501,15 @@ export const watchlistTools = {
 				const db = getDB();
 				const uId = String(ctx.userId);
 				const industryData = db.prepare(`
-					SELECT c.name as category_name, i.weight, w.stockCode
-					FROM watchlist_categories c
+					SELECT c.name as category_name, i.weight, w.stock_code
+					FROM industry_theme_categories c
 					JOIN watchlist_industry_items i ON c.id = i.categoryId
 					JOIN watchlist w ON i.watchlistId = w.id
 					WHERE i.userId = ? AND w.isDeleted = 0
 				`).all(uId) as any[];
 				const themeData = db.prepare(`
-					SELECT c.name as category_name, t.weight, w.stockCode
-					FROM watchlist_categories c
+					SELECT c.name as category_name, t.weight, w.stock_code
+					FROM industry_theme_categories c
 					JOIN watchlist_theme_items t ON c.id = t.categoryId
 					JOIN watchlist w ON t.watchlistId = w.id
 					WHERE t.userId = ? AND w.isDeleted = 0
@@ -443,8 +520,8 @@ export const watchlistTools = {
 					const name = item.category_name;
 					const existing: { category_name: string, weight_total: number, stocks: string[] } = aggMap.get(name) || { category_name: name, weight_total: 0, stocks: [] };
 					existing.weight_total += (item.weight || 0);
-					if (!existing.stocks.includes(item.stockCode)) {
-						existing.stocks.push(item.stockCode);
+					if (!existing.stocks.includes(item.stock_code)) {
+						existing.stocks.push(item.stock_code);
 					}
 					aggMap.set(name, existing);
 				});
@@ -467,14 +544,14 @@ export const watchlistTools = {
 				const uId = String(ctx.userId);
 				const industries = db.prepare(`
 					SELECT DISTINCT c.id, c.name, c.type, c.sortOrder
-					FROM watchlist_categories c
+					FROM industry_theme_categories c
 					JOIN watchlist_industry_items i ON c.id = i.categoryId
 					WHERE i.userId = ?
 					ORDER BY c.sortOrder ASC, c.name ASC
 				`).all(uId) as any[];
 				const themes = db.prepare(`
 					SELECT DISTINCT c.id, c.name, c.type, c.sortOrder
-					FROM watchlist_categories c
+					FROM industry_theme_categories c
 					JOIN watchlist_theme_items t ON c.id = t.categoryId
 					WHERE t.userId = ?
 					ORDER BY c.sortOrder ASC, c.name ASC
@@ -501,14 +578,14 @@ export const watchlistTools = {
 			const db = getDB();
 			const uId = String(ctx.userId);
 			try {
-				const stocks = db.prepare("SELECT stockCode as code, stockName as name FROM watchlist WHERE userId = ? AND isDeleted = 0").all(uId) as any[];
+				const stocks = db.prepare("SELECT stock_code as code, stock_name as name FROM watchlist WHERE userId = ? AND isDeleted = 0").all(uId) as any[];
 				if (stocks.length === 0) return JSON.stringify({ success: true, message: "没有可排序的股票" });
 				const sortedResults = await watchlistLogic.applySmartSort(ctx.runtime, `sort-${uId}`, stocks);
 				if (sortedResults.length > 0) {
 					db.exec("BEGIN TRANSACTION");
 					const stmt = db.prepare(`
 						UPDATE watchlist SET sortOrder = ?, updatedAt = STRFTIME('%Y-%m-%dT%H:%M:%fZ', 'NOW')
-						WHERE userId = ? AND stockCode = ? AND isDeleted = 0
+						WHERE userId = ? AND stock_code = ? AND isDeleted = 0
 					`);
 					sortedResults.forEach((item: any, i: number) => {
 						const currentOrder = i * 10;
@@ -535,30 +612,38 @@ export const watchlistTools = {
 			const uId = String(ctx.userId);
 			db.exec("BEGIN TRANSACTION");
 			try {
+				const existing = db.prepare("SELECT name, type FROM industry_theme_categories WHERE userId = ?").all(uId) as { name: string, type: string }[];
+				const existingSet = new Set(existing.map(row => `${row.name}:${row.type}`));
+
+				const stmt = db.prepare(`
+					INSERT INTO industry_theme_categories (id, remoteId, userId, name, type, weight, sortOrder)
+					VALUES (?, ?, ?, ?, ?, 0, 0)
+					ON CONFLICT(userId, remoteId) DO UPDATE SET name = excluded.name, updatedAt = STRFTIME('%Y-%m-%dT%H:%M:%fZ', 'NOW')
+					ON CONFLICT(userId, name, type) DO UPDATE SET remoteId = excluded.remoteId, updatedAt = STRFTIME('%Y-%m-%dT%H:%M:%fZ', 'NOW')
+				`);
+
 				if (args.industries) {
 					for (const name of args.industries) {
-						db.prepare(`
-							INSERT INTO watchlist_categories (id, remoteId, userId, name, type, weight, sortOrder)
-							VALUES (?, ?, ?, ?, 'industry', 0, 0)
-							ON CONFLICT(userId, remoteId) DO UPDATE SET name = excluded.name, updatedAt = STRFTIME('%Y-%m-%dT%H:%M:%fZ', 'NOW')
-							ON CONFLICT(userId, name, type) DO UPDATE SET remoteId = excluded.remoteId, updatedAt = STRFTIME('%Y-%m-%dT%H:%M:%fZ', 'NOW')
-						`).run(randomUUID(), name, uId, name);
+						const key = `${name}:industry`;
+						if (!existingSet.has(key)) {
+							stmt.run(randomUUID(), name, uId, name, 'industry');
+							existingSet.add(key);
+						}
 					}
 				}
 				if (args.themes) {
 					for (const name of args.themes) {
-						db.prepare(`
-							INSERT INTO watchlist_categories (id, remoteId, userId, name, type, weight, sortOrder)
-							VALUES (?, ?, ?, ?, 'theme', 0, 0)
-							ON CONFLICT(userId, remoteId) DO UPDATE SET name = excluded.name, updatedAt = STRFTIME('%Y-%m-%dT%H:%M:%fZ', 'NOW')
-							ON CONFLICT(userId, name, type) DO UPDATE SET remoteId = excluded.remoteId, updatedAt = STRFTIME('%Y-%m-%dT%H:%M:%fZ', 'NOW')
-						`).run(randomUUID(), name, uId, name);
+						const key = `${name}:theme`;
+						if (!existingSet.has(key)) {
+							stmt.run(randomUUID(), name, uId, name, 'theme');
+							existingSet.add(key);
+						}
 					}
 				}
 				db.exec("COMMIT");
 				return JSON.stringify({ success: true });
 			} catch (e: any) {
-				db.exec("ROLLBACK");
+				if (db.inTransaction) db.exec("ROLLBACK");
 				return JSON.stringify({ success: false, error: e.message });
 			}
 		}
@@ -598,7 +683,7 @@ export const watchlistTools = {
 			const db = getDB();
 			const uId = String(ctx.userId);
 			try {
-				const stocks = db.prepare("SELECT stockName, stockCode, exchange, market FROM watchlist WHERE userId = ? AND isDeleted = 0").all(uId) as any[];
+				const stocks = db.prepare("SELECT stock_name, stock_code, exchange, market FROM watchlist WHERE userId = ? AND isDeleted = 0").all(uId) as any[];
 				if (stocks.length > 30) {
 					return JSON.stringify({ success: false, error: "自选股数量超过 30 只，暂不支持批量重置分类" });
 				}
@@ -615,15 +700,15 @@ export const watchlistTools = {
 						db.prepare("DELETE FROM watchlist_industry_items WHERE userId = ?").run(uId);
 						db.prepare("DELETE FROM watchlist_theme_items WHERE userId = ?").run(uId);
 
-						const userStocks = db.prepare("SELECT id, stockCode, exchange FROM watchlist WHERE userId = ? AND isDeleted = 0").all(uId) as any[];
+						const userStocks = db.prepare("SELECT id, stock_code, exchange FROM watchlist WHERE userId = ? AND isDeleted = 0").all(uId) as any[];
 						batchResults.forEach((res, i) => {
 							if (res) {
 								const s = stocks[i];
-								const match = userStocks.find(us => us.stockCode === s.stockCode && us.exchange === s.exchange);
+								const match = userStocks.find(us => us.stock_code === s.stock_code && us.exchange === s.exchange);
 								if (match) {
 									upsertStockClassificationCache(db, {
-										stockName: s.stockName,
-										stockCode: s.stockCode,
+										stock_name: s.stock_name,
+										stock_code: s.stock_code,
 										exchange: s.exchange,
 										market: s.market
 									}, res);

package/src/openclawConfig.ts

@@ -0,0 +1,101 @@
+import type { OpenClawConfig } from "openclaw/plugin-sdk/core";
+import {
+	HEDGEHOG_AGENT_ID,
+	MAIN_AGENT_EXTRA_TOOL_NAMES,
+	MAIN_AGENT_ID,
+	listRegisteredAgentToolNames
+} from "./openclawConstants.js";
+
+type AgentEntry = {
+	id: string;
+	tools?: {
+		allow?: string[];
+		alsoAllow?: string[];
+		[key: string]: unknown;
+	};
+	[key: string]: unknown;
+};
+
+export type HedgehogAgentToolAllowMigration = {
+	config: OpenClawConfig;
+	changes: string[];
+};
+
+function uniqueStrings(values: readonly string[]): string[] {
+	return Array.from(new Set(values));
+}
+
+function withAllowedTools(agent: AgentEntry, toolNames: string[]): { agent: AgentEntry; changed: boolean; added: string[] } {
+	const tools = agent.tools || {};
+	const existing = Array.isArray(tools.alsoAllow) ? tools.alsoAllow : [];
+	const existingSet = new Set(existing);
+	const added = toolNames.filter((name) => !existingSet.has(name));
+	const nextAlsoAllow = uniqueStrings([...existing, ...toolNames]);
+	return {
+		agent: {
+			...agent,
+			tools: {
+				...tools,
+				alsoAllow: nextAlsoAllow
+			}
+		},
+		changed: added.length > 0,
+		added
+	};
+}
+
+function withRegisteredTools(agent: AgentEntry): { agent: AgentEntry; changed: boolean; added: string[] } {
+	return withAllowedTools(agent, listRegisteredAgentToolNames());
+}
+
+function upsertAgentTools(list: AgentEntry[], agentId: string, toolNames: string[]) {
+	const existingIndex = list.findIndex((agent) => agent?.id === agentId);
+	if (existingIndex >= 0) {
+		const result = withAllowedTools(list[existingIndex], toolNames);
+		if (!result.changed) return { changed: false, added: [] };
+		list[existingIndex] = result.agent;
+		return { changed: true, added: result.added };
+	}
+
+	const result = withAllowedTools({ id: agentId }, toolNames);
+	list.push(result.agent);
+	return { changed: result.changed, added: result.added };
+}
+
+export function ensureRegisteredToolsAllowedInConfig(config: OpenClawConfig): HedgehogAgentToolAllowMigration | null {
+	const agents = config.agents || {};
+	const list = Array.isArray(agents.list) ? agents.list as AgentEntry[] : [];
+	const existingIndex = list.findIndex((agent) => agent?.id === HEDGEHOG_AGENT_ID);
+	const nextList = [...list];
+	const changes: string[] = [];
+
+	if (existingIndex >= 0) {
+		const result = withRegisteredTools(nextList[existingIndex]);
+		if (result.changed) {
+			nextList[existingIndex] = result.agent;
+			changes.push(`Added ${result.added.join(", ")} to agents.list[id=${HEDGEHOG_AGENT_ID}].tools.alsoAllow.`);
+		}
+	} else {
+		const result = withRegisteredTools({ id: HEDGEHOG_AGENT_ID });
+		nextList.push(result.agent);
+		changes.push(`Added ${result.added.join(", ")} to agents.list[id=${HEDGEHOG_AGENT_ID}].tools.alsoAllow.`);
+	}
+
+	const mainResult = upsertAgentTools(nextList, MAIN_AGENT_ID, MAIN_AGENT_EXTRA_TOOL_NAMES);
+	if (mainResult.changed) {
+		changes.push(`Added ${mainResult.added.join(", ")} to agents.list[id=${MAIN_AGENT_ID}].tools.alsoAllow.`);
+	}
+
+	if (changes.length === 0) return null;
+
+	return {
+		config: {
+			...config,
+			agents: {
+				...agents,
+				list: nextList
+			}
+		},
+		changes
+	};
+}

package/src/openclawConstants.ts

@@ -0,0 +1,11 @@
+import { allFeaturesTools } from "./features/index.js";
+
+export const HEDGEHOG_AGENT_ID = "hedgehog-finance";
+export const MAIN_AGENT_ID = "main";
+export const MAIN_AGENT_EXTRA_TOOL_NAMES = ["update_hedgehog_skill_versions"];
+
+export function listRegisteredAgentToolNames(): string[] {
+	return Object.entries(allFeaturesTools)
+		.filter(([, tool]) => tool.registerTool !== false && tool.agentToolTarget !== "main")
+		.map(([name]) => name);
+}