browser-use 0.2.3

use crate::error::Result;
use crate::tools::{Tool, ToolContext, ToolResult};
use schemars::JsonSchema;
use serde::{Deserialize, Serialize};

/// Information about a browser tab
#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema)]
pub struct TabInfo {
    /// Tab index
    pub index: usize,
    /// Whether this is the active tab
    pub active: bool,
    /// Tab title
    pub title: String,
    /// Tab URL
    pub url: String,
}

/// Parameters for the tab_list tool (no parameters needed)
#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema)]
pub struct TabListParams {}

/// Tool for listing all browser tabs
#[derive(Default)]
pub struct TabListTool;

impl Tool for TabListTool {
    type Params = TabListParams;

    fn name(&self) -> &str {
        "tab_list"
    }

    fn execute_typed(
        &self,
        _params: TabListParams,
        context: &mut ToolContext,
    ) -> Result<ToolResult> {
        // Get all tabs
        let tabs = context.session.get_tabs()?;
        let active_tab = context.session.tab()?;

        // Build tab info list
        let mut tab_list = Vec::new();
        for (index, tab) in tabs.iter().enumerate() {
            // Check if this is the active tab by comparing Arc pointers
            let is_active = std::sync::Arc::ptr_eq(tab, &active_tab);

            // Get tab title (fallback to empty string on error)
            let title = tab.get_title().unwrap_or_default();

            // Get tab URL (not a Result, returns String directly)
            let url = tab.get_url();

            tab_list.push(TabInfo {
                index,
                active: is_active,
                title,
                url,
            });
        }

        // Build summary text
        let active_index = tab_list.iter().position(|t| t.active).unwrap_or(0);
        let active_info = &tab_list[active_index];

        let summary = if !tab_list.is_empty() {
            let all_tabs_str = tab_list
                .iter()
                .map(|tab| format!("[{}] Title: {} (URL: {})", tab.index, tab.title, tab.url))
                .collect::<Vec<_>>()
                .join("\n");

            format!(
                "Current Tab: [{}] {}\nAll Tabs:\n{}",
                active_index, active_info.title, all_tabs_str
            )
        } else {
            "No tabs available".to_string()
        };

        Ok(ToolResult::success_with(serde_json::json!({
            "tab_list": tab_list,
            "count": tab_list.len(),
            "summary": summary
        })))
    }
}