vm.nix

{ pkgs }:
let

  kernel_5_15 = pkgs.stdenv.mkDerivation {
    name = "download-kernel-5.15";
    src = pkgs.fetchurl {
      url = "https://github.com/javierhonduco/lightswitch-kernels/raw/cfb94e9/bzImage-v5.15-fedora38";
      hash = "sha256-pfQUxfBbDcLwpYFwwMubA4tMw5I60WYb7pKCvkPbgXY";
    };
    dontUnpack = true;
    installPhase = ''
      mkdir -p $out
      cp -r $src $out/bzImage
    '';
  };

  kernel_6_0 = pkgs.stdenv.mkDerivation {
    name = "download-kernel-6.0";
    src = pkgs.fetchurl {
      url = "https://github.com/javierhonduco/lightswitch-kernels/raw/cfb94e9/bzImage-v6.0-fedora38";
      hash = "sha256-EgUIm9QUuTFZZp6qYX6YW2PWW63x/whRXajArHqpqfo=";
    };
    dontUnpack = true;
    installPhase = ''
      mkdir -p $out
      cp -r $src $out/bzImage
    '';
  };

  kernel_6_2 = pkgs.stdenv.mkDerivation {
    name = "download-kernel-6.2";
    src = pkgs.fetchurl {
      url = "https://github.com/javierhonduco/lightswitch-kernels/raw/cfb94e9/bzImage-v6.2-fedora38";
      hash = "sha256-FDIkraUFzgJf5rJYQliqS8F7jeO4KWc5OeZDXqu/0uA=";
    };
    dontUnpack = true;
    installPhase = ''
      mkdir -p $out
      cp -r $src $out/bzImage
    '';
  };

  kernel_6_6 = pkgs.stdenv.mkDerivation {
    name = "download-kernel-6.6";
    src = pkgs.fetchurl {
      url = "https://github.com/javierhonduco/lightswitch-kernels/raw/cfb94e9/bzImage-v6.6-fedora38";
      hash = "sha256-9SNl69kpyyPzfxGk8jdPBAXT929a4Ep7vwkerP7SVl8=";
    };
    dontUnpack = true;
    installPhase = ''
      mkdir -p $out
      cp -r $src $out/bzImage
    '';
  };

  kernel_6_8_7 = pkgs.stdenv.mkDerivation {
    name = "download-kernel-6.8.7";
    src = pkgs.fetchurl {
      url = "https://github.com/javierhonduco/lightswitch-kernels/raw/cfb94e9/bzImage_v6.8.7";
      hash = "sha256-fZwGajRi9+otzokRxoss99aH9PLRuyl2UfJ5Echehdo=";
    };
    dontUnpack = true;
    installPhase = ''
      mkdir -p $out
      cp -r $src $out/bzImage
    '';
  };

  kernel_6_9_rc5 = pkgs.stdenv.mkDerivation {
    name = "download-kernel-6.9-rc5";
    src = pkgs.fetchurl {
      url = "https://github.com/javierhonduco/lightswitch-kernels/raw/cfb94e9/bzImage_v6.9-rc5";
      hash = "sha256-EA+nJ1M0/6QFPVA+fYkvXDhBcsmTnALpGr+tCJZsVyw=";
    };
    dontUnpack = true;
    installPhase = ''
      mkdir -p $out
      cp -r $src $out/bzImage
    '';
  };

  vmtest-create-config = lightswitch: pkgs.stdenv.mkDerivation {
    name = "vmtest-dump-config";
    dontUnpack = true;

    src = pkgs.writeText "vmtest.toml" ''
      [[target]]
      name = "Fedora 5.15"
      kernel = "${kernel_5_15}/bzImage"
      arch = "x86_64"
      command = "${lightswitch}/bin/lightswitch --duration 0 --sender=none --cache-dir-base=/tmp"

      [[target]]
      name = "Fedora 6.0"
      kernel = "${kernel_6_0}/bzImage"
      arch = "x86_64"
      command = "${lightswitch}/bin/lightswitch --duration 0 --sender=none --cache-dir-base=/tmp"

      [[target]]
      name = "Fedora 6.2"
      kernel = "${kernel_6_2}/bzImage"
      arch = "x86_64"
      command = "${lightswitch}/bin/lightswitch --duration 0 --sender=none --cache-dir-base=/tmp"

      [[target]]
      name = "Fedora 6.6"
      kernel = "${kernel_6_6}/bzImage"
      arch = "x86_64"
      command = "${lightswitch}/bin/lightswitch --duration 0 --sender=none --cache-dir-base=/tmp"

      [[target]]
      name = "Upstream 6.8.7"
      kernel = "${kernel_6_8_7}/bzImage"
      arch = "x86_64"
      command = "${lightswitch}/bin/lightswitch --duration 0 --sender=none --cache-dir-base=/tmp"

      [[target]]
      name = "Upstream v6.9-rc5"
      kernel = "${kernel_6_9_rc5}/bzImage"
      arch = "x86_64"
      command = "${lightswitch}/bin/lightswitch --duration 0 --sender=none --cache-dir-base=/tmp"
    '';
    nativeBuildInputs = [ ];
    installPhase = ''
      mkdir -p $out
      cp -r $src $out/vmtest.toml
    '';
  };

  vmtest = pkgs.rustPlatform.buildRustPackage {
    name = "vmtest";
    src = pkgs.fetchFromGitHub {
      owner = "danobi";
      repo = "vmtest";
      rev = "51f11bf301fea054342996802a16ed21fb5054f4";
      sha256 = "sha256-qtTq0dnDHi1ITfQzKrXz+1dRMymAFBivWpjXntD09+A=";
    };
    cargoHash = "sha256-SHjjCWz4FVVk1cczkMltRVEB3GK8jz2tVABNSlSZiUc=";
    # nativeCheckInputs = [ pkgs.qemu ];

    # There are some errors trying to access `/build/source/tests/*`.
    doCheck = false;

    meta = with pkgs.lib; {
      description = "Helps run tests in virtual machines";
      homepage = "https://github.com/danobi/vmtest/";
      license = licenses.asl20;
      mainProgram = "";
      maintainers = with maintainers; [ ];
      platforms = platforms.linux;
    };
  };
in
{
  run-vmtest = lightswitch:
    pkgs.writeShellScriptBin "run-vmtests" ''
      ${vmtest}/bin/vmtest --config ${vmtest-create-config lightswitch}/vmtest.toml
    '';
}