# -*- 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 pylons.decorators import jsonify

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

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

        hist_l = []

        branches_group = ([('branch:%s:%s' % (k, v), k) for

                         k, v in repo.branches.iteritems()], _("Branches"))

        bookmarks_group = ([('book:%s:%s' % (k, v), k) for

                         k, v in repo.bookmarks.iteritems()], _("Bookmarks"))

        tags_group = ([('tag:%s:%s' % (k, v), k) for

                         k, v in repo.tags.iteritems()], _("Tags"))

        hist_l.append(bookmarks_group)

        hist_l.append(branches_group)

        hist_l.append(tags_group)

        return hist_l

    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

        other_repos_info = {}

        c.org_refs = self._get_repo_refs(c.rhodecode_repo)

        c.org_repos = []

        c.other_repos = []

        c.org_repos.append((org_repo.repo_name, '%s/%s' % (

                                org_repo.user.username, c.repo_name))

                           )

        # add org repo to other so we can open pull request agains itself

        c.other_repos.extend(c.org_repos)

        c.default_revs = self._get_repo_refs(org_repo.scm_instance)

        #add orginal repo

        other_repos_info[org_repo.repo_name] = {

            'gravatar': h.gravatar_url(org_repo.user.email, 24),

            'description': org_repo.description,

            'revs': h.select('other_ref', '', c.default_revs, class_='refs')

        }

        #gather forks and add to this list

        for fork in org_repo.forks:

            c.other_repos.append((fork.repo_name, '%s/%s' % (

                                    fork.user.username, fork.repo_name))

                                 )

            other_repos_info[fork.repo_name] = {

                'gravatar': h.gravatar_url(fork.user.email, 24),

                'description': fork.description,

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

                                 self._get_repo_refs(fork.scm_instance),

                                 class_='refs')

            }

        #add parents of this fork also

        if org_repo.parent:

            c.default_pull_request = org_repo.parent.repo_name

            c.other_repos.append((org_repo.parent.repo_name, '%s/%s' % (

                                        org_repo.parent.user.username,

                                        org_repo.parent.repo_name))

                                     )

            other_repos_info[org_repo.parent.repo_name] = {

                'gravatar': h.gravatar_url(org_repo.parent.user.email, 24),

                'description': org_repo.parent.description,

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

                                 self._get_repo_refs(org_repo.parent.scm_instance),

                                 class_='refs')

            }

        c.other_repos_info = json.dumps(other_repos_info)

        c.review_members = [org_repo.user]

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

    @NotAnonymous()

    def create(self, repo_name):

        try:

            _form = PullRequestForm()().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')

            else:

                msg = _('error during creation of pull request')

            h.flash(msg, 'error')

            return redirect(url('pullrequest_home', repo_name=repo_name))

        org_repo = _form['org_repo']

        org_ref = _form['org_ref']

        other_repo = _form['other_repo']

        other_ref = _form['other_ref']

        revisions = _form['revisions']

        reviewers = _form['review_members']

        title = _form['pullrequest_title']

        description = _form['pullrequest_desc']

        try:

            pull_request = PullRequestModel().create(

                self.rhodecode_user.user_id, org_repo, org_ref, other_repo,

                other_ref, revisions, reviewers, title, description

            )

            Session().commit()

            h.flash(_('Successfully opened new pull request'),

                    category='success')

        except Exception:

            h.flash(_('Error occurred during sending pull request'),

                    category='error')

            log.error(traceback.format_exc())

            return redirect(url('pullrequest_home', repo_name=repo_name))

        return redirect(url('pullrequest_show', repo_name=other_repo,

                            pull_request_id=pull_request.pull_request_id))

    @NotAnonymous()

    @jsonify

    def update(self, repo_name, pull_request_id):

        pull_request = PullRequest.get_or_404(pull_request_id)

        if pull_request.is_closed():

            raise HTTPForbidden()

        #only owner or admin can update it

        owner = pull_request.author.user_id == c.rhodecode_user.user_id

        if h.HasPermissionAny('hg.admin', 'repository.admin')() or owner:

            reviewers_ids = map(int, filter(lambda v: v not in [None, ''],

                       request.POST.get('reviewers_ids', '').split(',')))

            PullRequestModel().update_reviewers(pull_request_id, reviewers_ids)

            Session.commit()

            return True

        raise HTTPForbidden()

    @NotAnonymous()

    @jsonify

    def delete(self, repo_name, pull_request_id):

        pull_request = PullRequest.get_or_404(pull_request_id)

        #only owner can delete it !

        if pull_request.author.user_id == c.rhodecode_user.user_id:

            PullRequestModel().delete(pull_request)

            Session().commit()

            h.flash(_('Successfully deleted pull request'),

                    category='success')

            return redirect(url('admin_settings_my_account'))

        raise HTTPForbidden()

    def _load_compare_data(self, pull_request, enable_comments=True):

        """

        Load context data needed for generating compare diff

        :param pull_request:

        :type pull_request:

        """

/**

RhodeCode JS Files

**/

if (typeof console == "undefined" || typeof console.log == "undefined"){

	console = { log: function() {} }

}

var str_repeat = function(i, m) {

	for (var o = []; m > 0; o[--m] = i);

	return o.join('');

};

/**

 * INJECT .format function into String

 * Usage: "My name is {0} {1}".format("Johny","Bravo")

 * Return "My name is Johny Bravo"

 * Inspired by https://gist.github.com/1049426

 */

String.prototype.format = function() {

	  function format() {

	    var str = this;

	    var len = arguments.length+1;

	    var safe = undefined;

	    var arg = undefined;

	    // For each {0} {1} {n...} replace with the argument in that position.  If 

	    // the argument is an object or an array it will be stringified to JSON.

	    for (var i=0; i < len; arg = arguments[i++]) {

	      safe = typeof arg === 'object' ? JSON.stringify(arg) : arg;

	      str = str.replace(RegExp('\\{'+(i-1)+'\\}', 'g'), safe);

	    }

	    return str;

	  }

	  // Save a reference of what may already exist under the property native.  

	  // Allows for doing something like: if("".format.native) { /* use native */ }

	  format.native = String.prototype.format;

	  // Replace the prototype property

	  return format;

}();

String.prototype.strip = function(char) {

	if(char === undefined){

	    char = '\\s';

	}

	return this.replace(new RegExp('^'+char+'+|'+char+'+$','g'), '');

}

String.prototype.lstrip = function(char) {

	if(char === undefined){

	    char = '\\s';

	}

	return this.replace(new RegExp('^'+char+'+'),'');

}

String.prototype.rstrip = function(char) {

	if(char === undefined){

	    char = '\\s';

	}

	return this.replace(new RegExp(''+char+'+$'),'');

}

if(!Array.prototype.indexOf) {

    Array.prototype.indexOf = function(needle) {

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

            if(this[i] === needle) {

                return i;

            }

        }

        return -1;

    };

}

// IE(CRAP) doesn't support previousElementSibling

var prevElementSibling = function( el ) {

    if( el.previousElementSibling ) {

        return el.previousElementSibling;

    } else {

        while( el = el.previousSibling ) {

            if( el.nodeType === 1 ) return el;

        }

    }

}

/**

 * SmartColorGenerator

 *

 *usage::

 *	var CG = new ColorGenerator();

 *  var col = CG.getColor(key); //returns array of RGB

 *  'rgb({0})'.format(col.join(',')

 * 

 * @returns {ColorGenerator}

 */

var ColorGenerator = function(){

	this.GOLDEN_RATIO = 0.618033988749895;

	this.CURRENT_RATIO = 0.22717784590367374 // this can be random

	this.HSV_1 = 0.75;//saturation

	this.HSV_2 = 0.95;

	this.color;

	this.cacheColorMap = {};

};

ColorGenerator.prototype = {

    getColor:function(key){

    	if(this.cacheColorMap[key] !== undefined){

    		return this.cacheColorMap[key];

    	}

    	else{

    		this.cacheColorMap[key] = this.generateColor();

    		return this.cacheColorMap[key];

    	}

    },

    _hsvToRgb:function(h,s,v){

        if (s == 0.0)

            return [v, v, v];

        i = parseInt(h * 6.0)

        f = (h * 6.0) - i

        p = v * (1.0 - s)

        q = v * (1.0 - s * f)

        t = v * (1.0 - s * (1.0 - f))

        i = i % 6

        if (i == 0) 

            return [v, t, p]

        if (i == 1) 

            return [q, v, p]

        if (i == 2) 

            return [p, v, t]

        if (i == 3)

            return [p, q, v]

        if (i == 4) 

            return [t, p, v]

        if (i == 5)

            return [v, p, q]            	

    },

    generateColor:function(){

        this.CURRENT_RATIO = this.CURRENT_RATIO+this.GOLDEN_RATIO;

        this.CURRENT_RATIO = this.CURRENT_RATIO %= 1;

        HSV_tuple = [this.CURRENT_RATIO, this.HSV_1, this.HSV_2]

        RGB_tuple = this._hsvToRgb(HSV_tuple[0],HSV_tuple[1],HSV_tuple[2]);

        function toRgb(v){

        	return ""+parseInt(v*256)

        }

        return [toRgb(RGB_tuple[0]),toRgb(RGB_tuple[1]),toRgb(RGB_tuple[2])];

    }

}

/**

 * GLOBAL YUI Shortcuts

 */

var YUC = YAHOO.util.Connect;

var YUD = YAHOO.util.Dom;

var YUE = YAHOO.util.Event;

var YUQ = YAHOO.util.Selector.query;

// defines if push state is enabled for this browser ?

var push_state_enabled = Boolean(

		window.history && window.history.pushState && window.history.replaceState

		&& !(   /* disable for versions of iOS before version 4.3 (8F190) */

				(/ Mobile\/([1-7][a-z]|(8([abcde]|f(1[0-8]))))/i).test(navigator.userAgent)

				/* disable for the mercury iOS browser, or at least older versions of the webkit engine */

				|| (/AppleWebKit\/5([0-2]|3[0-2])/i).test(navigator.userAgent)

		)

);

var _run_callbacks = function(callbacks){

	if (callbacks !== undefined){

		var _l = callbacks.length;

	    for (var i=0;i<_l;i++){

	    	var func = callbacks[i];

	    	if(typeof(func)=='function'){

	            try{

	          	    func();

<%inherit file="/base/base.html"/>

<%def name="title()">

    ${c.repo_name} ${_('New pull request')}

</%def>

<%def name="breadcrumbs_links()">

    ${h.link_to(_(u'Home'),h.url('/'))}

    »

    ${h.link_to(c.repo_name,h.url('changelog_home',repo_name=c.repo_name))}

    »

    ${_('New pull request')}

</%def>

<%def name="main()">

<div class="box">

    <!-- box / title -->

    <div class="title">

        ${self.breadcrumbs()}

    </div>

    ${h.form(url('pullrequest', repo_name=c.repo_name), method='post', id='pull_request_form')}

    <div style="float:left;padding:0px 30px 30px 30px">

       <div style="padding:0px 5px 5px 5px">

         <span>

           <a id="refresh" href="#">

             <img class="icon" title="${_('Refresh')}" alt="${_('Refresh')}" src="${h.url('/images/icons/arrow_refresh.png')}"/>

             ${_('refresh overview')}

           </a>

         </span>

       </div>

        ##ORG

        <div style="float:left">

            <div class="fork_user">

                <div class="gravatar">

                    <img alt="gravatar" src="${h.gravatar_url(c.rhodecode_db_repo.user.email,24)}"/>

                </div>

                <span style="font-size: 20px">

                ${h.select('org_repo','',c.org_repos,class_='refs')}:${h.select('org_ref','',c.org_refs,class_='refs')}

                </span>

                 <div style="padding:5px 3px 3px 42px;">${c.rhodecode_db_repo.description}</div>

            </div>

            <div style="clear:both;padding-top: 10px"></div>

        </div>

          <div style="float:left;font-size:24px;padding:0px 20px">

          <img height=32 width=32 src="${h.url('/images/arrow_right_64.png')}"/>

          </div>

        ##OTHER, most Probably the PARENT OF THIS FORK

        <div style="float:left">

            <div class="fork_user">

                <div class="gravatar">

                    <img id="other_repo_gravatar" alt="gravatar" src=""/>

                </div>

                <span style="font-size: 20px">

                </span>

                 <div id="other_repo_desc" style="padding:5px 3px 3px 42px;"></div>

            </div>

            <div style="clear:both;padding-top: 10px"></div>

        </div>

       <div style="clear:both;padding-top: 10px"></div>

       ## overview pulled by ajax

       <div style="float:left" id="pull_request_overview"></div>

       <div style="float:left;clear:both;padding:10px 10px 10px 0px;display:none">

            <a id="pull_request_overview_url" href="#">${_('Detailed compare view')}</a>

       </div>

     </div>

    <div style="float:left; border-left:1px dashed #eee">

        <h4>${_('Pull request reviewers')}</h4>

        <div id="reviewers" style="padding:0px 0px 0px 15px">

          ## members goes here !

          <div class="group_members_wrap">

            <ul id="review_members" class="group_members">

            %for member in c.review_members:

              <li id="reviewer_${member.user_id}">

                <div class="reviewers_member">

                  <div class="gravatar"><img alt="gravatar" src="${h.gravatar_url(member.email,14)}"/> </div>

                  <div style="float:left">${member.full_name} (${_('owner')})</div>

                  <input type="hidden" value="${member.user_id}" name="review_members" />

                  <span class="delete_icon action_button" onclick="removeReviewer(${member.user_id})"></span>

                </div>

              </li>

            %endfor

            </ul>

          </div>

          <div class='ac'>

            <div class="reviewer_ac">

               ${h.text('user', class_='yui-ac-input')}

               <span class="help-block">${_('Add reviewer to this pull request.')}</span>

               <div id="reviewers_container"></div>

            </div>

          </div>

        </div>

    </div>

    <h3>${_('Create new pull request')}</h3>

    <div class="form">

        <!-- fields -->

        <div class="fields">

             <div class="field">

                <div class="label">

                    <label for="pullrequest_title">${_('Title')}:</label>

                </div>

                <div class="input">

                    ${h.text('pullrequest_title',size=30)}

                </div>

             </div>

            <div class="field">

                <div class="label label-textarea">

                    <label for="pullrequest_desc">${_('description')}:</label>

                </div>

                <div class="textarea text-area editor">

                    ${h.textarea('pullrequest_desc',size=30)}

                </div>

            </div>

            <div class="buttons">

                ${h.submit('save',_('Send pull request'),class_="ui-btn large")}

                ${h.reset('reset',_('Reset'),class_="ui-btn large")}

           </div>

        </div>

    </div>

    ${h.end_form()}

</div>

<script type="text/javascript">

  var _USERS_AC_DATA = ${c.users_array|n};

  var _GROUPS_AC_DATA = ${c.users_groups_array|n};

  PullRequestAutoComplete('user', 'reviewers_container', _USERS_AC_DATA, _GROUPS_AC_DATA);

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

  var loadPreview = function(){

	  YUD.setStyle(YUD.get('pull_request_overview_url').parentElement,'display','none');

      var url = "${h.url('compare_url',

    	                 repo_name='org_repo',

    	                 org_ref_type='org_ref_type', org_ref='org_ref',

                         other_ref_type='other_ref_type', other_ref='other_ref',

                         repo='other_repo',

                         as_form=True)}";

      var select_refs = YUQ('#pull_request_form select.refs')

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

        var select_ref = select_refs[i];

        var select_ref_data = select_ref.value.split(':');

        var key = null;

        var val = null;

        if(select_ref_data.length>1){

          key = select_ref.name+"_type";

          val = select_ref_data[0];

          url = url.replace(key,val);

          key = select_ref.name;

          val = select_ref_data[1];

          url = url.replace(key,val);

        }else{

          key = select_ref.name;

          val = select_ref.value;

          url = url.replace(key,val);

        }

      }

      YUE.on('other_repo', 'change', function(e){

    	  var repo_name = e.currentTarget.value;

    	  // replace the <select> of changed repo

    	  YUD.get('other_ref').innerHTML = other_repos_info[repo_name]['revs'];

      });

      ypjax(url,'pull_request_overview', function(data){

    	  var sel_box = YUQ('#pull_request_form #other_repo')[0];

    	  var repo_name = sel_box.options[sel_box.selectedIndex].value;

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

    	  YUD.setStyle(YUD.get('pull_request_overview_url').parentElement,'display','');

    	  YUD.get('other_repo_gravatar').src = other_repos_info[repo_name]['gravatar'];

    	  YUD.get('other_repo_desc').innerHTML = other_repos_info[repo_name]['description'];

    	  YUD.get('other_ref').innerHTML = other_repos_info[repo_name]['revs'];

      })

  }

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

     loadPreview()

  })

  //lazy load overview after 0.5s

  setTimeout(loadPreview, 500)

</script>

</%def>