dsh-sql 0.4.5 → 0.4.6

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
package/CHANGELOG.md CHANGED
@@ -2,6 +2,14 @@
2
2
 
3
3
  [返回简介](README.md) · [使用说明](docs/USAGE.md) · [验证记录](docs/VALIDATION.md)
4
4
 
5
+ ## 0.4.6 (2026-10-20)
6
+
7
+ - 修复 sql_query 只读校验与 PostgreSQL 反斜杠语义不一致的问题:词法去噪改为按引擎处理反斜杠(MySQL 默认转义 / PostgreSQL 只有 `E'…'`、`U&'…'` 转义 / SQLite 与引号标识符里都不转义)。`SELECT '\'; DELETE FROM users; --'` 这类双语句载荷会被拒绝,不再能在 readOnly=true 下删表
8
+ - 修复 PRAGMA 函数式写法绕过只读校验的问题:`PRAGMA foreign_keys(0)`、`writable_schema(1)`、`user_version(4242)` 等与 `name=value` 等价,现在改为只读 PRAGMA 白名单,只放行 table_info / index_list / page_count 等内省语句
9
+ - readOnly=true 时把只读约束下沉到数据库自身:SQLite 以只读方式打开文件、PostgreSQL 加 `-c default_transaction_read_only=on`、MySQL 每个新连接执行 `SET SESSION TRANSACTION READ ONLY`。词法校验被绕过时服务器仍会拒绝写入
10
+ - sql_exec 审批文本不再截断到 200 字符:逐条列出每条语句并附语句类型统计;超过 800 字符时把完整 SQL 指回工具参数,200 字符之后的 DROP TABLE 不会再从审批卡里消失
11
+ - 新增 `test/readonly-hardening.test.mjs`(10 项回归),`tools.test.mjs` 的 PRAGMA 断言同步更新
12
+
5
13
  ## 0.4.5 (2026-10-05)
6
14
 
7
15
  - 保留 SQLite 64 位整数与三种数据库的同名列;零行结果保留列名。同名列 JSON 导出使用 columns / rows,避免后列覆盖前列。
package/README.en.md CHANGED
@@ -8,6 +8,8 @@ Connect SQLite, MySQL or PostgreSQL to query data and inspect schemas.
8
8
 
9
9
  [![npm](https://img.shields.io/npm/v/dsh-sql)](https://www.npmjs.com/package/dsh-sql) [![downloads](https://raw.githubusercontent.com/STARDUSTLC666/dsh-suite/npm-downloads/assets/dsh-sql-downloads.svg)](https://www.npmjs.com/package/dsh-sql)
10
10
 
11
+ Feedback and contributions are welcome: report [issues](https://github.com/STARDUSTLC666/dsh-sql/issues) or submit [pull requests](https://github.com/STARDUSTLC666/dsh-sql/pulls).
12
+
11
13
  ## What it does
12
14
 
13
15
  - Manage connections and inspect schemas or database summaries.
package/README.md CHANGED
@@ -8,6 +8,8 @@
8
8
 
9
9
  [![npm](https://img.shields.io/npm/v/dsh-sql)](https://www.npmjs.com/package/dsh-sql) [![downloads](https://raw.githubusercontent.com/STARDUSTLC666/dsh-suite/npm-downloads/assets/dsh-sql-downloads.svg)](https://www.npmjs.com/package/dsh-sql)
10
10
 
11
+ 欢迎使用,遇到问题或有改进建议,请提交 [issues](https://github.com/STARDUSTLC666/dsh-sql/issues) 和 [PR](https://github.com/STARDUSTLC666/dsh-sql/pulls)。
12
+
11
13
  ## 功能
12
14
 
13
15
  - 管理连接,查询结构和数据概览。
package/docs/USAGE.md CHANGED
@@ -44,7 +44,7 @@ dsh plugin --profile web remove dsh-sql
44
44
  maxRows: 1000 # 查询返回行数上限(1-10000)
45
45
  queryTimeoutMs: 60000 # 单次查询超时(默认 60 秒,5 秒 - 10 分钟)
46
46
  execTimeoutMs: 120000 # 单次写操作超时(默认 120 秒,5 秒 - 10 分钟)
47
- readOnly: false # true 时禁用 sql_exec
47
+ readOnly: false # true 时禁用 sql_exec,并让数据库自身以只读方式连接
48
48
  writeApproval: true # 写操作先弹审批(默认 true)
49
49
  ```
50
50
 
@@ -55,7 +55,7 @@ dsh plugin --profile web remove dsh-sql
55
55
  | 工具 | 作用 | 安全 |
56
56
  | :-- | :-- | :-- |
57
57
  | `sql_list` | 列出连接 + 连通性测试 | — |
58
- | `sql_query` | 只读查询(SELECT/PRAGMA/EXPLAIN/SHOW/DESCRIBE/WITH)| 关键字白名单 + 拒绝多语句 |
58
+ | `sql_query` | 只读查询(SELECT/PRAGMA/EXPLAIN/SHOW/DESCRIBE/WITH)| 按引擎去噪的关键字白名单 + 拒绝多语句 + PRAGMA 白名单 |
59
59
  | `sql_exec` | 写操作/DDL(可多语句脚本)| readOnly 禁用 + 审批门 |
60
60
  | `sql_schema` | 表清单 / 表结构 | 标识符白名单校验 |
61
61
  | `sql_stats` | 表数量、行数与库体积概览 | 表名引用 + 查询失败隔离 |
@@ -75,9 +75,10 @@ sql_exec { sql: UPDATE orders SET status = 'paid' WHERE id = 42 }
75
75
 
76
76
  ## 安全设计
77
77
 
78
- - **词法级只读保护**:sql_query 先剥离字符串/注释再校验,拒绝 data-modifying CTE(WITH…DELETE/UPDATE)、SELECT INTO、FOR UPDATE/FOR SHARE、PRAGMA 赋值与多语句
78
+ - **词法级只读保护**:sql_query 先剥离字符串/注释再校验,拒绝 data-modifying CTE(WITH…DELETE/UPDATE)、SELECT INTO、FOR UPDATE/FOR SHARE、多语句,以及不在只读白名单里的 PRAGMA。去噪按引擎处理反斜杠(MySQL 默认转义、PostgreSQL 只有 `E'…'`/`U&'…'` 转义、SQLite 与引号标识符里都不转义),避免「校验看到一条、服务器执行两条」
79
79
  - **写审批门**:sql_exec 默认弹审批(对齐 dsh-email 的发信审批),headless 环境无审批通道时拒绝执行
80
- - **readOnly 模式**:生产库可整体禁用写
80
+ - **readOnly 模式**:生产库可整体禁用写。除了禁用 sql_exec,还会把只读约束下沉到数据库自身:SQLite 以只读方式打开文件,PostgreSQL 用 `-c default_transaction_read_only=on` 启动参数,MySQL 在每个新连接上执行 `SET SESSION TRANSACTION READ ONLY`(需要 MySQL 5.6.5+ / MariaDB 10.0+)。即使词法校验被绕过,服务器也会拒绝写入
81
+ - **审批卡不藏语句**:sql_exec 的审批文本逐条列出每条语句,并给出一行语句类型统计;超过 800 字符时把完整 SQL 指回工具参数,第二条 DROP / DELETE 不会因为截断而从审批卡里消失
81
82
  - **流式行数钳制**:SQLite 迭代器 / MySQL Readable / PostgreSQL Query 行事件最多收集 maxRows+1 行,超量标记 truncated;MySQL 和 PostgreSQL 在达到上限时关闭该查询的专用连接,未达上限则正常归还连接池,避免全量结果驻留内存
82
83
  - **可取消执行**:查询与写操作遵守 Harness 的 `exec.signal`;取消时会中止等待并销毁正在工作的 MySQL/PostgreSQL 专用连接
83
84
  - **大整数无损**:数据库返回的 bigint 在 JavaScript 安全整数范围内输出 number,超出范围则输出十进制字符串,避免静默丢精度
package/lib/adapters.d.ts CHANGED
@@ -1,3 +1,5 @@
1
+ import mysql from 'mysql2/promise';
2
+ import pg from 'pg';
1
3
  import { type SqlConnectionConfig } from './config.js';
2
4
  /** 查询结果:列名 + 行(值数组,无损 JSON 友好)。 */
3
5
  export interface QueryResult {
@@ -21,5 +23,17 @@ export interface DatabaseAdapter {
21
23
  ping(signal?: AbortSignal): Promise<void>;
22
24
  close(): Promise<void>;
23
25
  }
24
- /** 按连接配置创建适配器。 */
25
- export declare function createAdapter(connection: SqlConnectionConfig): DatabaseAdapter;
26
+ /** 适配器额外选项:readOnly 时把只读约束下沉到数据库自身。 */
27
+ export interface AdapterOptions {
28
+ readOnly?: boolean;
29
+ }
30
+ /** MySQL 连接池配置(导出以便测试断言,不含连接)。 */
31
+ export declare function mysqlPoolConfig(connection: SqlConnectionConfig): mysql.PoolOptions;
32
+ /**
33
+ * PostgreSQL 连接池配置。readOnly 时用启动参数把会话默认事务设为只读:
34
+ * 即使词法校验被绕过(多语句、反斜杠、注释技巧),服务器也会以
35
+ * "cannot execute ... in a read-only transaction" 拒绝写操作。
36
+ */
37
+ export declare function postgresPoolConfig(connection: SqlConnectionConfig, readOnly?: boolean): pg.PoolConfig;
38
+ /** 按连接配置创建适配器;readOnly 会下沉到引擎自身的只读约束。 */
39
+ export declare function createAdapter(connection: SqlConnectionConfig, options?: AdapterOptions): DatabaseAdapter;
package/lib/adapters.js CHANGED
@@ -117,12 +117,56 @@ function streamPostgresQuery(client, sql, limit, discard) {
117
117
  client.query(query);
118
118
  });
119
119
  }
120
+ /** MySQL 连接池配置(导出以便测试断言,不含连接)。 */
121
+ export function mysqlPoolConfig(connection) {
122
+ return {
123
+ host: connection.host ?? 'localhost',
124
+ port: connection.port ?? 3306,
125
+ user: connection.user ?? '',
126
+ password: connection.password ?? '',
127
+ database: connection.database ?? '',
128
+ connectionLimit: 5,
129
+ enableKeepAlive: true,
130
+ supportBigNumbers: true,
131
+ bigNumberStrings: true,
132
+ // 显式关掉多语句:即使词法校验被绕过,服务器也不会把第二句当命令执行。
133
+ multipleStatements: false,
134
+ };
135
+ }
136
+ /**
137
+ * PostgreSQL 连接池配置。readOnly 时用启动参数把会话默认事务设为只读:
138
+ * 即使词法校验被绕过(多语句、反斜杠、注释技巧),服务器也会以
139
+ * "cannot execute ... in a read-only transaction" 拒绝写操作。
140
+ */
141
+ export function postgresPoolConfig(connection, readOnly = false) {
142
+ const config = {
143
+ host: connection.host ?? 'localhost',
144
+ port: connection.port ?? 5432,
145
+ user: connection.user ?? '',
146
+ password: connection.password ?? '',
147
+ database: connection.database ?? '',
148
+ max: 5,
149
+ };
150
+ if (readOnly)
151
+ config.options = '-c default_transaction_read_only=on';
152
+ return config;
153
+ }
120
154
  /** SQLite 适配器(node:sqlite,零依赖)。 */
121
155
  class SqliteAdapter {
122
156
  engine = 'sqlite';
123
157
  db;
124
- constructor(file) {
125
- this.db = new DatabaseSync(file === ':memory:' ? ':memory:' : file);
158
+ constructor(file, readOnly = false) {
159
+ const memory = file === ':memory:';
160
+ // 内存库没有落盘风险,也不支持只读打开;文件库在 readOnly 下由 SQLite
161
+ // 自己拒绝任何写入(PRAGMA 写、DDL、DML 都会报 attempt to write a readonly database)。
162
+ const options = readOnly && !memory ? { readOnly: true } : {};
163
+ try {
164
+ this.db = new DatabaseSync(memory ? ':memory:' : file, options);
165
+ }
166
+ catch (error) {
167
+ const detail = error instanceof Error ? error.message : String(error);
168
+ throw new Error('无法打开 SQLite 文件 ' + file + '(readOnly=' + readOnly + '):' + detail);
169
+ }
126
170
  this.db.exec('PRAGMA busy_timeout = 5000');
127
171
  }
128
172
  async listTables(signal) {
@@ -188,18 +232,15 @@ class SqliteAdapter {
188
232
  class MysqlAdapter {
189
233
  engine = 'mysql';
190
234
  pool;
191
- constructor(connection) {
192
- this.pool = mysql.createPool({
193
- host: connection.host ?? 'localhost',
194
- port: connection.port ?? 3306,
195
- user: connection.user ?? '',
196
- password: connection.password ?? '',
197
- database: connection.database ?? '',
198
- connectionLimit: 5,
199
- enableKeepAlive: true,
200
- supportBigNumbers: true,
201
- bigNumberStrings: true,
202
- });
235
+ constructor(connection, readOnly = false) {
236
+ this.pool = mysql.createPool(mysqlPoolConfig(connection));
237
+ if (readOnly) {
238
+ // 会话级只读:MySQL 5.6.5+ 的 SET SESSION TRANSACTION READ ONLY 会让
239
+ // 该连接上后续任何 DML/DDL 被服务器拒绝,不依赖词法校验。
240
+ this.pool.on('connection', (session) => {
241
+ void session.query('SET SESSION TRANSACTION READ ONLY').catch(() => { });
242
+ });
243
+ }
203
244
  }
204
245
  async withSignalConnection(signal, work) {
205
246
  signal?.throwIfAborted();
@@ -274,15 +315,8 @@ class MysqlAdapter {
274
315
  class PostgresAdapter {
275
316
  engine = 'postgres';
276
317
  pool;
277
- constructor(connection) {
278
- this.pool = new pg.Pool({
279
- host: connection.host ?? 'localhost',
280
- port: connection.port ?? 5432,
281
- user: connection.user ?? '',
282
- password: connection.password ?? '',
283
- database: connection.database ?? '',
284
- max: 5,
285
- });
318
+ constructor(connection, readOnly = false) {
319
+ this.pool = new pg.Pool(postgresPoolConfig(connection, readOnly));
286
320
  }
287
321
  async withSignalClient(signal, work) {
288
322
  signal?.throwIfAborted();
@@ -371,11 +405,12 @@ class PostgresAdapter {
371
405
  await this.pool.end();
372
406
  }
373
407
  }
374
- /** 按连接配置创建适配器。 */
375
- export function createAdapter(connection) {
408
+ /** 按连接配置创建适配器;readOnly 会下沉到引擎自身的只读约束。 */
409
+ export function createAdapter(connection, options = {}) {
410
+ const readOnly = options.readOnly === true;
376
411
  if (connection.engine === 'sqlite')
377
- return new SqliteAdapter(connection.file ?? ':memory:');
412
+ return new SqliteAdapter(connection.file ?? ':memory:', readOnly);
378
413
  if (connection.engine === 'mysql')
379
- return new MysqlAdapter(connection);
380
- return new PostgresAdapter(connection);
414
+ return new MysqlAdapter(connection, readOnly);
415
+ return new PostgresAdapter(connection, readOnly);
381
416
  }
package/lib/index.js CHANGED
@@ -9,7 +9,7 @@
9
9
  * @module dsh-sql
10
10
  */
11
11
  import { resolveConfig } from './config.js';
12
- import { buildSqlTools } from './tools.js';
12
+ import { buildSqlTools, summarizeWriteScript } from './tools.js';
13
13
  /** cordis 服务注入:apply 里要用 ctx.tools,必须显式声明。 */
14
14
  export const name = 'sql';
15
15
  export const inject = ['tools'];
@@ -27,7 +27,9 @@ export function apply(ctx, config) {
27
27
  const sql = typeof args.sql === 'string' ? args.sql : '';
28
28
  return {
29
29
  kind: 'ask',
30
- reason: 'sql_exec 写操作需要确认:' + sql.slice(0, 200) + (sql.length > 200 ? '…' : ''),
30
+ // 不截断整段 SQL:先逐条列出每条语句与语句类型统计,再附完整 SQL。
31
+ // 只截 200 字符会把第二条 DROP / DELETE 挡在审批卡之外。
32
+ reason: 'sql_exec 写操作需要确认:\n' + summarizeWriteScript(sql),
31
33
  };
32
34
  });
33
35
  }
package/lib/tools.d.ts CHANGED
@@ -26,8 +26,22 @@ export interface SqlToolDefinition {
26
26
  execute(args: unknown, exec: unknown): Promise<unknown>;
27
27
  timeoutMs?: number;
28
28
  }
29
- /** 校验只读查询:词法去噪后白名单开头 + 写关键字扫描 + 单语句。 */
30
- export declare function assertReadQuery(sql: string): string;
29
+ /** 词法校验需要区分的三种引擎。 */
30
+ type SqlEngine = 'sqlite' | 'mysql' | 'postgres';
31
+ /**
32
+ * 校验只读查询:词法去噪后白名单开头 + 写关键字扫描 + 单语句 + PRAGMA 白名单。
33
+ *
34
+ * `engine` 决定反斜杠转义规则(见 stripSqlNoise);不传时按最保守的方式把
35
+ * 反斜杠当字面量,宁可误拒也不放过第二条语句。
36
+ */
37
+ export declare function assertReadQuery(sql: string, engine?: SqlEngine): string;
38
+ /**
39
+ * 把一段写操作脚本压成审批卡可读的摘要。
40
+ *
41
+ * 关键点:逐条列出每条语句,并额外给一行语句类型统计。即使脚本体被省略,
42
+ * 第二条、第 N 条语句的 DROP / DELETE 也不会从审批文本里消失。
43
+ */
44
+ export declare function summarizeWriteScript(sql: string): string;
31
45
  /** 查询结果转 CSV 文本(RFC 4180 风格转义)。 */
32
46
  export declare function toCsv(columns: string[], rows: unknown[][]): string;
33
47
  /** 审批执行上下文的最小面。 */
@@ -42,3 +56,4 @@ export declare function buildSqlTools(config: ResolvedSqlConfig): {
42
56
  tools: SqlToolDefinition[];
43
57
  adapters: Map<string, DatabaseAdapter>;
44
58
  };
59
+ export {};
package/lib/tools.js CHANGED
@@ -40,8 +40,30 @@ function executionSignal(exec) {
40
40
  }
41
41
  /** 只读语句关键字白名单。 */
42
42
  const READ_KEYWORDS = /^(select|pragma|explain|show|describe|desc|with)\b/i;
43
- /** 去掉字符串、引号标识符与注释,保留真实 SQL 关键字与分号。 */
44
- function stripSqlNoise(sql) {
43
+ /**
44
+ * PostgreSQL 的 E'…' / U&'…' 会把反斜杠当转义;普通 '…' 在
45
+ * standard_conforming_strings=on(9.1 起的默认值)下反斜杠是普通字符。
46
+ */
47
+ function isPostgresEscapeString(sql, quoteIndex) {
48
+ const prev = sql[quoteIndex - 1];
49
+ const isWordChar = (ch) => ch !== undefined && /[A-Za-z0-9_$]/.test(ch);
50
+ if (prev === 'E' || prev === 'e')
51
+ return !isWordChar(sql[quoteIndex - 2]);
52
+ if (prev === '&') {
53
+ const u = sql[quoteIndex - 2];
54
+ return (u === 'U' || u === 'u') && !isWordChar(sql[quoteIndex - 3]);
55
+ }
56
+ return false;
57
+ }
58
+ /**
59
+ * 去掉字符串、引号标识符与注释,保留真实 SQL 关键字与分号。
60
+ *
61
+ * 反斜杠规则按引擎对齐,否则去噪结果会和服务器看到的分割方式不一致:
62
+ * MySQL 默认把 \ 当转义;PostgreSQL 只有 E'…' / U&'…' 才转义;
63
+ * SQLite 字符串里根本没有反斜杠转义。引号标识符("…" 与 `…`)在三种
64
+ * 引擎里都不把反斜杠当转义,永远按字面量处理。
65
+ */
66
+ function stripSqlNoise(sql, engine) {
45
67
  let out = '';
46
68
  let i = 0;
47
69
  while (i < sql.length) {
@@ -74,6 +96,7 @@ function stripSqlNoise(sql) {
74
96
  if (ch === "'") {
75
97
  out += ' ';
76
98
  i += 1;
99
+ const backslashEscapes = engine === 'mysql' || (engine === 'postgres' && isPostgresEscapeString(sql, i - 1));
77
100
  while (i < sql.length) {
78
101
  if (sql[i] === "'" && sql[i + 1] === "'") {
79
102
  i += 2;
@@ -83,7 +106,7 @@ function stripSqlNoise(sql) {
83
106
  i += 1;
84
107
  break;
85
108
  }
86
- if (sql[i] === '\\') {
109
+ if (backslashEscapes && sql[i] === '\\') {
87
110
  i += 2;
88
111
  continue;
89
112
  }
@@ -99,10 +122,6 @@ function stripSqlNoise(sql) {
99
122
  i += 1;
100
123
  break;
101
124
  }
102
- if (sql[i] === '\\') {
103
- i += 2;
104
- continue;
105
- }
106
125
  i += 1;
107
126
  }
108
127
  continue;
@@ -132,10 +151,74 @@ function stripSqlNoise(sql) {
132
151
  }
133
152
  /** 写操作关键字:出在 SELECT/EXPLAIN/WITH 语句里即拒绝。 */
134
153
  const WRITE_KEYWORDS = /\b(insert|update|delete|replace|merge|drop|alter|create|truncate|call|execute|copy|grant|revoke|attach|detach|vacuum|reindex|refresh|set|reset|begin|commit|rollback|savepoint|release|analyze|load_extension)\b/i;
135
- /** 校验只读查询:词法去噪后白名单开头 + 写关键字扫描 + 单语句。 */
136
- export function assertReadQuery(sql) {
154
+ /** 允许不带参数的只读 PRAGMA(含查询型与会话状态查询)。 */
155
+ const READONLY_PRAGMAS = new Set([
156
+ 'application_id',
157
+ 'auto_vacuum',
158
+ 'cell_size_check',
159
+ 'collation_list',
160
+ 'compile_options',
161
+ 'data_version',
162
+ 'database_list',
163
+ 'encoding',
164
+ 'freelist_count',
165
+ 'function_list',
166
+ 'journal_mode',
167
+ 'legacy_file_format',
168
+ 'mmap_size',
169
+ 'module_list',
170
+ 'page_count',
171
+ 'page_size',
172
+ 'pragma_list',
173
+ 'schema_version',
174
+ 'sqlite_version',
175
+ 'user_version',
176
+ ]);
177
+ /** 允许带一个标识符参数的只读内省 PRAGMA。 */
178
+ const READONLY_PRAGMAS_WITH_ARG = new Set([
179
+ 'foreign_key_check',
180
+ 'foreign_key_list',
181
+ 'index_info',
182
+ 'index_list',
183
+ 'index_xinfo',
184
+ 'integrity_check',
185
+ 'quick_check',
186
+ 'table_info',
187
+ 'table_list',
188
+ 'table_xinfo',
189
+ ]);
190
+ /**
191
+ * PRAGMA 白名单:只放行只读内省。
192
+ *
193
+ * SQLite 里 `PRAGMA name(value)` 与 `PRAGMA name = value` 完全等价,
194
+ * 只拦改号会让 `PRAGMA foreign_keys(0)` / `PRAGMA writable_schema(1)`
195
+ * 这类会改连接或库文件状态的调用漏过去,所以按名字白名单放行。
196
+ */
197
+ function assertReadOnlyPragma(statement) {
198
+ const match = /^pragma\s+(?:[a-z_][a-z0-9_]*\.)?([a-z_][a-z0-9_]*)\s*(.*)$/i.exec(statement);
199
+ if (match === null)
200
+ throw new Error('无法解析这条 PRAGMA,sql_query 只接受只读的 PRAGMA 内省语句。');
201
+ const pragmaName = match[1].toLowerCase();
202
+ const rest = match[2].trim();
203
+ if (rest !== '' && /^[=(]/.test(rest)) {
204
+ if (!READONLY_PRAGMAS_WITH_ARG.has(pragmaName)) {
205
+ throw new Error('sql_query 不接受带参数的 PRAGMA ' + pragmaName.toUpperCase() + '(PRAGMA name(value) 与 name=value 等价,会改连接或数据库状态)。请用 sql_exec。');
206
+ }
207
+ return;
208
+ }
209
+ if (!READONLY_PRAGMAS.has(pragmaName) && !READONLY_PRAGMAS_WITH_ARG.has(pragmaName)) {
210
+ throw new Error('sql_query 只放行只读 PRAGMA 白名单(table_info / index_list / page_count / user_version 等),' + pragmaName.toUpperCase() + ' 不在白名单内。需要修改请用 sql_exec。');
211
+ }
212
+ }
213
+ /**
214
+ * 校验只读查询:词法去噪后白名单开头 + 写关键字扫描 + 单语句 + PRAGMA 白名单。
215
+ *
216
+ * `engine` 决定反斜杠转义规则(见 stripSqlNoise);不传时按最保守的方式把
217
+ * 反斜杠当字面量,宁可误拒也不放过第二条语句。
218
+ */
219
+ export function assertReadQuery(sql, engine) {
137
220
  const trimmed = sql.trim();
138
- const clean = stripSqlNoise(trimmed);
221
+ const clean = stripSqlNoise(trimmed, engine);
139
222
  const first = /^[a-z]+/i.exec(clean.trim())?.[0]?.toLowerCase() ?? '';
140
223
  if (!READ_KEYWORDS.test(clean.trim())) {
141
224
  throw new Error('sql_query 只接受只读语句(SELECT / PRAGMA / EXPLAIN / SHOW / DESCRIBE / WITH)。写操作请用 sql_exec。');
@@ -145,8 +228,7 @@ export function assertReadQuery(sql) {
145
228
  throw new Error('sql_query 一次只允许一条语句。');
146
229
  const single = statements[0]?.trim() ?? '';
147
230
  if (first === 'pragma') {
148
- if (/=/.test(single))
149
- throw new Error('sql_query 不接受带赋值参数的 PRAGMA 写操作(如 PRAGMA journal_mode=WAL),请用 sql_exec。');
231
+ assertReadOnlyPragma(single);
150
232
  }
151
233
  else if (first === 'show' || first === 'describe' || first === 'desc') {
152
234
  // SHOW / DESCRIBE 本身只读,不再扫写关键字(避免误伤 SHOW CREATE TABLE)。
@@ -163,6 +245,40 @@ export function assertReadQuery(sql) {
163
245
  }
164
246
  return trimmed.replace(/;+\s*$/, '').trim();
165
247
  }
248
+ /**
249
+ * 把一段写操作脚本压成审批卡可读的摘要。
250
+ *
251
+ * 关键点:逐条列出每条语句,并额外给一行语句类型统计。即使脚本体被省略,
252
+ * 第二条、第 N 条语句的 DROP / DELETE 也不会从审批文本里消失。
253
+ */
254
+ export function summarizeWriteScript(sql) {
255
+ const trimmed = sql.trim();
256
+ const clean = stripSqlNoise(trimmed);
257
+ const statements = clean
258
+ .split(';')
259
+ .map((part) => part.replace(/\s+/g, ' ').trim())
260
+ .filter((part) => part !== '');
261
+ const effective = statements.length === 0 ? ['(空语句)'] : statements;
262
+ const tally = new Map();
263
+ for (const statement of effective) {
264
+ const keyword = /^[a-z]+/i.exec(statement)?.[0]?.toUpperCase() ?? '?';
265
+ tally.set(keyword, (tally.get(keyword) ?? 0) + 1);
266
+ }
267
+ const lines = ['共 ' + effective.length + ' 条语句:'];
268
+ effective.slice(0, 100).forEach((statement, index) => {
269
+ lines.push(' ' + (index + 1) + '. ' + (statement.length > 120 ? statement.slice(0, 120) + '…' : statement));
270
+ });
271
+ if (effective.length > 100)
272
+ lines.push(' …另有 ' + (effective.length - 100) + ' 条语句');
273
+ lines.push('语句类型:' + [...tally.entries()].map(([keyword, count]) => keyword + '×' + count).join(','));
274
+ if (trimmed.length <= 800) {
275
+ lines.push('', '完整 SQL:' + trimmed);
276
+ }
277
+ else {
278
+ lines.push('', '完整 SQL 共 ' + trimmed.length + ' 个字符,见本次工具调用参数。');
279
+ }
280
+ return lines.join('\n');
281
+ }
166
282
  const querySchema = {
167
283
  type: 'object',
168
284
  properties: {
@@ -289,15 +405,20 @@ const healthSchema = {
289
405
  export function buildSqlTools(config) {
290
406
  const cfg = config;
291
407
  const adapters = new Map();
292
- const getAdapter = (name) => {
408
+ /** 只解析连接配置(不建连接),用于在打开数据库前挑选词法规则。 */
409
+ const findConnection = (name) => {
293
410
  const target = name ?? cfg.connections[0].name;
294
411
  const connection = cfg.connections.find((item) => item.name.toLowerCase() === target.toLowerCase());
295
412
  if (connection === undefined) {
296
413
  throw new Error('未找到名为 ' + target + ' 的数据库连接。可用 sql_list 查看连接清单。');
297
414
  }
415
+ return connection;
416
+ };
417
+ const getAdapter = (name) => {
418
+ const connection = findConnection(name);
298
419
  let adapter = adapters.get(connection.name);
299
420
  if (adapter === undefined) {
300
- adapter = createAdapter(connection);
421
+ adapter = createAdapter(connection, { readOnly: cfg.readOnly });
301
422
  adapters.set(connection.name, adapter);
302
423
  }
303
424
  return { adapter, name: connection.name };
@@ -352,7 +473,7 @@ export function buildSqlTools(config) {
352
473
  };
353
474
  const sqlQuery = {
354
475
  name: 'sql_query',
355
- description: '执行只读 SQL 查询(SELECT / PRAGMA / EXPLAIN / SHOW / DESCRIBE / WITH)。词法级校验会拒绝 data-modifying CTE、SELECT INTO、FOR UPDATE/FOR SHARE、PRAGMA 赋值与多语句。connection 为连接名(缺省第一个连接)。返回列名与行数据,最多 maxRows 行(超出 truncated=true)。写操作请用 sql_exec。',
476
+ description: '执行只读 SQL 查询(SELECT / PRAGMA / EXPLAIN / SHOW / DESCRIBE / WITH)。词法级校验按引擎处理反斜杠转义,拒绝 data-modifying CTE、SELECT INTO、FOR UPDATE/FOR SHARE、非白名单 PRAGMA 与多语句;readOnly=true 时还会让数据库自身以只读方式连接。connection 为连接名(缺省第一个连接)。返回列名与行数据,最多 maxRows 行(超出 truncated=true)。写操作请用 sql_exec。',
356
477
  parameters: compileParameters({
357
478
  sql: { type: 'string', required: true, description: '只读 SQL 语句(必填,单条)。' },
358
479
  connection: { type: 'string', description: '连接名(可选,缺省第一个连接)。' },
@@ -378,8 +499,11 @@ export function buildSqlTools(config) {
378
499
  },
379
500
  async execute(rawArgs, exec) {
380
501
  const args = asRecord(rawArgs);
381
- const sql = assertReadQuery(requiredString(args, 'sql', 'SQL 语句'));
382
- const { adapter, name } = getAdapter(optionalString(args, 'connection'));
502
+ const rawSql = requiredString(args, 'sql', 'SQL 语句');
503
+ const connectionName = optionalString(args, 'connection');
504
+ // 先按目标连接的引擎做词法校验,再打开数据库:被拒绝的 SQL 不该去建连接或建库文件。
505
+ const sql = assertReadQuery(rawSql, findConnection(connectionName).engine);
506
+ const { adapter, name } = getAdapter(connectionName);
383
507
  const result = await adapter.query(sql, cfg.maxRows + 1, executionSignal(exec));
384
508
  const total = result.rows.length;
385
509
  const rows = result.rows.slice(0, cfg.maxRows);
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "dsh-sql",
3
- "version": "0.4.5",
3
+ "version": "0.4.6",
4
4
  "description": "连接 SQLite、MySQL 或 PostgreSQL,查询数据并检查数据库结构。",
5
5
  "type": "module",
6
6
  "main": "lib/index.js",