update docker image workflow

Author: HenryHengZJ

.github/workflows/docker-image.yml

@@ -19,6 +19,14 @@ on:
                 options:
                     - dev
                     - prod
+            image_type:
+                description: 'Type of image to build and push.'
+                type: choice
+                required: true
+                default: 'main'
+                options:
+                    - 'main'
+                    - 'worker'
             node_version:
                 description: 'Node.js version to build this image with.'
                 type: choice
@@ -41,6 +49,7 @@ jobs:
               id: defaults
               run: |
                   echo "registry=${{ github.event.inputs.registry || 'aws_ecr' }}" >> $GITHUB_OUTPUT
+                  echo "image_type=${{ github.event.inputs.image_type || 'main' }}" >> $GITHUB_OUTPUT
                   echo "node_version=${{ github.event.inputs.node_version || '20' }}" >> $GITHUB_OUTPUT
                   echo "tag_version=${{ github.event.inputs.tag_version || 'latest' }}" >> $GITHUB_OUTPUT
 
@@ -82,10 +91,24 @@ jobs:
               uses: docker/build-push-action@v5.3.0
               with:
                   context: .
-                  file: ${{ steps.defaults.outputs.registry == 'docker_hub' && './docker/Dockerfile' || 'Dockerfile' }}
+                  file: |
+                      ${{ 
+                        steps.defaults.outputs.image_type == 'worker' && 'docker/worker/Dockerfile.worker' ||
+                        (steps.defaults.outputs.registry == 'docker_hub' && './docker/Dockerfile' || 'Dockerfile') 
+                      }}
                   build-args: |
                       NODE_VERSION=${{ steps.defaults.outputs.node_version }}
                   platforms: linux/amd64,linux/arm64
                   push: true
                   tags: |
-                      ${{ steps.defaults.outputs.registry == 'docker_hub' && format('flowiseai/flowise-bullmq:{0}', steps.defaults.outputs.tag_version) || format('{0}.dkr.ecr.{1}.amazonaws.com/flowise-bullmq:{2}', secrets.AWS_ACCOUNT_ID, secrets.AWS_REGION, steps.defaults.outputs.tag_version) }}
+                      ${{ 
+                          steps.defaults.outputs.registry == 'docker_hub' && 
+                          format('flowiseai/flowise-bullmq{0}:{1}', 
+                              steps.defaults.outputs.image_type == 'worker' && '-worker' || '', 
+                              steps.defaults.outputs.tag_version) ||
+                          format('{0}.dkr.ecr.{1}.amazonaws.com/flowise-bullmq{2}:{3}', 
+                              secrets.AWS_ACCOUNT_ID, 
+                              secrets.AWS_REGION, 
+                              steps.defaults.outputs.image_type == 'worker' && '-worker' || '',
+                              steps.defaults.outputs.tag_version) 
+                      }}