aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/pull/repo.py
blob: a128dbfc57b6046737075fb7755a36f3839448ca (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
# Copyright (c) 2018 Egor Tensin <Egor.Tensin@gmail.com>
# This file is part of the "cgit repos" project.
# For details, see https://github.com/egor-tensin/cgit-repos.
# Distributed under the MIT License.

import os.path


class Repo:
    DEFAULT_OWNER = None

    @staticmethod
    def extract_repo_name(repo_id):
        return os.path.basename(repo_id)

    def __init__(self, repo_id, clone_url, owner=None, desc=None,
                 homepage=None):
        self.repo_id = repo_id
        self.repo_name = self.extract_repo_name(repo_id)
        self.clone_url = clone_url
        if owner is None:
            owner = Repo.DEFAULT_OWNER
        self.owner = owner
        if desc is None:
            if homepage is not None:
                desc = homepage
            elif clone_url is not None:
                desc = clone_url
            else:
                desc = self.repo_name
        self.desc = desc
        self.homepage = homepage


class GithubRepo(Repo):
    DEFAULT_USER = None

    def __init__(self, repo_id, clone_url=None, owner=None, desc=None,
                 homepage=None, user=DEFAULT_USER, via_ssh=True):
        if user is None:
            if GithubRepo.DEFAULT_USER is None:
                raise RuntimeError('neither explicit or default GitHub username was provided')
            user = GithubRepo.DEFAULT_USER
        name = Repo.extract_repo_name(repo_id)
        if clone_url is None:
            if via_ssh:
                clone_url = self.build_clone_url_ssh(user, name)
            else:
                clone_url = self.build_clone_url_https(user, name)
        if homepage is None:
            homepage = self.build_homepage_url(user, name)
        super().__init__(repo_id, clone_url, owner=owner, desc=desc,
                         homepage=homepage)

    @staticmethod
    def build_clone_url_ssh(user, name):
        return f'ssh://git@github.com/{user}/{name}.git'

    @staticmethod
    def build_clone_url_https(user, name):
        return f'https://github.com/{user}/{name}.git'

    @staticmethod
    def build_homepage_url(user, name):
        return f'https://github.com/{user}/{name}'


class BitbucketRepo(Repo):
    DEFAULT_USER = None

    def __init__(self, repo_id, clone_url=None, owner=None, desc=None,
                 homepage=None, user=DEFAULT_USER, via_ssh=True):
        if user is None:
            if BitbucketRepo.DEFAULT_USER is None:
                raise RuntimeError('neither explicit or default Bitbucket username was provided')
            user = BitbucketRepo.DEFAULT_USER
        name = Repo.extract_repo_name(repo_id)
        if clone_url is None:
            if via_ssh:
                clone_url = self.build_clone_url_ssh(user, name)
            else:
                clone_url = self.build_clone_url_https(user, name)
        if homepage is None:
            homepage = self.build_homepage_url(user, name)
        super().__init__(repo_id, clone_url, owner=owner, desc=desc,
                         homepage=homepage)

    @staticmethod
    def build_clone_url_ssh(user, name):
        return f'ssh://git@bitbucket.org/{user}/{name}.git'

    @staticmethod
    def build_clone_url_https(user, name):
        return f'https://bitbucket.org/{user}/{name}.git'

    @staticmethod
    def build_homepage_url(user, name):
        return f'https://bitbucket.org/{user}/{name.lower()}'