forked from gnolang/gno
-
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat: add 'gno env' subcommand (gnolang#1233)
- Loading branch information
Showing
26 changed files
with
485 additions
and
127 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,113 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "context" | ||
| "flag" | ||
|
|
||
| "github.com/gnolang/gno/gnovm/pkg/gnoenv" | ||
| "github.com/gnolang/gno/tm2/pkg/commands" | ||
| ) | ||
|
|
||
| type envCfg struct { | ||
| json bool | ||
| } | ||
|
|
||
| func newEnvCmd(io commands.IO) *commands.Command { | ||
| c := &envCfg{} | ||
| return commands.NewCommand( | ||
| commands.Metadata{ | ||
| Name: "env", | ||
| ShortUsage: "env [flags] <pkgsym>", | ||
| ShortHelp: "`env` prints Gno environment information", | ||
| }, | ||
| c, | ||
| func(_ context.Context, args []string) error { | ||
| return execEnv(c, args, io) | ||
| }, | ||
| ) | ||
| } | ||
|
|
||
| func (c *envCfg) RegisterFlags(fs *flag.FlagSet) { | ||
| fs.BoolVar( | ||
| &c.json, | ||
| "json", | ||
| false, | ||
| "Prints the environment in JSON format instead of as a shell script.", | ||
| ) | ||
| } | ||
|
|
||
| type envVar struct { | ||
| Key string | ||
| Value string | ||
| } | ||
|
|
||
| func findEnv(env []envVar, name string) string { | ||
| for _, e := range env { | ||
| if e.Key == name { | ||
| return e.Value | ||
| } | ||
| } | ||
| return "" | ||
| } | ||
|
|
||
| type envPrinter func(vars []envVar, io commands.IO) | ||
|
|
||
| func execEnv(cfg *envCfg, args []string, io commands.IO) error { | ||
| envs := []envVar{ | ||
| // GNOROOT Should point to the local location of the GNO repository. | ||
| // It serves as the gno equivalent of `GOROOT`. | ||
| {Key: "GNOROOT", Value: gnoenv.RootDir()}, | ||
| // GNOHOME Should point to the user local configuration. | ||
| // The most common place for this should be $HOME/gno. | ||
| {Key: "GNOHOME", Value: gnoenv.HomeDir()}, | ||
| } | ||
|
|
||
| // Setup filters | ||
| filters := make([]envVar, len(args)) | ||
| for i, arg := range args { | ||
| filters[i] = envVar{Key: arg, Value: findEnv(envs, arg)} | ||
| } | ||
|
|
||
| // Setup printer | ||
| var printerEnv envPrinter | ||
| if cfg.json { | ||
| printerEnv = printJSON | ||
| } else { | ||
| printerEnv = getPrinterShell(len(args) == 0) | ||
| } | ||
|
|
||
| // Print environements | ||
| if len(filters) > 0 { | ||
| printerEnv(filters, io) | ||
| } else { | ||
| printerEnv(envs, io) | ||
| } | ||
|
|
||
| return nil | ||
| } | ||
|
|
||
| func getPrinterShell(printkeys bool) envPrinter { | ||
| return func(vars []envVar, io commands.IO) { | ||
| for _, env := range vars { | ||
| if printkeys { | ||
| io.Printf("%s=%q\n", env.Key, env.Value) | ||
| } else { | ||
| io.Printf("%s\n", env.Value) | ||
| } | ||
| } | ||
| } | ||
| } | ||
|
|
||
| func printJSON(vars []envVar, io commands.IO) { | ||
| io.Println("{") | ||
| for i, env := range vars { | ||
| io.Printf("\t%q: %q", env.Key, env.Value) | ||
| if i != len(vars)-1 { | ||
| io.Printf(",") | ||
| } | ||
|
|
||
| // Jump to next line | ||
| io.Printf("\n") | ||
| } | ||
| io.Println("}") | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,43 @@ | ||
| package main | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "testing" | ||
| ) | ||
|
|
||
| func TestEnvApp(t *testing.T) { | ||
| const ( | ||
| testGnoRootEnv = "/faster/better/stronger" | ||
| testGnoHomeEnv = "/around/the/world" | ||
| ) | ||
|
|
||
| t.Setenv("GNOROOT", testGnoRootEnv) | ||
| t.Setenv("GNOHOME", testGnoHomeEnv) | ||
| tc := []testMainCase{ | ||
| // shell | ||
| {args: []string{"env", "foo"}, stdoutShouldBe: "\n"}, | ||
| {args: []string{"env", "foo", "bar"}, stdoutShouldBe: "\n\n"}, | ||
| {args: []string{"env", "GNOROOT"}, stdoutShouldBe: testGnoRootEnv + "\n"}, | ||
| {args: []string{"env", "GNOHOME", "storm"}, stdoutShouldBe: testGnoHomeEnv + "\n\n"}, | ||
| {args: []string{"env"}, stdoutShouldContain: fmt.Sprintf("GNOROOT=%q", testGnoRootEnv)}, | ||
| {args: []string{"env"}, stdoutShouldContain: fmt.Sprintf("GNOHOME=%q", testGnoHomeEnv)}, | ||
|
|
||
| // json | ||
| {args: []string{"env", "-json"}, stdoutShouldContain: fmt.Sprintf("\"GNOROOT\": %q", testGnoRootEnv)}, | ||
| {args: []string{"env", "-json"}, stdoutShouldContain: fmt.Sprintf("\"GNOHOME\": %q", testGnoHomeEnv)}, | ||
| { | ||
| args: []string{"env", "-json", "GNOROOT"}, | ||
| stdoutShouldBe: fmt.Sprintf("{\n\t\"GNOROOT\": %q\n}\n", testGnoRootEnv), | ||
| }, | ||
| { | ||
| args: []string{"env", "-json", "GNOROOT", "storm"}, | ||
| stdoutShouldBe: fmt.Sprintf("{\n\t\"GNOROOT\": %q,\n\t\"storm\": \"\"\n}\n", testGnoRootEnv), | ||
| }, | ||
| { | ||
| args: []string{"env", "-json", "storm"}, | ||
| stdoutShouldBe: "{\n\t\"storm\": \"\"\n}\n", | ||
| }, | ||
| } | ||
|
|
||
| testMainCaseRun(t, tc) | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.