summaryrefslogtreecommitdiffhomepage
path: root/cmd/non_interactive_mode.go
diff options
context:
space:
mode:
Diffstat (limited to 'cmd/non_interactive_mode.go')
-rw-r--r--cmd/non_interactive_mode.go292
1 files changed, 0 insertions, 292 deletions
diff --git a/cmd/non_interactive_mode.go b/cmd/non_interactive_mode.go
deleted file mode 100644
index 5023839c3..000000000
--- a/cmd/non_interactive_mode.go
+++ /dev/null
@@ -1,292 +0,0 @@
-package cmd
-
-import (
- "context"
- "fmt"
- "io"
- "os"
- "sync"
- "time"
-
- "log/slog"
-
- charmlog "github.com/charmbracelet/log"
- "github.com/sst/opencode/internal/app"
- "github.com/sst/opencode/internal/config"
- "github.com/sst/opencode/internal/db"
- "github.com/sst/opencode/internal/format"
- "github.com/sst/opencode/internal/llm/agent"
- "github.com/sst/opencode/internal/llm/tools"
- "github.com/sst/opencode/internal/message"
- "github.com/sst/opencode/internal/permission"
- "github.com/sst/opencode/internal/tui/components/spinner"
- "github.com/sst/opencode/internal/tui/theme"
-)
-
-// syncWriter is a thread-safe writer that prevents interleaved output
-type syncWriter struct {
- w io.Writer
- mu sync.Mutex
-}
-
-// Write implements io.Writer
-func (sw *syncWriter) Write(p []byte) (n int, err error) {
- sw.mu.Lock()
- defer sw.mu.Unlock()
- return sw.w.Write(p)
-}
-
-// newSyncWriter creates a new synchronized writer
-func newSyncWriter(w io.Writer) io.Writer {
- return &syncWriter{w: w}
-}
-
-// filterTools filters the provided tools based on allowed or excluded tool names
-func filterTools(allTools []tools.BaseTool, allowedTools, excludedTools []string) []tools.BaseTool {
- // If neither allowed nor excluded tools are specified, return all tools
- if len(allowedTools) == 0 && len(excludedTools) == 0 {
- return allTools
- }
-
- // Create a map for faster lookups
- allowedMap := make(map[string]bool)
- for _, name := range allowedTools {
- allowedMap[name] = true
- }
-
- excludedMap := make(map[string]bool)
- for _, name := range excludedTools {
- excludedMap[name] = true
- }
-
- var filteredTools []tools.BaseTool
-
- for _, tool := range allTools {
- toolName := tool.Info().Name
-
- // If we have an allowed list, only include tools in that list
- if len(allowedTools) > 0 {
- if allowedMap[toolName] {
- filteredTools = append(filteredTools, tool)
- }
- } else if len(excludedTools) > 0 {
- // If we have an excluded list, include all tools except those in the list
- if !excludedMap[toolName] {
- filteredTools = append(filteredTools, tool)
- }
- }
- }
-
- return filteredTools
-}
-
-// handleNonInteractiveMode processes a single prompt in non-interactive mode
-func handleNonInteractiveMode(ctx context.Context, prompt string, outputFormat format.OutputFormat, quiet bool, verbose bool, allowedTools, excludedTools []string) error {
- // Initial log message using standard slog
- slog.Info("Running in non-interactive mode", "prompt", prompt, "format", outputFormat, "quiet", quiet, "verbose", verbose,
- "allowedTools", allowedTools, "excludedTools", excludedTools)
-
- // Sanity check for mutually exclusive flags
- if quiet && verbose {
- return fmt.Errorf("--quiet and --verbose flags cannot be used together")
- }
-
- // Set up logging to stderr if verbose mode is enabled
- if verbose {
- // Create a synchronized writer to prevent interleaved output
- syncWriter := newSyncWriter(os.Stderr)
-
- // Create a charmbracelet/log logger that writes to the synchronized writer
- charmLogger := charmlog.NewWithOptions(syncWriter, charmlog.Options{
- Level: charmlog.DebugLevel,
- ReportCaller: true,
- ReportTimestamp: true,
- TimeFormat: time.RFC3339,
- Prefix: "OpenCode",
- })
-
- // Set the global logger for charmbracelet/log
- charmlog.SetDefault(charmLogger)
-
- // Create a slog handler that uses charmbracelet/log
- // This will forward all slog logs to charmbracelet/log
- slog.SetDefault(slog.New(charmLogger))
-
- // Log a message to confirm verbose logging is enabled
- charmLogger.Info("Verbose logging enabled")
- }
-
- // Start spinner if not in quiet mode
- var s *spinner.Spinner
- if !quiet {
- // Get the current theme to style the spinner
- currentTheme := theme.CurrentTheme()
-
- // Create a themed spinner
- if currentTheme != nil {
- // Use the primary color from the theme
- s = spinner.NewThemedSpinner("Thinking...", currentTheme.Primary())
- } else {
- // Fallback to default spinner if no theme is available
- s = spinner.NewSpinner("Thinking...")
- }
-
- s.Start()
- defer s.Stop()
- }
-
- // Connect DB, this will also run migrations
- conn, err := db.Connect()
- if err != nil {
- return err
- }
-
- // Create a context with cancellation
- ctx, cancel := context.WithCancel(ctx)
- defer cancel()
-
- // Create the app
- app, err := app.New(ctx, conn)
- if err != nil {
- slog.Error("Failed to create app", "error", err)
- return err
- }
-
- // Create a new session for this prompt
- session, err := app.Sessions.Create(ctx, "Non-interactive prompt")
- if err != nil {
- return fmt.Errorf("failed to create session: %w", err)
- }
-
- // Set the session as current
- app.CurrentSession = &session
-
- // Auto-approve all permissions for this session
- permission.AutoApproveSession(ctx, session.ID)
-
- // Create the user message
- _, err = app.Messages.Create(ctx, session.ID, message.CreateMessageParams{
- Role: message.User,
- Parts: []message.ContentPart{message.TextContent{Text: prompt}},
- })
- if err != nil {
- return fmt.Errorf("failed to create message: %w", err)
- }
-
- // If tool restrictions are specified, create a new agent with filtered tools
- if len(allowedTools) > 0 || len(excludedTools) > 0 {
- // Initialize MCP tools synchronously to ensure they're included in filtering
- mcpCtx, mcpCancel := context.WithTimeout(ctx, 10*time.Second)
- agent.GetMcpTools(mcpCtx, app.Permissions)
- mcpCancel()
-
- // Get all available tools including MCP tools
- allTools := agent.PrimaryAgentTools(
- app.Permissions,
- app.Sessions,
- app.Messages,
- app.History,
- app.LSPClients,
- )
-
- // Filter tools based on allowed/excluded lists
- filteredTools := filterTools(allTools, allowedTools, excludedTools)
-
- // Log the filtered tools for debugging
- var toolNames []string
- for _, tool := range filteredTools {
- toolNames = append(toolNames, tool.Info().Name)
- }
- slog.Debug("Using filtered tools", "count", len(filteredTools), "tools", toolNames)
-
- // Create a new agent with the filtered tools
- restrictedAgent, err := agent.NewAgent(
- config.AgentPrimary,
- app.Sessions,
- app.Messages,
- filteredTools,
- )
- if err != nil {
- return fmt.Errorf("failed to create restricted agent: %w", err)
- }
-
- // Use the restricted agent for this request
- eventCh, err := restrictedAgent.Run(ctx, session.ID, prompt)
- if err != nil {
- return fmt.Errorf("failed to run restricted agent: %w", err)
- }
-
- // Wait for the response
- var response message.Message
- for event := range eventCh {
- if event.Err() != nil {
- return fmt.Errorf("agent error: %w", event.Err())
- }
- response = event.Response()
- }
-
- // Format and print the output
- content := ""
- if textContent := response.Content(); textContent != nil {
- content = textContent.Text
- }
-
- formattedOutput, err := format.FormatOutput(content, outputFormat)
- if err != nil {
- return fmt.Errorf("failed to format output: %w", err)
- }
-
- // Stop spinner before printing output
- if !quiet && s != nil {
- s.Stop()
- }
-
- // Print the formatted output to stdout
- fmt.Println(formattedOutput)
-
- // Shutdown the app
- app.Shutdown()
-
- return nil
- }
-
- // Run the default agent if no tool restrictions
- eventCh, err := app.PrimaryAgent.Run(ctx, session.ID, prompt)
- if err != nil {
- return fmt.Errorf("failed to run agent: %w", err)
- }
-
- // Wait for the response
- var response message.Message
- for event := range eventCh {
- if event.Err() != nil {
- return fmt.Errorf("agent error: %w", event.Err())
- }
- response = event.Response()
- }
-
- // Get the text content from the response
- content := ""
- if textContent := response.Content(); textContent != nil {
- content = textContent.Text
- }
-
- // Format the output according to the specified format
- formattedOutput, err := format.FormatOutput(content, outputFormat)
- if err != nil {
- return fmt.Errorf("failed to format output: %w", err)
- }
-
- // Stop spinner before printing output
- if !quiet && s != nil {
- s.Stop()
- }
-
- // Print the formatted output to stdout
- fmt.Println(formattedOutput)
-
- // Shutdown the app
- app.Shutdown()
-
- return nil
-}