Set bazel workspace root for the bazel query verbs

cmd/aspect/aquery/BUILD.bazel

@@ -8,6 +8,7 @@ go_library(
     deps = [
         "//pkg/aspect/aquery",
         "//pkg/bazel",
+        "//pkg/interceptors",
         "//pkg/ioutils",
         "@com_github_spf13_cobra//:cobra",
     ],

cmd/aspect/aquery/aquery.go

@@ -7,10 +7,13 @@ Not licensed for re-use
 package aquery
 
 import (
+	"context"
+
 	"github.com/spf13/cobra"
 
 	"aspect.build/cli/pkg/aspect/aquery"
 	"aspect.build/cli/pkg/bazel"
+	"aspect.build/cli/pkg/interceptors"
 	"aspect.build/cli/pkg/ioutils"
 )
 
@@ -19,13 +22,21 @@ func NewDefaultAQueryCmd() *cobra.Command {
 }
 
 func NewAQueryCommand(streams ioutils.Streams, bzl bazel.Bazel) *cobra.Command {
-	q := aquery.New(streams, bzl, true)
-
 	cmd := &cobra.Command{
 		Use:   "aquery",
 		Short: "Executes an aquery.",
 		Long:  "Executes a query language expression over a specified subgraph of the build dependency graph using aquery.",
-		RunE:  q.Run,
+		RunE: interceptors.Run(
+			[]interceptors.Interceptor{
+				interceptors.WorkspaceRootInterceptor(),
+			},
+			func(ctx context.Context, cmd *cobra.Command, args []string) (exitErr error) {
+				workspaceRoot := ctx.Value(interceptors.WorkspaceRootKey).(string)
+				bzl.SetWorkspaceRoot(workspaceRoot)
+				q := aquery.New(streams, bzl, true)
+				return q.Run(cmd, args)
+			},
+		),
 	}
 
 	return cmd

cmd/aspect/cquery/BUILD.bazel

@@ -8,6 +8,7 @@ go_library(
     deps = [
         "//pkg/aspect/cquery",
         "//pkg/bazel",
+        "//pkg/interceptors",
         "//pkg/ioutils",
         "@com_github_spf13_cobra//:cobra",
     ],

cmd/aspect/cquery/cquery.go

@@ -7,10 +7,13 @@ Not licensed for re-use
 package cquery
 
 import (
+	"context"
+
 	"github.com/spf13/cobra"
 
 	"aspect.build/cli/pkg/aspect/cquery"
 	"aspect.build/cli/pkg/bazel"
+	"aspect.build/cli/pkg/interceptors"
 	"aspect.build/cli/pkg/ioutils"
 )
 
@@ -19,13 +22,21 @@ func NewDefaultCQueryCmd() *cobra.Command {
 }
 
 func NewCQueryCommand(streams ioutils.Streams, bzl bazel.Bazel) *cobra.Command {
-	q := cquery.New(streams, bzl, true)
-
 	cmd := &cobra.Command{
 		Use:   "cquery",
 		Short: "Executes a cquery.",
 		Long:  "Executes a query language expression over a specified subgraph of the build dependency graph using cquery.",
-		RunE:  q.Run,
+		RunE: interceptors.Run(
+			[]interceptors.Interceptor{
+				interceptors.WorkspaceRootInterceptor(),
+			},
+			func(ctx context.Context, cmd *cobra.Command, args []string) (exitErr error) {
+				workspaceRoot := ctx.Value(interceptors.WorkspaceRootKey).(string)
+				bzl.SetWorkspaceRoot(workspaceRoot)
+				q := cquery.New(streams, bzl, true)
+				return q.Run(cmd, args)
+			},
+		),
 	}
 
 	return cmd

cmd/aspect/query/BUILD.bazel

@@ -8,6 +8,7 @@ go_library(
     deps = [
         "//pkg/aspect/query",
         "//pkg/bazel",
+        "//pkg/interceptors",
         "//pkg/ioutils",
         "@com_github_spf13_cobra//:cobra",
     ],

cmd/aspect/query/query.go

@@ -7,10 +7,13 @@ Not licensed for re-use
 package query
 
 import (
+	"context"
+
 	"github.com/spf13/cobra"
 
 	"aspect.build/cli/pkg/aspect/query"
 	"aspect.build/cli/pkg/bazel"
+	"aspect.build/cli/pkg/interceptors"
 	"aspect.build/cli/pkg/ioutils"
 )
 
@@ -19,13 +22,21 @@ func NewDefaultQueryCmd() *cobra.Command {
 }
 
 func NewQueryCommand(streams ioutils.Streams, bzl bazel.Bazel) *cobra.Command {
-	q := query.New(streams, bzl, true)
-
 	cmd := &cobra.Command{
 		Use:   "query",
 		Short: "Executes a dependency graph query.",
 		Long:  "Executes a query language expression over a specified subgraph of the build dependency graph.",
-		RunE:  q.Run,
+		RunE: interceptors.Run(
+			[]interceptors.Interceptor{
+				interceptors.WorkspaceRootInterceptor(),
+			},
+			func(ctx context.Context, cmd *cobra.Command, args []string) (exitErr error) {
+				workspaceRoot := ctx.Value(interceptors.WorkspaceRootKey).(string)
+				bzl.SetWorkspaceRoot(workspaceRoot)
+				q := query.New(streams, bzl, true)
+				return q.Run(cmd, args)
+			},
+		),
 	}
 
 	return cmd