Method execute

src/tools/database/db-tools.ts:219–290

(args: z.infer<typeof DbQueryArgs>, _ctx: ToolContext)

  argsSchema = DbQueryArgs;

  async execute(args: z.infer<typeof DbQueryArgs>, _ctx: ToolContext): Promise<ToolResult> {
    if (!args.allow_write && !READ_ONLY_PREFIXES.test(args.sql)) {
      return {
        content: `[DB_QUERY_REFUSED] Only read statements (SELECT/EXPLAIN/SHOW/DESCRIBE/PRAGMA/WITH) are allowed by default. Detected non-read prefix in: ${args.sql.slice(0, 80)}…\n\nPass allow_write=true if intentional. Note: changes to a remote DB cannot be /restored by QodeX.`,
        isError: true,
      };
    }

    const parsed = parseConn(args.connection_string);
    if (!parsed) return { content: `[DB_QUERY_ERROR] Bad connection_string`, isError: true };
    const g = await getConnection(parsed);
    if (g.kind === 'error') return { content: `[DB_QUERY_ERROR] ${g.message}`, isError: true };
    const { conn } = g;
    const maxRows = args.max_rows ?? 200;

    try {
      let rows: any[] = [];
      let affectedRows: number | undefined;

      if (parsed.dialect === 'mysql') {
        const [r] = await conn.execute(args.sql, args.params ?? []);
        if (Array.isArray(r)) rows = r as any[];
        else affectedRows = (r as any).affectedRows;
      } else if (parsed.dialect === 'postgres') {
        const r = await conn.query(args.sql, args.params ?? []);
        rows = r.rows ?? [];
        affectedRows = r.rowCount ?? undefined;
      } else {
        const stmt = conn.prepare(args.sql);
        if (stmt.reader) {
          rows = stmt.all(...(args.params ?? []));
        } else {
          const info = stmt.run(...(args.params ?? []));
          affectedRows = info.changes;
        }
      }

      const out: string[] = [];
      out.push(`# DB Query result`);
      out.push(`SQL: ${args.sql.slice(0, 200)}${args.sql.length > 200 ? '…' : ''}`);
      if (affectedRows !== undefined) {
        out.push(`Affected rows: ${affectedRows}`);
      }
      if (rows.length > 0) {
        const total = rows.length;
        const shown = rows.slice(0, maxRows);
        const cols = Object.keys(shown[0]!);
        // Pretty-print as a markdown table
        out.push('');
        out.push(`Rows: ${total}${total > maxRows ? ` (showing first ${maxRows})` : ''}`);
        out.push('');
        out.push('| ' + cols.join(' | ') + ' |');
        out.push('|' + cols.map(() => '---').join('|') + '|');
        for (const r of shown) {
          out.push('| ' + cols.map(c => {
            const v = (r as any)[c];
            if (v === null) return 'NULL';
            const s = typeof v === 'object' ? JSON.stringify(v) : String(v);