flake.nix

{
  description = "netprofiler_lite: shareable object storage throughput benchmark";

  nixConfig = {
    extra-substituters = [
      "https://nix-community.cachix.org"
    ];
    extra-trusted-public-keys = [
      "nix-community.cachix.org-1:mB9FSh9qf2dCimDSUQxH7u0yZ6u5nYzK5vC7g1VQJzY="
    ];
  };

  inputs = {
    nixpkgs.url = "github:NixOS/nixpkgs/nixos-24.11";
    flake-utils.url = "github:numtide/flake-utils";
    fenix.url = "github:nix-community/fenix";
  };

  outputs = { self, nixpkgs, flake-utils, fenix }:
    flake-utils.lib.eachDefaultSystem (system:
      let
        pkgs = import nixpkgs { inherit system; };
        # Fenix toolchain provides `cargo`, `rustc`, `clippy`, `rustfmt`, etc.
        stable = fenix.packages.${system}.stable;
        toolchain = stable.toolchain;
        rustPlatform = pkgs.makeRustPlatform {
          cargo = toolchain;
          rustc = toolchain;
        };
        cargoToml = builtins.fromTOML (builtins.readFile ./Cargo.toml);
        pkgVersion = cargoToml.package.version;
      in
      {
        # Package
        # - `nix build` produces: ./result/bin/netprofiler_lite
        packages.default = rustPlatform.buildRustPackage {
          pname = "netprofiler_lite";
          version = pkgVersion;
          src = ./.;
          cargoLock.lockFile = ./Cargo.lock;
          doCheck = false;

          nativeBuildInputs = [ pkgs.pkg-config ];
        };

        # Static Linux package
        # - `nix build .#linux-static` produces a statically linked musl binary
        # - uses the same Rust toolchain as the normal build (fenix stable)
        packages.linux-static =
          if pkgs.stdenv.isLinux then
            let
              pkgsStatic = pkgs.pkgsStatic;
              toolchainMusl = fenix.packages.${system}.combine [
                stable.cargo
                stable.rustc
                fenix.packages.${system}.targets.x86_64-unknown-linux-musl.stable.rust-std
              ];
              rustPlatformMusl = pkgsStatic.makeRustPlatform {
                cargo = toolchainMusl;
                rustc = toolchainMusl;
              };
            in
            rustPlatformMusl.buildRustPackage {
              pname = "netprofiler_lite";
              version = pkgVersion;
              src = ./.;
              cargoLock.lockFile = ./Cargo.lock;
              doCheck = false;

              # Force the musl target so the resulting binary is fully static.
              cargoBuildTarget = "x86_64-unknown-linux-musl";
              RUSTFLAGS = "-C target-feature=+crt-static";
            }
          else
            pkgs.writeText "linux-static-unavailable" "linux-static is only available on Linux";

        # Apps
        # - `nix run .` runs the binary
        # - `nix run .#bench` runs a default benchmark command
        # - `nix run .#seed` runs the maintainer preseeding script
        apps.default = flake-utils.lib.mkApp {
          drv = self.packages.${system}.default;
        };

        apps.bench = flake-utils.lib.mkApp {
          drv = pkgs.writeShellApplication {
            name = "netprofiler_lite_bench";
            runtimeInputs = [ pkgs.bash pkgs.coreutils ];
            text = ''
              backends="sf-netprofiler-lite-public-6f9c2e-eun1:eu-north-1,sf-netprofiler-lite-public-6f9c2e-euc1:eu-central-1,sf-netprofiler-lite-public-6f9c2e-usw2:us-west-2,sf-netprofiler-lite-public-6f9c2e-use1:us-east-1,https://pub-0323b6896e3e42cb8971495d2f9a2370.r2.dev,https://pub-c02404be13b644a1874a29231dfbe0d2.r2.dev"

              exec "${self.packages.${system}.default}/bin/netprofiler_lite" \
                --backends "$backends" \
                --direction download \
                --concurrency 256 \
                --duration 15 \
                --prefix data-8m \
                --file-count 100 \
                --file-size-mb 8 \
                "$@"
            '';
          };
        };

        # Dev shell
        # - `nix develop` provides the Rust toolchain + utilities used by scripts
        devShells.default = pkgs.mkShell {
          packages = [
            pkgs.pkg-config

            pkgs.awscli2
            pkgs.curl
            pkgs.doppler

            stable.cargo
            stable.clippy
            stable.rustc
            stable.rustfmt
          ];
        };

        apps.seed = flake-utils.lib.mkApp {
          drv = pkgs.writeShellApplication {
            name = "seed_artifacts";
            runtimeInputs = [ pkgs.bash pkgs.awscli2 pkgs.coreutils pkgs.curl pkgs.doppler ];
            text = ''
              exec bash "${self}/scripts/seed_artifacts.sh" "$@"
            '';
          };
        };

        formatter = pkgs.nixfmt-rfc-style;
      });
}