# -*- coding: utf-8 -*-

"""

    rhodecode.controllers.pullrequests

    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

    pull requests controller for rhodecode for initializing pull requests

    :created_on: May 7, 2012

    :author: marcink

    :copyright: (C) 2010-2012 Marcin Kuzminski <marcin@python-works.com>

    :license: GPLv3, see COPYING for more details.

"""

# This program is free software: you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

#

# You should have received a copy of the GNU General Public License

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import logging

import traceback

import formencode

from webob.exc import HTTPNotFound, HTTPForbidden

from collections import defaultdict

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

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

        """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

        special = [(selected, 'tip (%s)' % ', '.join(tips))]

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

            raise HTTPNotFound

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

        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)

        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

        c.default_other_refs, c.default_other_ref = self._get_repo_refs(org_repo.scm_instance)

        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

        if org_repo.parent and org_repo.parent.scm_instance.revisions:

            c.default_other_repo = org_repo.parent.repo_name

            c.default_other_refs, c.default_other_ref = self._get_repo_refs(org_repo.parent.scm_instance)

            c.other_repos.append((org_repo.parent.repo_name, org_repo.parent.repo_name))

            other_repos_info[org_repo.parent.repo_name] = {

                'user': usr_data(org_repo.parent.user),

                'description': org_repo.parent.description,

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

                                 c.default_other_refs, class_='refs')

            }

        c.other_repos_info = json.dumps(other_repos_info)

        # other repo owner

        c.review_members = []

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

    @NotAnonymous()

    def create(self, repo_name):

        repo = RepoModel()._get_repo(repo_name)

        try:

            _form = PullRequestForm(repo.repo_id)().to_python(request.POST)

        except formencode.Invalid, errors:

            log.error(traceback.format_exc())

            if errors.error_dict.get('revisions'):

                msg = 'Revisions: %s' % errors.error_dict['revisions']

            elif errors.error_dict.get('pullrequest_title'):

                msg = _('Pull request requires a title with min. 3 chars')

                                               <div class="comments-cnt">${len(c.comments[cs.raw_id])}</div>

                                               <img src="${h.url('/images/icons/comments.png')}">

                                              </a>

                                            </div>

                                        %endif

                                        </div>

                                        <div class="changeset-status-container">

                                            %if c.statuses.get(cs.raw_id):

                                              <div title="${_('Changeset status')}" class="changeset-status-lbl">${c.statuses.get(cs.raw_id)[1]}</div>

                                              <div class="changeset-status-ico">

                                              %if c.statuses.get(cs.raw_id)[2]:

                                                <a class="tooltip" title="${_('Click to open associated pull request #%s' % c.statuses.get(cs.raw_id)[2])}" href="${h.url('pullrequest_show',repo_name=c.statuses.get(cs.raw_id)[3],pull_request_id=c.statuses.get(cs.raw_id)[2])}"><img src="${h.url('/images/icons/flag_status_%s.png' % c.statuses.get(cs.raw_id)[0])}" /></a>

                                              %else:

                                                <img src="${h.url('/images/icons/flag_status_%s.png' % c.statuses.get(cs.raw_id)[0])}" />

                                              %endif

                                              </div>

                                            %endif

                                        </div>

                                    </div>

                                   %if cs.parents:

                                    %for p_cs in reversed(cs.parents):

                                        <div class="parent">${_('Parent')}

                                            <span class="changeset_id">${p_cs.revision}:<span class="changeset_hash">${h.link_to(h.short_id(p_cs.raw_id),

                                            h.url('changeset_home',repo_name=c.repo_name,revision=p_cs.raw_id),title=p_cs.message)}</span></span>

                                        </div>

                                    %endfor

                                   %else:

                                        <div class="parent">${_('No parents')}</div>

                                   %endif

                                <span class="logtags">

                                    %if len(cs.parents)>1:

                                    <span class="merge">${_('merge')}</span>

                                    %endif

                                    %if cs.branch:

                                    <span class="branchtag" title="${'%s %s' % (_('branch'),cs.branch)}">

                                       ${h.link_to(h.shorter(cs.branch),h.url('files_home',repo_name=c.repo_name,revision=cs.raw_id))}

                                    </span>

                                    %endif

                                    %if h.is_hg(c.rhodecode_repo):

                                      %for book in cs.bookmarks:

                                      <span class="bookbook" title="${'%s %s' % (_('bookmark'),book)}">

                                         ${h.link_to(h.shorter(book),h.url('files_home',repo_name=c.repo_name,revision=cs.raw_id))}

                                      </span>

                                      %endfor

                                    %endif

                                    %for tag in cs.tags:

                                        <span class="tagtag"  title="${'%s %s' % (_('tag'),tag)}">

                                        ${h.link_to(h.shorter(tag),h.url('files_home',repo_name=c.repo_name,revision=cs.raw_id))}</span>

                                    %endfor

                                </span>

                        </div>

                    </div>

                %endfor

                <div class="pagination-wh pagination-left">

                    ${c.pagination.pager('$link_previous ~2~ $link_next')}

                </div>

                </div>

            </div>

            <script type="text/javascript" src="${h.url('/js/graph.js')}"></script>

            <script type="text/javascript">

                YAHOO.util.Event.onDOMReady(function(){

                    //Monitor range checkboxes and build a link to changesets

                    //ranges

                    var checkboxes = YUD.getElementsByClassName('changeset_range');

                    var url_tmpl = "${h.url('changeset_home',repo_name=c.repo_name,revision='__REVRANGE__')}";

                    var pr_tmpl = "${h.url('pullrequest_home',repo_name=c.repo_name)}";

                    YUE.on(checkboxes,'click',function(e){

                        var clicked_cb = e.currentTarget;

                        var checked_checkboxes = [];

                        for (pos in checkboxes){

                            if(checkboxes[pos].checked){

                                checked_checkboxes.push(checkboxes[pos]);

                            }

                        }

                        if(YUD.get('open_new_pr')){

                          if(checked_checkboxes.length>0){

                            // modify open pull request to show we have selected cs

                            YUD.get('open_new_pr').innerHTML = _TM['Open new pull request for selected changesets'];

                          }else{

                            YUD.get('open_new_pr').innerHTML = _TM['Open new pull request'];

                          }

                        }

                        if(checked_checkboxes.length>0){

                            var rev_end = checked_checkboxes[0].name;

                            var rev_start = checked_checkboxes[checked_checkboxes.length-1].name;

                            var url = url_tmpl.replace('__REVRANGE__',

                                    rev_start+'...'+rev_end);

                            var link = (rev_start == rev_end)

                              	? _TM['Show selected change __S']

                            	: _TM['Show selected changes __S -> __E'];

                            link = link.replace('__S',rev_start.substr(0,6));

                            link = link.replace('__E',rev_end.substr(0,6));

                            YUD.get('rev_range_container').href = url;

                            YUD.get('rev_range_container').innerHTML = link;

                            YUD.setStyle('rev_range_container','display','');

                            YUD.setStyle('rev_range_clear','display','');

                            YUD.get('open_new_pr').href = pr_tmpl + '?rev_start={0}&rev_end={1}'.format(rev_start,rev_end);

                        }

                        else{

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

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

                        }

                    });

                    YUE.on('rev_range_clear','click',function(e){

                        for (var i=0; i<checkboxes.length; i++){

                            var cb = checkboxes[i];

                            cb.checked = false;

                        }

                        YUE.preventDefault(e);

                    })

                    var msgs = YUQ('.message');

                    // get first element height

                    var el = YUQ('#graph_content .container')[0];

                    var row_h = el.clientHeight;

                    for(var i=0;i<msgs.length;i++){

                        var m = msgs[i];

                        var h = m.clientHeight;

                        var pad = YUD.getStyle(m,'padding');

                        if(h > row_h){

                            var offset = row_h - (h+12);

                            YUD.setStyle(m.nextElementSibling,'display','block');

                            YUD.setStyle(m.nextElementSibling,'margin-top',offset+'px');

                        };

                    }

                    YUE.on(YUQ('.expand'),'click',function(e){

                        var elem = e.currentTarget.parentNode.parentNode;

                        YUD.setStyle(e.currentTarget,'display','none');

                        YUD.setStyle(elem,'height','auto');

                        //redraw the graph, line_count and jsdata are global vars

                        set_canvas(100);

                        var r = new BranchRenderer();

                        r.render(jsdata,100,line_count);

                    })

                    // Fetch changeset details

                    YUE.on(YUD.getElementsByClassName('changed_total'),'click',function(e){

                        var id = e.currentTarget.id;

                        var url = "${h.url('changelog_details',repo_name=c.repo_name,cs='__CS__')}";

                        var url = url.replace('__CS__',id.replace('changed_total_',''));

                        ypjax(url,id,function(){tooltip_activate()});

                    });

                    // change branch filter

                    YUE.on(YUD.get('branch_filter'),'change',function(e){

                        var selected_branch = e.currentTarget.options[e.currentTarget.selectedIndex].value;

                        var url_main = "${h.url('changelog_home',repo_name=c.repo_name)}";

                        var url = "${h.url('changelog_home',repo_name=c.repo_name,branch='__BRANCH__')}";

                        var url = url.replace('__BRANCH__',selected_branch);

                        if(selected_branch != ''){

                            window.location = url;

                        }else{

                            window.location = url_main;

                        }

                    });

                    function set_canvas(width) {

                        var c = document.getElementById('graph_nodes');

                        var t = document.getElementById('graph_content');

                        canvas = document.getElementById('graph_canvas');

                        var div_h = t.clientHeight;

                        c.style.height=div_h+'px';

                        canvas.setAttribute('height',div_h);

                        c.style.height=width+'px';

                        canvas.setAttribute('width',width);

                    };

                    var heads = 1;

                    var line_count = 0;

                    var jsdata = ${c.jsdata|n};

                    for (var i=0;i<jsdata.length;i++) {

                        var in_l = jsdata[i][2];

                        for (var j in in_l) {

                            var m = in_l[j][1];

                            if (m > line_count)

                                line_count = m;

                        }

                    }

                    set_canvas(100);