nixos-images/flake.nix

{
  description = "NixOS images";

  #inputs.nixos-unstable.url = "github:NixOS/nixpkgs/nixos-unstable-small";
  inputs.nixos-unstable.url = "github:jmbaur/nixpkgs/etc-overlay-cross";
  inputs.nixos-stable.url = "github:NixOS/nixpkgs/nixos-24.05";

  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;

      packages = forAllSystems (
        system:
        let
          netboot = nixpkgs: (import (nixpkgs + "/nixos/release.nix") { }).netboot.${system};
          kexec-installer =
            nixpkgs: modules:
            (nixpkgs.legacyPackages.${system}.nixos (modules ++ [ 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 [
            { system.kexec-installer.name = "nixos-kexec-installer-noninteractive"; }
            self.nixosModules.noninteractive
          ];
          kexec-installer-nixos-stable-noninteractive = kexec-installer nixos-stable [
            { system.kexec-installer.name = "nixos-kexec-installer-noninteractive"; }
            self.nixosModules.noninteractive
          ];

          netboot-installer-nixos-unstable = netboot-installer nixos-unstable;
          netboot-installer-nixos-stable = netboot-installer nixos-stable;
        }
      );

      crossPackages = {
        x86_64-linux =
          let
            kexec-installer =
              nixpkgs: modules:
              (nixpkgs.legacyPackages.x86_64-linux.pkgsCross.riscv64.nixos (
                modules
                ++ [
                  self.nixosModules.kexec-installer
                  self.nixosModules.noninteractive
                ]
              )).config.system.build.kexecTarball;
          in
          {
            kexec-installer-nixos-unstable-noninteractive-riscv64 = kexec-installer nixos-unstable [ ];
            # needs a backport to nixpkgs-24.05
            #kexec-installer-nixos-stable-noninteractive-riscv64 = kexec-installer nixos-stable [ ];
          };
      };
    in
    {
      packages = nixos-unstable.lib.recursiveUpdate packages crossPackages;
      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
              pkgs = nixos-unstable.legacyPackages.x86_64-linux;
            in
            {
              kexec-installer-unstable = pkgs.callPackage ./nix/kexec-installer/test.nix {
                kexecTarball = self.packages.x86_64-linux.kexec-installer-nixos-unstable-noninteractive;
              };
              shellcheck = pkgs.runCommand "shellcheck" { nativeBuildInputs = [ pkgs.shellcheck ]; } ''
                shellcheck ${(pkgs.nixos [ self.nixosModules.kexec-installer ]).config.system.build.kexecRun}
                touch $out
              '';
              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; };
            };
        in
        nixos-unstable.lib.recursiveUpdate packages { x86_64-linux = checks; };
    };
}