import os
from sqlmodel import create_engine, SQLModel
from sqlalchemy.orm import sessionmaker

# PostgreSQL example (or use "sqlite:///database.db" for SQLite)
# DATABASE_URL = "postgresql://user:password@localhost/db_name"
# DATABASE_URL = "sqlite:///data/wrongmove.db"
# DATABASE_URL = "mysql://wrongmove:wrongmove@localhost:3306/wrongmove"
DATABASE_URL = os.environ["DB_CONNECTION_STRING"]


engine = create_engine(DATABASE_URL, echo=False)  # `echo=True` for debug logs
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)


def init_db():
    """Create all tables (only for development; use migrations in production)."""
    SQLModel.metadata.create_all(engine)