nix-config/flake.nix

{
  inputs = {
    nixpkgs.url = github:NixOS/nixpkgs/nixos-21.11;

    naersk = {
      url = github:nix-community/naersk;
      #inputs.nixpkgs.follows = "nixpkgs";
    };

    microvm = {
      url = github:astro/microvm.nix;
      inputs.nixpkgs.follows = "nixpkgs";
    };

    radio-conf = {
      url = github:dump-dvb/radio-conf;
    };

    data-accumulator = {
      url = github:dump-dvb/data-accumulator;
    };

    decode-server = {
      url = github:dump-dvb/decode-server;
    };

    dvb-api = {
      url = github:dump-dvb/dvb-api;
    };

    stops = {
      url = github:dump-dvb/stop-names;
      flake = false;
    };

    windshield = {
      url = github:dump-dvb/windshield;
      inputs.nixpkgs.follows = "nixpkgs";
    };

    docs = {
      url = github:dump-dvb/documentation;
      inputs.nixpkgs.follows = "nixpkgs";
    };

    wartrammer = {
      url = github:dump-dvb/wartrammer-40k;
    };

    clicky-bunty-server = {
      url = github:dump-dvb/clicky-bunty-server;
    };
  };

  outputs = { self, nixpkgs, naersk, microvm, radio-conf, data-accumulator, decode-server, dvb-api, stops, windshield, docs, wartrammer, clicky-bunty-server, ... }@inputs:
    let
      pkgs = nixpkgs.legacyPackages."x86_64-linux";
      lib = pkgs.lib;

      # function that generates a system with the given number
      generate_system = (number:
        {
          "traffic-stop-box-${toString number}" = nixpkgs.lib.nixosSystem {
            system = "x86_64-linux";
            specialArgs = { inherit inputs; };
            modules = [
              ./hosts/traffic-stop-box/configuration.nix
              ./hosts/traffic-stop-box/hardware-configuration.nix
              ./hardware/configuration-dell-wyse-3040.nix
              ./modules/gnuradio.nix
              ./modules/radio_wireguard_client.nix
              ./modules/options.nix
              {
                nixpkgs.overlays = [ radio-conf.overlay."x86_64-linux" decode-server.overlay."x86_64-linux" ];
                dvb-dump.systemNumber = number;
                dvb-dump.stopsJson = "${stops}/stops.json";
              }
            ];
          };
        }
      );

      # increment this number if you want to add a new system
      numberOfSystems = 10;
      # list of accending system numbers
      #id_list = ((num: if num <= 0 then [ num ] else [ num ] ++ (id_list (num - 1))) (numberOfSystems - 1));
      id_list = [ 0 1 2 3 4 ];
      # list of nixos systems
      list_of_systems = builtins.map generate_system id_list;
      # attribute set of all traffic stop boxes
      stop_boxes = nixpkgs.lib.foldr (x: y: nixpkgs.lib.mergeAttrs x y) { } list_of_systems;

      boxes = id_list;

      installScript = (target: (pkgs.writeScriptBin "deploy" ''
        #!${pkgs.runtimeShell}
        ssh root@10.13.37.${toString (target + 100)} "ps cax | grep \"nixos-rebuild\" > /dev/null"
        if [ $? -eq 0 ]
        then
            echo "Process is running."
            exit
        else
            echo "Process is not running."
            nix copy --to ssh://root@10.13.37.${toString (target + 100)} ${self}
            ssh root@10.13.37.${toString (target + 100)} -- nixos-rebuild switch --flake ${self} -L
        fi
      ''));

      # concatanes commands together
      deployBoxes = (systems: lib.strings.concatStringsSep " "
        (builtins.map (system: "${(installScript system)}/bin/deploy") systems));

      deployAllScript = (pkgs.writeScriptBin "deploy-all" (
        '' 
                #!${pkgs.runtimeShell} -ex
                ${pkgs.parallel}/bin/parallel --will-cite -j10 ::: ${deployBoxes boxes} || echo "Some deployment failed"
        ''
      ));

      individualScripts = lib.foldl (x: y: lib.mergeAttrs x y) { } (builtins.map (number: { "deploy-box-${toString number}" = (installScript number); }) boxes);


      #deployScripts = pkgs.callPackage ./pkgs/deployment.nix {
      #  boxes = id_list;
      #  self = self;
      #};

      packages = ({
        traffic-stop-box = self.nixosConfigurations.traffic-stop-box-0.config.system.build.vm;
        data-hoarder = self.nixosConfigurations.data-hoarder.config.system.build.vm;
        mobile-box-vm = self.nixosConfigurations.mobile-box.config.system.build.vm;
        mobile-box-iso = self.nixosConfigurations.mobile-box.config.system.build.isoImage;
        staging-microvm = self.nixosConfigurations.staging-data-hoarder.config.microvm.declaredRunner;
      } // {
        deploy-all = deployAllScript;
      } // individualScripts);
    in
    {
      defaultPackage."x86_64-linux" = self.nixosConfigurations.traffic-stop-box-0.config.system.build.vm;
      packages."x86_64-linux" = packages;

      nixosConfigurations =
        let
          data-hoarder-modules = [
            ./modules/data-accumulator.nix
            ./modules/nginx.nix
            ./modules/public_api.nix
            ./modules/map.nix
            ./modules/file_sharing.nix
            ./modules/options.nix
            ./modules/grafana.nix
            ./modules/website.nix
            ./modules/documentation.nix
            ./modules/clicky-bunty.nix
            {
              nixpkgs.overlays = [
                data-accumulator.overlay."x86_64-linux"
                dvb-api.overlay."x86_64-linux"
                windshield.overlay."x86_64-linux"
                docs.overlay."x86_64-linux"
                clicky-bunty-server.overlay."x86_64-linux"
              ];
              dvb-dump.stopsJson = "${stops}/stops.json";
              dvb-dump.graphJson = "${stops}/graph.json";
            }
          ];
        in
        (stop_boxes // {
          mobile-box = nixpkgs.lib.nixosSystem {
            system = "x86_64-linux";
            specialArgs = { inherit inputs; };
            modules = [
              "${nixpkgs}/nixos/modules/installer/cd-dvd/installation-cd-base.nix"
              ./hosts/mobile-box/configuration.nix
              ./hosts/mobile-box/hardware-configuration.nix
              ./hardware/configuration-dell-wyse-3040.nix
              ./modules/options.nix
              ./modules/mobile-box.nix
              {
                nixpkgs.overlays = [
                  radio-conf.overlay."x86_64-linux"
                  decode-server.overlay."x86_64-linux"
                  data-accumulator.overlay."x86_64-linux"
                  wartrammer.overlay."x86_64-linux"
                ];
                dvb-dump.stopsJson = "${stops}/stops.json";
                dvb-dump.systemNumber = 130;
              }
            ];
          };
        } // {
          data-hoarder = nixpkgs.lib.nixosSystem {
            system = "x86_64-linux";
            specialArgs = { inherit inputs; };
            modules = ([
              ./hosts/data-hoarder/configuration.nix
              ./modules/wireguard_server.nix
            ] ++ data-hoarder-modules);
          };
          staging-data-hoarder = nixpkgs.lib.nixosSystem {
            system = "x86_64-linux";
            specialArgs = { inherit inputs; };
            modules = ([
              ./hosts/staging/configuration.nix
              microvm.nixosModules.microvm
            ] ++ data-hoarder-modules);
          };
        });

      hydraJobs = {
        data-hoarder."x86_64-linux" = self.nixosConfigurations.data-hoarder.config.system.build.toplevel;
        traffic-stop-box-0."x86_64-linux" = self.nixosConfigurations.traffic-stop-box-0.config.system.build.toplevel;
        mobile-box."x86_64-linux" = self.nixosConfigurations.mobile-box.config.system.build.toplevel;
        staging-data-hoarder."x86_64-linux" = self.nixosConfigurations.staging-data-hoarder.config.system.build.toplevel;
      };
    };
}