from pylons.i18n.translation import _

from rhodecode.lib.vcs.exceptions import EmptyRepositoryError, RepositoryError

from rhodecode.lib import helpers as h

from rhodecode.lib.base import BaseRepoController, render

from rhodecode.lib.auth import LoginRequired, HasRepoPermissionAnyDecorator

from rhodecode.lib import diffs

from rhodecode.model.db import Repository

from rhodecode.model.pull_request import PullRequestModel

from webob.exc import HTTPBadRequest

from rhodecode.lib.diffs import LimitedDiffContainer

log = logging.getLogger(__name__)

class CompareController(BaseRepoController):

    @LoginRequired()

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

                                   'repository.admin')

    def __before__(self):

        super(CompareController, self).__before__()

        c.org_ref_type = org_ref[0]

        c.other_ref_type = other_ref[0]

        if rev_start and rev_end:

            # swap revs with cherry picked ones, save them for display

            #org_ref = ('rev', rev_start)

            #other_ref = ('rev', rev_end)

            c.org_ref = rev_start[:12]

            c.other_ref = rev_end[:12]

            # get parent of

            # rev start to include it in the diff

            _cs = other_repo.scm_instance.get_changeset(rev_start)

            org_ref = ('rev', rev_start)

            other_ref = ('rev', rev_end)

            #if we cherry pick it's not remote, make the other_repo org_repo

            org_repo = other_repo

        c.cs_ranges, ancestor = PullRequestModel().get_compare_data(

            org_repo, org_ref, other_repo, other_ref)

        c.statuses = c.rhodecode_db_repo.statuses([x.raw_id for x in

                                                   c.cs_ranges])

        # defines that we need hidden inputs with changesets

        c.as_form = request.GET.get('as_form', False)