From c7481c5c96536a6bb8a0c32639706bada10346bb Mon Sep 17 00:00:00 2001
From: Guillaume MARTINEZ <lunik@tiwabbit.fr>
Date: Fri, 18 Nov 2022 23:49:25 +0100
Subject: [PATCH] [GitLab] Add modules to manager project badges (#5534)

* [GitLab] Add modules to manager project badges

Signed-off-by: Lunik <lunik@tiwabbit.fr>

* first review

Signed-off-by: Lunik <lunik@tiwabbit.fr>

* Update plugins/modules/gitlab_project_badge.py

Co-authored-by: Felix Fontein <felix@fontein.de>

Signed-off-by: Lunik <lunik@tiwabbit.fr>
Co-authored-by: Felix Fontein <felix@fontein.de>
---
 plugins/modules/gitlab_project_badge.py       | 216 ++++++++++++++++++
 .../targets/gitlab_project_badge/aliases      |   6 +
 .../gitlab_project_badge/defaults/main.yml    |  11 +
 .../gitlab_project_badge/tasks/main.yml       | 214 +++++++++++++++++
 4 files changed, 447 insertions(+)
 create mode 100644 plugins/modules/gitlab_project_badge.py
 create mode 100644 tests/integration/targets/gitlab_project_badge/aliases
 create mode 100644 tests/integration/targets/gitlab_project_badge/defaults/main.yml
 create mode 100644 tests/integration/targets/gitlab_project_badge/tasks/main.yml

diff --git a/plugins/modules/gitlab_project_badge.py b/plugins/modules/gitlab_project_badge.py
new file mode 100644
index 0000000000..5b1a8d3f1c
--- /dev/null
+++ b/plugins/modules/gitlab_project_badge.py
@@ -0,0 +1,216 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+
+# Copyright (c) 2022, Guillaume MARTINEZ (lunik@tiwabbit.fr)
+# GNU General Public License v3.0+ (see LICENSES/GPL-3.0-or-later.txt or https://www.gnu.org/licenses/gpl-3.0.txt)
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+from __future__ import absolute_import, division, print_function
+__metaclass__ = type
+
+DOCUMENTATION = r'''
+---
+module: gitlab_project_badge
+short_description: Manage project badges on GitLab Server
+version_added: 6.1.0
+description:
+  - This module allows to add and remove badges to/from a project.
+author: Guillaume MARTINEZ (@Lunik)
+requirements:
+  - C(owner) or C(maintainer) rights to project on the GitLab server
+extends_documentation_fragment:
+  - community.general.auth_basic
+  - community.general.gitlab
+  - community.general.attributes
+
+attributes:
+  check_mode:
+    support: full
+  diff_mode:
+    support: none
+
+options:
+  project:
+    description:
+      - The name (or full path) of the GitLab project the badge is added to/removed from.
+    required: true
+    type: str
+
+  state:
+    description:
+      - State of the badge in the project.
+      - On C(present), it adds a badge to a GitLab project.
+      - On C(absent), it removes a badge from a GitLab project.
+    choices: ['present', 'absent']
+    default: 'present'
+    type: str
+
+  link_url:
+    description:
+      - The URL associated with the badge.
+    required: true
+    type: str
+
+  image_url:
+    description:
+      - The image URL of the badge.
+      - A badge is identified by this URL.
+    required: true
+    type: str
+'''
+
+EXAMPLES = r'''
+- name: Add a badge to a GitLab Project
+  community.general.gitlab_project_badge:
+    api_url: 'https://example.gitlab.com'
+    api_token: 'Your-Private-Token'
+    project: projectname
+    state: present
+    link_url: 'https://example.gitlab.com/%{project_path}'
+    image_url: 'https://example.gitlab.com/%{project_path}/badges/%{default_branch}/pipeline.svg'
+
+- name: Remove a badge from a GitLab Project
+  community.general.gitlab_project_badge:
+    api_url: 'https://example.gitlab.com'
+    api_token: 'Your-Private-Token'
+    project: projectname
+    state: absent
+    link_url: 'https://example.gitlab.com/%{project_path}'
+    image_url: 'https://example.gitlab.com/%{project_path}/badges/%{default_branch}/pipeline.svg'
+'''
+
+RETURN = '''
+badge:
+  description: The badge information.
+  returned: when I(state=present)
+  type: dict
+  sample:
+    id: 1
+    link_url: 'http://example.com/ci_status.svg?project=%{project_path}&ref=%{default_branch}'
+    image_url: 'https://shields.io/my/badge'
+    rendered_link_url: 'http://example.com/ci_status.svg?project=example-org/example-project&ref=master'
+    rendered_image_url: 'https://shields.io/my/badge'
+    kind: project
+'''
+
+from ansible.module_utils.api import basic_auth_argument_spec
+from ansible.module_utils.basic import AnsibleModule
+
+from ansible_collections.community.general.plugins.module_utils.gitlab import (
+    auth_argument_spec, gitlab_authentication, find_project, ensure_gitlab_package
+)
+
+
+def present_strategy(module, gl, project, wished_badge):
+    changed = False
+
+    existing_badge = None
+    for badge in project.badges.list(iterator=True):
+        if badge.image_url == wished_badge["image_url"]:
+            existing_badge = badge
+            break
+
+    if not existing_badge:
+        changed = True
+        if module.check_mode:
+            return changed, {"status": "A project badge would be created."}
+
+        badge = project.badges.create(wished_badge)
+        return changed, badge.attributes
+
+    if existing_badge.link_url != wished_badge["link_url"]:
+        changed = True
+        existing_badge.link_url = wished_badge["link_url"]
+
+    if changed:
+        if module.check_mode:
+            return changed, {"status": "Project badge attributes would be changed."}
+
+        existing_badge.save()
+
+    return changed, existing_badge.attributes
+
+
+def absent_strategy(module, gl, project, wished_badge):
+    changed = False
+
+    existing_badge = None
+    for badge in project.badges.list(iterator=True):
+        if badge.image_url == wished_badge["image_url"]:
+            existing_badge = badge
+            break
+
+    if not existing_badge:
+        return changed, None
+
+    changed = True
+    if module.check_mode:
+        return changed, {"status": "Project badge would be destroyed."}
+
+    existing_badge.delete()
+
+    return changed, None
+
+
+state_strategy = {
+    "present": present_strategy,
+    "absent": absent_strategy
+}
+
+
+def core(module):
+    ensure_gitlab_package(module)
+
+    gitlab_project = module.params['project']
+    state = module.params['state']
+
+    gl = gitlab_authentication(module)
+
+    project = find_project(gl, gitlab_project)
+    # project doesn't exist
+    if not project:
+        module.fail_json(msg="project '%s' not found." % gitlab_project)
+
+    wished_badge = {
+        "link_url": module.params["link_url"],
+        "image_url": module.params["image_url"],
+    }
+
+    changed, summary = state_strategy[state](module=module, gl=gl, project=project, wished_badge=wished_badge)
+
+    module.exit_json(changed=changed, badge=summary)
+
+
+def main():
+    argument_spec = basic_auth_argument_spec()
+    argument_spec.update(auth_argument_spec())
+    argument_spec.update(dict(
+        project=dict(type='str', required=True),
+        state=dict(type='str', default='present', choices=['present', 'absent']),
+        link_url=dict(type='str', required=True),
+        image_url=dict(type='str', required=True),
+    ))
+
+    module = AnsibleModule(
+        argument_spec=argument_spec,
+        mutually_exclusive=[
+            ['api_username', 'api_token'],
+            ['api_username', 'api_oauth_token'],
+            ['api_username', 'api_job_token'],
+            ['api_token', 'api_oauth_token'],
+            ['api_token', 'api_job_token'],
+        ],
+        required_together=[
+            ['api_username', 'api_password'],
+        ],
+        required_one_of=[
+            ['api_username', 'api_token', 'api_oauth_token', 'api_job_token'],
+        ],
+        supports_check_mode=True,
+    )
+
+    core(module)
+
+
+if __name__ == '__main__':
+    main()
diff --git a/tests/integration/targets/gitlab_project_badge/aliases b/tests/integration/targets/gitlab_project_badge/aliases
new file mode 100644
index 0000000000..9f72f37111
--- /dev/null
+++ b/tests/integration/targets/gitlab_project_badge/aliases
@@ -0,0 +1,6 @@
+# Copyright (c) 2022, Guillaume MARTINEZ (lunik@tiwabbit.fr)
+# GNU General Public License v3.0+ (see LICENSES/GPL-3.0-or-later.txt or https://www.gnu.org/licenses/gpl-3.0.txt)
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+gitlab/ci
+disabled
diff --git a/tests/integration/targets/gitlab_project_badge/defaults/main.yml b/tests/integration/targets/gitlab_project_badge/defaults/main.yml
new file mode 100644
index 0000000000..bf84a4751a
--- /dev/null
+++ b/tests/integration/targets/gitlab_project_badge/defaults/main.yml
@@ -0,0 +1,11 @@
+---
+# Copyright (c) 2022, Guillaume MARTINEZ <lunik@tiwabbit.fr>
+# GNU General Public License v3.0+ (see LICENSES/GPL-3.0-or-later.txt or https://www.gnu.org/licenses/gpl-3.0.txt)
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+gitlab_api_token: glpat-XXXXXXXXXXXXXXXXXXXX
+gitlab_api_url: https://gitlab.com
+gitlab_project_name: ansible_test_project
+gitlab_badge_link_url: 'https://example.gitlab.com/%{project_path}'
+updated_gitlab_badge_link_url: 'https://test.gitlab.com/%{project_path}'
+gitlab_badge_image_url: 'https://example.gitlab.com/%{project_path}/badges/%{default_branch}/pipeline.svg'
\ No newline at end of file
diff --git a/tests/integration/targets/gitlab_project_badge/tasks/main.yml b/tests/integration/targets/gitlab_project_badge/tasks/main.yml
new file mode 100644
index 0000000000..fa8a806efe
--- /dev/null
+++ b/tests/integration/targets/gitlab_project_badge/tasks/main.yml
@@ -0,0 +1,214 @@
+---
+####################################################################
+# WARNING: These are designed specifically for Ansible tests       #
+# and should not be used as examples of how to write Ansible roles #
+####################################################################
+
+# Copyright (c) 2022, Guillaume MARTINEZ <lunik@tiwabbit.fr>
+# GNU General Public License v3.0+ (see LICENSES/GPL-3.0-or-later.txt or https://www.gnu.org/licenses/gpl-3.0.txt)
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+- name: Install required libs
+  pip:
+    name: python-gitlab
+    state: present
+
+- name: Create {{ gitlab_project_name }}
+  gitlab_project:
+    api_url: "{{ gitlab_api_url }}"
+    validate_certs: False
+    api_token: "{{ gitlab_api_token }}"
+    name: "{{ gitlab_project_name }}"
+    initialize_with_readme: True
+    state: present
+
+- name: Create Badge (check)
+  check_mode: yes
+  gitlab_project_badge:
+    api_url: "{{ gitlab_api_url }}"
+    validate_certs: False
+    api_token: "{{ gitlab_api_token }}"
+    project: "{{ gitlab_project_name }}"
+    state: present
+    link_url: "{{ gitlab_badge_link_url }}"
+    image_url: "{{ gitlab_badge_image_url }}"
+  register: gitlab_badge_create_check_task
+
+- ansible.builtin.debug:
+    var: gitlab_badge_create_check_task
+
+- name: Check module call result
+  assert:
+    that:
+      - gitlab_badge_create_check_task.changed
+      - not gitlab_badge_create_check_task.failed
+
+- name: Create Badge
+  gitlab_project_badge:
+    api_url: "{{ gitlab_api_url }}"
+    validate_certs: False
+    api_token: "{{ gitlab_api_token }}"
+    project: "{{ gitlab_project_name }}"
+    state: present
+    link_url: "{{ gitlab_badge_link_url }}"
+    image_url: "{{ gitlab_badge_image_url }}"
+  register: gitlab_badge_create_task
+
+- ansible.builtin.debug:
+    var: gitlab_badge_create_task
+
+- name: Check module call result
+  assert:
+    that:
+      - gitlab_badge_create_task.changed
+      - not gitlab_badge_create_task.failed
+
+- name: Create Badge (confirmation)
+  gitlab_project_badge:
+    api_url: "{{ gitlab_api_url }}"
+    validate_certs: False
+    api_token: "{{ gitlab_api_token }}"
+    project: "{{ gitlab_project_name }}"
+    state: present
+    link_url: "{{ gitlab_badge_link_url }}"
+    image_url: "{{ gitlab_badge_image_url }}"
+  register: gitlab_badge_create_confirmation_task
+
+- ansible.builtin.debug:
+    var: gitlab_badge_create_confirmation_task
+
+- name: Check module call result
+  assert:
+    that:
+      - not gitlab_badge_create_confirmation_task.changed
+      - not gitlab_badge_create_confirmation_task.failed
+
+- name: Update Badge (check)
+  check_mode: yes
+  gitlab_project_badge:
+    api_url: "{{ gitlab_api_url }}"
+    validate_certs: False
+    api_token: "{{ gitlab_api_token }}"
+    project: "{{ gitlab_project_name }}"
+    state: present
+    link_url: "{{ updated_gitlab_badge_link_url }}"
+    image_url: "{{ gitlab_badge_image_url }}"
+  register: gitlab_badge_update_check_task
+
+- ansible.builtin.debug:
+    var: gitlab_badge_update_check_task
+
+- name: Check module call result
+  assert:
+    that:
+      - gitlab_badge_update_check_task.changed
+      - not gitlab_badge_update_check_task.failed
+
+- name: Update Badge
+  gitlab_project_badge:
+    api_url: "{{ gitlab_api_url }}"
+    validate_certs: False
+    api_token: "{{ gitlab_api_token }}"
+    project: "{{ gitlab_project_name }}"
+    state: present
+    link_url: "{{ updated_gitlab_badge_link_url }}"
+    image_url: "{{ gitlab_badge_image_url }}"
+  register: gitlab_badge_update_task
+
+- ansible.builtin.debug:
+    var: gitlab_badge_update_task
+
+- name: Check module call result
+  assert:
+    that:
+      - gitlab_badge_update_task.changed
+      - not gitlab_badge_update_task.failed
+
+- name: Update Badge (confirmation)
+  gitlab_project_badge:
+    api_url: "{{ gitlab_api_url }}"
+    validate_certs: False
+    api_token: "{{ gitlab_api_token }}"
+    project: "{{ gitlab_project_name }}"
+    state: present
+    link_url: "{{ updated_gitlab_badge_link_url }}"
+    image_url: "{{ gitlab_badge_image_url }}"
+  register: gitlab_badge_update_confirmation_task
+
+- ansible.builtin.debug:
+    var: gitlab_badge_update_confirmation_task
+
+- name: Check module call result
+  assert:
+    that:
+      - not gitlab_badge_update_confirmation_task.changed
+      - not gitlab_badge_update_confirmation_task.failed
+
+- name: Delete Badge (check)
+  check_mode: yes
+  gitlab_project_badge:
+    api_url: "{{ gitlab_api_url }}"
+    validate_certs: False
+    api_token: "{{ gitlab_api_token }}"
+    project: "{{ gitlab_project_name }}"
+    state: absent
+    link_url: "{{ updated_gitlab_badge_link_url }}"
+    image_url: "{{ gitlab_badge_image_url }}"
+  register: gitlab_badge_delete_check_task
+
+- ansible.builtin.debug:
+    var: gitlab_badge_delete_check_task
+
+- name: Check module call result
+  assert:
+    that:
+      - gitlab_badge_delete_check_task.changed
+      - not gitlab_badge_delete_check_task.failed
+
+- name: Delete Badge
+  gitlab_project_badge:
+    api_url: "{{ gitlab_api_url }}"
+    validate_certs: False
+    api_token: "{{ gitlab_api_token }}"
+    project: "{{ gitlab_project_name }}"
+    state: absent
+    link_url: "{{ updated_gitlab_badge_link_url }}"
+    image_url: "{{ gitlab_badge_image_url }}"
+  register: gitlab_badge_delete_task
+
+- ansible.builtin.debug:
+    var: gitlab_badge_delete_task
+
+- name: Check module call result
+  assert:
+    that:
+      - gitlab_badge_delete_task.changed
+      - not gitlab_badge_delete_task.failed
+
+- name: Delete Badge (confirmation)
+  gitlab_project_badge:
+    api_url: "{{ gitlab_api_url }}"
+    validate_certs: False
+    api_token: "{{ gitlab_api_token }}"
+    project: "{{ gitlab_project_name }}"
+    state: absent
+    link_url: "{{ updated_gitlab_badge_link_url }}"
+    image_url: "{{ gitlab_badge_image_url }}"
+  register: gitlab_badge_delete_confirmation_task
+
+- ansible.builtin.debug:
+    var: gitlab_badge_delete_confirmation_task
+
+- name: Check module call result
+  assert:
+    that:
+      - not gitlab_badge_delete_confirmation_task.changed
+      - not gitlab_badge_delete_confirmation_task.failed
+
+- name: Clean up {{ gitlab_project_name }}
+  gitlab_project:
+    api_url: "{{ gitlab_api_url }}"
+    validate_certs: False
+    api_token: "{{ gitlab_api_token }}"
+    name: "{{ gitlab_project_name }}"
+    state: absent