From 4a153bda1b2108ff0a37bfe9a6c7f8019627ffa6 Mon Sep 17 00:00:00 2001
From: Victor Carlos Erenu <victor.erenu@wazuh.com>
Date: Tue, 20 Jan 2026 20:15:05 +0700
Subject: [PATCH] Add developement option when tag name is only version without
 stage

---
 .github/workflows/Procedure_push_docker_images.yml | 6 +++++-
 build-docker-images/build-images.sh                | 6 +++++-
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/.github/workflows/Procedure_push_docker_images.yml b/.github/workflows/Procedure_push_docker_images.yml
index b8661b46..5cd2e42a 100644
--- a/.github/workflows/Procedure_push_docker_images.yml
+++ b/.github/workflows/Procedure_push_docker_images.yml
@@ -166,7 +166,11 @@ jobs:
             ./build-images.sh -v $WAZUH_VER -r $REVISION -d $DEV_STAGE -rg $IMAGE_REGISTRY -m
           fi
         else
-          ./build-images.sh -v $IMAGE_TAG -r $REVISION -rg $IMAGE_REGISTRY -m
+          if [ "${{ inputs.dev }}" = true ]; then
+            ./build-images.sh -v $IMAGE_TAG -r $REVISION -rg $IMAGE_REGISTRY -m -ref ${{ inputs.reference }}
+          else
+            ./build-images.sh -v $IMAGE_TAG -r $REVISION -rg $IMAGE_REGISTRY -m
+          fi
         fi
         # Save .env file (generated by build-images.sh) contents to $GITHUB_ENV
         ENV_FILE_PATH="../.env"
diff --git a/build-docker-images/build-images.sh b/build-docker-images/build-images.sh
index a7c381ce..d12f2eb1 100755
--- a/build-docker-images/build-images.sh
+++ b/build-docker-images/build-images.sh
@@ -70,7 +70,11 @@ build() {
             IMAGE_TAG="${WAZUH_IMAGE_VERSION}-${WAZUH_DEV_STAGE,,}"
         fi
     else
-        IMAGE_TAG="${WAZUH_IMAGE_VERSION}"
+        if  [ "${WAZUH_TAG_REFERENCE}" ];then
+            IMAGE_TAG="${WAZUH_IMAGE_VERSION}-${WAZUH_TAG_REFERENCE}"
+        else
+            IMAGE_TAG="${WAZUH_IMAGE_VERSION}"
+        fi
     fi
 
     echo WAZUH_VERSION=$WAZUH_IMAGE_VERSION > ../.env