import asyncio
from io import BytesIO
import io
import os
import random
import string

from tortoise.models import Model
from tortoise import fields
from tortoise.contrib.pydantic import pydantic_model_creator
import async_to_sync as sync
from tortoise.manager import Manager
from generateur.generateur_main import Generateur

from services.io import get_abs_path_from_relative_to_root

from .validators import ExoSourceValidator, get_support_compatibility_for_exo_source_from_path, get_support_compatibility_for_exo_source_from_data

from .customField import FileField


class Tag(Model):
    id = fields.IntField(pk=True)
    id_code = fields.CharField(unique=True, default="", max_length=15)

    name = fields.CharField(max_length=35)
    color = fields.CharField(max_length=100)
    
    owner = fields.ForeignKeyField('models.UserModel')


class Exercice(Model):
    id = fields.IntField(pk=True)
    id_code = fields.CharField(default="", max_length=10, unique=True)

    name = fields.CharField(max_length=50)

    consigne = fields.CharField(max_length=200, null=True, default="")

    exo_source = FileField(upload_root="/uploads",
                           validators=[ExoSourceValidator()])
    
    updated_at = fields.DatetimeField(auto_now=True)

    private = fields.BooleanField(default=False)

    tags = fields.ManyToManyField('models.Tag', null=True)
    
    origin = fields.ForeignKeyField('models.Exercice', null = True)
    isOriginal = fields.BooleanField(default = True)
    
    author = fields.ForeignKeyField('models.UserModel')
    
    def pdfSupport(self) -> bool:
        if not isinstance(self.exo_source, io.BytesIO):
            if os.path.exists(get_abs_path_from_relative_to_root(self.exo_source)):
                support_compatibility = get_support_compatibility_for_exo_source_from_path(
                    get_abs_path_from_relative_to_root(self.exo_source))
                return support_compatibility['isPdf']
            return False
        else:
            self.exo_source.seek(0)
            support_compatibility = get_support_compatibility_for_exo_source_from_data(
                self.exo_source.read())
            return support_compatibility['isPdf']

    def csvSupport(self) -> bool:
        if not isinstance(self.exo_source, io.BytesIO):
            if os.path.exists(get_abs_path_from_relative_to_root(self.exo_source)):
                support_compatibility = get_support_compatibility_for_exo_source_from_path(
                    get_abs_path_from_relative_to_root(self.exo_source))
                return support_compatibility['isCsv']
            return False
        else:
            self.exo_source.seek(0)
            support_compatibility = get_support_compatibility_for_exo_source_from_data(
                self.exo_source.read())
            return support_compatibility['isCsv']

    def webSupport(self) -> bool:

        if not isinstance(self.exo_source, io.BytesIO):
            if os.path.exists(get_abs_path_from_relative_to_root(self.exo_source)):
                support_compatibility = get_support_compatibility_for_exo_source_from_path(
                    get_abs_path_from_relative_to_root(self.exo_source))
                return support_compatibility['isWeb']
            return False
        else:
            self.exo_source.seek(0)
            support_compatibility = get_support_compatibility_for_exo_source_from_data(
                self.exo_source.read())
            return support_compatibility['isWeb']
        
    def examples(self) -> dict:
        if not isinstance(self.exo_source, io.BytesIO):
            return {
                "type": "Csv" if self.csvSupport() else "web" if self.webSupport() else None,
                "data":  Generateur(get_abs_path_from_relative_to_root(self.exo_source), 3, "csv" if self.csvSupport() else "web" if self.pdfSupport() else None, True) if self.csvSupport() == True or self.webSupport() == True else None
            }
        return {}
    
    class PydanticMeta:
        computed = ["pdfSupport", "csvSupport", "webSupport", 'examples']
        exclude = ["exo_source", 'id', "exercices"]