from kallithea.lib.compat import json

from kallithea.lib.base import BaseRepoController, render

from kallithea.lib.auth import LoginRequired, HasRepoPermissionAnyDecorator,\

    NotAnonymous

from kallithea.lib.helpers import Page

from kallithea.lib import helpers as h

from kallithea.lib import diffs

from kallithea.lib.exceptions import UserInvalidException

from kallithea.lib.utils import action_logger, jsonify

from kallithea.lib.vcs.utils import safe_str

from kallithea.lib.vcs.exceptions import EmptyRepositoryError

from kallithea.lib.diffs import LimitedDiffContainer

from kallithea.model.db import PullRequest, ChangesetStatus, ChangesetComment,\

    PullRequestReviewers

from kallithea.model.pull_request import PullRequestModel

from kallithea.model.meta import Session

from kallithea.model.repo import RepoModel

from kallithea.model.comment import ChangesetCommentsModel

from kallithea.model.changeset_status import ChangesetStatusModel

from kallithea.model.forms import PullRequestForm, PullRequestPostForm

from kallithea.lib.utils2 import safe_int

from kallithea.controllers.changeset import _ignorews_url, _context_url

from kallithea.controllers.compare import CompareController

from kallithea.lib.graphmod import graph_data

log = logging.getLogger(__name__)

class PullrequestsController(BaseRepoController):

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

        """return a structure with repo's interesting changesets, suitable for

        the selectors in pullrequest.html

        rev: a revision that must be in the list somehow and selected by default

        branch: a branch that must be in the list and selected by default - even if closed

        branch_rev: a revision of which peers should be preferred and available."""

        # list named branches that has been merged to this named branch - it should probably merge back

        peers = []

        if rev:

            rev = safe_str(rev)

        if branch:

            branch = safe_str(branch)

        if branch_rev:

            branch_rev = safe_str(branch_rev)

    def post(self, repo_name, pull_request_id):

        pull_request = PullRequest.get_or_404(pull_request_id)

        if pull_request.is_closed():

            raise HTTPForbidden()

        assert pull_request.other_repo.repo_name == repo_name

        #only owner or admin can update it

        owner = pull_request.owner.user_id == c.authuser.user_id

        repo_admin = h.HasRepoPermissionAny('repository.admin')(c.repo_name)

        if not (h.HasPermissionAny('hg.admin') or repo_admin or owner):

            raise HTTPForbidden()

        _form = PullRequestPostForm()().to_python(request.POST)

        reviewers_ids = [int(s) for s in _form['review_members']]

        if _form['updaterev']:

            return self.create_update(pull_request,

                                      _form['updaterev'],

                                      _form['pullrequest_title'],

                                      _form['pullrequest_desc'],

                                      reviewers_ids)

        old_description = pull_request.description

        pull_request.title = _form['pullrequest_title']

        pull_request.description = _form['pullrequest_desc'].strip() or _('No description')

        try:

            PullRequestModel().mention_from_description(pull_request, old_description)

            PullRequestModel().update_reviewers(pull_request_id, reviewers_ids)

        except UserInvalidException as u:

            h.flash(_('Invalid reviewer "%s" specified') % u, category='error')

            raise HTTPBadRequest()

        Session().commit()

        h.flash(_('Pull request updated'), category='success')

        return redirect(pull_request.url())

    @LoginRequired()

    @NotAnonymous()

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

                                   'repository.admin')

    @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.owner.user_id == c.authuser.user_id:

            PullRequestModel().delete(pull_request)

            Session().commit()

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

        allow_extra_fields = True

        filter_extra_fields = True

        org_repo = v.UnicodeString(strip=True, required=True)

        org_ref = v.UnicodeString(strip=True, required=True)

        other_repo = v.UnicodeString(strip=True, required=True)

        other_ref = v.UnicodeString(strip=True, required=True)

        review_members = v.Set()

        pullrequest_title = v.UnicodeString(strip=True, required=True)

        pullrequest_desc = v.UnicodeString(strip=True, required=False)

    return _PullRequestForm

def PullRequestPostForm():

    class _PullRequestPostForm(formencode.Schema):

        allow_extra_fields = True

        filter_extra_fields = True

        pullrequest_title = v.UnicodeString(strip=True, required=True)

        pullrequest_desc = v.UnicodeString(strip=True, required=False)

        review_members = v.Set()

        updaterev = v.UnicodeString(strip=True, required=False, if_missing=None)

    return _PullRequestPostForm

def GistForm(lifetime_options):

    class _GistForm(formencode.Schema):

        allow_extra_fields = True

        filter_extra_fields = True

        filename = All(v.BasePath()(),

                       v.UnicodeString(strip=True, required=False))

        description = v.UnicodeString(required=False, if_missing=u'')

        lifetime = v.OneOf(lifetime_options)

        mimetype = v.UnicodeString(required=False, if_missing=None)

        content = v.UnicodeString(required=True, not_empty=True)

        public = v.UnicodeString(required=False, if_missing=u'')

        private = v.UnicodeString(required=False, if_missing=u'')

    return _GistForm

<%def name="main()">

<% editable = not c.pull_request.is_closed() and (h.HasPermissionAny('hg.admin')() or h.HasRepoPermissionAny('repository.admin')(c.repo_name) or c.pull_request.owner.user_id == c.authuser.user_id) %>

${self.repo_context_bar('showpullrequest')}

<div class="box">

  <!-- box / title -->

  <div class="title">

    ${self.breadcrumbs()}

  </div>

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

    <div class="form pr-box" style="float: left">

      <div class="pr-details-title ${'closed' if c.pull_request.is_closed() else ''}">

          ${_('Title')}: ${c.pull_request.title}

          %if c.pull_request.is_closed():

              (${_('Closed')})

          %endif

      </div>

      <div id="pr-summary" class="fields">

        <div class="field pr-not-edit" style="min-height:37px">

          <div class="label-summary">

            <label>${_('Description')}:</label>

            %if editable:

            <div style="margin: 5px">

            </div>

            %endif

          </div>

          <div class="input">

            <div style="white-space:pre-wrap; line-height: 14px">${h.urlify_commit(c.pull_request.description, c.pull_request.org_repo.repo_name)}</div>

          </div>

        </div>

        %if editable:

          <div class="field">

              <div class="label-summary">

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

              </div>

              <div class="input">

                  ${h.text('pullrequest_title',class_="large",value=c.pull_request.title,placeholder=_('Summarize the changes'))}

              </div>

          </div>

          <div class="field">

              <div class="label-summary label-textarea">

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

              </div>

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

                  ${h.textarea('pullrequest_desc',size=30,content=c.pull_request.description,placeholder=_('Write a short description on this pull request'))}

              </div>

          </div>

        </div>

        %endif

        <div class="field">

          <div class="label-summary">

              <label>${_('Reviewer voting result')}:</label>

          </div>

          </div>

          <div class="input">

              <div>

               ## TODO: use cs_ranges[-1] or org_ref_parts[1] in both cases?

               %if h.is_hg(c.pull_request.org_repo):

                 <span style="font-family: monospace">hg pull ${c.pull_request.org_repo.clone_url()} -r ${h.short_id(c.cs_ranges[-1].raw_id)}</span>

               %elif h.is_git(c.pull_request.org_repo):

                 <span style="font-family: monospace">git pull ${c.pull_request.org_repo.clone_url()} ${c.pull_request.org_ref_parts[1]}</span>

               %endif

              </div>

          </div>

        </div>

        <div class="field">

          <div class="label-summary">

              <label>${_('Created on')}:</label>

          </div>

          <div class="input">

              <div>${h.fmt_date(c.pull_request.created_on)}</div>

          </div>

        </div>

        <div class="field">

          <div class="label-summary">

              <label>${_('Owner')}:</label>

          </div>

                  <div class="gravatar">

                    ${h.gravatar(c.pull_request.owner.email, size=20)}

                  </div>

                  <span>${c.pull_request.owner.full_name_and_username}</span><br/>

                  <span><a href="mailto:${c.pull_request.owner.email}">${c.pull_request.owner.email}</a></span><br/>

          </div>

        </div>

        <div class="field">

            <div class="label-summary">

              <label>${_('Update')}:</label>

            </div>

            <div class="input">

              <div class="msg-div">${c.update_msg}</div>

              %if c.avail_revs:

              <div id="updaterevs" style="max-height:200px; overflow-y:auto; overflow-x:hidden; margin-bottom: 10px; padding: 1px 0">

                <div style="height:0">

                  <canvas id="avail_graph_canvas" style="width:0"></canvas>

                </div>

                <table id="updaterevs-table" class="noborder" style="padding-left:50px">

                  %for cnt, cs in enumerate(c.avail_cs):

                    <tr id="chg_available_${cnt+1}">

                      %if cs.revision == c.cs_ranges[-1].revision:

                        <td>

                          %if editable:

                            ${h.radio(name='updaterev', value='')}

                          %endif

                        </td>

                        <td colspan="2">${_("Current revision - no change")}</td>

                      %else:

    ${diff_block.diff_block_js()}

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

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

    %endfor

    % if c.limited_diff:

      <h4>${_('Changeset was too big and was cut off...')} <a href="${h.url.current(fulldiff=1, **request.GET.mixed())}">${_('Show full diff anyway')}</a></h4>

    % endif

    </div>

    ## template for inline comment form

    ${comment.comment_inline_form()}

    ## render comments and inlines

    ${comment.generate_comments()}

    ## main comment form and it status

    ${comment.comments(h.url('pullrequest_comment', repo_name=c.repo_name,

                              pull_request_id=c.pull_request.pull_request_id),

                       c.current_voting_result,

                       is_pr=True, change_status=c.allowed_to_change_status)}

    <script type="text/javascript">

      $(document).ready(function(){

          PullRequestAutoComplete($('#user'), $('#reviewers_container'), _USERS_AC_DATA);

          $('.code-difftable').on('click', '.add-bubble', function(e){

              show_comment_form($(this));

          });

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

          var avail_r = new BranchRenderer('avail_graph_canvas', 'updaterevs-table', 'chg_available_');

          avail_r.render(avail_jsdata,40);

          move_comments($(".comments .comments-list-chunk"));

          $('#updaterevs input').change(function(e){

              var update = !!e.target.value;

              $('#pr-form-save').prop('disabled',update);

              $('#pr-form-clone').prop('disabled',!update);

          });

          var $org_review_members = $('#review_members').clone();

          $('#pr-form-reset').click(function(e){

              $('.pr-not-edit').show();

              $('#pr-form-save').prop('disabled',false);

              $('#pr-form-clone').prop('disabled',true);

              $('#review_members').html($org_review_members);

          });

          // hack: re-navigate to target after JS is done ... if a target is set and setting href thus won't reload

          if (window.location.hash != "") {

              window.location.href = window.location.href;

          }

          $('.missing_reviewer').click(function(){

            var $this = $(this);

            addReviewMember($this.attr('user_id'), $this.attr('fname'), $this.attr('lname'), $this.attr('nname'), $this.attr('gravatar_lnk'), $this.attr('gravatar_size'));

          });

      });

    </script>

</div>

</%def>

                                 {

                                  'org_repo': HG_REPO,

                                  'org_ref': 'branch:default:default',

                                  'other_repo': HG_REPO,

                                  'other_ref': 'branch:default:default',

                                  'pullrequest_title': 'title',

                                  'pullrequest_desc': 'description',

                                  '_authentication_token': self.authentication_token(),

                                 }

                                )

        self.assertEqual(response.status, '302 Found')

        # location is of the form:

        # http://localhost/vcs_test_hg/pull-request/54/_/title

        m = re.search('/pull-request/(\d+)/', response.location)

        self.assertNotEqual(m, None)

        pull_request_id = m.group(1)

        # update it

        response = self.app.post(url(controller='pullrequests', action='post',

                                     repo_name=HG_REPO, pull_request_id=pull_request_id),

                                 {

                                  'updaterev': 'default',

                                  'pullrequest_title': 'title',

                                  'pullrequest_desc': 'description',

                                  '_authentication_token': self.authentication_token(),

                                  'review_members': invalid_user_id,

                                 },

                                 status=400)

        response.mustcontain('Invalid reviewer "%s" specified' % invalid_user_id)

    def test_edit_with_invalid_reviewer(self):

        invalid_user_id = 99999

        self.log_user()

        # create a valid pull request

        response = self.app.post(url(controller='pullrequests', action='create',

                                     repo_name=HG_REPO),

                                 {

                                  'org_repo': HG_REPO,

                                  'org_ref': 'branch:default:default',

                                  'other_repo': HG_REPO,

                                  'other_ref': 'branch:default:default',

                                  'pullrequest_title': 'title',

                                  'pullrequest_desc': 'description',

                                  '_authentication_token': self.authentication_token(),

                                 }

                                )

        self.assertEqual(response.status, '302 Found')

        # location is of the form:

        # http://localhost/vcs_test_hg/pull-request/54/_/title

        m = re.search('/pull-request/(\d+)/', response.location)

        self.assertNotEqual(m, None)

        pull_request_id = m.group(1)

        # edit it

        response = self.app.post(url(controller='pullrequests', action='post',

                                     repo_name=HG_REPO, pull_request_id=pull_request_id),

                                 {

                                  'pullrequest_title': 'title',

                                  'pullrequest_desc': 'description',

                                  '_authentication_token': self.authentication_token(),

                                  'review_members': invalid_user_id,

                                 },

                                 status=400)

        response.mustcontain('Invalid reviewer "%s" specified' % invalid_user_id)

class TestPullrequestsGetRepoRefs(TestController):

    def setUp(self):

        self.main = fixture.create_repo('main', repo_type='hg')

        Session.add(self.main)

        Session.commit()

        self.c = PullrequestsController()

    def tearDown(self):

        fixture.destroy_repo('main')

        Session.commit()

        Session.remove()

    def test_repo_refs_empty_repo(self):

        # empty repo with no commits, no branches, no bookmarks, just one tag

        refs, default = self.c._get_repo_refs(self.main.scm_instance)

        self.assertEqual(default, 'tag:null:0000000000000000000000000000000000000000')