From 237d17b61f8ac970bd077a870f71357f23c65906 Mon Sep 17 00:00:00 2001
From: pducolin <paola.ducolin@datadoghq.com>
Date: Mon, 16 Dec 2024 18:55:09 +0100
Subject: [PATCH] [gitlab] add manual trigger for bump in datadog-agent

---
 .gitlab-ci.yml | 38 +++++++++++++++++++++++++++++---------
 tasks/ci.py    |  8 ++++++++
 2 files changed, 37 insertions(+), 9 deletions(-)

diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index ddc2d5ce1..af5d5155b 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -100,14 +100,9 @@ release-runner-image:
     - if: $CI_COMMIT_BRANCH == "main"
       when: on_success
 
-bump-version-on-datadog-agent:
-  stage: post-release
-  image: ${CI_REGISTRY_IMAGE_TEST}:${CI_COMMIT_SHORT_SHA}
-  tags: ["arch:amd64"]
-  needs: ["release-runner-image"]
-  rules:
-    - if: $CI_COMMIT_BRANCH == "main"
-      when: on_success
+.bump-version-on-datadog-agent:
+  variables:
+    EXTRA_UPDATE_ARGS: ""
   before_script:
     - set +x
     - export GITHUB_APP_USER_ID=153269286 # Can be found on https://api.github.com/users/agent-platform-auto-pr[bot]
@@ -126,10 +121,35 @@ bump-version-on-datadog-agent:
     - pushd datadog-agent
     - git checkout -b auto-bump/bump-test-infra-$CI_COMMIT_SHORT_SHA
     - export PREVIOUS_SHA=$(cat .gitlab/common/test_infra_version.yml | grep 'TEST_INFRA_DEFINITIONS_BUILDIMAGES:' | awk -F " " '{print $NF}')
-    - inv -e buildimages.update-test-infra-definitions --commit-sha $CI_COMMIT_SHA
+    - inv -e buildimages.update-test-infra-definitions --commit-sha $CI_COMMIT_SHA $EXTRA_UPDATE_ARGS
     - git add test/new-e2e/go.mod test/new-e2e/go.sum .gitlab/common/test_infra_version.yml
     - git commit -m "[test-infra-definitions][automated] Bump test-infra-definitions to $CI_COMMIT_SHORT_SHA"
     - git push -f origin auto-bump/bump-test-infra-$CI_COMMIT_SHORT_SHA
     - popd
     - pip install -r requirements.txt
     - inv ci.create-bump-pr-and-close-stale-ones-on-datadog-agent --branch auto-bump/bump-test-infra-$CI_COMMIT_SHORT_SHA --new-commit-sha $CI_COMMIT_SHA --old-commit-sha $PREVIOUS_SHA
+
+bump-version-on-datadog-agent-dev:
+  extends: .bump-version-on-datadog-agent
+  stage: post-release
+  image: ${CI_REGISTRY_IMAGE_TEST}:${CI_COMMIT_SHORT_SHA}
+  tags: ["arch:amd64"]
+  needs: ["build-runner-image"]
+  rules:
+    - if: $CI_COMMIT_BRANCH == "main"
+      when: never
+    - when: manual
+      allow_failure: true
+  variables:
+    EXTRA_UPDATE_ARGS: "--is-dev-image"
+
+bump-version-on-datadog-agent:
+  extends: .bump-version-on-datadog-agent
+  stage: post-release
+  image: ${CI_REGISTRY_IMAGE_TEST}:${CI_COMMIT_SHORT_SHA}
+  tags: ["arch:amd64"]
+  needs: ["release-runner-image"]
+  rules:
+    - if: $CI_COMMIT_BRANCH == "main"
+      when: on_success
+  
\ No newline at end of file
diff --git a/tasks/ci.py b/tasks/ci.py
index d1f33cfb6..200dd8a11 100644
--- a/tasks/ci.py
+++ b/tasks/ci.py
@@ -9,12 +9,16 @@
 
 @task
 def create_bump_pr_and_close_stale_ones_on_datadog_agent(ctx, branch: str, new_commit_sha: str, old_commit_sha: str):
+    is_dev_branch = False
     if os.getenv("CI") != "true":
         print("This task should only be run in CI")
         return
     if os.getenv("GITHUB_TOKEN") is None:
         print("GITHUB_TOKEN is not set")
         return
+    if os.getenv("CI_COMMIT_BRANCH") != "main":
+        print("Running on a dev branch")
+        is_dev_branch = True
 
     repo = Github(auth=Auth.Token(os.environ["GITHUB_TOKEN"])).get_repo("DataDog/datadog-agent")
     pr_body = f"""
@@ -36,6 +40,10 @@ def create_bump_pr_and_close_stale_ones_on_datadog_agent(ctx, branch: str, new_c
 
     print(f"PR created: {new_pr.html_url}")
 
+    if is_dev_branch:
+        print("Skipping stale PRs check since this is a dev branch")
+        return
+
     print("Looking for stale auto bump PRs...")
 
     issues = repo.get_issues(state="open", labels=["automatic/test-infra-bump"])