from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine
from sqlalchemy.orm import DeclarativeBase

from scribe.config import Config

# Named rather than inlined so the deadline below is READABLE. SQLAlchemy
# captures `connect_args` in a closure and merges it at connect time, so an
# inline dict cannot be recovered from the engine — and a guard that cannot
# read the value it guards is a guard that passes forever.
_CONNECT_ARGS: dict = {
    # A DEADLINE ON ESTABLISHING A CONNECTION (#4181, rule 156).
    #
    # asyncpg's `timeout` bounds the CONNECT — the TCP handshake plus session
    # setup — and nothing else. Against a host whose storage has wedged, that
    # handshake does not fail, it waits, and without this the wait is
    # asyncpg's own 60-second default: the entire lifespan budget spent before
    # a single query is even sent. Ten seconds is far longer than a healthy
    # local connect (single-digit milliseconds) and short enough to leave room
    # to fail usefully rather than be killed.
    #
    # WHAT THIS DOES NOT COVER, said plainly so the next reader doesn't assume
    # it does: a query on an already-open connection, which includes
    # `pool_pre_ping`'s liveness check. Bounding those is `command_timeout`,
    # and that is deliberately NOT set here — it would apply to every
    # statement, and this app legitimately runs long ones (the embedding
    # backfills, VACUUM ANALYZE). A blanket statement deadline would trade
    # this failure mode for a worse one. Callers that must not hang — the
    # lifespan hook above all — bound their own await instead; see
    # `_STARTUP_READ_TIMEOUT` in services/db_maintenance_scheduler.py.
    "timeout": 10,
}

engine = create_async_engine(
    Config.DATABASE_URL,
    echo=False,
    pool_pre_ping=True,
    pool_recycle=1800,
    connect_args=_CONNECT_ARGS,
)
async_session = async_sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)


class Base(DeclarativeBase):
    pass


from scribe.models.base import CreatedAtMixin, TimestampMixin  # noqa: E402, F401


from scribe.models.note import Note, TaskPriority, TaskStatus  # noqa: E402, F401
from scribe.models.setting import Setting  # noqa: E402, F401
from scribe.models.user import User  # noqa: E402, F401
from scribe.models.app_log import AppLog  # noqa: E402, F401
from scribe.models.password_reset import PasswordResetToken  # noqa: E402, F401
from scribe.models.invitation import InvitationToken  # noqa: E402, F401
from scribe.models.embedding import (  # noqa: E402, F401
    MilestoneEmbedding, NoteEmbedding, RuleEmbedding, SystemEmbedding,
)
from scribe.models.retrieval_log import RetrievalLog  # noqa: E402, F401
from scribe.models.retrieval_tuning import RetrievalTuningEvent  # noqa: E402, F401
from scribe.models.note_usage import NoteUsageEvent  # noqa: E402, F401
from scribe.models.rule_usage import RuleUsageEvent  # noqa: E402, F401
from scribe.models.project import Project  # noqa: E402, F401
from scribe.models.milestone import Milestone  # noqa: E402, F401
from scribe.models.task_log import TaskLog  # noqa: E402, F401
from scribe.models.note_draft import NoteDraft  # noqa: E402, F401
from scribe.models.note_version import NoteVersion  # noqa: E402, F401
from scribe.models.rule_version import RuleVersion  # noqa: E402, F401
from scribe.models.note_supersession import NoteSupersession  # noqa: E402, F401
from scribe.models.group import Group, GroupMembership  # noqa: E402, F401
from scribe.models.share import NoteShare, ProjectShare  # noqa: E402, F401
from scribe.models.notification import Notification  # noqa: E402, F401
from scribe.models.api_key import ApiKey  # noqa: E402, F401
from scribe.models.user_profile import UserProfile  # noqa: E402, F401
# Imported before rulebook: rule_systems foreign-keys canonical_systems.
from scribe.models.canonical_system import CanonicalSystem  # noqa: E402, F401
from scribe.models.rulebook import (  # noqa: E402, F401
    Rulebook, RulebookTopic, Rule, RuleRelation, rule_systems,
)
# After notes and rules: it foreign-keys both (milestone 440).
from scribe.models.lesson_rule_link import LessonRuleLink  # noqa: E402, F401
from scribe.models.repo_binding import RepoBinding  # noqa: E402, F401
from scribe.models.forge_connection import ForgeConnection  # noqa: E402, F401
from scribe.models.code_shape import CodeShape, CodeShapeConsumer, CodeShapeEvent, CodeShapeUse  # noqa: E402, F401
from scribe.models.system import System, RecordSystem  # noqa: E402, F401
from scribe.models.design_system import DesignSystem, DesignToken  # noqa: E402, F401