Add ability to load azd env vars into shell

.vscode/cspell.global.yaml

@@ -80,6 +80,7 @@ ignoreWords:
     - infradelete
     - inputhelper
     - ippre
+    - jmespath
     - JOBOBJECT
     - kubernetes
     - kusto

cli/azd/.vscode/launch.json

@@ -32,14 +32,14 @@
             "program": "${workspaceFolder}",
             "console": "integratedTerminal",
             "args": [
-                "auth", "login"
+                "env", "list", "-o", "json", "-q", "[?IsDefault].DotEnvPath"
                 //"pipeline", "config", "--principal-name", "foo", "--provider", "github"
                 //"package", "api", "--debug"
                 //"provision"
                 //"up"
                 //"env", "new"
             ],
-            //"cwd": "~/workspace/cwd/path",
+            "cwd": "/workspaces/todo-csharp-sql",
             // "env": {
             //     "INT_TAG_VALUE":"1989"
             // }

cli/azd/cmd/env.go

@@ -75,7 +75,7 @@ func envActions(root *actions.ActionDescriptor) *actions.ActionDescriptor {
 		Command:        newEnvGetValuesCmd(),
 		FlagsResolver:  newEnvGetValuesFlags,
 		ActionResolver: newEnvGetValuesAction,
-		OutputFormats:  []output.Format{output.JsonFormat, output.EnvVarsFormat},
+		OutputFormats:  []output.Format{output.JsonFormat, output.EnvVarsFormat, output.ExportFormat},
 		DefaultFormat:  output.EnvVarsFormat,
 	})
 

cli/azd/cmd/middleware/query.go

@@ -0,0 +1,65 @@
+package middleware
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+
+	"github.com/azure/azure-dev/cli/azd/cmd/actions"
+	"github.com/azure/azure-dev/cli/azd/internal"
+	"github.com/jmespath/go-jmespath"
+)
+
+type QueryMiddleware struct {
+	global *internal.GlobalCommandOptions
+}
+
+func (m *QueryMiddleware) Run(ctx context.Context, next NextFn) (*actions.ActionResult, error) {
+	// Execute the next middleware or action in the chain
+	fmt.Println(m.global.Query)
+	result, err := next(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	if m.global.Query == "" {
+		return result, nil
+	}
+
+	// Convert the result to JSON
+	resultJSON, err := json.Marshal(result)
+	if err != nil {
+		return nil, fmt.Errorf("marshalling action result: %w", err)
+	}
+
+	// Apply the JMESPath query
+	var jsonResult interface{}
+	err = json.Unmarshal(resultJSON, &jsonResult)
+	if err != nil {
+		return nil, fmt.Errorf("unmarshalling action result: %w", err)
+	}
+
+	filteredResult, err := jmespath.Search(m.global.Query, jsonResult)
+	if err != nil {
+		return nil, fmt.Errorf("applying JMESPath query: %w", err)
+	}
+
+	// Convert the filtered result back to an ActionResult
+	filteredResultJSON, err := json.Marshal(filteredResult)
+	if err != nil {
+		return nil, fmt.Errorf("marshalling filtered result: %w", err)
+	}
+
+	var actionResult actions.ActionResult
+	err = json.Unmarshal(filteredResultJSON, &actionResult)
+	if err != nil {
+		return nil, fmt.Errorf("unmarshalling filtered result: %w", err)
+	}
+
+	return &actionResult, nil
+}
+
+// NewQueryMiddleware is the registration function for the QueryMiddleware
+func NewQueryMiddleware(global *internal.GlobalCommandOptions) Middleware {
+	return &QueryMiddleware{global: global}
+}

cli/azd/cmd/root.go

@@ -99,6 +99,7 @@ func NewRootCmd(
 					"no-prompt",
 					false,
 					"Accepts the default value instead of prompting, or it fails if there is no default.")
+			rootCmd.PersistentFlags().StringVarP(&opts.Query, "query", "q", "", "Run jmespath query on results")
 
 			// The telemetry system is responsible for reading these flags value and using it to configure the telemetry
 			// system, but we still need to add it to our flag set so that when we parse the command line with Cobra we
@@ -336,7 +337,6 @@ func NewRootCmd(
 		UseMiddlewareWhen("telemetry", middleware.NewTelemetryMiddleware, func(descriptor *actions.ActionDescriptor) bool {
 			return !descriptor.Options.DisableTelemetry
 		})
-
 	// Register common dependencies for the IoC rootContainer
 	if rootContainer == nil {
 		rootContainer = ioc.NewNestedContainer(nil)

cli/azd/cmd/testdata/TestUsage-azd-auth-login.snap

@@ -18,9 +18,10 @@ Flags
         --use-device-code                      	: When true, log in by using a device code instead of a browser.
 
 Global Flags
-    -C, --cwd string 	: Sets the current working directory.
-        --debug      	: Enables debugging and diagnostics logging.
-        --no-prompt  	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -C, --cwd string   	: Sets the current working directory.
+        --debug        	: Enables debugging and diagnostics logging.
+        --no-prompt    	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -q, --query string 	: Run jmespath query on results
 
 Find a bug? Want to let us know how we're doing? Fill out this brief survey: https://aka.ms/azure-dev/hats.
 

cli/azd/cmd/testdata/TestUsage-azd-auth-logout.snap

@@ -9,9 +9,10 @@ Flags
     -h, --help 	: Gets help for logout.
 
 Global Flags
-    -C, --cwd string 	: Sets the current working directory.
-        --debug      	: Enables debugging and diagnostics logging.
-        --no-prompt  	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -C, --cwd string   	: Sets the current working directory.
+        --debug        	: Enables debugging and diagnostics logging.
+        --no-prompt    	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -q, --query string 	: Run jmespath query on results
 
 Find a bug? Want to let us know how we're doing? Fill out this brief survey: https://aka.ms/azure-dev/hats.
 

cli/azd/cmd/testdata/TestUsage-azd-auth.snap

@@ -13,9 +13,10 @@ Flags
     -h, --help 	: Gets help for auth.
 
 Global Flags
-    -C, --cwd string 	: Sets the current working directory.
-        --debug      	: Enables debugging and diagnostics logging.
-        --no-prompt  	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -C, --cwd string   	: Sets the current working directory.
+        --debug        	: Enables debugging and diagnostics logging.
+        --no-prompt    	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -q, --query string 	: Run jmespath query on results
 
 Use azd auth [command] --help to view examples and more information about a specific command.
 

cli/azd/cmd/testdata/TestUsage-azd-config-get.snap

@@ -9,9 +9,10 @@ Flags
     -h, --help 	: Gets help for get.
 
 Global Flags
-    -C, --cwd string 	: Sets the current working directory.
-        --debug      	: Enables debugging and diagnostics logging.
-        --no-prompt  	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -C, --cwd string   	: Sets the current working directory.
+        --debug        	: Enables debugging and diagnostics logging.
+        --no-prompt    	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -q, --query string 	: Run jmespath query on results
 
 Find a bug? Want to let us know how we're doing? Fill out this brief survey: https://aka.ms/azure-dev/hats.
 

cli/azd/cmd/testdata/TestUsage-azd-config-list-alpha.snap

@@ -9,9 +9,10 @@ Flags
     -h, --help 	: Gets help for list-alpha.
 
 Global Flags
-    -C, --cwd string 	: Sets the current working directory.
-        --debug      	: Enables debugging and diagnostics logging.
-        --no-prompt  	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -C, --cwd string   	: Sets the current working directory.
+        --debug        	: Enables debugging and diagnostics logging.
+        --no-prompt    	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -q, --query string 	: Run jmespath query on results
 
 Examples
   Displays a list of all available features in the alpha stage

cli/azd/cmd/testdata/TestUsage-azd-config-reset.snap

@@ -10,9 +10,10 @@ Flags
     -h, --help  	: Gets help for reset.
 
 Global Flags
-    -C, --cwd string 	: Sets the current working directory.
-        --debug      	: Enables debugging and diagnostics logging.
-        --no-prompt  	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -C, --cwd string   	: Sets the current working directory.
+        --debug        	: Enables debugging and diagnostics logging.
+        --no-prompt    	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -q, --query string 	: Run jmespath query on results
 
 Find a bug? Want to let us know how we're doing? Fill out this brief survey: https://aka.ms/azure-dev/hats.
 

cli/azd/cmd/testdata/TestUsage-azd-config-set.snap

@@ -9,9 +9,10 @@ Flags
     -h, --help 	: Gets help for set.
 
 Global Flags
-    -C, --cwd string 	: Sets the current working directory.
-        --debug      	: Enables debugging and diagnostics logging.
-        --no-prompt  	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -C, --cwd string   	: Sets the current working directory.
+        --debug        	: Enables debugging and diagnostics logging.
+        --no-prompt    	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -q, --query string 	: Run jmespath query on results
 
 Find a bug? Want to let us know how we're doing? Fill out this brief survey: https://aka.ms/azure-dev/hats.
 

cli/azd/cmd/testdata/TestUsage-azd-config-show.snap

@@ -9,9 +9,10 @@ Flags
     -h, --help 	: Gets help for show.
 
 Global Flags
-    -C, --cwd string 	: Sets the current working directory.
-        --debug      	: Enables debugging and diagnostics logging.
-        --no-prompt  	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -C, --cwd string   	: Sets the current working directory.
+        --debug        	: Enables debugging and diagnostics logging.
+        --no-prompt    	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -q, --query string 	: Run jmespath query on results
 
 Find a bug? Want to let us know how we're doing? Fill out this brief survey: https://aka.ms/azure-dev/hats.
 

cli/azd/cmd/testdata/TestUsage-azd-config-unset.snap

@@ -9,9 +9,10 @@ Flags
     -h, --help 	: Gets help for unset.
 
 Global Flags
-    -C, --cwd string 	: Sets the current working directory.
-        --debug      	: Enables debugging and diagnostics logging.
-        --no-prompt  	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -C, --cwd string   	: Sets the current working directory.
+        --debug        	: Enables debugging and diagnostics logging.
+        --no-prompt    	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -q, --query string 	: Run jmespath query on results
 
 Find a bug? Want to let us know how we're doing? Fill out this brief survey: https://aka.ms/azure-dev/hats.
 

cli/azd/cmd/testdata/TestUsage-azd-config.snap

@@ -21,9 +21,10 @@ Flags
     -h, --help 	: Gets help for config.
 
 Global Flags
-    -C, --cwd string 	: Sets the current working directory.
-        --debug      	: Enables debugging and diagnostics logging.
-        --no-prompt  	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -C, --cwd string   	: Sets the current working directory.
+        --debug        	: Enables debugging and diagnostics logging.
+        --no-prompt    	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -q, --query string 	: Run jmespath query on results
 
 Use azd config [command] --help to view examples and more information about a specific command.
 

cli/azd/cmd/testdata/TestUsage-azd-deploy.snap

@@ -16,9 +16,10 @@ Flags
     -h, --help                	: Gets help for deploy.
 
 Global Flags
-    -C, --cwd string 	: Sets the current working directory.
-        --debug      	: Enables debugging and diagnostics logging.
-        --no-prompt  	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -C, --cwd string   	: Sets the current working directory.
+        --debug        	: Enables debugging and diagnostics logging.
+        --no-prompt    	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -q, --query string 	: Run jmespath query on results
 
 Examples
   Deploy all services in the current project to Azure.

cli/azd/cmd/testdata/TestUsage-azd-down.snap

@@ -12,9 +12,10 @@ Flags
         --purge              	: Does not require confirmation before it permanently deletes resources that are soft-deleted by default (for example, key vaults).
 
 Global Flags
-    -C, --cwd string 	: Sets the current working directory.
-        --debug      	: Enables debugging and diagnostics logging.
-        --no-prompt  	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -C, --cwd string   	: Sets the current working directory.
+        --debug        	: Enables debugging and diagnostics logging.
+        --no-prompt    	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -q, --query string 	: Run jmespath query on results
 
 Examples
   Delete all resources for an application. You will be prompted to confirm your decision.

cli/azd/cmd/testdata/TestUsage-azd-env-get-value.snap

@@ -10,9 +10,10 @@ Flags
     -h, --help               	: Gets help for get-value.
 
 Global Flags
-    -C, --cwd string 	: Sets the current working directory.
-        --debug      	: Enables debugging and diagnostics logging.
-        --no-prompt  	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -C, --cwd string   	: Sets the current working directory.
+        --debug        	: Enables debugging and diagnostics logging.
+        --no-prompt    	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -q, --query string 	: Run jmespath query on results
 
 Find a bug? Want to let us know how we're doing? Fill out this brief survey: https://aka.ms/azure-dev/hats.
 

cli/azd/cmd/testdata/TestUsage-azd-env-get-values.snap

@@ -10,9 +10,10 @@ Flags
     -h, --help               	: Gets help for get-values.
 
 Global Flags
-    -C, --cwd string 	: Sets the current working directory.
-        --debug      	: Enables debugging and diagnostics logging.
-        --no-prompt  	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -C, --cwd string   	: Sets the current working directory.
+        --debug        	: Enables debugging and diagnostics logging.
+        --no-prompt    	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -q, --query string 	: Run jmespath query on results
 
 Find a bug? Want to let us know how we're doing? Fill out this brief survey: https://aka.ms/azure-dev/hats.
 

cli/azd/cmd/testdata/TestUsage-azd-env-list.snap

@@ -9,9 +9,10 @@ Flags
     -h, --help 	: Gets help for list.
 
 Global Flags
-    -C, --cwd string 	: Sets the current working directory.
-        --debug      	: Enables debugging and diagnostics logging.
-        --no-prompt  	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -C, --cwd string   	: Sets the current working directory.
+        --debug        	: Enables debugging and diagnostics logging.
+        --no-prompt    	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -q, --query string 	: Run jmespath query on results
 
 Find a bug? Want to let us know how we're doing? Fill out this brief survey: https://aka.ms/azure-dev/hats.
 

cli/azd/cmd/testdata/TestUsage-azd-env-new.snap

@@ -11,9 +11,10 @@ Flags
         --subscription string 	: Name or ID of an Azure subscription to use for the new environment
 
 Global Flags
-    -C, --cwd string 	: Sets the current working directory.
-        --debug      	: Enables debugging and diagnostics logging.
-        --no-prompt  	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -C, --cwd string   	: Sets the current working directory.
+        --debug        	: Enables debugging and diagnostics logging.
+        --no-prompt    	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -q, --query string 	: Run jmespath query on results
 
 Find a bug? Want to let us know how we're doing? Fill out this brief survey: https://aka.ms/azure-dev/hats.
 

cli/azd/cmd/testdata/TestUsage-azd-env-refresh.snap

@@ -11,9 +11,10 @@ Flags
         --hint string        	: Hint to help identify the environment to refresh
 
 Global Flags
-    -C, --cwd string 	: Sets the current working directory.
-        --debug      	: Enables debugging and diagnostics logging.
-        --no-prompt  	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -C, --cwd string   	: Sets the current working directory.
+        --debug        	: Enables debugging and diagnostics logging.
+        --no-prompt    	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -q, --query string 	: Run jmespath query on results
 
 Find a bug? Want to let us know how we're doing? Fill out this brief survey: https://aka.ms/azure-dev/hats.
 

cli/azd/cmd/testdata/TestUsage-azd-env-select.snap

@@ -9,9 +9,10 @@ Flags
     -h, --help 	: Gets help for select.
 
 Global Flags
-    -C, --cwd string 	: Sets the current working directory.
-        --debug      	: Enables debugging and diagnostics logging.
-        --no-prompt  	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -C, --cwd string   	: Sets the current working directory.
+        --debug        	: Enables debugging and diagnostics logging.
+        --no-prompt    	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -q, --query string 	: Run jmespath query on results
 
 Find a bug? Want to let us know how we're doing? Fill out this brief survey: https://aka.ms/azure-dev/hats.
 

cli/azd/cmd/testdata/TestUsage-azd-env-set-shell-context.snap

@@ -0,0 +1,20 @@
+
+Set the current environment variables to the shell context.
+
+Usage
+  azd env set-shell-context [flags]
+
+Flags
+        --docs               	: Opens the documentation for azd env set-shell-context in your web browser.
+    -e, --environment string 	: The name of the environment to use.
+    -h, --help               	: Gets help for set-shell-context.
+
+Global Flags
+    -C, --cwd string   	: Sets the current working directory.
+        --debug        	: Enables debugging and diagnostics logging.
+        --no-prompt    	: Accepts the default value instead of prompting, or it fails if there is no default.
+    -q, --query string 	: Run jmespath query on results
+
+Find a bug? Want to let us know how we're doing? Fill out this brief survey: https://aka.ms/azure-dev/hats.
+
+