from pathlib import Path

from sqlmodel import SQLModel, Session, create_engine

DB_PATH = Path.home() / ".debian-manager" / "data.db"
DB_PATH.parent.mkdir(parents=True, exist_ok=True)

engine = create_engine(
    f"sqlite:///{DB_PATH}",
    echo=False,
    connect_args={"check_same_thread": False},
)
# expire_on_commit=False — объекты остаются читаемыми после закрытия сессии
from sqlalchemy.orm import sessionmaker
from sqlmodel import Session as _Session

_SessionFactory = sessionmaker(bind=engine, class_=_Session, expire_on_commit=False)


def init_db() -> None:
    from app.db import models  # noqa: F401 — регистрация таблиц
    SQLModel.metadata.create_all(engine)
    _migrate()


def _migrate() -> None:
    """ALTER TABLE для колонок, добавленных после первого запуска."""
    import sqlite3
    conn = sqlite3.connect(DB_PATH)
    cols = [r[1] for r in conn.execute("PRAGMA table_info(server)").fetchall()]
    if "note" not in cols:
        conn.execute("ALTER TABLE server ADD COLUMN note TEXT DEFAULT ''")
        conn.commit()
    conn.close()


def get_session() -> Session:
    return _SessionFactory()