BrowserOS/packages/browseros-agent/apps/cli/cmd/snap.go

package cmd

import (
	"browseros-cli/output"

	"github.com/spf13/cobra"
)

func init() {
	cmd := &cobra.Command{
		Use:         "snap",
		Annotations: map[string]string{"group": "Observe:"},
		Short:       "Snapshot interactive elements on the page",
		Args:        cobra.NoArgs,
		Run: func(cmd *cobra.Command, args []string) {
			enhanced, _ := cmd.Flags().GetBool("enhanced")
			c := newClient()
			pageID, err := resolvePageID(c)
			if err != nil {
				output.Error(err.Error(), 2)
			}

			toolName := "take_snapshot"
			if enhanced {
				toolName = "take_enhanced_snapshot"
			}

			result, err := c.CallTool(toolName, map[string]any{"page": pageID})
			if err != nil {
				output.Error(err.Error(), 1)
			}
			if jsonOut {
				output.JSON(result)
			} else {
				output.Text(result)
			}
		},
	}

	cmd.Flags().BoolP("enhanced", "e", false, "Detailed accessibility tree with structural context")
	rootCmd.AddCommand(cmd)
}