from datetime import datetime

from sqlalchemy.orm import Session

from app.shared import schemas
from app.shared.db import models


# TODO: isolate database operations away from worker and into WEB
# ONLY WORKER
def update_sheet_last_url_archived_at(db: Session, sheet_id: str):
    db_sheet = (
        db.query(models.Sheet).filter(models.Sheet.id == sheet_id).first()
    )
    if db_sheet:
        db_sheet.last_url_archived_at = datetime.now()
        db.commit()
        return True
    return False


# ONLY WORKER and INTEROP


def get_group(db: Session, group_name: str) -> models.Group:
    return db.query(models.Group).filter(models.Group.id == group_name).first()


def create_or_get_user(db: Session, author_id: str) -> models.User:
    if isinstance(author_id, str):
        author_id = author_id.lower()
    db_user = (
        db.query(models.User).filter(models.User.email == author_id).first()
    )
    if not db_user:
        db_user = models.User(email=author_id)
        db.add(db_user)
        db.commit()
        db.refresh(db_user)
    return db_user


def create_tag(db: Session, tag: str) -> models.Tag:
    db_tag = db.query(models.Tag).filter(models.Tag.id == tag).first()
    if not db_tag:
        db_tag = models.Tag(id=tag)
        db.add(db_tag)
        db.commit()
        db.refresh(db_tag)
    return db_tag


def create_archive(
    db: Session,
    archive: schemas.ArchiveCreate,
    tags: list[models.Tag],
    urls: list[models.ArchiveUrl],
) -> models.Archive:
    db_archive = models.Archive(
        id=archive.id,
        url=archive.url,
        result=archive.result,
        public=archive.public,
        author_id=archive.author_id,
        group_id=archive.group_id,
        sheet_id=archive.sheet_id,
        store_until=archive.store_until,
    )
    db_archive.tags = tags
    db_archive.urls = urls
    db.add(db_archive)
    db.commit()
    db.refresh(db_archive)
    return db_archive


def store_archived_url(
    db: Session, archive: schemas.ArchiveCreate
) -> models.Archive:
    # create and load user, tags, if needed
    create_or_get_user(db, archive.author_id)
    db_tags = [create_tag(db, tag) for tag in (archive.tags or [])]
    # insert everything
    db_archive = create_archive(
        db, archive=archive, tags=db_tags, urls=archive.urls
    )
    return db_archive