mirror of
https://github.com/OpenHands/OpenHands.git
synced 2026-03-22 13:47:19 +08:00
d5e66b4f3a
Co-authored-by: openhands <openhands@all-hands.dev> Co-authored-by: rohitvinodmalhotra@gmail.com <rohitvinodmalhotra@gmail.com> Co-authored-by: Hiep Le <69354317+hieptl@users.noreply.github.com> Co-authored-by: Tim O'Farrell <tofarr@gmail.com>
109 lines
3.4 KiB
Python
109 lines
3.4 KiB
Python
import os
|
|
|
|
from pydantic import SecretStr
|
|
|
|
from openhands.integrations.gitlab.service import (
|
|
GitLabBranchesMixin,
|
|
GitLabFeaturesMixin,
|
|
GitLabPRsMixin,
|
|
GitLabReposMixin,
|
|
GitLabResolverMixin,
|
|
)
|
|
from openhands.integrations.service_types import (
|
|
BaseGitService,
|
|
GitService,
|
|
ProviderType,
|
|
)
|
|
from openhands.utils.import_utils import get_impl
|
|
|
|
|
|
class GitLabService(
|
|
GitLabBranchesMixin,
|
|
GitLabFeaturesMixin,
|
|
GitLabPRsMixin,
|
|
GitLabReposMixin,
|
|
GitLabResolverMixin,
|
|
BaseGitService,
|
|
GitService,
|
|
):
|
|
"""
|
|
Assembled GitLab service class combining mixins by feature area.
|
|
|
|
TODO: This doesn't seem a good candidate for the get_impl() pattern. What are the abstract methods we should actually separate and implement here?
|
|
This is an extension point in OpenHands that allows applications to customize GitLab
|
|
integration behavior. Applications can substitute their own implementation by:
|
|
1. Creating a class that inherits from GitService
|
|
2. Implementing all required methods
|
|
3. Setting server_config.gitlab_service_class to the fully qualified name of the class
|
|
|
|
The class is instantiated via get_impl() in openhands.server.shared.py.
|
|
"""
|
|
|
|
BASE_URL = 'https://gitlab.com/api/v4'
|
|
GRAPHQL_URL = 'https://gitlab.com/api/graphql'
|
|
|
|
def __init__(
|
|
self,
|
|
user_id: str | None = None,
|
|
external_auth_id: str | None = None,
|
|
external_auth_token: SecretStr | None = None,
|
|
token: SecretStr | None = None,
|
|
external_token_manager: bool = False,
|
|
base_domain: str | None = None,
|
|
) -> None:
|
|
self.user_id = user_id
|
|
self.external_token_manager = external_token_manager
|
|
self.external_auth_id = external_auth_id
|
|
self.external_auth_token = external_auth_token
|
|
|
|
if token:
|
|
self.token = token
|
|
|
|
if base_domain:
|
|
# Check if protocol is already included
|
|
if base_domain.startswith(('http://', 'https://')):
|
|
# Use the provided protocol
|
|
self.BASE_URL = f'{base_domain}/api/v4'
|
|
self.GRAPHQL_URL = f'{base_domain}/api/graphql'
|
|
else:
|
|
# Default to https if no protocol specified
|
|
self.BASE_URL = f'https://{base_domain}/api/v4'
|
|
self.GRAPHQL_URL = f'https://{base_domain}/api/graphql'
|
|
|
|
@property
|
|
def provider(self) -> str:
|
|
return ProviderType.GITLAB.value
|
|
|
|
|
|
gitlab_service_cls = os.environ.get(
|
|
'OPENHANDS_GITLAB_SERVICE_CLS',
|
|
'openhands.integrations.gitlab.gitlab_service.GitLabService',
|
|
)
|
|
|
|
# Lazy loading to avoid circular imports
|
|
_gitlab_service_impl = None
|
|
|
|
|
|
def get_gitlab_service_impl():
|
|
"""Get the GitLab service implementation with lazy loading."""
|
|
global _gitlab_service_impl
|
|
if _gitlab_service_impl is None:
|
|
_gitlab_service_impl = get_impl(GitLabService, gitlab_service_cls)
|
|
return _gitlab_service_impl
|
|
|
|
|
|
# For backward compatibility, provide the implementation as a property
|
|
class _GitLabServiceImplProxy:
|
|
"""Proxy class to provide lazy loading for GitLabServiceImpl."""
|
|
|
|
def __getattr__(self, name):
|
|
impl = get_gitlab_service_impl()
|
|
return getattr(impl, name)
|
|
|
|
def __call__(self, *args, **kwargs):
|
|
impl = get_gitlab_service_impl()
|
|
return impl(*args, **kwargs)
|
|
|
|
|
|
GitLabServiceImpl: type[GitLabService] = _GitLabServiceImplProxy() # type: ignore[assignment]
|