Method execute

src/tools/database/db-tools.ts:129–196

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

  argsSchema = DbSchemaArgs;

  async execute(args: z.infer<typeof DbSchemaArgs>, _ctx: ToolContext): Promise<ToolResult> {
    const parsed = parseConn(args.connection_string);
    if (!parsed) return { content: `[DB_SCHEMA_ERROR] Could not parse connection_string. Expected mysql://, postgres://, or sqlite:///path`, isError: true };
    const g = await getConnection(parsed);
    if (g.kind === 'error') return { content: `[DB_SCHEMA_ERROR] ${g.message}`, isError: true };
    const { conn } = g;

    try {
      const out: string[] = [];
      out.push(`# DB Schema: ${parsed.dialect}${parsed.database ? ` / ${parsed.database}` : ''}${parsed.filepath ? ` / ${parsed.filepath}` : ''}`);

      if (!args.table) {
        let tables: any[];
        if (parsed.dialect === 'mysql') {
          const [rows] = await conn.execute('SHOW TABLES');
          tables = (rows as any[]).map(r => Object.values(r)[0]);
        } else if (parsed.dialect === 'postgres') {
          const r = await conn.query(`SELECT tablename FROM pg_tables WHERE schemaname NOT IN ('pg_catalog','information_schema') ORDER BY tablename`);
          tables = r.rows.map((r: any) => r.tablename);
        } else {
          tables = conn.prepare(`SELECT name FROM sqlite_master WHERE type='table' AND name NOT LIKE 'sqlite_%' ORDER BY name`).all().map((r: any) => r.name);
        }
        out.push(`Tables (${tables.length}):`);
        for (const t of tables) out.push(`  - ${t}`);
      } else {
        // Table-specific detail
        if (parsed.dialect === 'mysql') {
          const [cols] = await conn.execute(`SHOW FULL COLUMNS FROM \`${args.table.replace(/`/g, '')}\``);
          out.push(`## Columns`);
          for (const c of cols as any[]) out.push(`  ${c.Field.padEnd(28)} ${String(c.Type).padEnd(20)} ${c.Null === 'NO' ? 'NOT NULL' : 'NULL    '} ${c.Key || ''} ${c.Default !== null ? `DEFAULT ${c.Default}` : ''}`);
          const [idx] = await conn.execute(`SHOW INDEX FROM \`${args.table.replace(/`/g, '')}\``);
          if ((idx as any[]).length > 0) {
            out.push('');
            out.push(`## Indexes`);
            for (const i of idx as any[]) out.push(`  ${i.Key_name.padEnd(28)} on ${i.Column_name}${i.Non_unique === 0 ? ' (UNIQUE)' : ''}`);
          }
        } else if (parsed.dialect === 'postgres') {
          const colsR = await conn.query(`
            SELECT column_name, data_type, is_nullable, column_default
            FROM information_schema.columns WHERE table_name = $1
            ORDER BY ordinal_position`, [args.table]);
          out.push(`## Columns`);
          for (const c of colsR.rows) out.push(`  ${c.column_name.padEnd(28)} ${String(c.data_type).padEnd(20)} ${c.is_nullable === 'NO' ? 'NOT NULL' : 'NULL    '} ${c.column_default !== null ? `DEFAULT ${c.column_default}` : ''}`);
          const idxR = await conn.query(`SELECT indexname, indexdef FROM pg_indexes WHERE tablename = $1`, [args.table]);
          if (idxR.rows.length > 0) {
            out.push('');
            out.push(`## Indexes`);
            for (const i of idxR.rows) out.push(`  ${i.indexname}: ${i.indexdef}`);
          }
        } else {
          const cols = conn.prepare(`PRAGMA table_info("${args.table.replace(/"/g, '')}")`).all();
          out.push(`## Columns`);
          for (const c of cols as any[]) out.push(`  ${c.name.padEnd(28)} ${String(c.type).padEnd(20)} ${c.notnull ? 'NOT NULL' : 'NULL    '} ${c.dflt_value ? `DEFAULT ${c.dflt_value}` : ''}${c.pk ? ' PK' : ''}`);
          const idx = conn.prepare(`PRAGMA index_list("${args.table.replace(/"/g, '')}")`).all();
          if ((idx as any[]).length > 0) {
            out.push('');
            out.push(`## Indexes`);
            for (const i of idx as any[]) out.push(`  ${i.name}${i.unique ? ' (UNIQUE)' : ''}`);