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.utils2 import str2bool

log = logging.getLogger(__name__)

class CompareController(BaseRepoController):

    @LoginRequired()

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

                                   'repository.admin')

    def __before__(self):

        super(CompareController, self).__before__()

            h.flash(str(e), category='warning')

            if not partial:

                redirect(h.url('summary_home', repo_name=repo.repo_name))

            raise HTTPBadRequest()

    def index(self, org_ref_type, org_ref, other_ref_type, other_ref):

        org_repo = c.rhodecode_db_repo.repo_name

        org_ref = (org_ref_type, org_ref)

        other_ref = (other_ref_type, other_ref)

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

        bundle_compare = str2bool(request.GET.get('bundle', True))

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

              org_ref_type=other_ref[0], org_ref=other_ref[1],

              other_ref_type=org_ref[0], other_ref=org_ref[1],

              repo=org_repo, as_form=request.GET.get('as_form'),

              bundle=bundle_compare)

        c.org_repo = org_repo = Repository.get_by_repo_name(org_repo)

        c.other_repo = other_repo = Repository.get_by_repo_name(other_repo)

        if c.org_repo is None or c.other_repo is None:

            log.error('Could not found repo %s or %s' % (org_repo, other_repo))

        if not bundle_compare and c.cs_ranges:

            # case we want a simple diff without incoming changesets, just

            # for review purposes. Make the diff on the forked repo, with

            # revision that is common ancestor

            other_ref = ('rev', c.cs_ranges[-1].parents[0].raw_id)

            other_repo = org_repo

        c.org_ref = org_ref[1]

        c.other_ref = other_ref[1]

        _diff = diffs.differ(other_repo, other_ref, org_repo, org_ref,

                             discovery_data, bundle_compare=bundle_compare)

        _parsed = diff_processor.prepare()

        c.files = []

        c.changes = {}

        for f in _parsed:

            fid = h.FID('', f['filename'])

            c.files.append([fid, f['operation'], f['filename'], f['stats']])

            diff = diff_processor.as_html(enable_comments=False, parsed_lines=[f])

            c.changes[fid] = [f['operation'], f['filename'], diff]

        return render('compare/compare_diff.html')

        <td><div class="gravatar"><img alt="gravatar" src="${h.gravatar_url(h.email_or_none(cs.author),14)}"/></div></td>

        <td>

          %if cs.raw_id in c.statuses:

            <div title="${c.statuses[cs.raw_id][1]}" class="changeset-status-ico"><img src="${h.url('/images/icons/flag_status_%s.png' % c.statuses[cs.raw_id][0])}" /></div>

          %endif

        </td>

        <td>${h.link_to('r%s:%s' % (cs.revision,h.short_id(cs.raw_id)),h.url('changeset_home',repo_name=c.target_repo,revision=cs.raw_id))}

        %if c.as_form:

          ${h.hidden('revisions',cs.raw_id)}

        %endif

        </td>

        <td><div class="author">${h.person(cs.author)}</div></td>

        </tr>

    %endfor

  %endif

  </table>

</div>

            <%include file="compare_cs.html" />

            ## FILES

            <div style="font-size:1.1em;font-weight: bold;clear:both;padding-top:10px">${_('Files affected')}</div>

            <div class="cs_files">

              %for fid, change, f, stat in c.files:

                  <div class="cs_${change}">

                    <div class="node">${h.link_to(h.safe_unicode(f),h.url.current(anchor=fid))}</div>

                    <div class="changes">${h.fancy_file_stats(stat)}</div>

                  </div>

              %endfor

            </div>

        </div>

    </div>

    ## diff block

    <%namespace name="diff_block" file="/changeset/diff_block.html"/>

    %for fid, change, f, stat in c.files:

      ${diff_block.diff_block_simple([c.changes[fid]])}

    %endfor

     <script type="text/javascript">

      YUE.onDOMReady(function(){

          YUE.on(YUQ('.diff-menu-activate'),'click',function(e){

              var act = e.currentTarget.nextElementSibling;

              if(YUD.hasClass(act,'active')){

                  YUD.removeClass(act,'active');

                  YUD.setStyle(act,'display','none');

              }else{

                  YUD.addClass(act,'active');