add Gitlab support

4 files changed, 88 insertions, 2 deletions

diff --git a/cgitize/config.py b/cgitize/config.py
index 036b855..0e42357 100644
--- a/cgitize/config.py
+++ b/cgitize/config.py
@@ -9,6 +9,7 @@ import tomli
 
 from cgitize.bitbucket import Bitbucket
 from cgitize.github import GitHub
+from cgitize.gitlab import Gitlab
 from cgitize.repo import Repo
 
 
@@ -95,6 +96,21 @@ class BitbucketSection(Section):
         return map(HostedRepo, self.repositories.enum_repositories())
 
 
+class GitlabSection(Section):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self.users = UsersSection(self.impl.get('users', {}))
+        self.repositories = RepositoriesSection(self.impl.get('repositories', {}))
+
+    @property
+    def access_token(self):
+        return self._get_config_or_env('access_token', 'CGITIZE_GITLAB_ACCESS_TOKEN')
+
+    @property
+    def url_auth(self):
+        return self.access_token
+
+
 class UsersSection(Section):
     def enum_users(self):
         return self.impl.values()
@@ -154,6 +170,7 @@ class Config:
         self.repositories = RepositoriesSection(self.impl.get('repositories', {}))
         self.github = GitHubSection(self.impl.get('github', {}))
         self.bitbucket = BitbucketSection(self.impl.get('bitbucket', {}))
+        self.gitlab = GitlabSection(self.impl.get('gitlab', {}))
 
     def _parse_explicit_repositories(self):
         for r in self.repositories.enum_repositories():
@@ -179,7 +196,12 @@ class Config:
         bitbucket = Bitbucket(self.bitbucket.username, self.bitbucket.app_password)
         return self._parse_hosted_repositories(self.bitbucket, bitbucket)
 
+    def _parse_gitlab_repositories(self):
+        gitlab = Gitlab(self.gitlab.access_token)
+        return self._parse_hosted_repositories(self.gitlab, gitlab)
+
     def parse_repositories(self):
         yield from self._parse_explicit_repositories()
         yield from self._parse_github_repositories()
         yield from self._parse_bitbucket_repositories()
+        yield from self._parse_gitlab_repositories()
diff --git a/cgitize/gitlab.py b/cgitize/gitlab.py
new file mode 100644
index 0000000..77612cb
--- /dev/null
+++ b/cgitize/gitlab.py
@@ -0,0 +1,41 @@
+# Copyright (c) 2021 Egor Tensin <Egor.Tensin@gmail.com>
+# This file is part of the "cgitize" project.
+# For details, see https://github.com/egor-tensin/cgitize.
+# Distributed under the MIT License.
+
+import logging
+
+import gitlab
+
+from cgitize.repo import Repo
+
+
+class Gitlab:
+    def __init__(self, access_token):
+        self._impl = gitlab.Gitlab('https://gitlab.com', private_token=access_token)
+
+    def get_repo(self, repo):
+        try:
+            return self._impl.projects.get(repo.id)
+        except gitlab.exceptions.GitlabGetError:
+            logging.error("Couldn't fetch repository: %s", repo.id)
+            raise
+
+    def get_user_repos(self, user):
+        try:
+            # Strictly speaking, Gitlab supports the /users/:username/projects
+            # endpoint, which means you shouldn't need to fetch the user first,
+            # but I don't think python-gitlab 2.10.0 supports that?
+            users = self._impl.users.list(username=user.name)
+            if not users:
+                raise RuntimeError(f"Couldn't find Gitlab user: {user.name}")
+            assert len(users) == 1
+            user = users[0]
+            return user.projects.list()
+        except gitlab.exceptions.GitlabGetError:
+            logging.error("Couldn't fetch user repositories: %s", user.name)
+            raise
+
+    @staticmethod
+    def convert_repo(repo, *args, **kwargs):
+        return Repo.from_gitlab(repo, *args, **kwargs)
diff --git a/cgitize/repo.py b/cgitize/repo.py
index d926ee9..079801f 100644
--- a/cgitize/repo.py
+++ b/cgitize/repo.py
@@ -63,6 +63,21 @@ class Repo:
         return Repo(name, clone_url, owner=owner, desc=desc, homepage=homepage,
                     url_auth=url_auth, subdir=subdir)
 
+    @staticmethod
+    def from_gitlab(src, config, subdir=None):
+        name = src.name
+        desc = src.description
+        homepage = src.web_url
+        owner = src.namespace['name']
+
+        https_url = src.http_url_to_repo
+        ssh_url = src.ssh_url_to_repo
+        clone_url = ssh_url if config.main.clone_via_ssh else https_url
+        url_auth = None if config.main.clone_via_ssh else config.gitlab.url_auth
+
+        return Repo(name, clone_url, owner=owner, desc=desc, homepage=homepage,
+                    url_auth=url_auth, subdir=subdir)
+
     def __init__(self, name, clone_url, owner=None, desc=None, homepage=None,
                  url_auth=None, subdir=None):
         self._name = name
diff --git a/test/integration/example/test.sh b/test/integration/example/test.sh
index f1f193d..6991697 100755
--- a/test/integration/example/test.sh
+++ b/test/integration/example/test.sh
@@ -123,13 +123,17 @@ test_ssh() {
     verify_repos \
         lens \
         chromiumembedded/cef \
+        inkscape \
         wireguard/wintun \
         github-dir/cgitize-test-repository \
-        bitbucket-dir/cgitize-test-repository
+        bitbucket-dir/cgitize-test-repository \
+        gitlab-dir/cgitize-test-repository
     verify_origin lens 'git@github.com:ekmett/lens.git'
     verify_origin chromiumembedded/cef 'git@bitbucket.org:chromiumembedded/cef.git'
+    verify_origin inkscape 'git@gitlab.com:inkscape/inkscape.git'
     verify_origin github-dir/cgitize-test-repository 'git@github.com:egor-tensin/cgitize-test-repository.git'
     verify_origin bitbucket-dir/cgitize-test-repository 'git@bitbucket.org:egor-tensin/cgitize-test-repository.git'
+    verify_origin gitlab-dir/cgitize-test-repository 'git@gitlab.com:egor-tensin/cgitize-test-repository.git'
     cleanup
 }
 
@@ -144,13 +148,17 @@ test_https() {
     verify_repos \
         lens \
         chromiumembedded/cef \
+        inkscape \
         wireguard/wintun \
         github-dir/cgitize-test-repository \
-        bitbucket-dir/cgitize-test-repository
+        bitbucket-dir/cgitize-test-repository \
+        gitlab-dir/cgitize-test-repository
     verify_origin lens 'https://github.com/ekmett/lens.git'
     verify_origin chromiumembedded/cef 'https://bitbucket.org/chromiumembedded/cef.git'
+    verify_origin inkscape 'https://gitlab.com/inkscape/inkscape.git'
     verify_origin github-dir/cgitize-test-repository 'https://github.com/egor-tensin/cgitize-test-repository.git'
     verify_origin bitbucket-dir/cgitize-test-repository 'https://bitbucket.org/egor-tensin/cgitize-test-repository.git'
+    verify_origin gitlab-dir/cgitize-test-repository 'https://gitlab.com/egor-tensin/cgitize-test-repository.git'
     cleanup
 }