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

from mercurial import scmutil

from rhodecode.lib.utils2 import safe_int

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()

    def _get_repo_refs(self, repo, rev=None, branch_rev=None):

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

        the selectors in pullrequest.html"""

        # list named branches that has been merged to this named branch - it should probably merge back

        peers = []

        if branch_rev:

            # not restricting to merge() would also get branch point and be better

            # (especially because it would get the branch point) ... but is currently too expensive

            revs = ["sort(parents(branch(id('%s')) and merge()) - branch(id('%s')))" %

                    (branch_rev, branch_rev)]

            otherbranches = {}

            for i in scmutil.revrange(repo._repo, revs):

                cs = repo.get_changeset(i)

                otherbranches[cs.branch] = cs.raw_id

            for branch, node in otherbranches.iteritems():

                selected = 'branch:%s:%s' % (branch, node)

                peers.append((selected, branch))

        selected = None

        branches = []

        for branch, branchrev in repo.branches.iteritems():

            n = 'branch:%s:%s' % (branch, branchrev)

            branches.append((n, branch))

            if rev == branchrev:

                selected = n

        bookmarks = []

        for bookmark, bookmarkrev in repo.bookmarks.iteritems():

            n = 'book:%s:%s' % (bookmark, bookmarkrev)

            bookmarks.append((n, bookmark))

            if rev == bookmarkrev: