|
|
# Copyright (C) 2011-2023 RhodeCode GmbH
|
|
|
#
|
|
|
# This program is free software: you can redistribute it and/or modify
|
|
|
# it under the terms of the GNU Affero General Public License, version 3
|
|
|
# (only), as published by the Free Software Foundation.
|
|
|
#
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
# GNU General Public License for more details.
|
|
|
#
|
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
#
|
|
|
# This program is dual-licensed. If you wish to learn more about the
|
|
|
# RhodeCode Enterprise Edition, including its added features, Support services,
|
|
|
# and proprietary license terms, please see https://rhodecode.com/licenses/
|
|
|
|
|
|
|
|
|
"""
|
|
|
Model for integrations
|
|
|
"""
|
|
|
|
|
|
|
|
|
import logging
|
|
|
|
|
|
from sqlalchemy import or_, and_
|
|
|
|
|
|
from rhodecode import events
|
|
|
from rhodecode.integrations.types.base import EEIntegration
|
|
|
from rhodecode.lib.caching_query import FromCache
|
|
|
from rhodecode.model import BaseModel
|
|
|
from rhodecode.model.db import Integration, Repository, RepoGroup, true, false, case, null
|
|
|
from rhodecode.integrations import integration_type_registry
|
|
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
|
|
|
class IntegrationModel(BaseModel):
|
|
|
|
|
|
cls = Integration
|
|
|
|
|
|
def __get_integration(self, integration):
|
|
|
if isinstance(integration, Integration):
|
|
|
return integration
|
|
|
elif isinstance(integration, int):
|
|
|
return self.sa.query(Integration).get(integration)
|
|
|
else:
|
|
|
if integration:
|
|
|
raise Exception('integration must be int or Instance'
|
|
|
' of Integration got %s' % type(integration))
|
|
|
|
|
|
def create(self, IntegrationType, name, enabled, repo, repo_group, child_repos_only, settings):
|
|
|
""" Create an IntegrationType integration """
|
|
|
integration = Integration()
|
|
|
integration.integration_type = IntegrationType.key
|
|
|
self.sa.add(integration)
|
|
|
self.update_integration(integration, name, enabled, repo, repo_group,
|
|
|
child_repos_only, settings)
|
|
|
self.sa.commit()
|
|
|
return integration
|
|
|
|
|
|
def update_integration(self, integration, name, enabled, repo, repo_group,
|
|
|
child_repos_only, settings):
|
|
|
integration = self.__get_integration(integration)
|
|
|
|
|
|
integration.repo = repo
|
|
|
integration.repo_group = repo_group
|
|
|
integration.child_repos_only = child_repos_only
|
|
|
integration.name = name
|
|
|
integration.enabled = enabled
|
|
|
integration.settings = settings
|
|
|
|
|
|
return integration
|
|
|
|
|
|
def delete(self, integration):
|
|
|
integration = self.__get_integration(integration)
|
|
|
if integration:
|
|
|
self.sa.delete(integration)
|
|
|
return True
|
|
|
return False
|
|
|
|
|
|
def get_integration_handler(self, integration):
|
|
|
TypeClass = integration_type_registry.get(integration.integration_type)
|
|
|
if not TypeClass:
|
|
|
log.error('No class could be found for integration type: {}'.format(
|
|
|
integration.integration_type))
|
|
|
return None
|
|
|
elif isinstance(TypeClass, EEIntegration) or issubclass(TypeClass, EEIntegration):
|
|
|
log.error('EE integration cannot be '
|
|
|
'executed for integration type: {}'.format(
|
|
|
integration.integration_type))
|
|
|
return None
|
|
|
|
|
|
return TypeClass(integration.settings)
|
|
|
|
|
|
def send_event(self, integration, event):
|
|
|
""" Send an event to an integration """
|
|
|
handler = self.get_integration_handler(integration)
|
|
|
if handler:
|
|
|
log.debug(
|
|
|
'events: sending event %s on integration %s using handler %s',
|
|
|
event, integration, handler)
|
|
|
handler.send_event(event)
|
|
|
|
|
|
def get_integrations(self, scope, IntegrationType=None):
|
|
|
"""
|
|
|
Return integrations for a scope, which must be one of:
|
|
|
|
|
|
'all' - every integration, global/repogroup/repo
|
|
|
'global' - global integrations only
|
|
|
<Repository> instance - integrations for this repo only
|
|
|
<RepoGroup> instance - integrations for this repogroup only
|
|
|
"""
|
|
|
|
|
|
if isinstance(scope, Repository):
|
|
|
query = self.sa.query(Integration).filter(
|
|
|
Integration.repo == scope)
|
|
|
elif isinstance(scope, RepoGroup):
|
|
|
query = self.sa.query(Integration).filter(
|
|
|
Integration.repo_group == scope)
|
|
|
elif scope == 'global':
|
|
|
# global integrations
|
|
|
query = self.sa.query(Integration).filter(
|
|
|
and_(Integration.repo_id == None, Integration.repo_group_id == None)
|
|
|
)
|
|
|
elif scope == 'root-repos':
|
|
|
query = self.sa.query(Integration).filter(
|
|
|
and_(Integration.repo_id == None,
|
|
|
Integration.repo_group_id == None,
|
|
|
Integration.child_repos_only == true())
|
|
|
)
|
|
|
elif scope == 'all':
|
|
|
query = self.sa.query(Integration)
|
|
|
else:
|
|
|
raise Exception(
|
|
|
"invalid `scope`, must be one of: "
|
|
|
"['global', 'all', <Repository>, <RepoGroup>]")
|
|
|
|
|
|
if IntegrationType is not None:
|
|
|
query = query.filter(
|
|
|
Integration.integration_type==IntegrationType.key)
|
|
|
|
|
|
result = []
|
|
|
for integration in query.all():
|
|
|
IntType = integration_type_registry.get(integration.integration_type)
|
|
|
result.append((IntType, integration))
|
|
|
return result
|
|
|
|
|
|
def get_for_event(self, event, cache=False):
|
|
|
"""
|
|
|
Get integrations that match an event
|
|
|
"""
|
|
|
# base query
|
|
|
query = self.sa.query(
|
|
|
Integration
|
|
|
).filter(
|
|
|
Integration.enabled == true()
|
|
|
)
|
|
|
|
|
|
global_integrations_filter = and_(
|
|
|
Integration.repo_id == null(),
|
|
|
Integration.repo_group_id == null(),
|
|
|
Integration.child_repos_only == false(),
|
|
|
)
|
|
|
|
|
|
if isinstance(event, events.RepoEvent):
|
|
|
root_repos_integrations_filter = and_(
|
|
|
Integration.repo_id == null(),
|
|
|
Integration.repo_group_id == null(),
|
|
|
Integration.child_repos_only == true(),
|
|
|
)
|
|
|
|
|
|
clauses = [
|
|
|
global_integrations_filter,
|
|
|
]
|
|
|
cases = [
|
|
|
(global_integrations_filter, 1),
|
|
|
(root_repos_integrations_filter, 2),
|
|
|
]
|
|
|
|
|
|
# repo group integrations
|
|
|
if event.repo.group:
|
|
|
# repo group with only root level repos
|
|
|
group_child_repos_filter = and_(
|
|
|
Integration.repo_group_id == event.repo.group.group_id,
|
|
|
Integration.child_repos_only == true()
|
|
|
)
|
|
|
|
|
|
clauses.append(group_child_repos_filter)
|
|
|
cases.append(
|
|
|
(group_child_repos_filter, 3),
|
|
|
)
|
|
|
|
|
|
# repo group cascade to kids
|
|
|
group_recursive_repos_filter = and_(
|
|
|
Integration.repo_group_id.in_(
|
|
|
[group.group_id for group in event.repo.groups_with_parents]
|
|
|
),
|
|
|
Integration.child_repos_only == false()
|
|
|
)
|
|
|
clauses.append(group_recursive_repos_filter)
|
|
|
cases.append(
|
|
|
(group_recursive_repos_filter, 4),
|
|
|
)
|
|
|
|
|
|
if not event.repo.group: # root repo
|
|
|
clauses.append(root_repos_integrations_filter)
|
|
|
|
|
|
# repo integrations
|
|
|
if event.repo.repo_id: # pre create events dont have a repo_id yet
|
|
|
specific_repo_filter = Integration.repo_id == event.repo.repo_id
|
|
|
clauses.append(specific_repo_filter)
|
|
|
cases.append(
|
|
|
(specific_repo_filter, 5),
|
|
|
)
|
|
|
|
|
|
order_by_criterion = case(cases)
|
|
|
|
|
|
query = query.filter(or_(*clauses))
|
|
|
query = query.order_by(order_by_criterion)
|
|
|
|
|
|
if cache:
|
|
|
cache_key = f"get_enabled_repo_integrations_{event.repo.repo_id}"
|
|
|
query = query.options(
|
|
|
FromCache("sql_cache_short", cache_key))
|
|
|
else: # only global integrations
|
|
|
order_by_criterion = Integration.integration_id
|
|
|
|
|
|
query = query.filter(global_integrations_filter)
|
|
|
query = query.order_by(order_by_criterion)
|
|
|
if cache:
|
|
|
query = query.options(
|
|
|
FromCache("sql_cache_short", "get_enabled_global_integrations"))
|
|
|
|
|
|
result = query.all()
|
|
|
return result
|
|
|
|
