gitlab: add avatar method

Author: wlcx

buildbot_nix/buildbot_nix/gitlab_project.py

@@ -1,17 +1,21 @@
 import os
 import signal
+from collections.abc import Generator
 from pathlib import Path
 from typing import Any
-from urllib.parse import urlparse
+from urllib.parse import urlencode, urlparse
 
 from buildbot.changes.base import ChangeSource
 from buildbot.config.builder import BuilderConfig
 from buildbot.plugins import util
 from buildbot.reporters.base import ReporterBase
 from buildbot.reporters.gitlab import GitLabStatusPush
+from buildbot.util import httpclientservice
+from buildbot.www import resource
 from buildbot.www.auth import AuthBase
 from buildbot.www.avatar import AvatarBase
 from pydantic import BaseModel
+from twisted.internet import defer
 from twisted.logger import Logger
 from twisted.python import log
 
@@ -186,7 +190,7 @@ def create_auth(self) -> AuthBase:
         raise NotImplementedError
 
     def create_avatar_method(self) -> AvatarBase | None:
-        return None
+        return AvatarGitlab(gitlab_base_url=self.config.instance_url)
 
     @property
     def reload_builder_name(self) -> str:
@@ -260,6 +264,90 @@ def run_post(self) -> Any:
         return util.SUCCESS
 
 
+class AvatarGitlab(AvatarBase):
+    name = "gitlab"
+
+    gitlab_base_url: str
+
+    def __init__(
+        self,
+        gitlab_base_url: str,
+        debug: bool = False,
+        verify: bool = True,
+    ) -> None:
+        self.gitlab_base_url = gitlab_base_url
+        self.debug = debug
+        self.verify = verify
+
+        self.master = None
+        self.client = None
+
+    @defer.inlineCallbacks
+    def _get_http_client(
+        self,
+    ) -> Generator[defer.Deferred, None, httpclientservice.HTTPSession]:
+        if self.client is not None:
+            return self.client
+
+        headers = {
+            "User-Agent": "Buildbot",
+        }
+
+        self.client = yield httpclientservice.HTTPSession(
+            self.master.httpservice,
+            self.gitlab_base_url,
+            headers=headers,
+            debug=self.debug,
+            verify=self.verify,
+        )
+
+        return self.client
+
+    @defer.inlineCallbacks
+    def getUserAvatar(  # noqa: N802
+        self,
+        email: str,
+        username: str,
+        size: str | int,
+        defaultAvatarUrl: str,  # noqa: N803
+    ) -> Generator[defer.Deferred, None, None]:
+        # avatar_url = self._get_avatar_by_username(username)
+        if isinstance(size, int):
+            size = str(size)
+        avatar_url = yield self._get_avatar_by_email(email, size)
+        if not avatar_url:
+            avatar_url = defaultAvatarUrl
+        raise resource.Redirect(avatar_url)
+
+    def _get_avatar_by_username(self, username: str) -> str | None:
+        qs = urlencode(dict(username=username))
+        users = paginated_github_request(
+            f"{self.config.instance_url}/api/v4/users?{qs}"
+        )
+        if len(users) == 1:
+            return users[0]["avatar_url"]
+        if len(users) > 1:
+            # TODO: log warning
+            ...
+        return None
+
+    @defer.inlineCallbacks
+    def _get_avatar_by_email(
+        self, email: str, size: str | None
+    ) -> Generator[defer.Deferred, None, str | None]:
+        http = yield self._get_http_client()
+        q = dict(email=email)
+        if size is not None:
+            q["size"] = size
+        qs = urlencode(q)
+        res = yield http.get(f"/api/v4/avatar?{qs}")
+        data = yield res.json()
+        if "avatar_url" in data:
+            return data["avatar_url"]
+        else:
+            return None
+
+
 def refresh_projects(config: GitlabConfig, cache_file: Path) -> list[RepoData]:
     # access level 40 == Maintainer. See https://docs.gitlab.com/api/members/#roles
     return [