from django.db import models
from django.utils.translation import gettext_lazy as _
from django.core.exceptions import ValidationError
from mptt.models import MPTTModel, TreeForeignKey
from django.utils.text import slugify


class HadisSect(models.Model):
    class SectType(models.TextChoices):
        SHIA = 'shia', _('Shia')
        SUNNI = 'sunni', _('Sunni')

    sect_type = models.CharField(max_length=10, choices=SectType.choices, unique=True, verbose_name=_('Sect Name'))
    title = models.CharField(max_length=256, verbose_name=_('Name'))
    description = models.TextField(verbose_name=_('Description'), null=True, blank=True)
    is_active = models.BooleanField(default=True, verbose_name=_('Is Active'))
    order = models.IntegerField(default=0, verbose_name=_('order'))

    def __str__(self):
        return f"{self.sect_type}: {self.title}"
    
    
    class Meta:
        verbose_name = _('Hadis Sect')
        verbose_name_plural = _('Hadis Sects')
        ordering = ('order',)


class HadisCategory(MPTTModel):
    class SourceType(models.TextChoices):
        QURAN = 'quran', _('Quran')
        HADITH = 'hadith', _('Hadith')
        HISTORY = 'history', _('History')
        FATWA = 'fatwa', _('Fatwa')
        QUOTE = 'quote', _('Quote')

    parent = TreeForeignKey('self', on_delete=models.CASCADE, null=True, blank=True, related_name='children')
    sect = models.ForeignKey(HadisSect, on_delete=models.PROTECT, verbose_name=_('Sect'), null=False, blank=False)
    source_type = models.CharField(max_length=10, choices=SourceType.choices, verbose_name=_('Source Type'))
    title = models.CharField(max_length=256, verbose_name=_('Title'))
    description = models.TextField(verbose_name=_('Description'), null=True, blank=True)
    order = models.IntegerField(default=0, verbose_name=_('order'))
    xmind_file = models.FileField(upload_to='hadis/xmind_files/', verbose_name=_('xmind file'), null=True, blank=True)
    slug = models.SlugField(max_length=255, null=True, blank=True)
    content_type = None
    language = None
    language_id = None

    def clean(self):
        super().clean()
        if self.parent and self.sect_id != self.parent.sect_id:
            raise ValidationError(
                _('Child category must have the same sect_type as its parent. '
                  f'Parent sect: {self.parent.sect.sect_type}, '
                  f'Your sect: {self.sect.sect_type}')
            )

    def save(self, *args, **kwargs):
        self.full_clean()
        if not self.slug:
            base_slug = slugify(self.title, allow_unicode=True)
            slug = base_slug
            while HadisCategory.objects.filter(slug=slug).exclude(pk=self.pk).exists():
                slug = f"{base_slug}"
            self.slug = slug
        super().save(*args, **kwargs)


    
    class Meta:
        verbose_name = _('Hadis Category')
        verbose_name_plural = _('Hadis Categories')
        ordering = ('order',)
        
    def __str__(self):
        return f"{self.sect.sect_type}: {self.source_type} - {self.title}"