diff options
Diffstat (limited to 'cmd/non_interactive_mode.go')
| -rw-r--r-- | cmd/non_interactive_mode.go | 292 |
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 -} |
