dejacode/workflow/integrations/__init__.py at 7ca6895acc511b5dbdf3d24294e881cb0b257348 · aboutcode-org/dejacode · GitHub

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
#
# Copyright (c) nexB Inc. and others. All rights reserved.
# DejaCode is a trademark of nexB Inc.
# SPDX-License-Identifier: AGPL-3.0-only
# See https://github.com/aboutcode-org/dejacode for support or download.
# See https://aboutcode.org for more information about AboutCode FOSS projects.
#

import re

from workflow.integrations.base import BaseIntegration
from workflow.integrations.forgejo import ForgejoIntegration
from workflow.integrations.github import GitHubIntegration
from workflow.integrations.gitlab import GitLabIntegration
from workflow.integrations.jira import JiraIntegration
from workflow.integrations.sourcehut import SourceHutIntegration

__all__ = [
    "BaseIntegration",
    "ForgejoIntegration",
    "GitHubIntegration",
    "GitLabIntegration",
    "JiraIntegration",
    "SourceHutIntegration",
    "is_valid_issue_tracker_id",
    "get_class_for_tracker",
    "get_class_for_platform",
]

FORGEJO_PATTERN = re.compile(r"^https://(?:[a-zA-Z0-9.-]*forgejo[a-zA-Z0-9.-]*)/[^/]+/[^/]+/?$")

GITHUB_PATTERN = re.compile(r"^https://github\.com/[^/]+/[^/]+/?$")

GITLAB_PATTERN = re.compile(r"^https://gitlab\.com/[^/]+/[^/]+/?$")

JIRA_PATTERN = re.compile(
    r"^https://[a-zA-Z0-9.-]+\.atlassian\.net(?:/[^/]+)*"
    r"/(?:projects|browse)/[A-Z][A-Z0-9]+(?:/[^/]*)*/*$"
)

SOURCEHUT_PATTERN = re.compile(r"^https://todo\.sr\.ht/~[^/]+/[^/]+/?$")

ISSUE_TRACKER_PATTERNS = [
    FORGEJO_PATTERN,
    GITHUB_PATTERN,
    GITLAB_PATTERN,
    JIRA_PATTERN,
    SOURCEHUT_PATTERN,
]


def is_valid_issue_tracker_id(issue_tracker_id):
    return any(pattern.match(issue_tracker_id) for pattern in ISSUE_TRACKER_PATTERNS)


def get_class_for_tracker(issue_tracker_id):
    if "github.com" in issue_tracker_id:
        return GitHubIntegration
    elif "gitlab.com" in issue_tracker_id:
        return GitLabIntegration
    elif "atlassian.net" in issue_tracker_id:
        return JiraIntegration
    elif "forgejo" in issue_tracker_id:
        return ForgejoIntegration
    elif "todo.sr.ht" in issue_tracker_id:
        return SourceHutIntegration


def get_class_for_platform(platform):
    return {
        "forgejo": ForgejoIntegration,
        "github": GitHubIntegration,
        "gitlab": GitLabIntegration,
        "jira": JiraIntegration,
        "sourcehut": SourceHutIntegration,
    }.get(platform)