move prefix out of

Author: donbobka

commands/push.sh

@@ -26,6 +26,8 @@ if plugin_read_list_into_result BUILDKITE_PLUGIN_DOCKER_COMPOSE_BUILD; then
   build_services=("${result[@]}")
 fi
 
+metadata_key_prefix="$(plugin_read_config METADATA_KEY_PREFIX '')"
+
 # Then we figure out what to push, and where
 for line in $(plugin_read_list PUSH) ; do
   IFS=':' read -r -a tokens <<< "$line"
@@ -37,7 +39,7 @@ for line in $(plugin_read_list PUSH) ; do
   elif in_array "${service_name}" "${build_services[@]}"; then
     echo "~~~ :docker: Service was built in this step, using that image"
     service_image="$(default_compose_image_for_service "${service_name}")"
-  elif prebuilt_image="$(get_prebuilt_image "$service_name")"; then
+  elif prebuilt_image="$(get_prebuilt_image "$metadata_key_prefix" "$service_name")"; then
     echo "~~~ :docker: Using pre-built image ${prebuilt_image}"
 
     # Only pull it down once
@@ -63,15 +65,15 @@ for line in $(plugin_read_list PUSH) ; do
   if [[ ${#tokens[@]} -eq 1 ]] ; then
     echo "${group_type} :docker: Pushing images for ${service_name}" >&2;
     retry "$push_retries" run_docker_compose push "${service_name}"
-    set_prebuilt_image "${service_name}" "${service_image}"
+    set_prebuilt_image "${metadata_key_prefix}" "${service_name}" "${service_image}"
     target_image="${service_image}" # necessary for build-alias
   # push: "service-name:repo:tag"
   else
     target_image="$(IFS=:; echo "${tokens[*]:1}")"
     echo "${group_type} :docker: Pushing image $target_image" >&2;
     plugin_prompt_and_run docker tag "$service_image" "$target_image"
     retry "$push_retries" plugin_prompt_and_run docker push "$target_image"
-    set_prebuilt_image "${service_name}" "${target_image}"
+    set_prebuilt_image "${metadata_key_prefix}" "${service_name}" "${target_image}"
   fi
 done
 
@@ -82,5 +84,5 @@ for service_alias in $(plugin_read_list BUILD_ALIAS) ; do
     exit 1
   fi
 
-  set_prebuilt_image "$service_alias" "${target_image}"
+  set_prebuilt_image "${metadata_key_prefix}" "$service_alias" "${target_image}"
 done

commands/run.sh

@@ -35,6 +35,8 @@ while read -r name ; do
   fi
 done <<< "$(plugin_read_list PULL)"
 
+metadata_key_prefix="$(plugin_read_config METADATA_KEY_PREFIX '')"
+
 # A list of tuples of [service image cache_from] for build_image_override_file
 prebuilt_service_overrides=()
 prebuilt_services=()
@@ -45,7 +47,7 @@ for service_name in "${prebuilt_candidates[@]}" ; do
   if [[ -n "$prebuilt_image_override" ]] && [[ "$service_name" == "$run_service" ]] ; then
     echo "~~~ :docker: Overriding run image for $service_name"
     prebuilt_image="$prebuilt_image_override"
-  elif prebuilt_image=$(get_prebuilt_image "$service_name") ; then
+  elif prebuilt_image=$(get_prebuilt_image "$metadata_key_prefix" "$service_name") ; then
      echo "~~~ :docker: Found a pre-built image for $service_name"
   fi
 

lib/metadata.bash

@@ -7,7 +7,15 @@ function plugin_check_metadata_exists() {
 
 # Read agent metadata for the plugin
 function plugin_get_metadata() {
-  local key="docker-compose-plugin-$1"
+  local metadata_key_prefix="$1"
+  local key="$2"
+
+  if [ -n "$metadata_key_prefix" ]; then
+    metadata_key_prefix="$metadata_key_prefix-"
+  fi
+
+  key="docker-compose-plugin-$metadata_key_prefix$key"
+
   if plugin_check_metadata_exists "$key"; then
     plugin_prompt buildkite-agent meta-data get "$key"
     buildkite-agent meta-data get "$key"
@@ -18,8 +26,16 @@ function plugin_get_metadata() {
 
 # Write agent metadata for the plugin
 function plugin_set_metadata() {
-  local key="docker-compose-plugin-$1"
-  local value="$2"
+  local metadata_key_prefix="$1"
+  local key="$2"
+  local value="$3"
+
+  if [ -n "$metadata_key_prefix" ]; then
+    metadata_key_prefix="$metadata_key_prefix-"
+  fi
+
+  key="docker-compose-plugin-$metadata_key_prefix$key"
+
   plugin_prompt_and_must_run buildkite-agent meta-data set "$key" "$value"
 }
 
@@ -44,13 +60,17 @@ function prebuilt_image_meta_data_key() {
 
 # Sets a prebuilt image for a service name
 function set_prebuilt_image() {
-  local service="$1"
-  local image="$2"
-  plugin_set_metadata "$(prebuilt_image_meta_data_key "$service")" "$image"
+  local metadata_key_prefix="$1"
+  local service="$2"
+  local image="$3"
+
+  plugin_set_metadata "$metadata_key_prefix" "$(prebuilt_image_meta_data_key "$service")" "$image"
 }
 
 # Gets a prebuilt image for a service name
 function get_prebuilt_image() {
-  local service="$1"
-  plugin_get_metadata "$(prebuilt_image_meta_data_key "$service")"
+  local metadata_key_prefix="$1"
+  local service="$2"
+
+  plugin_get_metadata "$metadata_key_prefix" "$(prebuilt_image_meta_data_key "$service")"
 }

plugin.yml

@@ -131,6 +131,8 @@ configuration:
       type: boolean
     workdir:
       type: string
+    metadata-key-prefix:
+      type: string
   anyOf:
     - required:
       - run

tests/metadata.bats

@@ -36,7 +36,7 @@ load '../lib/metadata'
   # Only expect the 'exists' command to be called, not the 'get'
   stub buildkite-agent "meta-data exists docker-compose-plugin-built-image-tag-test : exit 1"
 
-  run get_prebuilt_image "test"
+  run get_prebuilt_image "" "test"
 
   assert_failure
   unstub buildkite-agent
@@ -47,7 +47,7 @@ load '../lib/metadata'
     "meta-data exists docker-compose-plugin-built-image-tag-test : exit 0" \
     "meta-data get docker-compose-plugin-built-image-tag-test : exit 0" 
 
-  run get_prebuilt_image "test"
+  run get_prebuilt_image "" "test"
 
   assert_success
   assert_output --partial "buildkite-agent meta-data get docker-compose-plugin-built-image-tag-test"