import random
from datetime import timedelta

from django.db import models
from django.contrib.auth import get_user_model
from django.utils import timezone 

from core.apps.shared.models.base import BaseModel
from core.apps.contracts.enums.contract import SIDES, STATUS
from core.apps.contracts.enums.contract_side import ROLE
from core.apps.contracts.enums.contract_signature import SIGNATURE_TYPE, SIGNATURE_STATUS


class Contract(BaseModel):
    file = models.FileField(upload_to="contract/files/%Y/%m/")
    contract_number = models.PositiveIntegerField()
    name = models.CharField(max_length=200)
    
    sides = models.CharField(max_length=13, choices=SIDES) # choices
    status = models.CharField(max_length=15, choices=STATUS, default='created') # choices
 
    face_id = models.BooleanField(default=False)
    attach_file = models.BooleanField(default=False)
    add_folder = models.BooleanField(default=False)
    add_notification = models.BooleanField(default=False)

    company = models.ForeignKey(get_user_model(), on_delete=models.CASCADE, related_name='c')

    def __str__(self):
        return f'{self.name}'
    
    class Meta:
        verbose_name = 'contract'
        verbose_name_plural = 'contracts'
        db_table = 'contracts'


class ContractSide(BaseModel):
    full_name = models.CharField(max_length=70)
    indentification = models.CharField(max_length=20)
    position = models.CharField(max_length=50, null=True, blank=True)

    has_indentification = models.BooleanField(default=False)

    user_role = models.CharField(max_length=9, choices=ROLE)
    
    contract = models.ForeignKey(Contract, on_delete=models.CASCADE, related_name='contract_sides')
    user = models.ForeignKey(get_user_model(), on_delete=models.CASCADE, related_name='contract_sides')

    def __str__(self):
        return f"{self.full_name} - {self.contract}"
    
    class Meta:
        verbose_name = 'contract side'
        verbose_name_plural = 'contract sides'
        db_table = 'contracts_sides'
        unique_together = ['contract', 'user']


class ContractSignature(BaseModel):
    contract = models.ForeignKey(Contract, on_delete=models.CASCADE, related_name='contract_signatures')
    contract_side = models.OneToOneField(ContractSide, on_delete=models.CASCADE, related_name='contract_signatures')

    status = models.CharField(max_length=20, choices=SIGNATURE_STATUS, default='organized') 
    signature_type = models.CharField(max_length=20, choices=SIGNATURE_TYPE, null=True, blank=True)

    is_signature = models.BooleanField(default=False)

    def __str__(self):
        return f'{self.contract_side} user signature for {self.contract} contract'
    
    def generate_code(self):
        code = ''.join([str(random.randint(1, 9) % 10) for _ in range(4)])
        ContractSignatureCode.objects.create(
            code=code,
            signature=self,
            expiration_time = timezone.now() + timedelta(minutes=2)
        )
        return code
    
    class Meta:
        verbose_name = 'contract signature'
        verbose_name_plural = 'contract signatures'
        db_table = 'contract_signatures'
        unique_together = ['contract', 'contract_side']


class ContractSignatureCode(BaseModel):
    code = models.PositiveSmallIntegerField()
    signature = models.ForeignKey(ContractSignature, on_delete=models.CASCADE, related_name='signature_codes')
    expiration_time = models.DateTimeField()

    def __str__(self):
        return f'{self.code} - {self.signature}'

    class Meta:
        verbose_name = 'contract signature code'
        verbose_name_plural = 'contract signature codes'
        db_table = 'contract_signature_codes'