# Copyright (C) 2015 Andrey Antukh <niwi@niwi.be>
# Copyright (C) 2015 Jesús Espino <jespinog@gmail.com>
# Copyright (C) 2015 David Barragán <bameda@dbarragan.com>
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

from django.db import models
from django.utils.translation import ugettext_lazy as _
from django.utils import timezone

from django_pgjson.fields import JsonField

from taiga.projects.occ.mixins import OCCModelMixin


######################################################
#  Custom Attribute Models
#######################################################


class AbstractCustomAttribute(models.Model):
    FIELD_TYPES = (
        ("TEXT", _("Text")),
        ("MULTI", _("Multi-Line Text"))
    )
    name = models.CharField(null=False, blank=False, max_length=64, verbose_name=_("name"))
    description = models.TextField(null=False, blank=True, verbose_name=_("description"))
    field_type = models.CharField(null=False, blank=False, choices=FIELD_TYPES, max_length=5, default="TEXT", verbose_name=_("type"))
    order = models.IntegerField(null=False, blank=False, default=10000, verbose_name=_("order"))
    project = models.ForeignKey("projects.Project", null=False, blank=False, related_name="%(class)ss",
                                verbose_name=_("project"))

    created_date = models.DateTimeField(null=False, blank=False, default=timezone.now,
                                        verbose_name=_("created date"))
    modified_date = models.DateTimeField(null=False, blank=False,
                                         verbose_name=_("modified date"))
    _importing = None

    class Meta:
        abstract = True
        ordering = ["project", "order", "name"]
        unique_together = ("project", "name")

    def __str__(self):
        return self.name

    def save(self, *args, **kwargs):
        if not self._importing or not self.modified_date:
            self.modified_date = timezone.now()

        return super().save(*args, **kwargs)


class UserStoryCustomAttribute(AbstractCustomAttribute):
    class Meta(AbstractCustomAttribute.Meta):
        verbose_name = "user story custom attribute"
        verbose_name_plural = "user story custom attributes"


class TaskCustomAttribute(AbstractCustomAttribute):
    class Meta(AbstractCustomAttribute.Meta):
        verbose_name = "task custom attribute"
        verbose_name_plural = "task custom attributes"


class IssueCustomAttribute(AbstractCustomAttribute):
    class Meta(AbstractCustomAttribute.Meta):
        verbose_name = "issue custom attribute"
        verbose_name_plural = "issue custom attributes"


######################################################
#  Custom Attributes Values Models
#######################################################

class AbstractCustomAttributesValues(OCCModelMixin, models.Model):
    attributes_values = JsonField(null=False, blank=False, default={}, verbose_name=_("values"))

    class Meta:
        abstract = True
        ordering = ["id"]


class UserStoryCustomAttributesValues(AbstractCustomAttributesValues):
    user_story = models.OneToOneField("userstories.UserStory",
                                      null=False, blank=False, related_name="custom_attributes_values",
                                      verbose_name=_("user story"))

    class Meta(AbstractCustomAttributesValues.Meta):
        verbose_name = "user story ustom attributes values"
        verbose_name_plural = "user story custom attributes values"

    @property
    def project(self):
        # NOTE: This property simplifies checking permissions
        return self.user_story.project


class TaskCustomAttributesValues(AbstractCustomAttributesValues):
    task = models.OneToOneField("tasks.Task",
                                null=False, blank=False, related_name="custom_attributes_values",
                                verbose_name=_("task"))

    class Meta(AbstractCustomAttributesValues.Meta):
        verbose_name = "task ustom attributes values"
        verbose_name_plural = "task custom attributes values"

    @property
    def project(self):
        # NOTE: This property simplifies checking permissions
        return self.task.project


class IssueCustomAttributesValues(AbstractCustomAttributesValues):
    issue = models.OneToOneField("issues.Issue",
                                 null=False, blank=False, related_name="custom_attributes_values",
                                 verbose_name=_("issue"))

    class Meta(AbstractCustomAttributesValues.Meta):
        verbose_name = "issue ustom attributes values"
        verbose_name_plural = "issue custom attributes values"

    @property
    def project(self):
        # NOTE: This property simplifies checking permissions
        return self.issue.project