from kallithea.lib.vcs.utils.hgcompat import unionrepo

from kallithea.lib import helpers as h

from kallithea.lib.base import BaseRepoController, render

from kallithea.lib.auth import LoginRequired, HasRepoPermissionAnyDecorator

from kallithea.lib import diffs

from kallithea.model.db import Repository

from kallithea.lib.diffs import LimitedDiffContainer

from kallithea.controllers.changeset import anchor_url, _ignorews_url,\

    _context_url, get_line_ctx, get_ignore_ws

from kallithea.lib.graphmod import graph_data

from kallithea.lib.compat import json

log = logging.getLogger(__name__)

class CompareController(BaseRepoController):

    def __before__(self):

        super(CompareController, self).__before__()

    @staticmethod

    def _get_changesets(alias, org_repo, org_rev, other_repo, other_rev):

        """

        Returns lists of changesets that can be merged from org_repo@org_rev

        to other_repo@other_rev

        ... and the other way

        ... and the ancestor that would be used for merge

        :param org_repo: repo object, that is most likely the orginal repo we forked from

        :param org_rev: the revision we want our compare to be made

        :param other_repo: repo object, mostl likely the fork of org_repo. It hass

            all changesets that we need to obtain

        :param other_rev: revision we want out compare to be made on other_repo

        """

        ancestor = None

        if org_rev == other_rev:

            org_changesets = []

            other_changesets = []

            ancestor = org_rev

        elif alias == 'hg':

            #case two independent repos

            if org_repo != other_repo:

                hgrepo = unionrepo.unionrepository(other_repo.baseui,

                                                   other_repo.path,

                                                   org_repo.path)

                # all ancestors of other_rev will be in other_repo and

                # rev numbers from hgrepo can be used in other_repo - org_rev ancestors cannot

            #no remote compare do it on the same repository

            else:

                hgrepo = other_repo._repo

            ancestors = hgrepo.revs("ancestor(id(%s), id(%s))", org_rev, other_rev)

            if ancestors:

                # pick arbitrary ancestor - but there is usually only one

                ancestor = hgrepo[ancestors[0]].hex()

            other_revs = hgrepo.revs("ancestors(id(%s)) and not ancestors(id(%s)) and not id(%s)",

                                     other_rev, org_rev, org_rev)

            other_changesets = [other_repo.get_changeset(rev) for rev in other_revs]

            org_revs = hgrepo.revs("ancestors(id(%s)) and not ancestors(id(%s)) and not id(%s)",

                                   org_rev, other_rev, other_rev)

            org_changesets = [org_repo.get_changeset(hgrepo[rev].hex()) for rev in org_revs]

        elif alias == 'git':

            if org_repo != other_repo:

                from dulwich.repo import Repo

                from dulwich.client import SubprocessGitClient

                gitrepo = Repo(org_repo.path)

                SubprocessGitClient(thin_packs=False).fetch(other_repo.path, gitrepo)

                gitrepo_remote = Repo(other_repo.path)

                SubprocessGitClient(thin_packs=False).fetch(org_repo.path, gitrepo_remote)

                revs = []

                for x in gitrepo_remote.get_walker(include=[other_rev],

                                                   exclude=[org_rev]):

                    revs.append(x.commit.id)

                other_changesets = [other_repo.get_changeset(rev) for rev in reversed(revs)]

                if other_changesets:

                    ancestor = other_changesets[0].parents[0].raw_id

                else:

                    # no changesets from other repo, ancestor is the other_rev

                    ancestor = other_rev

            else:

                so, se = org_repo.run_git_command(

                    'log --reverse --pretty="format: %%H" -s %s..%s'

                        % (org_rev, other_rev)

                )

                other_changesets = [org_repo.get_changeset(cs)

                              for cs in re.findall(r'[0-9a-fA-F]{40}', so)]

            org_changesets = []

        else:

            raise Exception('Bad alias only git and hg is allowed')

        return other_changesets, org_changesets, ancestor

    @LoginRequired()

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

                                   'repository.admin')

    def index(self, repo_name):

        c.compare_home = True

        org_repo = c.db_repo.repo_name

        other_repo = request.GET.get('other_repo', org_repo)

        c.org_repo = Repository.get_by_repo_name(org_repo)

        c.other_repo = Repository.get_by_repo_name(other_repo)

        c.org_ref_name = c.other_ref_name = _('Select changeset')

        return render('compare/compare_diff.html')

    @LoginRequired()

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

                                   'repository.admin')

    def compare(self, repo_name, org_ref_type, org_ref_name, other_ref_type, other_ref_name):

        org_repo = c.db_repo.repo_name

        other_repo = request.GET.get('other_repo', org_repo)

        # If merge is True:

        #   Show what org would get if merged with other:

        #   List changesets that are ancestors of other but not of org.

        #   New changesets in org is thus ignored.

        #   Diff will be from common ancestor, and merges of org to other will thus be ignored.

        # If merge is False:

        #   Make a raw diff from org to other, no matter if related or not.

        #   Changesets in one and not in the other will be ignored

        merge = bool(request.GET.get('merge'))

        # fulldiff disables cut_off_limit

        c.fulldiff = request.GET.get('fulldiff')

        # partial uses compare_cs.html template directly

        partial = request.environ.get('HTTP_X_PARTIAL_XHR')

        # as_form puts hidden input field with changeset revisions

        c.as_form = partial and request.GET.get('as_form')

        # swap url for compare_diff page - never partial and never as_form

        c.swap_url = h.url('compare_url',

            repo_name=other_repo,

            org_ref_type=other_ref_type, org_ref_name=other_ref_name,

            other_repo=org_repo,

            other_ref_type=org_ref_type, other_ref_name=org_ref_name,

            merge=merge or '')

        # set callbacks for generating markup for icons

        c.ignorews_url = _ignorews_url

        c.context_url = _context_url

        ignore_whitespace = request.GET.get('ignorews') == '1'

        line_context = request.GET.get('context', 3)

        org_repo = Repository.get_by_repo_name(org_repo)

        other_repo = Repository.get_by_repo_name(other_repo)

        if org_repo is None:

            msg = 'Could not find org repo %s' % org_repo

            log.error(msg)

            h.flash(msg, category='error')

            return redirect(url('compare_home', repo_name=c.repo_name))

        if other_repo is None:

            msg = 'Could not find other repo %s' % other_repo

            log.error(msg)

            h.flash(msg, category='error')

            return redirect(url('compare_home', repo_name=c.repo_name))

        if org_repo.scm_instance.alias != other_repo.scm_instance.alias:

            msg = 'compare of two different kind of remote repos not available'

            log.error(msg)

            h.flash(msg, category='error')

            return redirect(url('compare_home', repo_name=c.repo_name))

        c.org_rev = self._get_ref_rev(org_repo, org_ref_type, org_ref_name)

        c.other_rev = self._get_ref_rev(other_repo, other_ref_type, other_ref_name)

        c.compare_home = False

        c.org_repo = org_repo

        c.other_repo = other_repo

        c.org_ref_name = org_ref_name

        c.other_ref_name = other_ref_name

        c.org_ref_type = org_ref_type

        c.other_ref_type = other_ref_type

        c.cs_repo = other_repo

        c.cs_ranges, c.cs_ranges_org, c.ancestor = self._get_changesets(

            org_repo.scm_instance.alias, org_repo.scm_instance, c.org_rev,

            other_repo.scm_instance, c.other_rev)

        raw_ids = [x.raw_id for x in c.cs_ranges]

        c.cs_comments = other_repo.get_comments(raw_ids)

        c.statuses = other_repo.statuses(raw_ids)

        revs = [ctx.revision for ctx in reversed(c.cs_ranges)]

        c.jsdata = json.dumps(graph_data(c.other_repo.scm_instance, revs))

        response.mustcontain('''<a href="/%s/changeset/5c9ff4f6d7508db0e72b1d2991c357d0d8e07af2">r116:5c9ff4f6d750</a>''' % GIT_REPO)

        response.mustcontain('''<a href="/%s/changeset/b7187fa2b8c1d773ec35e9dee12f01f74808c879">r117:b7187fa2b8c1</a>''' % GIT_REPO)

        response.mustcontain('''<a href="/%s/changeset/5f3b74262014a8de2dc7dade1152de9fd0c8efef">r118:5f3b74262014</a>''' % GIT_REPO)

        response.mustcontain('''<a href="/%s/changeset/17438a11f72b93f56d0e08e7d1fa79a378578a82">r119:17438a11f72b</a>''' % GIT_REPO)

        response.mustcontain('''<a href="/%s/changeset/5a3a8fb005554692b16e21dee62bf02667d8dc3e">r120:5a3a8fb00555</a>''' % GIT_REPO)

        response.mustcontain('11 files changed with 94 insertions and 64 deletions')

        #files

        response.mustcontain('''<a href="#C--1c5cf9e91c12">docs/api/utils/index.rst</a>''')

        response.mustcontain('''<a href="#C--e3305437df55">test_and_report.sh</a>''')

        response.mustcontain('''<a href="#C--c8e92ef85cd1">.hgignore</a>''')

        response.mustcontain('''<a href="#C--6e08b694d687">.hgtags</a>''')

        response.mustcontain('''<a href="#C--2c14b00f3393">docs/api/index.rst</a>''')

        response.mustcontain('''<a href="#C--430ccbc82bdf">vcs/__init__.py</a>''')

        response.mustcontain('''<a href="#C--9c390eb52cd6">vcs/backends/hg.py</a>''')

        response.mustcontain('''<a href="#C--ebb592c595c0">vcs/utils/__init__.py</a>''')

        response.mustcontain('''<a href="#C--7abc741b5052">vcs/utils/annotate.py</a>''')

        response.mustcontain('''<a href="#C--2ef0ef106c56">vcs/utils/diffs.py</a>''')

        response.mustcontain('''<a href="#C--3150cb87d4b7">vcs/utils/lazy.py</a>''')

    def test_index_branch_hg(self):

        self.log_user()

        response = self.app.get(url('compare_url',

                                    repo_name=HG_REPO,

                                    org_ref_type="branch",

                                    org_ref_name='default',

                                    other_ref_type="branch",

                                    other_ref_name='default',

                                    ))

        response.mustcontain('%s@default' % (HG_REPO))

        response.mustcontain('%s@default' % (HG_REPO))

        # branch are equal

        response.mustcontain('<span class="empty_data">No files</span>')

        response.mustcontain('<span class="empty_data">No changesets</span>')

    def test_index_branch_git(self):

        self.log_user()

        response = self.app.get(url('compare_url',

                                    repo_name=GIT_REPO,

                                    org_ref_type="branch",

                                    org_ref_name='master',

                                    other_ref_type="branch",

                                    other_ref_name='master',

                                    ))

        response.mustcontain('%s@master' % (GIT_REPO))

        response.mustcontain('%s@master' % (GIT_REPO))

        # branch are equal

        response.mustcontain('<span class="empty_data">No files</span>')

        response.mustcontain('<span class="empty_data">No changesets</span>')

    def test_compare_revisions_hg(self):

        self.log_user()

        rev1 = 'b986218ba1c9'

        rev2 = '3d8f361e72ab'

        response = self.app.get(url('compare_url',

                                    repo_name=HG_REPO,

                                    org_ref_type="rev",

                                    org_ref_name=rev1,

                                    other_ref_type="rev",

                                    other_ref_name=rev2,

                                    ))

        response.mustcontain('%s@%s' % (HG_REPO, rev1))

        response.mustcontain('%s@%s' % (HG_REPO, rev2))

        ## outgoing changesets between those revisions

        response.mustcontain("""<a href="/%s/changeset/3d8f361e72ab303da48d799ff1ac40d5ac37c67e">r1:%s</a>""" % (HG_REPO, rev2))

        response.mustcontain('1 file changed with 7 insertions and 0 deletions')

        ## files

        response.mustcontain("""<a href="#C--c8e92ef85cd1">.hgignore</a>""")

    def test_compare_revisions_git(self):

        self.log_user()

        rev1 = 'c1214f7e79e02fc37156ff215cd71275450cffc3'

        rev2 = '38b5fe81f109cb111f549bfe9bb6b267e10bc557'

        response = self.app.get(url('compare_url',

                                    repo_name=GIT_REPO,

                                    org_ref_type="rev",

                                    org_ref_name=rev1,

                                    other_ref_type="rev",

                                    other_ref_name=rev2,

                                    ))

        response.mustcontain('%s@%s' % (GIT_REPO, rev1))

        response.mustcontain('%s@%s' % (GIT_REPO, rev2))

        ## outgoing changesets between those revisions

        response.mustcontain("""<a href="/%s/changeset/38b5fe81f109cb111f549bfe9bb6b267e10bc557">r1:%s</a>""" % (GIT_REPO, rev2[:12]))

        response.mustcontain('1 file changed with 7 insertions and 0 deletions')

        ## files

        response.mustcontain("""<a href="#C--c8e92ef85cd1">.hgignore</a>""")