{ lib, stdenv, fetchurl, jdk17_headless, jdk11_headless, makeWrapper, bash, coreutils, gnugrep, gnused, ps, nixosTests }:

let
  versionMap = {
    "3_8" = {
      kafkaVersion = "3.8.0";
      scalaVersion = "2.13";
      sha256 = "sha256-4Cl8xv2wnvnZkFdRsl0rYpwXUo+GKbYFYe7/h84pCZw=";
      jre = jdk17_headless;
      nixosTest = nixosTests.kafka.kafka_3_8;
    };
    "3_7" = {
      kafkaVersion = "3.7.1";
      scalaVersion = "2.13";
      sha256 = "sha256-YqyuShQ92YPcfrSATVdEugxQsZm1CPWZ7wAQIOJVj8k=";
      jre = jdk17_headless;
      nixosTest = nixosTests.kafka.kafka_3_7;
    };
    "3_6" = {
      kafkaVersion = "3.6.2";
      scalaVersion = "2.13";
      sha256 = "sha256-wxfkf3cUHTFG6VY9nLodZIbIHmcLIR7OasRqn3Lkqqw=";
      jre = jdk17_headless;
      nixosTest = nixosTests.kafka.kafka_3_6;
    };
  };

  build = versionInfo: stdenv.mkDerivation rec {
    version = "${versionInfo.scalaVersion}-${versionInfo.kafkaVersion}";
    pname = "apache-kafka";

    src = fetchurl {
      url = "mirror://apache/kafka/${versionInfo.kafkaVersion}/kafka_${version}.tgz";
      inherit (versionInfo) sha256;
    };

    nativeBuildInputs = [ makeWrapper ];
    buildInputs = [ versionInfo.jre bash gnugrep gnused coreutils ps ];

    installPhase = ''
      mkdir -p $out
      cp -R config libs $out

      mkdir -p $out/bin
      cp bin/kafka* $out/bin
      cp bin/connect* $out/bin

      # allow us the specify logging directory using env
      substituteInPlace $out/bin/kafka-run-class.sh \
        --replace 'LOG_DIR="$base_dir/logs"' 'LOG_DIR="$KAFKA_LOG_DIR"'

      substituteInPlace $out/bin/kafka-server-stop.sh \
        --replace 'ps' '${ps}/bin/ps'

      for p in $out/bin\/*.sh; do
        wrapProgram $p \
          --set JAVA_HOME "${versionInfo.jre}" \
          --set KAFKA_LOG_DIR "/tmp/apache-kafka-logs" \
          --prefix PATH : "${bash}/bin:${coreutils}/bin:${gnugrep}/bin:${gnused}/bin"
      done
      chmod +x $out/bin\/*
    '';

    passthru = {
      inherit (versionInfo) jre; # Used by the NixOS module to select the supported JRE
      tests.nixos = versionInfo.nixosTest;
    };

    meta = {
      homepage = "https://kafka.apache.org";
      description = "High-throughput distributed messaging system";
      license = lib.licenses.asl20;
      sourceProvenance = [ lib.sourceTypes.binaryBytecode ];
      maintainers = [ lib.maintainers.ragge ];
      platforms = lib.platforms.unix;
    };
  };
in lib.mapAttrs'
  (majorVersion: versionInfo: lib.nameValuePair "apacheKafka_${majorVersion}" (build versionInfo))
  versionMap