/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to breezy/plugins/propose/gitlabs.py

  • Committer: Jelmer Vernooij
  • Date: 2020-06-01 14:06:02 UTC
  • mto: (7490.29.9 work)
  • mto: This revision was merged to the branch mainline in revision 7507.
  • Revision ID: jelmer@jelmer.uk-20200601140602-uhg6azhxqnmwx7lm
Split github support out into a separate plugin.

Show diffs side-by-side

added added

removed removed

Lines of Context:
breezy/plugins/propose/gitlabs.py
16
16
 
17
17
"""Support for GitLab."""
18
18
 
 
19
from __future__ import absolute_import
 
20
 
19
21
import json
20
22
import os
21
23
import time
28
30
    urlutils,
29
31
    )
30
32
from ...git.urls import git_url_to_bzr_url
 
33
from ...sixish import PY3
31
34
from ...trace import mutter
32
35
from ...transport import get_transport
33
36
 
97
100
        self.error = error
98
101
 
99
102
 
100
 
class GitLabConflict(errors.BzrError):
101
 
 
102
 
    _fmt = "Conflict during operation: %(reason)s"
103
 
 
104
 
    def __init__(self, reason):
105
 
        errors.BzrError(self, reason=reason)
106
 
 
107
 
 
108
 
class ForkingDisabled(errors.BzrError):
109
 
 
110
 
    _fmt = ("Forking on project %(project)s is disabled.")
111
 
 
112
 
    def __init__(self, project):
113
 
        self.project = project
114
 
 
115
 
 
116
103
class MergeRequestExists(Exception):
117
104
    """Raised when a merge requests already exists."""
118
105
 
231
218
        return self._branch_url_from_project(
232
219
            self._mr['source_project_id'], self._mr['source_branch'])
233
220
 
234
 
    def get_source_revision(self):
235
 
        from breezy.git.mapping import default_mapping
236
 
        sha = self._mr['sha']
237
 
        if sha is None:
238
 
            return None
239
 
        return default_mapping.revision_id_foreign_to_bzr(sha.encode('ascii'))
240
 
 
241
221
    def get_target_branch_url(self):
242
222
        return self._branch_url_from_project(
243
223
            self._mr['target_project_id'], self._mr['target_branch'])
295
275
        import iso8601
296
276
        return iso8601.parse_date(merged_at)
297
277
 
298
 
    def post_comment(self, body):
299
 
        kwargs = {'body': body}
300
 
        self.gl._post_merge_request_note(
301
 
            self._mr['project_id'], self._mr['iid'], kwargs)
302
 
 
303
278
 
304
279
def gitlab_url_to_bzr_url(url, name):
 
280
    if not PY3:
 
281
        name = name.encode('utf-8')
305
282
    return git_url_to_bzr_url(url, branch=name)
306
283
 
307
284
 
341
318
            raise KeyError('no such user %s' % username)
342
319
        if response.status == 200:
343
320
            return json.loads(response.data)
344
 
        raise errors.UnexpectedHttpStatus(path, response.status)
 
321
        raise errors.InvalidHttpResponse(path, response.text)
345
322
 
346
323
    def _get_user_by_email(self, email):
347
324
        path = 'users?search=%s' % urlutils.quote(str(email), '')
353
330
            if len(ret) != 1:
354
331
                raise ValueError('unexpected number of results; %r' % ret)
355
332
            return ret[0]
356
 
        raise errors.UnexpectedHttpStatus(path, response.status)
 
333
        raise errors.InvalidHttpResponse(path, response.text)
357
334
 
358
335
    def _get_project(self, project_name):
359
336
        path = 'projects/%s' % urlutils.quote(str(project_name), '')
362
339
            raise NoSuchProject(project_name)
363
340
        if response.status == 200:
364
341
            return json.loads(response.data)
365
 
        raise errors.UnexpectedHttpStatus(path, response.status)
 
342
        raise errors.InvalidHttpResponse(path, response.text)
366
343
 
367
 
    def _fork_project(self, project_name, timeout=50, interval=5, owner=None):
 
344
    def _fork_project(self, project_name, timeout=50, interval=5):
368
345
        path = 'projects/%s/fork' % urlutils.quote(str(project_name), '')
369
 
        fields = {}
370
 
        if owner is not None:
371
 
            fields['namespace'] = owner
372
 
        response = self._api_request('POST', path, fields=fields)
373
 
        if response.status == 404:
374
 
            raise ForkingDisabled(project_name)
375
 
        if response.status == 409:
376
 
            resp = json.loads(response.data)
377
 
            raise GitLabConflict(resp.get('message'))
 
346
        response = self._api_request('POST', path)
378
347
        if response.status not in (200, 201):
379
 
            raise errors.UnexpectedHttpStatus(path, response.status)
 
348
            raise errors.InvalidHttpResponse(path, response.text)
380
349
        # The response should be valid JSON, but let's ignore it
381
350
        project = json.loads(response.data)
382
351
        # Spin and wait until import_status for new project
390
359
            project = self._get_project(project['path_with_namespace'])
391
360
        return project
392
361
 
393
 
    def get_current_user(self):
 
362
    def _get_logged_in_username(self):
394
363
        return self._current_user['username']
395
364
 
396
 
    def get_user_url(self, username):
397
 
        return urlutils.join(self.base_url, username)
398
 
 
399
365
    def _list_paged(self, path, parameters=None, per_page=None):
400
366
        if parameters is None:
401
367
            parameters = {}
412
378
            if response.status == 403:
413
379
                raise errors.PermissionDenied(response.text)
414
380
            if response.status != 200:
415
 
                raise errors.UnexpectedHttpStatus(path, response.status)
 
381
                raise errors.InvalidHttpResponse(path, response.text)
416
382
            page = response.getheader("X-Next-Page")
417
383
            for entry in json.loads(response.data):
418
384
                yield entry
435
401
        if response.status == 403:
436
402
            raise errors.PermissionDenied(response.text)
437
403
        if response.status != 200:
438
 
            raise errors.UnexpectedHttpStatus(path, response.status)
 
404
            raise errors.InvalidHttpResponse(path, response.text)
439
405
        return json.loads(response.data)
440
406
 
441
407
    def _list_projects(self, owner):
449
415
        response = self._api_request('PUT', path, fields=mr)
450
416
        if response.status == 200:
451
417
            return json.loads(response.data)
452
 
        raise errors.UnexpectedHttpStatus(path, response.status)
453
 
 
454
 
    def _post_merge_request_note(self, project_id, iid, kwargs):
455
 
        path = 'projects/%s/merge_requests/%s/notes' % (
456
 
            urlutils.quote(str(project_id), ''), iid)
457
 
        response = self._api_request('POST', path, fields=kwargs)
458
 
        if response.status == 201:
459
 
            json.loads(response.data)
460
 
            return
461
 
        raise errors.UnexpectedHttpStatus(path, response.status)
 
418
        raise errors.InvalidHttpResponse(path, response.text)
462
419
 
463
420
    def _create_mergerequest(
464
421
            self, title, source_project_id, target_project_id,
481
438
        if response.status == 409:
482
439
            raise MergeRequestExists()
483
440
        if response.status != 201:
484
 
            raise errors.UnexpectedHttpStatus(path, response.status)
 
441
            raise errors.InvalidHttpResponse(path, response.text)
485
442
        return json.loads(response.data)
486
443
 
487
444
    def get_push_url(self, branch):
495
452
                        allow_lossy=True, tag_selector=None):
496
453
        (host, base_project, base_branch_name) = parse_gitlab_branch_url(base_branch)
497
454
        if owner is None:
498
 
            owner = self.get_current_user()
 
455
            owner = self._get_logged_in_username()
499
456
        if project is None:
500
457
            project = self._get_project(base_project)['path']
501
458
        try:
502
459
            target_project = self._get_project('%s/%s' % (owner, project))
503
460
        except NoSuchProject:
504
 
            target_project = self._fork_project(base_project, owner=owner)
 
461
            target_project = self._fork_project(base_project)
505
462
        remote_repo_url = git_url_to_bzr_url(target_project['ssh_url_to_repo'])
506
463
        remote_dir = controldir.ControlDir.open(remote_repo_url)
507
464
        try:
521
478
    def get_derived_branch(self, base_branch, name, project=None, owner=None):
522
479
        (host, base_project, base_branch_name) = parse_gitlab_branch_url(base_branch)
523
480
        if owner is None:
524
 
            owner = self.get_current_user()
 
481
            owner = self._get_logged_in_username()
525
482
        if project is None:
526
483
            project = self._get_project(base_project)['path']
527
484
        try:
570
527
                raise GitLabLoginMissing()
571
528
            else:
572
529
                raise GitlabLoginError(response.text)
573
 
        raise UnsupportedHoster(self.base_url)
 
530
        raise UnsupportedHoster(url)
574
531
 
575
532
    @classmethod
576
533
    def probe_from_url(cls, url, possible_transports=None):
597
554
    def iter_my_proposals(self, status='open'):
598
555
        state = mp_status_to_status(status)
599
556
        for mp in self._list_merge_requests(
600
 
                owner=self.get_current_user(), state=state):
 
557
                owner=self._get_logged_in_username(), state=state):
601
558
            yield GitLabMergeProposal(self, mp)
602
559
 
603
560
    def iter_my_forks(self):
604
 
        for project in self._list_projects(owner=self.get_current_user()):
 
561
        for project in self._list_projects(owner=self._get_logged_in_username()):
605
562
            base_project = project.get('forked_from_project')
606
563
            if not base_project:
607
564
                continue
629
586
        if response.status == 404:
630
587
            raise NoSuchProject(project)
631
588
        if response.status != 202:
632
 
            raise errors.UnexpectedHttpStatus(path, response.status)
 
589
            raise errors.InvalidHttpResponse(path, response.text)
633
590
 
634
591
 
635
592
class GitlabMergeProposalBuilder(MergeProposalBuilder):