import org.elasticsearch.gradle.LoggedExec
import org.elasticsearch.gradle.VersionProperties
import org.elasticsearch.gradle.internal.DockerBase
import org.elasticsearch.gradle.internal.distribution.InternalElasticsearchDistributionTypes
import org.elasticsearch.gradle.internal.ExportElasticsearchBuildResourcesTask
import org.elasticsearch.gradle.internal.docker.DockerBuildTask
import org.elasticsearch.gradle.internal.docker.DockerSupportPlugin
import org.elasticsearch.gradle.internal.docker.DockerSupportService
import org.elasticsearch.gradle.internal.docker.ShellRetry
import org.elasticsearch.gradle.internal.docker.TransformLog4jConfigFilter
import org.elasticsearch.gradle.internal.docker.*
import org.elasticsearch.gradle.util.GradleUtils
import org.elasticsearch.gradle.Architecture
import java.nio.file.Path
import java.time.temporal.ChronoUnit

apply plugin: 'elasticsearch.legacy-yaml-rest-test'
apply plugin: 'elasticsearch.test.fixtures'
apply plugin: 'elasticsearch.internal-distribution-download'
apply plugin: 'elasticsearch.dra-artifacts'
apply plugin: 'elasticsearch.jdk-download'
apply plugin: 'elasticsearch.repositories'

String buildId = providers.systemProperty('build.id').getOrNull()
boolean useLocalArtifacts = buildId != null && buildId.isBlank() == false && useDra == false

repositories {
  // Define a repository that allows Gradle to fetch a resource from GitHub. This
  // is only used to fetch the `tini` binary, when building the Iron Bank docker image
  // for testing purposes. While in theory we could download `tini` this way for the
  // other Docker variants, the need for the main image to be rebuildable by Docker Hub
  // means that the Dockerfile itself has to fetch the binary.
  ivy {
    url = 'https://github.com/'
    patternLayout {
      artifact '/[organisation]/[module]/releases/download/v[revision]/[module]-[classifier]'
    }
    metadataSources { artifact() }
    content { includeGroup 'krallin' }
  }
}

if (useDra == false) {
  repositories {
    exclusiveContent {
      // Cloud builds bundle some beats
      forRepository {
        ivy {
          name = 'beats'
          if (useLocalArtifacts) {
            url = getLayout().getBuildDirectory().dir("artifacts").get().asFile
            patternLayout {
              artifact '/[organisation]/[module]-[revision]-[classifier].[ext]'
            }
          } else {
            url = "https://artifacts-snapshot.elastic.co/"
            patternLayout {
              if (VersionProperties.isElasticsearchSnapshot()) {
                artifact '/[organization]/[revision]/downloads/[organization]/[module]/[module]-[revision]-[classifier].[ext]'
                artifact '/[organization]/[revision]/downloads/[organization]/filebeat/[module]-[revision]-[classifier].[ext]'
                artifact '/[organization]/[revision]/downloads/[organization]/metricbeat/[module]-[revision]-[classifier].[ext]'
              } else {
                // When building locally we always use snapshot artifacts even if passing `-Dbuild.snapshot=false`.
                // Release builds are always done with a local repo.
                artifact '/[organization]/[revision]-SNAPSHOT/downloads/[organization]/[module]/[module]-[revision]-SNAPSHOT-[classifier].[ext]'
                artifact '/[organization]/[revision]-SNAPSHOT/downloads/[organization]/filebeat/[module]-[revision]-SNAPSHOT-[classifier].[ext]'
                artifact '/[organization]/[revision]-SNAPSHOT/downloads/[organization]/metricbeat/[module]-[revision]-SNAPSHOT-[classifier].[ext]'
              }
            }
          }
          metadataSources { artifact() }
        }
      }
      filter {
        includeGroup("beats")
      }
    }
  }
}

configurations {
  aarch64DockerSource {
    attributes {
      attribute(ArtifactTypeDefinition.ARTIFACT_TYPE_ATTRIBUTE, ArtifactTypeDefinition.DIRECTORY_TYPE)
    }
  }
  dockerSource {
    attributes {
      attribute(ArtifactTypeDefinition.ARTIFACT_TYPE_ATTRIBUTE, ArtifactTypeDefinition.DIRECTORY_TYPE)
    }
  }
  // Iron bank images require a tarball
  aarch64DockerSourceTar
  dockerSourceTar

  log4jConfig
  tini
  allPlugins
  filebeat_aarch64
  filebeat_fips_aarch64
  filebeat_x86_64
  filebeat_fips_x86_64
  metricbeat_aarch64
  metricbeat_fips_aarch64
  metricbeat_x86_64
  metricbeat_fips_x86_64
  fips
}

String tiniArch = Architecture.current() == Architecture.AARCH64 ? 'arm64' : 'amd64'

dependencies {
  aarch64DockerSource project(":distribution:archives:linux-aarch64-tar")
  aarch64DockerSourceTar project(path: ":distribution:archives:linux-aarch64-tar", configuration: "default")
  dockerSource project(":distribution:archives:linux-tar")
  dockerSourceTar project(path: ":distribution:archives:linux-tar", configuration: "default")
  log4jConfig project(path: ":distribution", configuration: 'log4jConfig')
  tini "krallin:tini:0.19.0:${tiniArch}"
  allPlugins project(path: ':plugins', configuration: 'allPlugins')
  allPlugins project(path: ':x-pack:extras:plugins', configuration: 'allPlugins')
  filebeat_aarch64 "beats:filebeat:${VersionProperties.elasticsearch}:linux-arm64@tar.gz"
  filebeat_x86_64 "beats:filebeat:${VersionProperties.elasticsearch}:linux-x86_64@tar.gz"
  filebeat_fips_aarch64 "beats:filebeat-fips:${VersionProperties.elasticsearch}:linux-arm64@tar.gz"
  filebeat_fips_x86_64 "beats:filebeat-fips:${VersionProperties.elasticsearch}:linux-x86_64@tar.gz"

  metricbeat_aarch64 "beats:metricbeat:${VersionProperties.elasticsearch}:linux-arm64@tar.gz"
  metricbeat_x86_64 "beats:metricbeat:${VersionProperties.elasticsearch}:linux-x86_64@tar.gz"

  metricbeat_fips_aarch64 "beats:metricbeat-fips:${VersionProperties.elasticsearch}:linux-arm64@tar.gz"
  metricbeat_fips_x86_64 "beats:metricbeat-fips:${VersionProperties.elasticsearch}:linux-x86_64@tar.gz"

  fips "org.bouncycastle:bc-fips:1.0.2.5"
  fips "org.bouncycastle:bctls-fips:1.0.19"
}

ext.expansions = { Architecture architecture, DockerBase base, String publicationContext = '' ->
  def (major, minor) = VersionProperties.elasticsearch.split("\\.")

  // We tag our Docker images with various pieces of information, including a timestamp
  // for when the image was built. However, this makes it impossible completely cache
  // the image. When developing the Docker images, it's very tedious to completely rebuild
  // an image for every single change. Therefore, outside of CI, we fix the
  // build time to midnight so that the Docker build cache is usable.
  def buildDate = buildParams.ci ? buildParams.buildDate : buildParams.buildDate.truncatedTo(ChronoUnit.DAYS).toString()

  return [
    'arch'               : architecture.classifier,
    'base_image'         : base.image,
    'bin_dir'            : base == DockerBase.IRON_BANK ? 'scripts' : 'bin',
    'build_date'         : buildDate,
    'config_dir'         : base == DockerBase.IRON_BANK ? 'scripts' : 'config',
    'git_revision'       : buildParams.gitRevision.get(),
    'license'            : base == DockerBase.IRON_BANK ? 'Elastic License 2.0' : 'Elastic-License-2.0',
    'package_manager'    : base.packageManager,
    'docker_base'        : base.name().toLowerCase(),
    'docker_context'     : publicationContext,
    'version'            : VersionProperties.elasticsearch,
    'major_minor_version': "${major}.${minor}",
    'retry'              : ShellRetry
  ]
}

/**
 * This filter squashes long runs of newlines so that the output
 * is a little more aesthetically pleasing.
 */
class SquashNewlinesFilter extends FilterReader {
  SquashNewlinesFilter(Reader input) {
    super(new StringReader(input.text.replaceAll("\n{2,}", "\n\n")))
  }
}

private static String toCamel(String input) {
  return input.split("[^a-zA-Z0-9]").collect({ it.substring(0, 1) + it.substring(1).toLowerCase(Locale.ROOT) }).join("")
}

private static String taskName(String prefix, Architecture architecture, DockerBase base, String suffix) {
  return prefix +
    (architecture == Architecture.AARCH64 ? 'Aarch64' : '') +
    (base == DockerBase.DEFAULT ? "" : toCamel(base.name())) +
    suffix
}

ext.dockerBuildContext = { Architecture architecture, DockerBase base, String publicationContext = '' ->
  copySpec {
    final Map<String, String> varExpansions = expansions(architecture, base, publicationContext)
    final Path projectDir = project.projectDir.toPath()

    if (base == DockerBase.IRON_BANK) {
      into('scripts') {
        from projectDir.resolve("src/docker/bin")
        from projectDir.resolve("src/docker/config")
      }
      from(projectDir.resolve("src/docker/iron_bank")) {
        expand(varExpansions)
      }
    } else {
      into('bin') {
        from projectDir.resolve("src/docker/bin")
      }

      into('config') {
        from projectDir.resolve("src/docker/config")
      }
    }
    from(projectDir.resolve("src/docker/${base.dockerfile}")) {
      expand(varExpansions)
      filter SquashNewlinesFilter
      rename base.dockerfile, "Dockerfile"
    }
  }
}

tasks.register("copyNodeKeyMaterial", Sync) {
  def certsDir = file("build/certs")
  def pemFile = file("build/certs/testnode.pem")
  def crtFile = file("build/certs/testnode.crt")
  from project(':x-pack:plugin:core')
    .files(
      'src/test/resources/org/elasticsearch/xpack/security/transport/ssl/certs/simple/testnode.pem',
      'src/test/resources/org/elasticsearch/xpack/security/transport/ssl/certs/simple/testnode.crt'
    )
  into "build/certs"
  doLast {
    certsDir.setReadable(true, false)
    pemFile.setReadable(true, false)
    crtFile.setReadable(true, false)
  }
}

elasticsearch_distributions {
  Architecture.values().each { eachArchitecture ->
    "docker_${eachArchitecture == Architecture.AARCH64 ? '_aarch64' : ''}" {
      architecture = eachArchitecture
      type = InternalElasticsearchDistributionTypes.DOCKER
      version = VersionProperties.getElasticsearch()
      failIfUnavailable = false // This ensures we don't attempt to build images if docker is unavailable
    }
  }
}

interface Injected {
  @Inject
  FileSystemOperations getFs()
}

tasks.named("preProcessFixture").configure {
  dependsOn elasticsearch_distributions.matching { it.architecture == Architecture.current() }
  dependsOn "copyNodeKeyMaterial"
  def injected = project.objects.newInstance(Injected)
  def testFixturesFolder = project.testFixturesDir.absoluteFile
  doLast {
    // tests expect to have an empty repo
    injected.fs.delete {
      it.delete("${testFixturesFolder}/repo")
    }
    ["${testFixturesFolder}/repo",
     "${testFixturesFolder}/logs/default-1",
     "${testFixturesFolder}/logs/default-2"].each { location ->
      File file = new File(location)
      file.mkdirs()
      file.setWritable(true, false)
    }
  }
}

tasks.named("processYamlRestTestResources").configure {
  from project(':x-pack:plugin:core')
    .files(
      'src/test/resources/org/elasticsearch/xpack/security/transport/ssl/certs/simple/testnode.pem',
      'src/test/resources/org/elasticsearch/xpack/security/transport/ssl/certs/simple/testnode.crt'
    )
}

tasks.named("yamlRestTest").configure {
  outputs.doNotCacheIf('Build cache is disabled for Docker tests') { true }
  maxParallelForks = '1'
  include '**/*IT.class'
}

tasks.named("check").configure {
  dependsOn "yamlRestTest"
}

// We build the images used in compose locally, but the pull command insists on using a repository
// thus we must disable it to prevent it from doing so.
// Everything will still be pulled since we will build the local images on a pull
tasks.named("composePull").configure {
  enabled = false
}

tasks.named("composeUp").configure {
  dependsOn tasks.named("preProcessFixture")
}


def exportDockerImages = tasks.register("exportDockerImages")
def exportCompressedDockerImages = tasks.register("exportCompressedDockerImages")
def exportDockerContexts = tasks.register("exportDockerContexts")

void addBuildDockerContextTask(Architecture architecture, DockerBase base, String taskSuffix = 'DockerContext', String classifier = "docker-build-context") {
  String configDirectory = base == DockerBase.IRON_BANK ? 'scripts' : 'config'
  String arch = architecture == Architecture.AARCH64 ? '-aarch64' : ''

  final TaskProvider<Tar> buildDockerContextTask =
    tasks.register(taskName('build', architecture, base, taskSuffix), Tar) {
      archiveExtension = 'tar.gz'
      compression = Compression.GZIP
      archiveClassifier = "${classifier}${arch}"
      archiveBaseName = "elasticsearch${base.suffix}"
      with dockerBuildContext(architecture, base, classifier)

      into(configDirectory) {
        from(configurations.log4jConfig) {
          filter TransformLog4jConfigFilter
        }
      }
      if(base == DockerBase.CLOUD_ESS_FIPS) {

        // If we're performing a release build, but `build.id` hasn't been set, we can
        // infer that we're not at the Docker building stage of the build, and therefore
        // we should skip the beats part of the build.
        String buildId = providers.systemProperty('build.id').getOrNull()
        boolean includeBeats = VersionProperties.isElasticsearchSnapshot() == true || buildId != null || useDra

        if (includeBeats) {
          from configurations.getByName("filebeat_fips_${architecture.classifier}")
          from configurations.getByName("metricbeat_fips_${architecture.classifier}")
          // For some reason, the artifact name can differ depending on what repository we used.
          rename ~/((?:file|metric)beat)-.*\.tar\.gz$/, "\$1-${VersionProperties.elasticsearch}.tar.gz"
        }

        into("plugins") {
          from configurations.allPlugins
        }

        into("fips") {
          into("libs") {
            from configurations.fips
          }
          into("resources") {
            from tasks.named('fipsDockerResources')
          }
        }
      }

      Provider<DockerSupportService> serviceProvider = GradleUtils.getBuildService(
        project.gradle.sharedServices,
        DockerSupportPlugin.DOCKER_SUPPORT_SERVICE_NAME
      )
      onlyIf("$architecture supported") { serviceProvider.get().isArchitectureSupported(architecture) }
    }

  exportDockerContexts.configure {
    dependsOn buildDockerContextTask
  }

  if (base == DockerBase.IRON_BANK) {
    tasks.named("assemble").configure {
      dependsOn(buildDockerContextTask)
    }
  }
}

void addTransformDockerContextTask(Architecture architecture, DockerBase base) {
  def transformTask = tasks.register(taskName("transform", architecture, base, "DockerContext"), Sync) {
    TaskProvider<Tar> buildContextTask = tasks.named(taskName("build", architecture, base, "DockerContext"))
    dependsOn(buildContextTask)

    String arch = architecture == Architecture.AARCH64 ? '-aarch64' : ''
    String archiveName = "elasticsearch${base.suffix}-${VersionProperties.elasticsearch}-docker-build-context${arch}"
    String distributionFolderName = "elasticsearch-${VersionProperties.elasticsearch}"

    from(tarTree("${project.buildDir}/distributions/${archiveName}.tar.gz")) {
      if (base != DockerBase.IRON_BANK && base != DockerBase.DEFAULT) {
        // iron bank always needs a COPY with the tarball file path
        eachFile { FileCopyDetails details ->
          if (details.name.equals("Dockerfile")) {
            filter { String contents ->
              return contents.replaceAll('^RUN curl.*artifacts-no-kpi.*$', "COPY $distributionFolderName .")
                .replaceAll('^RUN tar -zxf /tmp/elasticsearch.tar.gz --strip-components=1$', "")
            }
          }
        }
      }
      if (base == DockerBase.DEFAULT) {
        // iron bank always needs a COPY with the tarball file path
        eachFile { FileCopyDetails details ->
          if (details.name.equals("Dockerfile")) {
            filter { String contents ->
              return contents.replaceAll('^RUN *.*artifacts-no-kpi.*$', "COPY $distributionFolderName .")
                .replaceAll('^RUN tar -zxf /tmp/elasticsearch.tar.gz --strip-components=1 &&', "RUN ")
            }
          }
        }
      }
    }
    into "${project.buildDir}/docker-context/${archiveName}"

    // Since we replaced the remote URL in the Dockerfile, copy in the required file
    if (base == DockerBase.IRON_BANK) {
      from(architecture == Architecture.AARCH64 ? configurations.aarch64DockerSourceTar : configurations.dockerSourceTar)
      from(configurations.tini) {
        rename { _ -> 'tini' }
      }
    } else {
      from(architecture == Architecture.AARCH64 ? configurations.aarch64DockerSource : configurations.dockerSource)
    }

    expansions(architecture, base).findAll { it.key != 'build_date' }.each { k, v ->
      inputs.property(k, { v.toString() })
    }
    Provider<DockerSupportService> serviceProvider = GradleUtils.getBuildService(
      project.gradle.sharedServices,
      DockerSupportPlugin.DOCKER_SUPPORT_SERVICE_NAME
    )
    onlyIf("$architecture supported") { serviceProvider.get().isArchitectureSupported(architecture) }
  }

  // Register transformed context as a project artifact
  def contextConfig = configurations.create(taskName("docker", architecture, base, "Context")) {
    canBeResolved = false
    attributes {
      attribute(ArtifactTypeDefinition.ARTIFACT_TYPE_ATTRIBUTE, ArtifactTypeDefinition.DIRECTORY_TYPE)
      attribute(Attribute.of(Architecture), architecture)
      attribute(Attribute.of(DockerBase), base)
    }
  }

  artifacts.add(contextConfig.name, transformTask)
}


private static List<String> generateTags(DockerBase base, Architecture architecture) {
  final String version = VersionProperties.elasticsearch

  String image = "elasticsearch${base.suffix}"

  String namespace = 'elasticsearch'
  if (base == base == DockerBase.CLOUD_ESS) {
    namespace += '-ci'
  }

  def tags = ["${image}:${architecture.classifier}"]

  if (architecture == Architecture.current()) {
    tags.addAll(["${image}:test", "${image}:${version}", "docker.elastic.co/${namespace}/${image}:${version}"])
  }

  return tags
}

void addBuildDockerImageTask(Architecture architecture, DockerBase base) {

  final TaskProvider<DockerBuildTask> buildDockerImageTask =
    tasks.register(taskName("build", architecture, base, "DockerImage"), DockerBuildTask) {

      TaskProvider<Sync> transformTask = tasks.named(taskName("transform", architecture, base, "DockerContext"))
      dependsOn(transformTask)

      dockerContext.fileProvider(transformTask.map { Sync task -> task.getDestinationDir() })

      noCache = buildParams.ci
      tags = generateTags(base, architecture)
      platforms.add(architecture.dockerPlatform)

      // We don't build the Iron Bank image when we release Elasticsearch, as there's
      // separate process for submitting new releases. However, for testing we do a
      // "test build" so that we can check that the Docker build at least works and
      // is functional.
      if (base == DockerBase.IRON_BANK) {
        Map<String, String> buildArgsMap = [
          'BASE_REGISTRY': 'docker.io',
          'BASE_IMAGE'   : 'redhat/ubi9',
          'BASE_TAG'     : 'latest'
        ]

        // Iron Bank has a single, parameterized base image
        String baseImage = base.image
        for (String key : buildArgsMap.keySet()) {
          baseImage = baseImage.replace('${' + key + '}', buildArgsMap.get(key))
        }

        baseImages = [baseImage]
        buildArgs = buildArgsMap
      } else if(base.image != null) {
        baseImages = [base.image]
      } else {
        baseImages = []
      }

      Provider<DockerSupportService> serviceProvider = GradleUtils.getBuildService(
        project.gradle.sharedServices,
        DockerSupportPlugin.DOCKER_SUPPORT_SERVICE_NAME
      )
      onlyIf("$architecture supported") { serviceProvider.get().isArchitectureSupported(architecture) }

    }

  if (base != DockerBase.IRON_BANK && base != DockerBase.CLOUD_ESS) {
    tasks.named("assemble").configure {
      dependsOn(buildDockerImageTask)
    }
  }
}

void addBuildCloudDockerImageTasks(Architecture architecture) {
  DockerBase dockerBase = DockerBase.CLOUD_ESS
  String arch = architecture == Architecture.AARCH64 ? '-aarch64' : ''
  String contextDir = "${project.buildDir}/docker-context/elasticsearch${dockerBase.suffix}-${VersionProperties.elasticsearch}-docker-build-context${arch}"

  final TaskProvider<Sync> buildContextTask =
    tasks.register(taskName('build', architecture, dockerBase, 'DockerContext'), Sync) {
      into contextDir

      final Path projectDir = project.projectDir.toPath()

      into("plugins") {
        from configurations.allPlugins
      }

      // If we're performing a release build, but `build.id` hasn't been set, we can
      // infer that we're not at the Docker building stage of the build, and therefore
      // we should skip the beats part of the build.
      String buildId = providers.systemProperty('build.id').getOrNull()
      boolean includeBeats = VersionProperties.isElasticsearchSnapshot() == true || buildId != null || useDra

      if (includeBeats) {
        from configurations.getByName("filebeat_${architecture.classifier}")
        from configurations.getByName("metricbeat_${architecture.classifier}")
      }
      // For some reason, the artifact name can differ depending on what repository we used.
      rename ~/((?:file|metric)beat)-.*\.tar\.gz$/, "\$1-${VersionProperties.elasticsearch}.tar.gz"

      String baseSuffix = DockerBase.WOLFI.suffix
      from(projectDir.resolve("src/docker/Dockerfile.ess")) {
        expand(
          [
            base_image : "elasticsearch${baseSuffix}:${architecture.classifier}",
            docker_base: "${dockerBase.name().toLowerCase()}",
            version    : "${VersionProperties.elasticsearch}",
            retry      : ShellRetry
          ]
        )
        filter SquashNewlinesFilter
        rename ~/Dockerfile\.ess$/, 'Dockerfile'
      }
    }

  final TaskProvider<DockerBuildTask> buildDockerImageTask =
    tasks.register(taskName("build", architecture, dockerBase, "DockerImage"), DockerBuildTask) {

      DockerBase base = DockerBase.WOLFI

      TaskProvider<DockerBuildTask> buildBaseTask = tasks.named(taskName("build", architecture, base, "DockerImage"))
      inputs.files(buildBaseTask)

      dockerContext.fileProvider(buildContextTask.map { it.getDestinationDir() })

      noCache = buildParams.ci
      baseImages = []
      tags = generateTags(dockerBase, architecture)
      platforms.add(architecture.dockerPlatform)
      Provider<DockerSupportService> serviceProvider = GradleUtils.getBuildService(
        project.gradle.sharedServices,
        DockerSupportPlugin.DOCKER_SUPPORT_SERVICE_NAME
      )
      onlyIf("$architecture supported") { serviceProvider.get().isArchitectureSupported(architecture) }

    }

  tasks.named("assemble").configure {
    dependsOn(buildDockerImageTask)
  }
}

// fips
TaskProvider<ExportElasticsearchBuildResourcesTask> fipsResourcesTask = tasks.register('fipsDockerResources', ExportElasticsearchBuildResourcesTask)
fipsResourcesTask.configure {
  outputDir = project.layout.buildDirectory.dir('fips-docker-resources').get().asFile
  copy 'fips_java.security'
  copy 'fips_java.policy'
}

for (final Architecture architecture : Architecture.values()) {
  for (final DockerBase base : DockerBase.values()) {
    if (base == DockerBase.CLOUD_ESS) {
      addBuildCloudDockerImageTasks(architecture)
    } else {
      addBuildDockerContextTask(architecture, base)
      addTransformDockerContextTask(architecture, base)
      addBuildDockerImageTask(architecture, base)
    }
    if(base == DockerBase.DEFAULT) {
      // Add additional docker hub specific context which we use solely for publishing to docker hub.
      // At the moment it is exactly the same as the default context.
      addBuildDockerContextTask(architecture, base, 'DockerHubContext', "docker-hub-build-context")
    }
  }
}

/*
 * The export subprojects write out the generated Docker images to disk, so
 * that they can be easily reloaded, for example into a VM for distribution testing
 */
subprojects { Project subProject ->
  if (subProject.name.endsWith('-export')) {
    apply plugin: 'distribution'

    final Architecture architecture = subProject.name.contains('aarch64-') ? Architecture.AARCH64 : Architecture.X64
    DockerBase base = DockerBase.DEFAULT
    if (subProject.name.contains('ironbank-')) {
      base = DockerBase.IRON_BANK
    } else if (subProject.name.contains('cloud-ess-docker')) {
      base = DockerBase.CLOUD_ESS
    } else if (subProject.name.contains('wolfi-')) {
      base = DockerBase.WOLFI
    } else if (subProject.name.contains('cloud-ess-fips-docker')) {
      base = DockerBase.CLOUD_ESS_FIPS
    }

    final String arch = architecture == Architecture.AARCH64 ? '-aarch64' : ''
    final String extension =
      (base == DockerBase.IRON_BANK ? 'ironbank.tar' :
        (base == DockerBase.CLOUD_ESS ? 'cloud-ess.tar' :
          (base == DockerBase.CLOUD_ESS_FIPS ? 'cloud-ess-fips.tar' :
            (base == DockerBase.WOLFI ? 'wolfi.tar' :
              'docker.tar'))))
    final String artifactName = "elasticsearch${arch}${base.suffix}_test"

    final String exportTaskName = taskName("export", architecture, base, 'DockerImage')
    final String buildTaskName = taskName('build', architecture, base, 'DockerImage')
    final String tarFile = "${parent.projectDir}/build/${artifactName}_${VersionProperties.elasticsearch}.${extension}"

    def exportTask = tasks.register(exportTaskName, LoggedExec) {
      inputs.file("${parent.projectDir}/build/markers/${buildTaskName}.marker")
      executable = 'docker'
      outputs.file(tarFile)
      outputs.doNotCacheIf("Build cache is disabled for export tasks") { true }
      args "save",
        "-o",
        tarFile,
        "elasticsearch${base.suffix}:${architecture.classifier}"
      dependsOn(parent.path + ":" + buildTaskName)
      Provider<DockerSupportService> serviceProvider = GradleUtils.getBuildService(
        project.gradle.sharedServices,
        DockerSupportPlugin.DOCKER_SUPPORT_SERVICE_NAME
      )
      onlyIf("$architecture supported") { serviceProvider.get().isArchitectureSupported(architecture) }
    }

    exportDockerImages.configure {
      dependsOn exportTask
    }

    def compressExportTask = tasks.register(taskName("compress", architecture, base, 'DockerImageExport'), Tar) {
      it.from(project.tarTree(tarFile))
      archiveExtension = 'tar.gz'
      it.setCompression(Compression.GZIP)
      it.getArchiveBaseName().set("elasticsearch${base.suffix}-${VersionProperties.elasticsearch}-docker-image")
      it.getArchiveVersion().set("")
      it.getArchiveClassifier().set(architecture == Architecture.AARCH64 ? 'aarch64' : '')
      it.getDestinationDirectory().set(new File(project.parent.buildDir, 'distributions'))
      it.dependsOn(exportTask)
    }

    exportCompressedDockerImages.configure {
      dependsOn compressExportTask
    }

    artifacts.add('default', file(tarFile)) {
      type = 'tar'
      name = artifactName
      builtBy exportTaskName
    }
  }
}

tasks.named('resolveAllDependencies') {
  // Don't try and resolve filebeat or metricbeat snapshots as they may not always be available
  // Also skip building Elasticsearch distributions
  configs = configurations.matching { it.name.contains('beat') == false && it.name.toLowerCase().contains('dockersource') == false }
}