from celery.decorators import task
from eve_api.models import EVEAccount, EVEPlayerCorporation, EVEPlayerAlliance
from sso.models import ServiceAccount
from django.contrib.auth.models import User
from django.db.models import signals


# Signals that the tasks need to listen for
def eveapi_deleted(sender, instance, **kwargs):
    if instance.user:
        update_user_access.delay(user=instance.user.id)

signals.post_delete.connect(eveapi_deleted, sender=EVEAccount)


@task()
def update_user_access(user, **kwargs):
    """
    Process all corporate and alliance entries and correct
    access groups.
    """

    user = User.objects.get(id=user)

    # Create a list of all Corp and Alliance groups
    corpgroups = []
    corpgroups.extend(EVEPlayerCorporation.objects.filter(group__isnull=False))
    corpgroups.extend(EVEPlayerAlliance.objects.filter(group__isnull=False))

    # Create a list of Char groups
    chargroups = []
    for eacc in EVEAccount.objects.filter(user=user):
        if eacc.api_status in [1, 3]:
            for char in eacc.characters.all():
                if char.corporation.group:
                    chargroups.append(char.corporation.group)
                if char.corporation.alliance and char.corporation.alliance.group:
                        chargroups.append(char.corporation.alliance.group)

    # Generate the list of groups to add/remove
    delgroups = set(set(user.groups.all()) & set(corpgroups)) - set(chargroups)
    addgroups = set(chargroups) - set(set(user.groups.all()) & set(corpgroups))

    for g in delgroups:
        if g in user.groups.all():
            user.groups.remove(g)

    for g in addgroups:
        if not g in user.groups.all():
            user.groups.add(g)

    # For users set to not active, delete all accounts
    if not user.is_active:
        for servacc in ServiceAccount.objects.filter(user=user):
            servacc.active = 0
            servacc.save()
            pass

    # For each of the user's services, check they're in a valid group for it and enable/disable as needed.
    for servacc in ServiceAccount.objects.filter(user=user):
        if not (set(user.groups.all()) & set(servacc.service.groups.all())):
            if servacc.active:
                servacc.active = 0
                servacc.save()
                pass
        else:
            if not servacc.active:
                servacc.active = 1
                servacc.save()
                pass

    update_service_groups.delay(user_id=user.id)


@task(ignore_result=True)
def update_service_groups(user_id):
    for service in ServiceAccount.objects.filter(user=user_id, active=True).select_related('service__api'):
        api = service.service.api_class
        try:
            api.update_groups(service.service_uid, service.user.groups.all(), service.character)
        except:
            print "Error updating %s %s" % (service.service, service.service_uid)
            pass