from sqlalchemy import Column, Integer, String, Text, DateTime, func
from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession, async_sessionmaker
from sqlalchemy.orm import DeclarativeBase
import config

engine = create_async_engine(config.DATABASE_URL, echo=False)
async_session = async_sessionmaker(engine, class_=AsyncSession, expire_on_commit=False)


class Base(DeclarativeBase):
    pass


class Book(Base):
    __tablename__ = "books"

    id = Column(Integer, primary_key=True, autoincrement=True)
    book_id = Column(String(100), unique=True, nullable=False, index=True)  # 平台书籍ID
    title = Column(String(500), nullable=False)  # 书名
    author = Column(String(200), default="")  # 作者
    created_at = Column(DateTime, server_default=func.now())


class Chapter(Base):
    __tablename__ = "chapters"

    id = Column(Integer, primary_key=True, autoincrement=True)
    book_id = Column(String(100), nullable=False, index=True)  # 关联的书籍ID
    chapter_id = Column(String(100), nullable=False, index=True)  # 平台音频ID
    app_chapter_id = Column(String(100), default="")  # App章节ID
    title = Column(String(500), default="")  # 章节标题
    text_content = Column(Text, default="")  # TTS 文本内容
    audio_file = Column(String(500), default="")  # 生成的音频文件路径
    status = Column(String(20), default="pending")  # pending / generating / ready / error
    error_msg = Column(Text, default="")
    created_at = Column(DateTime, server_default=func.now())
    updated_at = Column(DateTime, server_default=func.now(), onupdate=func.now())


async def init_db():
    async with engine.begin() as conn:
        await conn.run_sync(Base.metadata.create_all)