from django.db import models
from django.utils.translation import gettext_lazy as _
from django.core.exceptions import ValidationError
from dj_category.models import BaseCategoryAbstract


class HadisCategory(BaseCategoryAbstract):
    class SourceType(models.TextChoices):
        SHIA = 'shia', _('Shia')
        SUNNI = 'sunni', _('Sunni')

    class ContentType(models.TextChoices):
        QURAN = 'quran', _('Quran')
        HADITH = 'hadith', _('Hadith')

    class LevelChoices(models.IntegerChoices):
        LEVEL_1 = 1, _('Level 1 (Root)')
        LEVEL_2 = 2, _('Level 2 (Child)')
        LEVEL_3 = 3, _('Level 3 (Grandchild)')

    source_type = models.CharField(max_length=10, choices=SourceType.choices, default=SourceType.SHIA, verbose_name=_('Source Type'), blank=True)
    category_type = models.CharField(max_length=10, choices=ContentType.choices, verbose_name=_('Category Content Type'), blank=True, null=True)
    name = models.CharField(max_length=355, verbose_name=_('name'))
    order = models.IntegerField(default=0, verbose_name=_('order'))
    slug = None
    content_type = None
    language = None
    language_id = None

    # This field is not stored in the database, it's only used for the form
    level_choice = None
    
    class Meta:
        verbose_name = _('Hadis Category')
        verbose_name_plural = _('Hadis Categories')
        ordering = ('order',)
        
    def __str__(self):
        return f'<{str(self.level_p)}>{self.name}'

    def __repr__(self):
        return f'<{str(self.level_p)}>{self.name}'
    
    def clean(self):
        super().clean()

        # Skip validation for new objects that haven't been saved yet
        # This allows the admin form to set these values properly
        if self.pk is None:
            return

        # For existing objects, apply the validation rules
        if self.level_p == 1 and self.category_type:
            raise ValidationError(_("Level 1 cannot have content type"))

        if self.level_p == 2 and not self.category_type:
            raise ValidationError(_("Level 2 must have content type"))

        if self.level_p == 3 and (self.source_type or self.category_type):
            raise ValidationError(_("Level 3 cannot have source/content type"))


    def save(self, *args, **kwargs):
        self.clean()

        # Get the level from the parent structure
        level = self.level_p

        # Apply level-specific logic
        # if level == 2 and self.parent:
            # For level 2, inherit source_type from parent
            # self.source_type = self.parent.source_type
        # elif level == 3:
            # For level 3, inherit both from parent
            # if self.parent and self.parent.parent:
                # self.source_type = self.parent.source_type
                # self.category_type = self.parent.category_type

        # Call the parent class's save method
        super().save(*args, **kwargs)
        
    @property
    def level_p(self):
        if not self.parent:
            return 1
        elif not self.parent.parent:
            return 2
        else:
            return 3
    def get_level_info(self):
        info = {
            'level': self.level_p,
            'source_type': None,
            'category_type': None,
        }
        if self.level_p == 1:
            info['source_type'] = self.source_type
        elif self.level_p == 2:
            info['source_type'] = self.parent.source_type
            info['category_type'] = self.category_type
        return info