ICTU/quality-time

View on GitHub
components/shared_code/src/shared_data_model/sources/gitlab.py

Summary

Maintainability
C
1 day
Test Coverage
"""GitLab source."""

from pydantic import HttpUrl

from shared_data_model.meta.entity import Color, Entity, EntityAttribute, EntityAttributeType
from shared_data_model.meta.source import Source
from shared_data_model.parameters import (
    URL,
    Branch,
    Branches,
    BranchesToIgnore,
    Days,
    FailureType,
    MergeRequestState,
    MultipleChoiceParameter,
    MultipleChoiceWithAdditionParameter,
    PrivateToken,
    StringParameter,
    TargetBranchesToInclude,
    Upvotes,
)

ALL_GITLAB_METRICS = [
    "change_failure_rate",
    "failed_jobs",
    "job_runs_within_time_period",
    "merge_requests",
    "pipeline_duration",
    "source_up_to_dateness",
    "source_version",
    "unmerged_branches",
    "unused_jobs",
]

JOB_ENTITY = Entity(
    name="job",
    attributes=[
        EntityAttribute(name="Job name", key="name", url="url"),
        EntityAttribute(name="Job stage", key="stage"),
        EntityAttribute(name="Branch or tag", key="branch"),
        EntityAttribute(
            name="Status of most recent build",
            key="build_status",
            color={"canceled": Color.ACTIVE, "failed": Color.NEGATIVE, "success": Color.POSITIVE},
        ),
        EntityAttribute(name="Date of most recent build", key="build_date", type=EntityAttributeType.DATE),
    ],
)

GITLAB_BRANCH_HELP_URL = HttpUrl("https://docs.gitlab.com/ee/user/project/repository/branches/")

GITLAB = Source(
    name="GitLab",
    description="GitLab provides Git-repositories, wiki's, issue-tracking and continuous integration/continuous "
    "deployment pipelines.",
    url=HttpUrl("https://about.gitlab.com/"),
    documentation={
        "generic": """```{note}
Some metric sources are documents in JSON, XML, CSV, or HTML format. Examples include JUnit XML reports, JaCoCo XML
reports and Axe CSV reports. Usually, you add a JUnit (or JaCoCo, or Axe...) source and then simply configure the same
URL that you use to access the document via the browser. Unfortunately, this does not work if the document is stored in
GitLab. In that case, you still use the JUnit (or JaCoCo, or Axe...) source, but provide a GitLab API URL as URL.
Depending on where the document is stored in GitLab, there are two scenarios; the source is a build artifact of a GitLab
CI pipeline, or the source is stored in a GitLab repository:

1. When the metric source is a build artifact of a GitLab CI pipeline, use [URLs of the following format](https://docs.\
gitlab.com/ee/api/job_artifacts.html#download-a-single-artifact-file-from-specific-tag-or-branch):

    `https://<gitlab-server>/api/v4/projects/<project-id>/jobs/artifacts/<branch>/raw/<path>/<to>/<file-name>?\
job=<job-name>`

    The project id can be found under the
    [project's general settings](https://docs.gitlab.com/ee/user/project/settings/).

    If the repository is private, you also need to enter an [personal access token](https://docs.gitlab.com/ee/user/\
profile/personal_access_tokens.html) with the scope `read_api` in the private token field.

2.  When the metric source is a file stored in a GitLab repository, use [URLs of the following format](https://docs.\
gitlab.com/ee/api/repository_files.html#get-raw-file-from-repository):

    `https://<gitlab-server>/api/v4/projects/<project-id>/repository/files/<file-path-with-slashes-%2F-encoded>/raw?\
ref=<branch>`

    The project id can be found under the
    [project's general settings](https://docs.gitlab.com/ee/user/project/settings/).

    If the repository is private, you also need to enter an [personal access token](https://docs.gitlab.com/ee/user/\
profile/personal_access_tokens.html) with the scope `read_repository` in the private token field.
```""",
    },
    parameters={
        "url": URL(
            name="GitLab instance URL",
            help="URL of the GitLab instance, with port if necessary, but without path. For example, "
            "'https://gitlab.com'.",
            validate_on=["private_token"],
            metrics=ALL_GITLAB_METRICS,
        ),
        "project": StringParameter(
            name="Project (name with namespace or id)",
            short_name="project",
            mandatory=True,
            help_url=HttpUrl("https://docs.gitlab.com/ee/user/project/"),
            metrics=[
                "change_failure_rate",
                "failed_jobs",
                "job_runs_within_time_period",
                "pipeline_duration",
                "merge_requests",
                "source_up_to_dateness",
                "unmerged_branches",
                "unused_jobs",
            ],
        ),
        "private_token": PrivateToken(
            name="Private token (with read_api scope)",
            help_url=HttpUrl("https://docs.gitlab.com/ee/user/profile/personal_access_tokens.html"),
            metrics=ALL_GITLAB_METRICS,
        ),
        "file_path": StringParameter(
            name="File or folder path",
            short_name="path",
            help="Use the date and time the path was last changed to determine the up-to-dateness. If no path "
            "is specified, the pipeline is used to determine the up-to-dateness.",
            metrics=["source_up_to_dateness"],
        ),
        "branch": Branch(help_url=GITLAB_BRANCH_HELP_URL),
        "branches": Branches(help_url=GITLAB_BRANCH_HELP_URL),
        "branches_to_ignore": BranchesToIgnore(help_url=GITLAB_BRANCH_HELP_URL),
        "refs_to_ignore": MultipleChoiceWithAdditionParameter(
            name="Branches and tags to ignore (regular expressions, branch names or tag names)",
            short_name="branches and tags to ignore",
            help_url=GITLAB_BRANCH_HELP_URL,
            metrics=["change_failure_rate", "failed_jobs", "job_runs_within_time_period", "unused_jobs"],
        ),
        "inactive_days": Days(
            name="Number of days since last commit after which to consider branches inactive",
            short_name="number of days since last commit",
            default_value="7",
            metrics=["unmerged_branches"],
        ),
        "inactive_job_days": Days(
            name="Number of days without builds after which to consider CI-jobs unused",
            short_name="number of days without builds",
            default_value="90",
            metrics=["unused_jobs"],
        ),
        "failure_type": FailureType(values=["canceled", "failed", "skipped"]),
        "jobs_to_ignore": MultipleChoiceWithAdditionParameter(
            name="Jobs to ignore (regular expressions or job names)",
            short_name="jobs to ignore",
            help="Jobs to ignore can be specified by job name or by regular expression.",
            metrics=["change_failure_rate", "failed_jobs", "job_runs_within_time_period", "unused_jobs"],
        ),
        "lookback_days": Days(
            name="Number of days to look back for selecting pipeline jobs",
            short_name="number of days to look back",
            default_value="90",
            metrics=[
                "change_failure_rate",
                "failed_jobs",
                "job_runs_within_time_period",
                "source_up_to_dateness",
                "unused_jobs",
            ],
        ),
        "lookback_days_pipelines": Days(
            name="Number of days to look back for selecting pipelines",
            short_name="number of days to look back",
            default_value="7",
            metrics=["pipeline_duration"],
        ),
        "merge_request_state": MergeRequestState(values=["opened", "locked", "merged", "closed"]),
        "approval_state": MultipleChoiceParameter(
            name="Approval states to include (requires GitLab Premium)",
            short_name="approval states",
            help_url=HttpUrl("https://docs.gitlab.com/ee/user/project/merge_requests/approvals/"),
            values=["approved", "not approved", "unknown"],
            api_values={"approved": "yes", "not approved": "no", "unknown": "?"},
            placeholder="all approval states",
            metrics=["merge_requests"],
        ),
        "pipeline_statuses_to_include": MultipleChoiceParameter(
            name="Pipeline statuses to include",
            short_name="pipeline statuses",
            values=[
                "created",
                "waiting for resource",
                "preparing",
                "pending",
                "running",
                "success",
                "failed",
                "canceled",
                "skipped",
                "manual",
                "scheduled",
            ],
            api_values={"waiting for resource": "waiting_for_resource"},
            placeholder="all pipeline statuses",
            metrics=["pipeline_duration", "source_up_to_dateness"],
        ),
        "pipeline_triggers_to_include": MultipleChoiceParameter(
            name="Pipeline triggers to include",
            short_name="pipeline triggers",
            values=[
                "push",
                "web",
                "trigger",
                "schedule",
                "api",
                "external",
                "pipeline",
                "chat",
                "web-IDE",
                "merge request event",
                "external pull request event",
                "parent pipeline",
                "ondemand DAST scan",
                "ondemand DAST validation",
            ],
            api_values={
                "merge request event": "merge_request_event",
                "external pull request event": "external_pull_request_event",
                "parent pipeline": "parent_pipeline",
                "ondemand DAST scan": "ondemand_dast_scan",
                "ondemand DAST validation": "ondemand_dast_validation",
                "web-IDE": "webide",
            },
            placeholder="all pipeline triggers",
            metrics=["pipeline_duration", "source_up_to_dateness"],
        ),
        "upvotes": Upvotes(),
        "target_branches_to_include": TargetBranchesToInclude(help_url=GITLAB_BRANCH_HELP_URL),
    },
    entities={
        "change_failure_rate": Entity(
            name="deployment",
            attributes=[
                EntityAttribute(name="Job name", key="name", url="url"),
                EntityAttribute(name="Job stage", key="stage"),
                EntityAttribute(name="Branch or tag", key="branch"),
                EntityAttribute(name="Status of most recent build", key="build_status"),
                EntityAttribute(name="Date of most recent build", key="build_date", type=EntityAttributeType.DATE),
            ],
        ),
        "failed_jobs": JOB_ENTITY,
        "job_runs_within_time_period": JOB_ENTITY,
        "merge_requests": Entity(
            name="merge request",
            attributes=[
                EntityAttribute(name="Merge request", key="title", url="url"),
                EntityAttribute(name="Target branch"),
                EntityAttribute(name="State"),
                EntityAttribute(name="Approved"),
                EntityAttribute(name="Upvotes", type=EntityAttributeType.INTEGER),
                EntityAttribute(name="Downvotes", type=EntityAttributeType.INTEGER),
                EntityAttribute(name="Created", type=EntityAttributeType.DATETIME),
                EntityAttribute(name="Updated", type=EntityAttributeType.DATETIME),
                EntityAttribute(name="Merged", type=EntityAttributeType.DATETIME),
            ],
        ),
        "unmerged_branches": Entity(
            name="branch",
            name_plural="branches",
            attributes=[
                EntityAttribute(name="Branch name", key="name", url="url"),
                EntityAttribute(
                    name="Date of most recent commit",
                    key="commit_date",
                    type=EntityAttributeType.DATE,
                ),
            ],
        ),
        "unused_jobs": JOB_ENTITY,
    },
)