Merge pull request #2 from suyog19/dev

feat: main deploy workflow + repo mapping model

Author: suyog19

app/database.py

@@ -64,4 +64,14 @@ def init_db(retries: int = 5, delay: int = 3):
                 )
             """)
 
+            cur.execute("""
+                CREATE TABLE IF NOT EXISTS repo_mappings (
+                    id            SERIAL PRIMARY KEY,
+                    issue_key     VARCHAR(50)   NOT NULL UNIQUE,
+                    repo_name     VARCHAR(200)  NOT NULL,
+                    target_branch VARCHAR(100)  NOT NULL DEFAULT 'main',
+                    created_at    TIMESTAMP     NOT NULL DEFAULT NOW()
+                )
+            """)
+
     logger.info("Database initialized — tables ready")

app/main.py

@@ -3,9 +3,13 @@
 from fastapi import FastAPI
 from dotenv import load_dotenv
 
+from fastapi import HTTPException
+from pydantic import BaseModel
+
 from app.database import init_db
 from app.telegram import send_message
 from app.webhooks import router as webhooks_router
+from app.repo_mapping import get_mapping, get_all_mappings, add_mapping
 
 load_dotenv()
 
@@ -50,3 +54,31 @@ def health_check():
 def debug_telegram():
     send_message("debug", "test", "Manual test from /debug/send-telegram")
     return {"sent": True}
+
+
+# ---------------------------------------------------------------------------
+# Repo mapping endpoints
+# ---------------------------------------------------------------------------
+
+class RepoMappingIn(BaseModel):
+    issue_key: str
+    repo_name: str
+    target_branch: str = "main"
+
+
+@app.get("/debug/repo-mappings")
+def list_repo_mappings():
+    return get_all_mappings()
+
+
+@app.get("/debug/repo-mappings/{issue_key}")
+def inspect_repo_mapping(issue_key: str):
+    mapping = get_mapping(issue_key)
+    if not mapping:
+        raise HTTPException(status_code=404, detail=f"No mapping found for '{issue_key}'")
+    return mapping
+
+
+@app.post("/debug/repo-mappings")
+def create_repo_mapping(body: RepoMappingIn):
+    return add_mapping(body.issue_key, body.repo_name, body.target_branch)

app/repo_mapping.py

@@ -0,0 +1,44 @@
+import logging
+from app.database import get_conn
+
+logger = logging.getLogger("orchestrator")
+
+
+def get_mapping(issue_key: str) -> dict | None:
+    with get_conn() as conn:
+        with conn.cursor() as cur:
+            cur.execute(
+                "SELECT issue_key, repo_name, target_branch FROM repo_mappings WHERE issue_key = %s",
+                (issue_key,),
+            )
+            row = cur.fetchone()
+    if not row:
+        logger.warning("No repo mapping found for issue_key: %s", issue_key)
+        return None
+    return {"issue_key": row[0], "repo_name": row[1], "target_branch": row[2]}
+
+
+def get_all_mappings() -> list[dict]:
+    with get_conn() as conn:
+        with conn.cursor() as cur:
+            cur.execute("SELECT issue_key, repo_name, target_branch FROM repo_mappings ORDER BY id")
+            rows = cur.fetchall()
+    return [{"issue_key": r[0], "repo_name": r[1], "target_branch": r[2]} for r in rows]
+
+
+def add_mapping(issue_key: str, repo_name: str, target_branch: str = "main") -> dict:
+    with get_conn() as conn:
+        with conn.cursor() as cur:
+            cur.execute(
+                """
+                INSERT INTO repo_mappings (issue_key, repo_name, target_branch)
+                VALUES (%s, %s, %s)
+                ON CONFLICT (issue_key) DO UPDATE
+                  SET repo_name = EXCLUDED.repo_name,
+                      target_branch = EXCLUDED.target_branch
+                RETURNING issue_key, repo_name, target_branch
+                """,
+                (issue_key, repo_name, target_branch),
+            )
+            row = cur.fetchone()
+    return {"issue_key": row[0], "repo_name": row[1], "target_branch": row[2]}