---
# This job will create or update a badge when extended using the following template.
# It requires a Project Access Token or Personal Access Token with api scope set
# as a CI/CD variable called GL_TOKEN.
# To create your own automatic badges using anybadge, add the following to your
# .gitlab-ci.yml:

# badge:your_custom_job:
#   extends: .badge
#   variables:
#     VALUE: "1.2.3"
#     LABEL: "version"  # Manditory unless an anybadge template is provided through BADGE_EXTRA_ARGS.
#     COLOR: "red"  # Optional.
#     URL: "${CI_PROJECT_URL}"
#     BADGE_EXTRA_ARG: "" # Optional. Additional argument to pass to anybadge.
#     BADGE_NAME: ""  # Optional. Name of the badge in Settings. Defaults to $LABEL

# Choose a color from this list: https://github.com/jongracecox/anybadge#colors
# Even fancier usage is displayed here: https://github.com/jongracecox/anybadge#examples

# Alternatively, you may provide your own IMAGE_URL directly:

# badge:your_custom_job:
#   extends: .badge-raw
#   variables:
#     IMAGE_URL: "https://img.shields.io/pypi/v/${PYTHON_PACKAGE}"
#     URL: "https://pypi.org/project/${PYTHON_PACKAGE}"
#     BADGE_NAME: "PyPI"  # Name of the badge in Settings.

# Make sure your_custom_job is replaced and unique, and you've changed the variables
# according to your preferences.

include:
  - remote: https://jobs.just-ci.dev/HEAD/project-automation/generic.yaml

.badge:
  extends: .badge-raw
  script:
    - BADGE_NAME=${BADGE_NAME:-${LABEL}}
    - IMAGE_URL=${CI_PROJECT_URL}/-/jobs/artifacts/${CI_DEFAULT_BRANCH}/raw/ci_badges/${BADGE_NAME}.svg?job=${CI_JOB_NAME}
    - set --
    - if [ "${LABEL}" != "" ]; then set -- -l "$LABEL" "$@"; fi
    - if [ "${COLOR}" != "" ]; then set -- -c "$COLOR" "$@"; fi
    - mkdir -p ci_badges
    - |
      echo "[+] Created ci_badges/${BADGE_NAME}.svg"
      echo "Label: ${LABEL}"
      echo "Value: ${VALUE}"
      echo "Color: ${COLOR}"
      echo "Extra argument: ${BADGE_EXTRA_ARG}"
      echo "URL: ${URL}"
      echo "Image URL: ${IMAGE_URL}"
      echo "You can use the image URL to point to this badge from, for example, a readme."
    - anybadge --version
    - anybadge -v "${VALUE}" "$@" -f "ci_badges/${BADGE_NAME}.svg"
      ${BADGE_EXTRA_ARG}
    - !reference [.badge-raw, script]
  artifacts:
    paths:
      - ci_badges/*.svg
    when: always

.badge-raw:
  image: registry.gitlab.com/just-ci/images/anybadge:latest
  stage: .post
  variables:
    GIT_STRATEGY: none
  script:
    - |
      if [ "${DISABLE_BADGE_CREATION}" != "" ]; then
        echo "[*] Badge creation is disabled."
        exit 0
      elif [ "${GL_TOKEN}" == "" ]; then
        echo "[!] No GL_TOKEN set. If you want to disable badge creation, set a variable called DISABLE_BADGE_CREATION to any value."
        exit 1
      fi
    - |
      echo "[*] Creating/updating badge..."
      BADGE_ID=$(curl -SsLf -H "PRIVATE-TOKEN: ${GL_TOKEN}" "${CI_API_V4_URL}/projects/${CI_PROJECT_ID}/badges" | jq --arg BADGE_NAME "${BADGE_NAME}" '.[] | select(.name==$BADGE_NAME) | .id') || true
      if [ -z "${BADGE_ID}" ]; then
        echo "[-] Badge doesn't exist yet. We'll create a new one..."
        STATUS_CODE=$(curl -SsL -w "%{http_code}" -X POST -H "PRIVATE-TOKEN: ${GL_TOKEN}" -d "image_url=${IMAGE_URL}&link_url=${URL:-$CI_PROJECT_URL}&name=${BADGE_NAME}" "${CI_API_V4_URL}/projects/${CI_PROJECT_ID}/badges" -o /dev/null)
      else
        echo "[*] Updating badge ID ${BADGE_ID}..."
        STATUS_CODE=$(curl -SsL -w "%{http_code}" -X PUT -H "PRIVATE-TOKEN: ${GL_TOKEN}" -d "image_url=${IMAGE_URL}&link_url=${URL:-$CI_PROJECT_URL}" "${CI_API_V4_URL}/projects/${CI_PROJECT_ID}/badges/${BADGE_ID}" -o /dev/null)
      fi

      if [ "${STATUS_CODE:0:1}" = "2" ]; then
        echo "[+] Badge set successfully."
      elif [ "${STATUS_CODE}" = "403" ]; then
        echo "[!] Access Forbidden. Does your GL_TOKEN have access to badge creation?"
        exit 1
      else
        echo "[!] Something went wrong. We received the status code ${STATUS_CODE}."
        exit 1
      fi
  allow_failure: true
  rules:
    - if: "$CI_COMMIT_BRANCH == $CI_DEFAULT_BRANCH"

badge:GL_TOKEN:
  extends: .fix:GL_TOKEN
  variables:
    JOB_NAME: badge
  rules:
    - if:
        "$CI_COMMIT_BRANCH == $CI_DEFAULT_BRANCH && $GL_TOKEN == null &&
        $DISABLE_BADGE_CREATION == null"
