Fix panic in auto-completion handler for bundle run/open (#3358)

## Why
Regression introduced in #3175 /
version 0.259.0.

Fixes #3338

## Tests
New acceptance tests that trigger completion handler for bundle/open.

Author: denik

NEXT_CHANGELOG.md

@@ -10,5 +10,6 @@
 * Show "DEFAULT" as the default profile for `databricks auth login` [#3252](https://github.com/databricks/cli/pull/3252)
 
 ### Bundles
+* Fix panic in auto competion handler for bundler run/open. Regression introduced in 0.259.0.
 
 ### API Changes

acceptance/bundle/open/output.txt

@@ -24,3 +24,9 @@ Opening browser at [DATABRICKS_URL]/jobs/[NUMID]?o=[NUMID]
 Error: exec: "open": cannot run executable found relative to current directory
 
 Exit code (musterr): 1
+
+=== test auto-completion handler
+>>> [CLI] __complete bundle open ,
+foo
+:4
+Completion ended with directive: ShellCompDirectiveNoFileComp

acceptance/bundle/open/script

@@ -11,3 +11,6 @@ export PATH=.:$PATH
 
 title "open after deployment. This will fail to open browser and complain, that's ok, we only want the message"
 musterr trace $CLI bundle open foo
+
+title "test auto-completion handler"
+trace $CLI __complete bundle open ,

acceptance/bundle/run/basic/output.txt

@@ -32,3 +32,9 @@ Run URL: [DATABRICKS_URL]/job/run/[NUMID]
 === inline script
 >>> [CLI] bundle run -- echo hello
 hello
+
+=== completion handler
+>>> [CLI] __complete bundle run ,
+foo
+:4
+Completion ended with directive: ShellCompDirectiveNoFileComp

acceptance/bundle/run/basic/script

@@ -14,3 +14,6 @@ errcode trace $CLI bundle run foo -- arg1 arg2
 
 title "inline script"
 errcode trace $CLI bundle run -- echo "hello"
+
+title "completion handler"
+trace $CLI __complete bundle run ,

cmd/bundle/open.go

@@ -134,6 +134,9 @@ func newOpenCommand() *cobra.Command {
 	}
 
 	cmd.ValidArgsFunction = func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+		ctx := logdiag.InitContext(cmd.Context())
+		cmd.SetContext(ctx)
+
 		b := root.MustConfigureBundle(cmd)
 		if logdiag.HasError(cmd.Context()) {
 			return nil, cobra.ShellCompDirectiveError

cmd/bundle/run.go

@@ -225,6 +225,9 @@ Example usage:
 	}
 
 	cmd.ValidArgsFunction = func(cmd *cobra.Command, args []string, toComplete string) ([]string, cobra.ShellCompDirective) {
+		ctx := logdiag.InitContext(cmd.Context())
+		cmd.SetContext(ctx)
+
 		b := root.MustConfigureBundle(cmd)
 		if logdiag.HasError(cmd.Context()) {
 			return nil, cobra.ShellCompDirectiveError