| 30 | } |
| 31 | |
| 32 | func prepareEnv() { |
| 33 | // Unset CODER_ environment variables |
| 34 | for _, env := range os.Environ() { |
| 35 | if strings.HasPrefix(env, "CODER_") { |
| 36 | split := strings.SplitN(env, "=", 2) |
| 37 | if err := os.Unsetenv(split[0]); err != nil { |
| 38 | panic(err) |
| 39 | } |
| 40 | } |
| 41 | } |
| 42 | |
| 43 | // Override default OS values to ensure the same generated results. |
| 44 | err := os.Setenv("CLIDOCGEN_CACHE_DIRECTORY", "~/.cache") |
| 45 | if err != nil { |
| 46 | panic(err) |
| 47 | } |
| 48 | err = os.Setenv("CLIDOCGEN_CONFIG_DIRECTORY", "~/.config/coderv2") |
| 49 | if err != nil { |
| 50 | panic(err) |
| 51 | } |
| 52 | err = os.Setenv("TMPDIR", "/tmp") |
| 53 | if err != nil { |
| 54 | panic(err) |
| 55 | } |
| 56 | } |
| 57 | |
| 58 | func deleteEmptyDirs(dir string) error { |
| 59 | return filepath.Walk(dir, func(path string, info os.FileInfo, err error) error { |