speed-up builds with buildGoCache

flake.nix

@@ -7,10 +7,14 @@
   inputs.nix-editor.url = "github:replit/nix-editor";
   inputs.nix-editor.inputs.nixpkgs.follows = "nixpkgs";
 
+  inputs.build-go-cache.url = "github:numtide/build-go-cache";
+  inputs.build-go-cache.inputs.nixpkgs.follows = "nixpkgs";
+
   outputs = {
     self,
     nixpkgs,
     nix-editor,
+    build-go-cache,
   }: let
     systems = [
       "aarch64-darwin"
@@ -27,6 +31,7 @@
     packages = eachSystem (system:
       import ./nix {
         inherit self nixpkgs rev system nix-editor;
+        inherit (build-go-cache.legacyPackages.${system}) buildGoCache;
       });
     devShells = eachSystem (system: {
       default = self.packages.${system}.devShell;

nix/default.nix

@@ -4,12 +4,13 @@
   rev,
   self,
   system,
+  buildGoCache
 }: let
   pkgs = nixpkgs.legacyPackages.${system};
   nix-editor-pkg = nix-editor.packages.${system}.nix-editor;
 in rec {
   default = upm;
   devShell = pkgs.callPackage ./devshell {nix-editor = nix-editor-pkg;};
   fmt = pkgs.callPackage ./fmt {};
-  upm = pkgs.callPackage ./upm {inherit rev;};
+  upm = pkgs.callPackage ./upm {inherit rev; inherit buildGoCache;};
 }

nix/upm/default.nix

@@ -1,15 +1,31 @@
 {
-  lib,
   buildGoModule,
   rev,
   makeWrapper,
+  buildGoCache,
+  lib,
+  runCommand,
 }:
+let
+  vendorHash = "sha256-7t87/28pBxU7A+SRvw6xxJQ2wErZ/76OJC8feNEQ80E=";
+
+  goCache = buildGoCache {
+    # keep this up-to-date in CI with:
+    # $ nix run 'github:numtide/build-go-cache#get-external-imports' -- ./. imported-packages
+    importPackagesFile = ./imported-packages;
+    # FIXME: Somehow we get cache invalidation everytime if we don't do this, also it uses a source filter
+    src = runCommand "go-mod" {} ''
+      install -D ${../../go.mod} $out/go.mod
+      install -D ${../../go.sum} $out/go.sum
+    '';
+    inherit vendorHash;
+  };
+in
 buildGoModule rec {
   pname = "upm";
   version = rev;
-
   src = builtins.path {
-    name = "${pname}-${version}-src";
+    name = "${pname}-src";
     path = ../../.;
     filter = path: _:
       builtins.all (block: (builtins.baseNameOf path) != block) [
@@ -24,8 +40,6 @@ buildGoModule rec {
       ];
   };
 
-  vendorHash = "sha256-7t87/28pBxU7A+SRvw6xxJQ2wErZ/76OJC8feNEQ80E=";
-
   ldflags = [
     "-X github.com/replit/upm/internal/cli.version=${rev}"
   ];
@@ -34,7 +48,7 @@ buildGoModule rec {
     go generate ./internal/backends/python
   '';
 
-  buildInputs = [makeWrapper];
+  buildInputs = [makeWrapper goCache];
 
   subPackages = ["cmd/upm"];
 
@@ -46,5 +60,8 @@ buildGoModule rec {
       --set PYPI_MAP_DB "$out/pypi_map.sqlite"
   '';
 
+  inherit vendorHash;
+  proxyVendor = true; # we only support proxyVendor with buildGoCache just now
+
   doCheck = false;
 }

nix/upm/imported-packages

@@ -0,0 +1,53 @@
+archive/tar
+archive/zip
+bufio
+bytes
+cloud.google.com/go/bigquery
+compress/gzip
+context
+crypto/md5
+database/sql
+embed
+encoding/binary
+encoding/hex
+encoding/json
+encoding/xml
+errors
+flag
+fmt
+github.com/BurntSushi/toml
+github.com/hashicorp/go-version
+github.com/kballard/go-shellquote
+github.com/mattn/go-sqlite3
+github.com/natefinch/atomic
+github.com/smacker/go-tree-sitter
+github.com/smacker/go-tree-sitter/javascript
+github.com/smacker/go-tree-sitter/python
+github.com/smacker/go-tree-sitter/typescript/tsx
+github.com/smacker/go-tree-sitter/typescript/typescript
+github.com/spf13/cobra
+golang.org/x/net/context
+golang.org/x/net/html
+golang.org/x/term
+google.golang.org/api/iterator
+gopkg.in/DataDog/dd-trace-go.v1/ddtrace
+gopkg.in/DataDog/dd-trace-go.v1/ddtrace/tracer
+gopkg.in/yaml.v2
+io
+io/fs
+log
+net/http
+net/url
+os
+os/exec
+path
+path/filepath
+reflect
+regexp
+runtime
+sort
+strconv
+strings
+sync
+testing
+time