import aiosqlite
import json
import re

from workflow_service.config import Settings

DATABASE_FILE = Settings().DB_FILE

async def init_db():
  async with aiosqlite.connect(DATABASE_FILE) as db:
    await db.execute("""
      CREATE TABLE IF NOT EXISTS workflows (
        name TEXT PRIMARY KEY,
        base_name TEXT NOT NULL,
        version TEXT NOT NULL,
        workflow_json TEXT NOT NULL
      )
    """)
    await db.commit()
  print(f"数据库 '{DATABASE_FILE}' 已准备就绪。")

async def save_workflow(name: str, workflow_json: str):
  # 正则表达式修正为匹配更精确的版本格式 [YYYYMMDDHHMMSS]
  version_match = re.search(r"\[(\d{14})\]$", name)
  if not version_match:
    raise ValueError("Workflow name must have a version suffix like [YYYYMMDDHHMMSS]")
  version = version_match.group(1)
  base_name = re.sub(r"\s*\[\d{14}\]$", "", name).strip()
  async with aiosqlite.connect(DATABASE_FILE) as db:
    await db.execute(
      "INSERT OR REPLACE INTO workflows (name, base_name, version, workflow_json) VALUES (?, ?, ?, ?)",
      (name, base_name, version, workflow_json)
    )
    await db.commit()

async def get_all_workflows() -> list[dict]:
  async with aiosqlite.connect(DATABASE_FILE) as db:
    db.row_factory = aiosqlite.Row
    cursor = await db.execute("SELECT name, workflow_json FROM workflows")
    rows = await cursor.fetchall()
    return [{"name": row["name"], "workflow": json.loads(row["workflow_json"])} for row in rows]

async def get_latest_workflow_by_base_name(base_name: str) -> dict | None:
  async with aiosqlite.connect(DATABASE_FILE) as db:
    db.row_factory = aiosqlite.Row
    cursor = await db.execute("SELECT * FROM workflows WHERE base_name = ? ORDER BY version DESC LIMIT 1", (base_name,))
    row = await cursor.fetchone()
    return dict(row) if row else None

async def get_workflow_by_version(base_name: str, version: str) -> dict | None:
  # [BUG修复] 修复了工作流名称拼接错误
  name = f"{base_name} [{version}]"
  async with aiosqlite.connect(DATABASE_FILE) as db:
    db.row_factory = aiosqlite.Row
    cursor = await db.execute("SELECT * FROM workflows WHERE name = ?", (name,))
    row = await cursor.fetchone()
    return dict(row) if row else None

async def delete_workflow(name: str) -> bool:
  async with aiosqlite.connect(DATABASE_FILE) as db:
    cursor = await db.execute("DELETE FROM workflows WHERE name = ?", (name,))
    await db.commit()
    return cursor.rowcount > 0