buildai-cli 0.3.67__tar.gz → 0.3.68__tar.gz

{buildai_cli-0.3.67 → buildai_cli-0.3.68}/.gitignore

@@ -54,6 +54,7 @@ Thumbs.db
 .vercel
 
 # Local tool state
+.superpowers/
 .entire/
 .codex/config.toml
 .codex-bin/

{buildai_cli-0.3.67 → buildai_cli-0.3.68}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: buildai-cli
-Version: 0.3.67
+Version: 0.3.68
 Summary: Build AI CLI (Typer)
 Requires-Python: >=3.11
 Requires-Dist: httpx>=0.27.0

{buildai_cli-0.3.67 → buildai_cli-0.3.68}/cli/commands/db/common.py

@@ -2,6 +2,7 @@
 
 from __future__ import annotations
 
+import re
 from dataclasses import dataclass
 from pathlib import Path
 from typing import Any
@@ -20,6 +21,11 @@ MIGRATIONS_SA_DB_USER = "buildai-migrations-sa@data-470400.iam"
 REPO_ROOT = Path(__file__).resolve().parents[5]
 MIGRATIONS_DIR = REPO_ROOT / "migrations"
 MIGRATIONS_TABLE = "public._migrations"
+_NO_TRANSACTION_MARKER = "-- requires-no-transaction: true"
+_CONCURRENT_DDL_PATTERN = re.compile(
+    r"\b(?:CREATE|DROP|REINDEX|REFRESH)\s+(?:MATERIALIZED\s+VIEW\s+)?(?:INDEX|TABLE|VIEW)?\s*CONCURRENTLY\b",
+    flags=re.IGNORECASE,
+)
 
 
 @dataclass
@@ -140,3 +146,133 @@ def migration_requires_system_admin(path: Path) -> bool:
     """Detect migrations that must run with the DB admin connection."""
 
     return "requires-system-admin" in path.read_text(encoding="utf-8", errors="ignore")
+
+
+def migration_requires_manual_apply(path: Path) -> bool:
+    """Detect migrations that must be explicitly selected by an operator."""
+
+    for line in path.read_text(encoding="utf-8", errors="ignore").splitlines()[:10]:
+        if line.strip().lower() == "-- requires-manual-apply: true":
+            return True
+    return False
+
+
+def migration_requires_no_transaction(path: Path, sql: str) -> bool:
+    """Return whether a migration must be executed statement-by-statement."""
+
+    header = "\n".join(sql.splitlines()[:10]).lower()
+    if _NO_TRANSACTION_MARKER in header:
+        return True
+    return _CONCURRENT_DDL_PATTERN.search(sql) is not None
+
+
+def split_sql_statements(sql: str) -> list[str]:
+    """Split SQL into executable statements while respecting quotes and DO blocks."""
+
+    statements: list[str] = []
+    chunk: list[str] = []
+    in_single_quote = False
+    in_double_quote = False
+    in_line_comment = False
+    in_block_comment = False
+    dollar_tag: str | None = None
+    i = 0
+
+    while i < len(sql):
+        char = sql[i]
+        next_char = sql[i + 1] if i + 1 < len(sql) else ""
+
+        if in_line_comment:
+            chunk.append(char)
+            if char == "\n":
+                in_line_comment = False
+            i += 1
+            continue
+
+        if in_block_comment:
+            chunk.append(char)
+            if char == "*" and next_char == "/":
+                chunk.append(next_char)
+                in_block_comment = False
+                i += 2
+            else:
+                i += 1
+            continue
+
+        if dollar_tag is not None:
+            if sql.startswith(dollar_tag, i):
+                chunk.append(dollar_tag)
+                i += len(dollar_tag)
+                dollar_tag = None
+            else:
+                chunk.append(char)
+                i += 1
+            continue
+
+        if in_single_quote:
+            chunk.append(char)
+            if char == "'" and next_char == "'":
+                chunk.append(next_char)
+                i += 2
+                continue
+            if char == "'":
+                in_single_quote = False
+            i += 1
+            continue
+
+        if in_double_quote:
+            chunk.append(char)
+            if char == '"':
+                in_double_quote = False
+            i += 1
+            continue
+
+        if char == "-" and next_char == "-":
+            chunk.append(char)
+            chunk.append(next_char)
+            in_line_comment = True
+            i += 2
+            continue
+
+        if char == "/" and next_char == "*":
+            chunk.append(char)
+            chunk.append(next_char)
+            in_block_comment = True
+            i += 2
+            continue
+
+        if char == "'":
+            chunk.append(char)
+            in_single_quote = True
+            i += 1
+            continue
+
+        if char == '"':
+            chunk.append(char)
+            in_double_quote = True
+            i += 1
+            continue
+
+        if char == "$":
+            match = re.match(r"\$[A-Za-z_][A-Za-z0-9_]*\$|\$\$", sql[i:])
+            if match:
+                dollar_tag = match.group(0)
+                chunk.append(dollar_tag)
+                i += len(dollar_tag)
+                continue
+
+        if char == ";":
+            statement = "".join(chunk).strip()
+            if statement:
+                statements.append(statement)
+            chunk = []
+            i += 1
+            continue
+
+        chunk.append(char)
+        i += 1
+
+    trailing = "".join(chunk).strip()
+    if trailing:
+        statements.append(trailing)
+    return statements

{buildai_cli-0.3.67 → buildai_cli-0.3.68}/cli/commands/db/migrate.py

@@ -20,8 +20,11 @@ from .common import (
     get_migration_status,
     get_migrations_connection,
     migration_label,
+    migration_requires_manual_apply,
+    migration_requires_no_transaction,
     migration_requires_system_admin,
     set_owner_role,
+    split_sql_statements,
 )
 
 _MIGRATION_WRITE_PROFILES = frozenset(
@@ -125,6 +128,17 @@ def migrate(
                 if target == "all"
                 else [f for f in migration_files if target and target in f.name]
             )
+            if target == "all":
+                manual_apply = [f for f in to_run if migration_requires_manual_apply(f)]
+                if manual_apply:
+                    warning(
+                        "Skipping manual-apply migration(s): "
+                        + ", ".join(item.name for item in manual_apply)
+                    )
+                    to_run = [f for f in to_run if f not in manual_apply]
+            elif any(migration_requires_manual_apply(f) for f in to_run):
+                warning("Selected manual-apply migration explicitly; proceeding.")
+
             if not to_run:
                 success("No migrations to run")
                 return
@@ -157,7 +171,11 @@ def migrate(
                     await conn.execute("SET statement_timeout = '0'")
                 else:
                     await conn.execute(f"SET statement_timeout = '{timeout}s'")
-                await conn.execute(sql, timeout=None)
+                if migration_requires_no_transaction(migration, sql):
+                    for statement in split_sql_statements(sql):
+                        await conn.execute(statement, timeout=None)
+                else:
+                    await conn.execute(sql, timeout=None)
                 await conn.execute(
                     f"INSERT INTO {MIGRATIONS_TABLE} (filename) VALUES ($1) ON CONFLICT DO NOTHING",
                     migration.name,

{buildai_cli-0.3.67 → buildai_cli-0.3.68}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "buildai-cli"
-version = "0.3.67"
+version = "0.3.68"
 description = "Build AI CLI (Typer)"
 requires-python = ">=3.11"
 dependencies = [