from __future__ import annotations

from app.db.session import SessionLocal

from app.workers.celery_app import celery_app

@celery_app.task(name="sightline.process_import")

def process_import_task(import_id: str) -> dict:

    from app.services.ingest import process_import

    db = SessionLocal()

    try:

        return process_import(db, import_id)

    finally:

        db.close()

@celery_app.task(name="sightline.run_enrichment")

def run_enrichment_task(job_id: str) -> dict:

    """Background enrichment for slow/network-bound enrichers."""

    from datetime import datetime, timezone

    from app.models import EnrichmentJob

    from app.services.enrichment import run_enricher

    db = SessionLocal()

    try:

        job = db.get(EnrichmentJob, job_id)

        if not job:

            return {"error": "job not found"}

        result = run_enricher(job.enricher, job.target)

        job.result = result.to_dict()

        job.confidence = result.confidence

        job.status = "done" if not result.error else "error"

        job.finished_at = datetime.now(timezone.utc)

        db.commit()

        return result.to_dict()

    finally:

        db.close()

@celery_app.task(name="sightline.apply_retention")

def apply_retention_task() -> dict:

    """Scheduled retention enforcement - delete reads older than the configured window."""

    from datetime import datetime, timedelta, timezone

    from app.core.config import settings

    from app.services.retention import purge_expired

    db = SessionLocal()

    try:

        cutoff = datetime.now(timezone.utc) - timedelta(days=settings.default_retention_days)

        deleted = purge_expired(db, cutoff)

        db.commit()

        return {"deleted": deleted, "cutoff": cutoff.isoformat()}

    finally:

        db.close()