# Create your tasks here
import uuid
import logging
from typing import List, Dict
from django.core import signals
from celery import shared_task

from api import utils, manager
from api.exceptions import ServiceUnavailable
logger = logging.getLogger(__name__)


@shared_task(
    autoretry_for=(Exception, ),
    retry_backoff=8,
    retry_jitter=True,
    retry_backoff_max=3600,
    retry_kwargs={'max_retries': 32}
)
def send_usage(usage: List[Dict[str, str]]):
    task_id = uuid.uuid4().hex
    signals.request_started.send(sender=task_id)
    try:
        manager.UsageAPI().post(usage)
    except Exception as e:
        signals.got_request_exception.send(sender=task_id)
        raise e
    else:
        signals.request_finished.send(sender=task_id)


@shared_task(retry_kwargs={'max_retries': 3})
def send_app_log(app_id, msg, level=logging.INFO):
    task_id = uuid.uuid4().hex
    signals.request_started.send(sender=task_id)
    try:
        utils.send_app_log(app_id, msg, level)
    except Exception as e:
        signals.got_request_exception.send(sender=task_id)
        raise e
    else:
        signals.request_finished.send(sender=task_id)


@shared_task(
    autoretry_for=(ServiceUnavailable, ),
    retry_kwargs={'max_retries': 3}
)
def scale_app(app, user, structure):
    task_id = uuid.uuid4().hex
    signals.request_started.send(sender=task_id)
    try:
        app.scale(user, structure)
    except Exception as e:
        signals.got_request_exception.send(sender=task_id)
        raise e
    else:
        signals.request_finished.send(sender=task_id)


@shared_task(
    autoretry_for=(ServiceUnavailable, ),
    retry_kwargs={'max_retries': 3}
)
def run_pipeline(release, *args, **kwargs):
    task_id = uuid.uuid4().hex
    signals.request_started.send(sender=task_id)
    try:
        release.app.pipeline(release, *args, **kwargs)
    except Exception as e:
        signals.got_request_exception.send(sender=task_id)
        raise e
    else:
        signals.request_finished.send(sender=task_id)


@shared_task(
    autoretry_for=(ServiceUnavailable, ),
    retry_jitter=True,
    retry_kwargs={'max_retries': 3}
)
def restart_app(app, **kwargs):
    task_id = uuid.uuid4().hex
    signals.request_started.send(sender=task_id)
    try:
        app.restart(**kwargs)
    except Exception as e:
        signals.got_request_exception.send(sender=task_id)
        raise e
    else:
        signals.request_finished.send(sender=task_id)


@shared_task(
    autoretry_for=(ServiceUnavailable, ),
    retry_jitter=True,
    retry_kwargs={'max_retries': 3}
)
def delete_pod(app, **kwargs):
    task_id = uuid.uuid4().hex
    signals.request_started.send(sender=task_id)
    try:
        app.delete_pod(**kwargs)
    except Exception as e:
        signals.got_request_exception.send(sender=task_id)
        raise e
    else:
        signals.request_finished.send(sender=task_id)


@shared_task(
    autoretry_for=(ServiceUnavailable, ),
    retry_jitter=True,
    retry_kwargs={'max_retries': 3}
)
def mount_app(app, user, volume, path):
    task_id = uuid.uuid4().hex
    signals.request_started.send(sender=task_id)
    try:
        # merge mount volume path and remove keys if a null value is provided
        for key, value in path.items():
            if value is None:
                if key not in volume.path:
                    continue
                volume.path.pop(key)
            else:
                volume.path[key] = value
        volume.save()
        structure = {}
        for scale_type, replicas in app.structure.items():
            if scale_type in path:
                structure[scale_type] = replicas
        app.mount(user, volume, structure)
    except Exception as e:
        signals.got_request_exception.send(sender=task_id)
        logger.exception(e)
    else:
        signals.request_finished.send(sender=task_id)


@shared_task(
    autoretry_for=(ServiceUnavailable, ),
    retry_kwargs={'max_retries': 3}
)
def scale_resources(blocklist, app, suspended_state, scale_type):
    task_id = uuid.uuid4().hex
    signals.request_started.send(sender=task_id)
    try:
        blocklist.scale_resources(app, suspended_state, scale_type)
    except Exception as e:
        signals.got_request_exception.send(sender=task_id)
        raise e
    else:
        signals.request_finished.send(sender=task_id)