import os

from ckeditor.fields import RichTextField
from django.db import models

from infrastructure.files.file_storage import OverwriteStorage
from infrastructure.persistence.models.user.user import User


class Idea(models.Model):
    user = models.ForeignKey(
        User, verbose_name="Пользователь", related_name="ideas", on_delete=models.SET_NULL, null=True
    )

    CATEGORIES = [
        ("errors", "Ошибка в работе"),
        ("correction", "Исправление"),
        ("modernization", "Улучшение"),
        ("new_feature", "Новая возможность"),
    ]

    category = models.CharField(max_length=100, verbose_name="категория", choices=CATEGORIES)

    STATUSES = [
        ("new", "Новое"),
        ("in_pregress", "В работе"),
        ("planned", "Запланировано"),
        ("done", "Реализовано"),
        ("reject", "Отклонено"),
        ("repeat", "Повтор"),
    ]

    status = models.CharField(max_length=100, verbose_name="статус", choices=STATUSES, default="new")

    finishe_date = models.DateField(null=True, verbose_name="Срок")

    created_at = models.DateField(auto_now_add=True)

    title = models.CharField(max_length=60, verbose_name="Тема")
    description = models.TextField(max_length=1000, verbose_name="Описание")

    admin_answer = RichTextField(max_length=1000, verbose_name="Ответ", null=True, blank=True)

    class Meta:
        app_label = "user"
        verbose_name = "Предложение"
        verbose_name_plural = "Предложения"

    def __str__(self):
        return f"{str(self.user)} - {self.title}"


class Like(models.Model):
    user = models.ForeignKey(User, on_delete=models.SET_NULL, null=True)
    idea = models.ForeignKey(Idea, related_name="likes", on_delete=models.CASCADE)

    class Meta:
        app_label = "user"


def get_upload_to_idea_screen(instance, filename):
    return os.path.join("images", str(instance.idea_id), filename)


class IdeaScreen(models.Model):
    screen = models.ImageField(upload_to=get_upload_to_idea_screen, storage=OverwriteStorage())
    idea = models.ForeignKey(Idea, on_delete=models.CASCADE, related_name="screens")

    class Meta:
        app_label = "user"