Add option to control the cache location.

namespacelabs · Jun 27, 2024 · b8c110f · b8c110f
1 parent 90fe717
commit b8c110f
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/internal/cli/fncobra/main.go b/internal/cli/fncobra/main.go
@@ -166,6 +166,8 @@ func doMain(opts MainOpts) (colors.Style, error) {
  "If set to true, we also output debug log messages for API responses.")
  rootCmd.PersistentFlags().BoolVar(&enableErrorTracing, "error_tracing", enableErrorTracing,
  "If set to true, prints a trace of foundation errors leading to the root cause with source info.")
+ rootCmd.PersistentFlags().StringVar(&dirs.CacheDir, "cache_dir", dirs.CacheDir,
+ "Where to place cache contents.")
 
  storedrun.SetupFlags(rootCmd.PersistentFlags())
 
@@ -178,6 +180,7 @@ func doMain(opts MainOpts) (colors.Style, error) {
  "debug_to_console",
  "debug_to_file",
  "debug_api_response",
+ "cache_dir",
  } {
  _ = rootCmd.PersistentFlags().MarkHidden(noisy)
  }

diff --git a/internal/workspace/dirs/dirs.go b/internal/workspace/dirs/dirs.go
@@ -10,11 +10,18 @@ import (
  "strings"
 )
 
+var CacheDir string
+
 func Cache() (string, error) {
+ if CacheDir != "" {
+ return filepath.Join(CacheDir, "ns"), nil
+ }
+
  dir, err := os.UserCacheDir()
  if err != nil {
  return dir, err
  }
+
  return filepath.Join(dir, "ns"), nil
 }