fix(cli): check for query_file when running queries (#8166)

Synced from #814 by
fzakaria:

IntelliJ IDE uses a query with no arguments which seems to cause havoc
when this is run via aspect.

I added support to check for `--tool_tag` and not prompt the user for a
query selection to better handle this usecase.

Fixes #813

### Changes are visible to end-users: yes

- Searched for relevant documentation and updated as needed: yes
- Breaking change (forces users to change their own code or config): no
- Suggested release notes appear below: yes

Support running queries via `--query_file`

### Test plan

```
> /Users/fzakaria/code/github.com/aspect-build/aspect-cli/bazel-bin/cmd/aspect/aspect_/aspect query --tool_tag=ijwb:IDEA:ultimate "--override_repository=intellij_aspect=/Users/fzakaria/Library/Application Support/JetBrains/IntelliJIdea2024.1/plugins/ijwb/aspect" --override_repository=intellij_aspect_template=/Users/fzakaria/code/github.com/confluentinc/ce-kafka/.ijwb/aspect --output=label_kind --keep_going --query_file=/Users/fzakaria/code/github.com/confluentinc/ce-kafka/.ijwb/queries/query-13438905584701813446
> echo $?
2
```

Closes #814

Co-authored-by: Farid Zakaria <fzakaria@confluent.io>
GitOrigin-RevId: bf38a5b5e7d9ace6eb7d7719887cde69fc125783

Author: jbedard

pkg/aspect/aquery/aquery.go

@@ -69,7 +69,7 @@ func (runner *AQuery) Run(ctx context.Context, cmd *cobra.Command, args []string
 		return shared.GetPrettyError(cmd, err)
 	}
 
-	command, query, runReplacements, err := shared.SelectQuery(cmd.CalledAs(), presets, runner.Presets, presetNames, runner.Streams, nonFlags, runner.Select)
+	command, query, runReplacements, err := shared.SelectQuery(cmd.CalledAs(), presets, runner.Presets, presetNames, runner.Streams, nonFlags, flags, runner.Select)
 	if err != nil {
 		return shared.GetPrettyError(cmd, err)
 	}

pkg/aspect/cquery/cquery.go

@@ -69,7 +69,7 @@ func (runner *CQuery) Run(ctx context.Context, cmd *cobra.Command, args []string
 		return shared.GetPrettyError(cmd, err)
 	}
 
-	command, query, runReplacements, err := shared.SelectQuery(cmd.CalledAs(), presets, runner.Presets, presetNames, runner.Streams, nonFlags, runner.Select)
+	command, query, runReplacements, err := shared.SelectQuery(cmd.CalledAs(), presets, runner.Presets, presetNames, runner.Streams, nonFlags, flags, runner.Select)
 	if err != nil {
 		return shared.GetPrettyError(cmd, err)
 	}

pkg/aspect/query/query.go

@@ -97,7 +97,7 @@ func (runner *Query) Run(ctx context.Context, cmd *cobra.Command, args []string)
 		return shared.GetPrettyError(cmd, err)
 	}
 
-	command, query, runReplacements, err := shared.SelectQuery(cmd.CalledAs(), presets, runner.Presets, presetNames, runner.Streams, nonFlags, runner.Select)
+	command, query, runReplacements, err := shared.SelectQuery(cmd.CalledAs(), presets, runner.Presets, presetNames, runner.Streams, nonFlags, flags, runner.Select)
 	if err != nil {
 		return shared.GetPrettyError(cmd, err)
 	}

pkg/aspect/query/shared/query.go

@@ -202,11 +202,20 @@ func SelectQuery(
 	presetNames []string,
 	streams ioutils.Streams,
 	args []string,
+	flags []string,
 	s func(presetNames []string) SelectRunner,
 ) (string, string, bool, error) {
 
+	hasQueryFile := false
+	for _, flag := range flags {
+		if strings.Contains(flag, "--query_file") {
+			hasQueryFile = true
+			break
+		}
+	}
+
 	var preset *PresetQuery
-	if len(args) == 0 {
+	if len(args) == 0 && !hasQueryFile {
 		selectQueryPrompt := s(presetNames)
 
 		i, _, err := selectQueryPrompt.Run()
@@ -217,7 +226,10 @@ func SelectQuery(
 
 		preset = rawPresets[i]
 	} else {
-		maybeQueryOrPreset := args[0]
+		maybeQueryOrPreset := ""
+		if len(args) > 0 {
+			maybeQueryOrPreset = args[0]
+		}
 		if value, ok := processedPresets[maybeQueryOrPreset]; ok {
 			// Treat this as the name of the preset query, so don't prompt for it.
 			fmt.Fprintf(streams.Stdout, "Preset query \"%s\" selected\n", value.Name)