#!/usr/bin/env python3
"""Import SQL from a directory (dev). Usage: python scripts/import_local.py <dir> <subject_name>"""
import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parents[1]))

import io
import zipfile

from app.database import SessionLocal
from app.import_service import load_dir_sql, run_import
from app.models import User


def dir_to_zip_bytes(dir_path: Path) -> bytes:
    buf = io.BytesIO()
    with zipfile.ZipFile(buf, "w", zipfile.ZIP_DEFLATED) as zf:
        for name, data in load_dir_sql(dir_path).items():
            zf.writestr(Path(name).name, data)
    return buf.getvalue()


def main():
    if len(sys.argv) < 3:
        print("Usage: import_local.py <sql_dir> <subject_name>")
        sys.exit(1)
    dir_path = Path(sys.argv[1])
    subject = sys.argv[2]
    db = SessionLocal()
    user = db.query(User).filter(User.username == "admin").first()
    uid = user.id if user else None
    result = run_import(
        db,
        subject_name=subject,
        zip_filename=dir_path.name + ".zip",
        file_bytes=dir_to_zip_bytes(dir_path),
        user_id=uid,
    )
    print(result)


if __name__ == "__main__":
    main()