Add Nix flake for installing the CLI on macOS and Linux

README.md

@@ -27,6 +27,35 @@ brew update
 brew upgrade baseten
 ```
 
+### Nix (macOS and Linux)
+
+This repository is a Nix flake. Run the CLI without installing anything:
+
+```bash
+nix run github:basetenlabs/baseten-cli -- --help
+```
+
+or install it into your profile:
+
+```bash
+nix profile install github:basetenlabs/baseten-cli
+```
+
+To install with [Home Manager](https://github.com/nix-community/home-manager),
+add the flake as an input and put the package in `home.packages`:
+
+```nix
+{
+  inputs.baseten-cli.url = "github:basetenlabs/baseten-cli";
+
+  # In your home-manager configuration (with `inputs` passed through):
+  # home.packages = [ inputs.baseten-cli.packages.${pkgs.system}.default ];
+}
+```
+
+The flake also exposes `overlays.default`, which adds the package as
+`pkgs.baseten`.
+
 ### Prebuilt binaries
 
 Download the [latest release](https://github.com/basetenlabs/baseten-cli/releases/latest) for Windows, macOS, or Linux, extract the archive, and place the `baseten` executable on your `PATH`.

flake.nix

@@ -0,0 +1,50 @@
+{
+  description = "CLI for Baseten";
+
+  inputs = {
+    nixpkgs.url = "github:nixos/nixpkgs/nixos-26.05";
+    flake-utils.url = "github:numtide/flake-utils";
+  };
+
+  outputs =
+    {
+      self,
+      nixpkgs,
+      flake-utils,
+    }:
+    flake-utils.lib.eachDefaultSystem (
+      system:
+      let
+        pkgs = import nixpkgs { inherit system; };
+        baseten = pkgs.callPackage ./nix/package.nix {
+          version = if self ? shortRev then "unstable-${self.shortRev}" else "dev";
+        };
+      in
+      {
+        packages = {
+          default = baseten;
+          inherit baseten;
+        };
+
+        checks = {
+          build = baseten;
+          # Smoke test: the built binary runs and reports the expected version.
+          version = pkgs.runCommand "baseten-version-check" { } ''
+            got="$(${pkgs.lib.getExe baseten} version)"
+            want=${pkgs.lib.escapeShellArg baseten.version}
+            if [ "$got" != "$want" ]; then
+              echo "baseten version reported '$got', expected '$want'" >&2
+              exit 1
+            fi
+            ${pkgs.lib.getExe baseten} --help > /dev/null
+            touch $out
+          '';
+        };
+      }
+    )
+    // {
+      overlays.default = final: prev: {
+        baseten = final.callPackage ./nix/package.nix { };
+      };
+    };
+}

nix/package.nix

@@ -0,0 +1,31 @@
+{
+  lib,
+  buildGoModule,
+  version ? "dev",
+}:
+buildGoModule {
+  pname = "baseten";
+  inherit version;
+
+  src = lib.cleanSource ../.;
+
+  # Update by setting to lib.fakeHash and copying the hash from the build error.
+  vendorHash = "sha256-ghtyvdt80fy6JCw0J20a70fai9fUQyJW5vS8gQibZWE=";
+
+  subPackages = [ "cmd/baseten" ];
+
+  env.CGO_ENABLED = 0;
+
+  ldflags = [
+    "-s"
+    "-w"
+    "-X github.com/basetenlabs/baseten-cli/internal/cmd.Version=${version}"
+  ];
+
+  meta = {
+    description = "CLI for Baseten";
+    homepage = "https://github.com/basetenlabs/baseten-cli";
+    license = lib.licenses.mit;
+    mainProgram = "baseten";
+  };
+}