flake.nix

{
  description = "NixOS images";

  # https://github.com/NixOS/nixpkgs/pull/364924
  inputs.nixos-unstable.url = "github:Mic92/nixpkgs/compression";
  inputs.nixos-stable.url = "github:Mic92/nixpkgs/compression-24.11";

  nixConfig.extra-substituters = [ "https://nix-community.cachix.org" ];
  nixConfig.extra-trusted-public-keys = [ "nix-community.cachix.org-1:mB9FSh9qf2dCimDSUo8Zy7bkq5CX+/rkCWyvRCYg3Fs=" ];

  outputs = { self, nixos-unstable, nixos-stable }:
    let
      supportedSystems = [ "aarch64-linux" "x86_64-linux" ];
      forAllSystems = nixos-unstable.lib.genAttrs supportedSystems;
    in
    {
      packages = forAllSystems (system:
        let
          netboot = nixpkgs: (import (nixpkgs + "/nixos/release.nix") { }).netboot.${system};
          kexec-installer = nixpkgs: module: (nixpkgs.legacyPackages.${system}.nixos [ module self.nixosModules.kexec-installer ]).config.system.build.kexecTarball;
          netboot-installer = nixpkgs: (nixpkgs.legacyPackages.${system}.nixos [ self.nixosModules.netboot-installer ]).config.system.build.netboot;
          image-installer = nixpkgs: (nixpkgs.legacyPackages.${system}.nixos [ self.nixosModules.image-installer ]).config.system.build.isoImage;
        in
        {
          netboot-nixos-unstable = netboot nixos-unstable;
          netboot-nixos-stable = netboot nixos-stable;
          kexec-installer-nixos-unstable = kexec-installer nixos-unstable {};
          kexec-installer-nixos-stable = kexec-installer nixos-stable {};

          image-installer-nixos-unstable = image-installer nixos-unstable;
          image-installer-nixos-stable = image-installer nixos-stable;

          kexec-installer-nixos-unstable-noninteractive = kexec-installer nixos-unstable {
            _file = __curPos.file;
            system.kexec-installer.name = "nixos-kexec-installer-noninteractive";
            imports = [
              self.nixosModules.noninteractive
            ];
          };
          kexec-installer-nixos-stable-noninteractive = kexec-installer nixos-stable {
            _file = __curPos.file;
            system.kexec-installer.name = "nixos-kexec-installer-noninteractive";
            imports = [
              self.nixosModules.noninteractive
            ];
          };

          netboot-installer-nixos-unstable = netboot-installer nixos-unstable;
          netboot-installer-nixos-stable = netboot-installer nixos-stable;
        });
      nixosModules = {
        kexec-installer = ./nix/kexec-installer/module.nix;
        noninteractive = ./nix/noninteractive.nix;
        # TODO: also add a test here once we have https://github.com/NixOS/nixpkgs/pull/228346 merged
        netboot-installer = ./nix/netboot-installer/module.nix;
        image-installer = ./nix/image-installer/module.nix;
      };
      checks =
        let
          # re-export the packages as checks
          packages = forAllSystems (system: nixos-unstable.lib.mapAttrs' (n: nixos-unstable.lib.nameValuePair "package-${n}") self.packages.${system});
          checks =
            let
              pkgsUnstable = nixos-unstable.legacyPackages.x86_64-linux;
              pkgsStable = nixos-stable.legacyPackages.x86_64-linux;

              bootTests = pkgs: channel: suffix: pkgs.lib.mapAttrs' (name: pkgs.lib.nameValuePair "${name}${suffix}") (pkgs.callPackages ./nix/image-installer/tests.nix {
                nixpkgs = channel;
                nixosModules = self.nixosModules;
              });
            in
            {
              kexec-installer-unstable = pkgsUnstable.callPackage ./nix/kexec-installer/test.nix {
                kexecTarball = self.packages.x86_64-linux.kexec-installer-nixos-unstable-noninteractive;
              };

              kexec-installer-stable = nixos-stable.legacyPackages.x86_64-linux.callPackage ./nix/kexec-installer/test.nix {
                kexecTarball = self.packages.x86_64-linux.kexec-installer-nixos-stable-noninteractive;
              };
              shellcheck = pkgsUnstable.runCommand "shellcheck"
                {
                  nativeBuildInputs = [ pkgsUnstable.shellcheck ];
                } ''
                shellcheck ${(pkgsUnstable.nixos [self.nixosModules.kexec-installer]).config.system.build.kexecRun}
                touch $out
              '';
            } // (bootTests pkgsUnstable nixos-unstable "-nixos-unstable")
              // (bootTests pkgsStable nixos-stable "-nixos-stable");
        in
        nixos-unstable.lib.recursiveUpdate packages { x86_64-linux = checks; };
    };
}