# Copyright (C) 2016 The Sipwise Team - http://sipwise.com

# This program is free software: you can redistribute it and/or modify it
# under the terms of the GNU 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 General Public License for
# more details.

# You should have received a copy of the GNU General Public License along
# with this program.  If not, see <http://www.gnu.org/licenses/>.
import logging
import re
import json
from django.db import models
from django_extensions.db.fields.json import JSONField
from django_extensions.db.fields import ModificationDateTimeField

logger = logging.getLogger(__name__)


class Project(models.Model):
    name = models.CharField(max_length=50, unique=True, null=False)
    json_tags = JSONField(null=True)
    json_branches = JSONField(null=True)
    modified = ModificationDateTimeField(null=True)

    @classmethod
    def _filter_values(cls, values, val_ok_filter, regex=None):
        res = set()

        for value in values:
            match = re.search(val_ok_filter, value["ref"])
            if match:
                val_ok = match.group(1)
                if regex is not None:
                    if re.search(regex, val_ok):
                        res.add(val_ok)
                else:
                    res.add(val_ok)
        return sorted(res, reverse=True)

    @classmethod
    def _get_filtered_json(cls, text):
        """gerrit responds with malformed json
           https://gerrit-review.googlesource.com/Documentation/rest-api.html#output
        """
        return json.loads(text[5:])

    def __str__(self):
        return self.name

    @property
    def tags(self):
        return Project._filter_values(self.json_tags, '^refs/tags/(.+)$')

    @tags.setter
    def tags(self, value):
        self.json_tags = Project._get_filtered_json(value)

    @property
    def branches(self):
        return Project._filter_values(self.json_branches, '^refs/heads/(.+)$')

    @branches.setter
    def branches(self, value):
        self.json_branches = Project._get_filtered_json(value)

    def filter_tags(self, regex):
        return Project._filter_values(self.json_tags,
                                      '^refs/tags/(.+)$', regex)

    def filter_branches(self, regex):
        return Project._filter_values(self.json_branches,
                                      '^refs/heads/(.+)$', regex)

    def branches_mrXX(self):
        return self.filter_branches(r'^mr[0-9]+\.[0-9]+$')

    def branches_mrXXX(self):
        return self.filter_branches(r'^mr[0-9]+\.[0-9]+\.[0-9]+$')