from itertools import groupby

from pylons import request, response, session, tmpl_context as c, url

from pylons.controllers.util import abort, redirect

from pylons.i18n.translation import _

from rhodecode.lib.compat import json

from rhodecode.lib.base import BaseRepoController, render

from rhodecode.lib.auth import LoginRequired, HasRepoPermissionAnyDecorator,\

    NotAnonymous

from rhodecode.lib import helpers as h

from rhodecode.lib import diffs

from rhodecode.lib.utils import action_logger, jsonify

from rhodecode.lib.vcs.exceptions import EmptyRepositoryError

from rhodecode.lib.vcs.backends.base import EmptyChangeset

from rhodecode.lib.diffs import LimitedDiffContainer

from rhodecode.model.db import User, PullRequest, ChangesetStatus,\

    ChangesetComment

from rhodecode.model.pull_request import PullRequestModel

from rhodecode.model.meta import Session

from rhodecode.model.repo import RepoModel

from rhodecode.model.comment import ChangesetCommentsModel

from rhodecode.model.changeset_status import ChangesetStatusModel

from rhodecode.model.forms import PullRequestForm

log = logging.getLogger(__name__)

class PullrequestsController(BaseRepoController):

    @LoginRequired()

    @HasRepoPermissionAnyDecorator('repository.read', 'repository.write',

                                   'repository.admin')

    def __before__(self):

        super(PullrequestsController, self).__before__()

        repo_model = RepoModel()

        c.users_array = repo_model.get_users_js()

        c.users_groups_array = repo_model.get_users_groups_js()

        """return a structure with repo's interesting changesets, suitable for

        the selectors in pullrequest.html"""

        branches = [('branch:%s:%s' % (k, v), k)

                    for k, v in repo.branches.iteritems()]

        bookmarks = [('book:%s:%s' % (k, v), k)

                     for k, v in repo.bookmarks.iteritems()]

        tags = [('tag:%s:%s' % (k, v), k)

                for k, v in repo.tags.iteritems()

                if k != 'tip']

        tip = repo.tags['tip']

        colontip = ':' + tip

        tips = [x[1] for x in branches + bookmarks + tags

                if x[0].endswith(colontip)]

        selected = 'tag:tip:%s' % tip

        if rev:

            selected = 'rev:%s:%s' % (rev, rev)

        return [(special, _("Special")),

                (bookmarks, _("Bookmarks")),

                (branches, _("Branches")),

                (tags, _("Tags")),

                ], selected

    def _get_is_allowed_change_status(self, pull_request):

        owner = self.rhodecode_user.user_id == pull_request.user_id

        reviewer = self.rhodecode_user.user_id in [x.user_id for x in

                                                   pull_request.reviewers]

        return (self.rhodecode_user.admin or owner or reviewer)

    def show_all(self, repo_name):

        c.pull_requests = PullRequestModel().get_all(repo_name)

        c.repo_name = repo_name

        return render('/pullrequests/pullrequest_show_all.html')

    @NotAnonymous()

    def index(self):

        org_repo = c.rhodecode_db_repo

        if org_repo.scm_instance.alias != 'hg':

            log.error('Review not available for GIT REPOS')

        try:

            org_repo.scm_instance.get_changeset()

        except EmptyRepositoryError, e:

            h.flash(h.literal(_('There are no changesets yet')),

                    category='warning')

            redirect(url('summary_home', repo_name=org_repo.repo_name))

        org_rev = request.GET.get('rev_end')

        # rev_start is not directly useful - its parent could however be used

        # as default for other and thus give a simple compare view

        #other_rev = request.POST.get('rev_start')

        other_repos_info = {}

        c.org_repos = []

        c.org_repos.append((org_repo.repo_name, org_repo.repo_name))

        c.default_org_repo = org_repo.repo_name

        c.org_refs, c.default_org_ref = self._get_repo_refs(org_repo.scm_instance, org_rev)

        c.other_repos = []

        # add org repo to other so we can open pull request against itself

        c.other_repos.extend(c.org_repos)

        c.default_other_repo = org_repo.repo_name

        usr_data = lambda usr: dict(user_id=usr.user_id,

                                    username=usr.username,

                                    firstname=usr.firstname,

                                    lastname=usr.lastname,

                                    gravatar_link=h.gravatar_url(usr.email, 14))

        other_repos_info[org_repo.repo_name] = {

            'user': usr_data(org_repo.user),

            'description': org_repo.description,

            'revs': h.select('other_ref', c.default_other_ref,

                             c.default_other_refs, class_='refs')

        }

        # gather forks and add to this list ... even though it is rare to

        # request forks to pull their parent

        for fork in org_repo.forks:

            c.other_repos.append((fork.repo_name, fork.repo_name))

            refs, default_ref = self._get_repo_refs(fork.scm_instance)

            other_repos_info[fork.repo_name] = {

                'user': usr_data(fork.user),

                'description': fork.description,

                'revs': h.select('other_ref', default_ref, refs, class_='refs')

            }

        # add parents of this fork also, but only if it's not empty