import uuid

from django.db import models
from django.utils.translation import gettext_lazy as _


class ClusterSettings(models.Model):
    name = models.CharField(default='cluster_settings', max_length=16, unique=True)
    enabled = models.BooleanField(default=False)
    primary_enable_wireguard = models.BooleanField(default=True)
    stats_sync_interval = models.IntegerField(default=60)
    stats_cache_interval = models.IntegerField(default=60)
    cluster_mode = models.CharField(default='mirror', max_length=16, choices=(('mirror', 'Mirror'), ))
    restart_mode = models.CharField(default='auto', max_length=16, choices=(('auto', 'Automatically restart/reload'), ('manual', 'Manual')))
    worker_display = models.CharField(
        default='server_address', max_length=16, choices=(
            ('name', 'Name'), ('server_address', 'Server Address'),
            ('location', 'Location'), ('address_location', 'Address + Location')
        )
    )
    config_version = models.PositiveIntegerField(default=0)
    dns_version = models.PositiveIntegerField(default=0)

    updated = models.DateTimeField(auto_now=True)
    created = models.DateTimeField(auto_now_add=True)
    uuid = models.UUIDField(default=uuid.uuid4, editable=False, unique=True)

    def __str__(self):
        return self.name


class Worker(models.Model):
    name = models.CharField(max_length=100, unique=True)
    enabled = models.BooleanField(default=True)
    token = models.UUIDField(default=uuid.uuid4, unique=True)
    ip_lock = models.BooleanField(default=False)
    ip_address = models.GenericIPAddressField(blank=True, null=True)

    force_reload = models.BooleanField(default=False)
    force_restart = models.BooleanField(default=False)

    country = models.CharField(max_length=100, blank=True, null=True)
    city = models.CharField(max_length=100, blank=True, null=True)
    hostname = models.CharField(max_length=100, blank=True, null=True)

    error_status = models.CharField(default='', max_length=32, choices=(
        ('', ''),
        ('ip_lock', _('IP lock is enabled, but the worker is attempting to access from a different IP address.')),
        ('worker_disabled', _('Worker is not enabled')),
        ('cluster_disabled', _('Cluster is not enabled')),
        ('missing_version', _('Please report worker_config_version, worker_dns_version and worker_version in the API request.')),
        ('update_required', _('Worker update is required.'))
    ))

    updated = models.DateTimeField(auto_now=True)
    created = models.DateTimeField(auto_now_add=True)
    uuid = models.UUIDField(default=uuid.uuid4, editable=False, unique=True)


class WorkerStatus(models.Model):
    worker = models.OneToOneField(Worker, on_delete=models.CASCADE)
    last_seen = models.DateTimeField(auto_now=True)
    last_reload = models.DateTimeField(blank=True, null=True)
    last_restart = models.DateTimeField(blank=True, null=True)
    dns_version = models.PositiveIntegerField(default=0)
    config_version = models.PositiveIntegerField(default=0)
    worker_version = models.PositiveIntegerField(default=0)
    active_peers = models.PositiveIntegerField(default=0)
    wireguard_status = models.JSONField(default=dict)

    updated = models.DateTimeField(auto_now=True)
    created = models.DateTimeField(auto_now_add=True)
    uuid = models.UUIDField(default=uuid.uuid4, editable=False, unique=True)