tftsr-devops_investigation/src-tauri/src/commands/integrations.rs

use crate::integrations::{ConnectionResult, PublishResult, TicketResult};
use crate::state::AppState;
use rusqlite::OptionalExtension;
use serde::{Deserialize, Serialize};
use std::collections::HashMap;
use std::sync::{Arc, Mutex};
use tauri::{Manager, State};
use tokio::sync::oneshot;

// Global OAuth state storage (verifier + service per state key)
lazy_static::lazy_static! {
    static ref OAUTH_STATE: Arc<Mutex<HashMap<String, (String, String)>>> = Arc::new(Mutex::new(HashMap::new()));
}

// Global callback server shutdown channel
lazy_static::lazy_static! {
    static ref CALLBACK_SERVER_SHUTDOWN: Arc<Mutex<Option<oneshot::Sender<()>>>> = Arc::new(Mutex::new(None));
}

#[tauri::command]
pub async fn test_confluence_connection(
    base_url: String,
    _credentials: serde_json::Value,
    app_handle: tauri::AppHandle,
    app_state: State<'_, AppState>,
) -> Result<ConnectionResult, String> {
    // Try to get fresh cookies from persistent webview
    let cookies = get_fresh_cookies_from_webview("confluence", &app_handle, &app_state).await?;

    if let Some(cookie_list) = cookies {
        // Use cookies for authentication
        let cookie_header = crate::integrations::webview_auth::cookies_to_header(&cookie_list);

        let client = reqwest::Client::new();
        let url = format!("{}/rest/api/user/current", base_url.trim_end_matches('/'));

        let resp = client
            .get(&url)
            .header("Cookie", cookie_header)
            .send()
            .await
            .map_err(|e| format!("Connection failed: {e}"))?;

        if resp.status().is_success() {
            Ok(ConnectionResult {
                success: true,
                message: "Successfully connected to Confluence using browser session".to_string(),
            })
        } else {
            let status = resp.status();
            let text = resp.text().await.unwrap_or_default();
            Ok(ConnectionResult {
                success: false,
                message: format!("Connection failed with status {status}: {text}"),
            })
        }
    } else {
        // No webview open, check if we have stored credentials
        let encrypted_token: Option<String> = {
            let db = app_state
                .db
                .lock()
                .map_err(|e| format!("Failed to lock database: {e}"))?;

            db.query_row(
                "SELECT encrypted_token FROM credentials WHERE service = ?1",
                ["confluence"],
                |row| row.get(0),
            )
            .optional()
            .map_err(|e| format!("Failed to query credentials: {e}"))?
        };

        if let Some(token) = encrypted_token {
            let decrypted = crate::integrations::auth::decrypt_token(&token)?;

            // Try to parse as cookies JSON first
            if let Ok(cookie_list) =
                serde_json::from_str::<Vec<crate::integrations::webview_auth::Cookie>>(&decrypted)
            {
                let cookie_header =
                    crate::integrations::webview_auth::cookies_to_header(&cookie_list);

                let client = reqwest::Client::new();
                let url = format!("{}/rest/api/user/current", base_url.trim_end_matches('/'));

                let resp = client
                    .get(&url)
                    .header("Cookie", cookie_header)
                    .send()
                    .await
                    .map_err(|e| format!("Connection failed: {e}"))?;

                if resp.status().is_success() {
                    Ok(ConnectionResult {
                        success: true,
                        message: "Successfully connected to Confluence using stored session"
                            .to_string(),
                    })
                } else {
                    let status = resp.status();
                    Ok(ConnectionResult {
                        success: false,
                        message: format!(
                            "Connection failed with status {status}. Session may have expired - try reopening the browser window."
                        ),
                    })
                }
            } else {
                // Treat as bearer token
                let config = crate::integrations::confluence::ConfluenceConfig {
                    base_url: base_url.clone(),
                    access_token: decrypted,
                };
                crate::integrations::confluence::test_connection(&config).await
            }
        } else {
            Err("Not authenticated. Please open the browser window and log in, or provide a manual token.".to_string())
        }
    }
}

#[tauri::command]
pub async fn publish_to_confluence(
    _doc_id: String,
    _space_key: String,
    _parent_page_id: Option<String>,
) -> Result<PublishResult, String> {
    Err("Integrations available in v0.2. Please update to the latest version.".to_string())
}

#[tauri::command]
pub async fn test_servicenow_connection(
    instance_url: String,
    _credentials: serde_json::Value,
    app_handle: tauri::AppHandle,
    app_state: State<'_, AppState>,
) -> Result<ConnectionResult, String> {
    // Try to get fresh cookies from persistent webview
    let cookies = get_fresh_cookies_from_webview("servicenow", &app_handle, &app_state).await?;

    if let Some(cookie_list) = cookies {
        let cookie_header = crate::integrations::webview_auth::cookies_to_header(&cookie_list);

        let client = reqwest::Client::new();
        let url = format!(
            "{}/api/now/table/sys_user?sysparm_limit=1",
            instance_url.trim_end_matches('/')
        );

        let resp = client
            .get(&url)
            .header("Cookie", cookie_header)
            .send()
            .await
            .map_err(|e| format!("Connection failed: {e}"))?;

        if resp.status().is_success() {
            Ok(ConnectionResult {
                success: true,
                message: "Successfully connected to ServiceNow using browser session".to_string(),
            })
        } else {
            let status = resp.status();
            Ok(ConnectionResult {
                success: false,
                message: format!("Connection failed with status {status}"),
            })
        }
    } else {
        // Check stored credentials
        let encrypted_token: Option<String> = {
            let db = app_state
                .db
                .lock()
                .map_err(|e| format!("Failed to lock database: {e}"))?;

            db.query_row(
                "SELECT encrypted_token FROM credentials WHERE service = ?1",
                ["servicenow"],
                |row| row.get(0),
            )
            .optional()
            .map_err(|e| format!("Failed to query credentials: {e}"))?
        };

        if let Some(token) = encrypted_token {
            let password = crate::integrations::auth::decrypt_token(&token)?;
            let config = crate::integrations::servicenow::ServiceNowConfig {
                instance_url: instance_url.clone(),
                username: "".to_string(),
                password,
            };
            crate::integrations::servicenow::test_connection(&config).await
        } else {
            Err("Not authenticated. Please open the browser window and log in, or provide a manual token.".to_string())
        }
    }
}

#[tauri::command]
pub async fn create_servicenow_incident(
    _issue_id: String,
    _config: serde_json::Value,
) -> Result<TicketResult, String> {
    Err("Integrations available in v0.2. Please update to the latest version.".to_string())
}

#[tauri::command]
pub async fn test_azuredevops_connection(
    org_url: String,
    _credentials: serde_json::Value,
    app_handle: tauri::AppHandle,
    app_state: State<'_, AppState>,
) -> Result<ConnectionResult, String> {
    // Try to get fresh cookies from persistent webview
    let cookies = get_fresh_cookies_from_webview("azuredevops", &app_handle, &app_state).await?;

    if let Some(cookie_list) = cookies {
        let cookie_header = crate::integrations::webview_auth::cookies_to_header(&cookie_list);

        let client = reqwest::Client::new();
        let url = format!(
            "{}/_apis/projects?api-version=6.0",
            org_url.trim_end_matches('/')
        );

        let resp = client
            .get(&url)
            .header("Cookie", cookie_header)
            .send()
            .await
            .map_err(|e| format!("Connection failed: {e}"))?;

        if resp.status().is_success() {
            Ok(ConnectionResult {
                success: true,
                message: "Successfully connected to Azure DevOps using browser session".to_string(),
            })
        } else {
            let status = resp.status();
            Ok(ConnectionResult {
                success: false,
                message: format!("Connection failed with status {status}"),
            })
        }
    } else {
        // Check stored credentials
        let encrypted_token: Option<String> = {
            let db = app_state
                .db
                .lock()
                .map_err(|e| format!("Failed to lock database: {e}"))?;

            db.query_row(
                "SELECT encrypted_token FROM credentials WHERE service = ?1",
                ["azuredevops"],
                |row| row.get(0),
            )
            .optional()
            .map_err(|e| format!("Failed to query credentials: {e}"))?
        };

        if let Some(token) = encrypted_token {
            let access_token = crate::integrations::auth::decrypt_token(&token)?;
            let config = crate::integrations::azuredevops::AzureDevOpsConfig {
                organization_url: org_url.clone(),
                access_token,
                project: "".to_string(),
            };
            crate::integrations::azuredevops::test_connection(&config).await
        } else {
            Err("Not authenticated. Please open the browser window and log in, or provide a manual token.".to_string())
        }
    }
}

#[tauri::command]
pub async fn create_azuredevops_workitem(
    _issue_id: String,
    _project: String,
    _config: serde_json::Value,
) -> Result<TicketResult, String> {
    Err("Integrations available in v0.2. Please update to the latest version.".to_string())
}

// ─── OAuth2 Commands ────────────────────────────────────────────────────────

#[derive(Debug, serde::Serialize, serde::Deserialize)]
pub struct OAuthInitResponse {
    pub auth_url: String,
    pub state: String,
}

/// Initiate OAuth2 authorization flow for a service.
/// Starts the callback server and returns the authorization URL.
#[tauri::command]
pub async fn initiate_oauth(
    service: String,
    app_state: State<'_, AppState>,
) -> Result<OAuthInitResponse, String> {
    // Start callback server if not already running
    let server_already_running = {
        let shutdown = CALLBACK_SERVER_SHUTDOWN.lock().map_err(|e| e.to_string())?;
        shutdown.is_some()
    };

    if !server_already_running {
        tracing::info!("Starting OAuth callback server");

        let (mut callback_rx, shutdown_tx) =
            crate::integrations::callback_server::start_callback_server(8765)
                .await
                .map_err(|e| format!("Failed to start callback server: {e}"))?;

        // Store shutdown channel
        {
            let mut shutdown = CALLBACK_SERVER_SHUTDOWN.lock().map_err(|e| e.to_string())?;
            *shutdown = Some(shutdown_tx);
        }

        // Clone the Arc fields from app_state for the spawned task
        let db = app_state.db.clone();
        let settings = app_state.settings.clone();
        let app_data_dir = app_state.app_data_dir.clone();
        let integration_webviews = app_state.integration_webviews.clone();
        let mcp_connections = app_state.mcp_connections.clone();
        let pending_approvals = app_state.pending_approvals.clone();

        tokio::spawn(async move {
            let app_state_for_callback = AppState {
                db,
                settings,
                app_data_dir,
                integration_webviews,
                mcp_connections,
                pending_approvals,
                clusters: Arc::new(tokio::sync::Mutex::new(std::collections::HashMap::new())),
                port_forwards: Arc::new(tokio::sync::Mutex::new(std::collections::HashMap::new())),
                refresh_registry: Arc::new(tokio::sync::Mutex::new(crate::kube::RefreshRegistry::new())),
            };
            while let Some(callback) = callback_rx.recv().await {
                tracing::info!("Received OAuth callback for state: {}", callback.state);

                // Retrieve service and verifier
                let (service, verifier) = {
                    let mut oauth_state = match OAUTH_STATE.lock() {
                        Ok(state) => state,
                        Err(e) => {
                            tracing::error!("Failed to lock OAuth state: {e}");
                            continue;
                        }
                    };

                    match oauth_state.remove(&callback.state) {
                        Some((svc, ver)) => (svc, ver),
                        None => {
                            tracing::warn!("Unknown OAuth state: {}", callback.state);
                            continue;
                        }
                    }
                };

                // Call handle_oauth_callback internally
                let result = handle_oauth_callback_internal(
                    service,
                    callback.code,
                    verifier,
                    &app_state_for_callback,
                )
                .await;

                match result {
                    Ok(_) => tracing::info!("OAuth callback handled successfully"),
                    Err(e) => tracing::error!("OAuth callback failed: {e}"),
                }
            }

            tracing::info!("OAuth callback listener stopped");
        });
    }

    // Generate PKCE challenge
    let pkce = crate::integrations::auth::generate_pkce();

    // Generate state key for this OAuth session
    let state_key = uuid::Uuid::now_v7().to_string();

    // Store verifier and service name
    {
        let mut oauth_state = OAUTH_STATE
            .lock()
            .map_err(|e| format!("Failed to lock OAuth state: {e}"))?;
        oauth_state.insert(
            state_key.clone(),
            (service.clone(), pkce.code_verifier.clone()),
        );
    }

    // Build authorization URL based on service
    let (auth_endpoint, client_id, scope, redirect_uri) = match service.as_str() {
        "confluence" => (
            "https://auth.atlassian.com/authorize",
            std::env::var("CONFLUENCE_CLIENT_ID")
                .unwrap_or_else(|_| "confluence-client-id-placeholder".to_string()),
            "read:confluence-space.summary read:confluence-content.summary write:confluence-content",
            "http://localhost:8765/callback",
        ),
        "azuredevops" => (
            "https://login.microsoftonline.com/common/oauth2/v2.0/authorize",
            std::env::var("ADO_CLIENT_ID")
                .unwrap_or_else(|_| "ado-client-id-placeholder".to_string()),
            "vso.work vso.work_write",
            "http://localhost:8765/callback",
        ),
        "servicenow" => {
            // ServiceNow uses basic auth, not OAuth2
            return Err("ServiceNow uses basic authentication, not OAuth2".to_string());
        }
        _ => return Err(format!("Unknown service: {service}")),
    };

    let auth_url = crate::integrations::auth::build_auth_url(
        auth_endpoint,
        &client_id,
        redirect_uri,
        scope,
        &pkce,
    );

    Ok(OAuthInitResponse {
        auth_url,
        state: state_key,
    })
}

/// Internal function to handle OAuth callback (used by callback server).
async fn handle_oauth_callback_internal(
    service: String,
    code: String,
    verifier: String,
    app_state: &AppState,
) -> Result<(), String> {
    // Get token endpoint and client_id based on service
    let (token_endpoint, client_id, redirect_uri) = match service.as_str() {
        "confluence" => (
            "https://auth.atlassian.com/oauth/token",
            std::env::var("CONFLUENCE_CLIENT_ID")
                .unwrap_or_else(|_| "confluence-client-id-placeholder".to_string()),
            "http://localhost:8765/callback",
        ),
        "azuredevops" => (
            "https://login.microsoftonline.com/common/oauth2/v2.0/token",
            std::env::var("ADO_CLIENT_ID")
                .unwrap_or_else(|_| "ado-client-id-placeholder".to_string()),
            "http://localhost:8765/callback",
        ),
        _ => return Err(format!("Unknown service: {service}")),
    };

    // Exchange authorization code for access token
    let oauth_token = crate::integrations::auth::exchange_code(
        token_endpoint,
        &client_id,
        &code,
        redirect_uri,
        &verifier,
    )
    .await?;

    // Store token in database with encryption
    let token_hash = {
        use sha2::{Digest, Sha256};
        let mut hasher = Sha256::new();
        hasher.update(oauth_token.access_token.as_bytes());
        format!("{:x}", hasher.finalize())
    };

    let encrypted_token = crate::integrations::auth::encrypt_token(&oauth_token.access_token)?;

    let expires_at = Some(
        chrono::DateTime::from_timestamp(oauth_token.expires_at, 0)
            .ok_or_else(|| "Invalid expires_at timestamp".to_string())?
            .format("%Y-%m-%d %H:%M:%S")
            .to_string(),
    );

    // Insert into credentials table
    let db = app_state
        .db
        .lock()
        .map_err(|e| format!("Failed to lock database: {e}"))?;

    db.execute(
        "INSERT OR REPLACE INTO credentials (id, service, token_hash, encrypted_token, created_at, expires_at)
         VALUES (?1, ?2, ?3, ?4, ?5, ?6)",
        rusqlite::params![
            uuid::Uuid::now_v7().to_string(),
            service,
            token_hash,
            encrypted_token,
            chrono::Utc::now().format("%Y-%m-%d %H:%M:%S").to_string(),
            expires_at,
        ],
    )
    .map_err(|e| format!("Failed to store credentials: {e}"))?;

    // Log audit event
    let audit_details = serde_json::json!({
        "service": service,
        "token_hash": token_hash,
        "expires_at": expires_at,
    });

    crate::audit::log::write_audit_event(
        &db,
        "oauth_callback_success",
        "credential",
        &service,
        &audit_details.to_string(),
    )
    .map_err(|e| format!("Failed to log audit event: {e}"))?;

    Ok(())
}

/// Handle OAuth2 callback (Tauri command for external/manual calls).
/// This is rarely used since callbacks are handled automatically by the callback server.
#[tauri::command]
pub async fn handle_oauth_callback(
    service: String,
    code: String,
    state_key: String,
    app_state: State<'_, AppState>,
) -> Result<(), String> {
    // Retrieve verifier from temporary state
    let verifier = {
        let mut oauth_state = OAUTH_STATE
            .lock()
            .map_err(|e| format!("Failed to lock OAuth state: {e}"))?;
        oauth_state
            .remove(&state_key)
            .map(|(_svc, ver)| ver)
            .ok_or_else(|| "Invalid or expired OAuth state".to_string())?
    };

    handle_oauth_callback_internal(service, code, verifier, app_state.inner()).await
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_oauth_state_storage() {
        let key = "test-key".to_string();
        let service = "confluence".to_string();
        let verifier = "test-verifier".to_string();

        // Store
        {
            let mut state = OAUTH_STATE.lock().unwrap();
            state.insert(key.clone(), (service.clone(), verifier.clone()));
        }

        // Retrieve
        {
            let state = OAUTH_STATE.lock().unwrap();
            assert_eq!(state.get(&key), Some(&(service.clone(), verifier.clone())));
        }

        // Remove
        {
            let mut state = OAUTH_STATE.lock().unwrap();
            state.remove(&key);
        }

        // Verify removed
        {
            let state = OAUTH_STATE.lock().unwrap();
            assert!(!state.contains_key(&key));
        }
    }

    #[test]
    fn test_oauth_state_multiple_keys() {
        let key1 = "key1".to_string();
        let key2 = "key2".to_string();

        {
            let mut state = OAUTH_STATE.lock().unwrap();
            state.insert(
                key1.clone(),
                ("confluence".to_string(), "verifier1".to_string()),
            );
            state.insert(
                key2.clone(),
                ("azuredevops".to_string(), "verifier2".to_string()),
            );
        }

        {
            let mut state = OAUTH_STATE.lock().unwrap();
            state.remove(&key1);
        }

        let state = OAUTH_STATE.lock().unwrap();
        assert!(!state.contains_key(&key1));
        assert!(state.contains_key(&key2));
        assert_eq!(
            state.get(&key2),
            Some(&("azuredevops".to_string(), "verifier2".to_string()))
        );
    }

    #[test]
    fn test_oauth_init_response_serialization() {
        let response = OAuthInitResponse {
            auth_url: "https://example.com/auth".to_string(),
            state: "state-123".to_string(),
        };

        let json = serde_json::to_string(&response).unwrap();
        assert!(json.contains("https://example.com/auth"));
        assert!(json.contains("state-123"));

        let deserialized: OAuthInitResponse = serde_json::from_str(&json).unwrap();
        assert_eq!(deserialized.auth_url, response.auth_url);
        assert_eq!(deserialized.state, response.state);
    }

    #[test]
    fn test_integration_config_serialization() {
        let config = IntegrationConfig {
            service: "confluence".to_string(),
            base_url: "https://example.atlassian.net".to_string(),
            username: Some("user@example.com".to_string()),
            project_name: None,
            space_key: Some("DEV".to_string()),
        };

        let json = serde_json::to_string(&config).unwrap();
        assert!(json.contains("confluence"));
        assert!(json.contains("https://example.atlassian.net"));
        assert!(json.contains("user@example.com"));
        assert!(json.contains("DEV"));

        let deserialized: IntegrationConfig = serde_json::from_str(&json).unwrap();
        assert_eq!(deserialized.service, config.service);
        assert_eq!(deserialized.base_url, config.base_url);
        assert_eq!(deserialized.username, config.username);
        assert_eq!(deserialized.space_key, config.space_key);
    }

    #[test]
    fn test_webview_tracking() {
        use std::collections::HashMap;
        use std::sync::{Arc, Mutex};

        let webview_tracking: Arc<Mutex<HashMap<String, String>>> =
            Arc::new(Mutex::new(HashMap::new()));

        // Add webview
        {
            let mut tracking = webview_tracking.lock().unwrap();
            tracking.insert("confluence".to_string(), "confluence-auth".to_string());
        }

        // Verify exists
        {
            let tracking = webview_tracking.lock().unwrap();
            assert_eq!(
                tracking.get("confluence"),
                Some(&"confluence-auth".to_string())
            );
        }

        // Remove webview
        {
            let mut tracking = webview_tracking.lock().unwrap();
            tracking.remove("confluence");
        }

        // Verify removed
        {
            let tracking = webview_tracking.lock().unwrap();
            assert!(!tracking.contains_key("confluence"));
        }
    }

    #[test]
    fn test_token_auth_request_serialization() {
        let request = TokenAuthRequest {
            service: "azuredevops".to_string(),
            token: "secret_token_123".to_string(),
            token_type: "Bearer".to_string(),
            base_url: "https://dev.azure.com/org".to_string(),
        };

        let json = serde_json::to_string(&request).unwrap();
        let deserialized: TokenAuthRequest = serde_json::from_str(&json).unwrap();

        assert_eq!(deserialized.service, request.service);
        assert_eq!(deserialized.token, request.token);
        assert_eq!(deserialized.token_type, request.token_type);
        assert_eq!(deserialized.base_url, request.base_url);
    }
}

// ─── Webview-Based Authentication (Option C) ────────────────────────────────

#[derive(Debug, Serialize, Deserialize)]
pub struct WebviewAuthRequest {
    pub service: String,
    pub base_url: String,
}

#[derive(Debug, Serialize, Deserialize)]
pub struct WebviewAuthResponse {
    pub success: bool,
    pub message: String,
    pub webview_id: String,
}

/// Open persistent browser window for user to log in.
/// Window stays open for browsing and fresh cookie extraction.
/// User can close it manually when no longer needed.
#[tauri::command]
pub async fn authenticate_with_webview(
    service: String,
    base_url: String,
    project_name: Option<String>,
    app_handle: tauri::AppHandle,
    app_state: State<'_, AppState>,
) -> Result<WebviewAuthResponse, String> {
    let webview_id = format!("{service}-auth");

    // Check if window already exists
    if let Some(existing_label) = app_state
        .integration_webviews
        .lock()
        .map_err(|e| format!("Failed to lock webviews: {e}"))?
        .get(&service)
    {
        if app_handle.get_webview_window(existing_label).is_some() {
            return Ok(WebviewAuthResponse {
                success: true,
                message: format!(
                    "{service} browser window is already open. Switch to it to log in."
                ),
                webview_id: existing_label.clone(),
            });
        }
    }

    // Open persistent browser window
    let _credentials = crate::integrations::webview_auth::authenticate_with_webview(
        app_handle.clone(),
        &service,
        &base_url,
        project_name.as_deref(),
    )
    .await?;

    // Store window reference in memory
    app_state
        .integration_webviews
        .lock()
        .map_err(|e| format!("Failed to lock webviews: {e}"))?
        .insert(service.clone(), webview_id.clone());

    // Persist to database for restoration on app restart
    let db = app_state
        .db
        .lock()
        .map_err(|e| format!("Failed to lock database: {e}"))?;

    db.execute(
        "INSERT OR REPLACE INTO persistent_webviews
         (id, service, webview_label, base_url, last_active)
         VALUES (?1, ?2, ?3, ?4, datetime('now'))",
        rusqlite::params![
            uuid::Uuid::now_v7().to_string(),
            service.clone(),
            webview_id.clone(),
            base_url.clone(),
        ],
    )
    .map_err(|e| format!("Failed to persist webview: {e}"))?;

    tracing::info!("Persisted webview {} for service {}", webview_id, service);

    // Set up window close handler to remove from tracking and database
    if let Some(webview_window) = app_handle.get_webview_window(&webview_id) {
        let service_clone = service.clone();
        let db_arc = app_state.db.clone();
        let webviews_arc = app_state.integration_webviews.clone();

        webview_window.on_window_event(move |event| {
            if let tauri::WindowEvent::CloseRequested { .. } = event {
                let service = service_clone.clone();
                let db = db_arc.clone();
                let webviews = webviews_arc.clone();

                // Spawn async task to clean up
                tauri::async_runtime::spawn(async move {
                    // Remove from in-memory tracking
                    if let Ok(mut webviews_lock) = webviews.lock() {
                        webviews_lock.remove(&service);
                        tracing::info!("Removed {} from webview tracking", service);
                    }

                    // Remove from database
                    if let Ok(db_lock) = db.lock() {
                        if let Err(e) = db_lock.execute(
                            "DELETE FROM persistent_webviews WHERE service = ?1",
                            rusqlite::params![service],
                        ) {
                            tracing::warn!("Failed to remove persistent webview from DB: {}", e);
                        } else {
                            tracing::info!("Removed {} from persistent webviews database", service);
                        }
                    }
                });
            }
        });
    }

    Ok(WebviewAuthResponse {
        success: true,
        message: format!(
            "{service} browser window opened. This window will stay open across app restarts - use it to browse and authenticate. Cookies are maintained automatically."
        ),
        webview_id,
    })
}

/// Extract cookies from webview after user completes login.
/// User should call this after they've successfully logged in.
#[tauri::command]
pub async fn extract_cookies_from_webview(
    service: String,
    webview_id: String,
    app_handle: tauri::AppHandle,
    app_state: State<'_, AppState>,
) -> Result<ConnectionResult, String> {
    // Get the webview window
    let webview_window = app_handle
        .get_webview_window(&webview_id)
        .ok_or_else(|| "Webview window not found".to_string())?;

    // Extract cookies using IPC mechanism (more reliable than platform-specific APIs)
    let cookies =
        crate::integrations::webview_auth::extract_cookies_via_ipc(&webview_window, &app_handle)
            .await?;

    if cookies.is_empty() {
        return Err("No cookies found. Make sure you completed the login.".to_string());
    }

    // Encrypt and store cookies in database
    let cookies_json =
        serde_json::to_string(&cookies).map_err(|e| format!("Failed to serialize cookies: {e}"))?;
    let encrypted_cookies = crate::integrations::auth::encrypt_token(&cookies_json)?;

    let token_hash = {
        use sha2::{Digest, Sha256};
        let mut hasher = Sha256::new();
        hasher.update(cookies_json.as_bytes());
        format!("{:x}", hasher.finalize())
    };

    // Store in database
    let db = app_state
        .db
        .lock()
        .map_err(|e| format!("Failed to lock database: {e}"))?;

    db.execute(
        "INSERT OR REPLACE INTO credentials (id, service, token_hash, encrypted_token, created_at, expires_at)
         VALUES (?1, ?2, ?3, ?4, ?5, ?6)",
        rusqlite::params![
            uuid::Uuid::now_v7().to_string(),
            service,
            token_hash,
            encrypted_cookies,
            chrono::Utc::now().format("%Y-%m-%d %H:%M:%S").to_string(),
            None::<String>, // Cookies don't have explicit expiry
        ],
    )
    .map_err(|e| format!("Failed to store cookies: {e}"))?;

    // NOTE: Window stays open for persistent browsing - no longer closing after cookie extraction

    Ok(ConnectionResult {
        success: true,
        message: format!("{service} authentication saved successfully. The browser window will stay open for future use."),
    })
}

// ─── Manual Token Authentication (Token Mode) ───────────────────────────────

#[derive(Debug, Serialize, Deserialize)]
pub struct TokenAuthRequest {
    pub service: String,
    pub token: String,
    pub token_type: String, // "Bearer", "Basic", "api_token"
    pub base_url: String,
}

/// Store a manually provided token (API key, PAT, etc.)
/// This is the fallback authentication method when OAuth2 and webview don't work.
#[tauri::command]
pub async fn save_manual_token(
    request: TokenAuthRequest,
    app_state: State<'_, AppState>,
) -> Result<ConnectionResult, String> {
    // Validate token by testing connection
    let test_result = match request.service.as_str() {
        "confluence" => {
            let config = crate::integrations::confluence::ConfluenceConfig {
                base_url: request.base_url.clone(),
                access_token: request.token.clone(),
            };
            crate::integrations::confluence::test_connection(&config).await
        }
        "azuredevops" => {
            let config = crate::integrations::azuredevops::AzureDevOpsConfig {
                organization_url: request.base_url.clone(),
                access_token: request.token.clone(),
                project: "".to_string(), // Project not needed for connection test
            };
            crate::integrations::azuredevops::test_connection(&config).await
        }
        "servicenow" => {
            // ServiceNow uses basic auth, token is base64(username:password)
            let config = crate::integrations::servicenow::ServiceNowConfig {
                instance_url: request.base_url.clone(),
                username: "".to_string(), // Encoded in token
                password: request.token.clone(),
            };
            crate::integrations::servicenow::test_connection(&config).await
        }
        _ => {
            return Err(format!(
                "Unknown service: {service}",
                service = request.service
            ))
        }
    };

    // If test fails, don't save the token
    if let Ok(result) = &test_result {
        if !result.success {
            return Ok(ConnectionResult {
                success: false,
                message: format!(
                    "Token validation failed: {}. Token not saved.",
                    result.message
                ),
            });
        }
    }

    // Encrypt and store token
    let encrypted_token = crate::integrations::auth::encrypt_token(&request.token)?;

    let token_hash = {
        use sha2::{Digest, Sha256};
        let mut hasher = Sha256::new();
        hasher.update(request.token.as_bytes());
        format!("{:x}", hasher.finalize())
    };

    let db = app_state
        .db
        .lock()
        .map_err(|e| format!("Failed to lock database: {e}"))?;

    db.execute(
        "INSERT OR REPLACE INTO credentials (id, service, token_hash, encrypted_token, created_at, expires_at)
         VALUES (?1, ?2, ?3, ?4, ?5, ?6)",
        rusqlite::params![
            uuid::Uuid::now_v7().to_string(),
            request.service,
            token_hash,
            encrypted_token,
            chrono::Utc::now().format("%Y-%m-%d %H:%M:%S").to_string(),
            None::<String>,
        ],
    )
    .map_err(|e| format!("Failed to store token: {e}"))?;

    // Log audit event
    crate::audit::log::write_audit_event(
        &db,
        "manual_token_saved",
        "credential",
        &request.service,
        &serde_json::json!({
            "token_type": request.token_type,
            "token_hash": token_hash,
        })
        .to_string(),
    )
    .map_err(|e| format!("Failed to log audit event: {e}"))?;

    Ok(ConnectionResult {
        success: true,
        message: format!(
            "{service} token saved and validated successfully",
            service = request.service
        ),
    })
}

// ============================================================================
// Fresh Cookie Extraction (called before each API request)
// ============================================================================

/// Get fresh cookies from an open webview window for immediate use.
/// This is called before each integration API call to handle token rotation.
/// Returns None if window is closed or cookies unavailable.
pub async fn get_fresh_cookies_from_webview(
    service: &str,
    app_handle: &tauri::AppHandle,
    app_state: &State<'_, AppState>,
) -> Result<Option<Vec<crate::integrations::webview_auth::Cookie>>, String> {
    // Check if webview exists for this service
    let webview_label = {
        let webviews = app_state
            .integration_webviews
            .lock()
            .map_err(|e| format!("Failed to lock webviews: {e}"))?;

        match webviews.get(service) {
            Some(label) => label.clone(),
            None => return Ok(None), // No webview open for this service
        }
    };

    // Get window handle
    let webview_window = match app_handle.get_webview_window(&webview_label) {
        Some(window) => window,
        None => {
            // Window was closed, remove from tracking
            app_state
                .integration_webviews
                .lock()
                .map_err(|e| format!("Failed to lock webviews: {e}"))?
                .remove(service);
            return Ok(None);
        }
    };

    // Extract current cookies
    match crate::integrations::webview_auth::extract_cookies_via_ipc(&webview_window, app_handle)
        .await
    {
        Ok(cookies) if !cookies.is_empty() => Ok(Some(cookies)),
        Ok(_) => Ok(None), // No cookies available
        Err(e) => {
            tracing::warn!("Failed to extract cookies from {}: {}", service, e);
            Ok(None)
        }
    }
}

// ============================================================================
// Persistent Webview Restoration
// ============================================================================

/// Restore persistent browser windows from database on app startup.
/// This recreates integration browser windows that were open when the app last closed.
pub async fn restore_persistent_webviews(
    app_handle: &tauri::AppHandle,
    app_state: &AppState,
) -> Result<(), String> {
    let webviews_to_restore: Vec<(String, String, String)> = {
        let db = app_state
            .db
            .lock()
            .map_err(|e| format!("Failed to lock database: {e}"))?;

        let mut stmt = db
            .prepare("SELECT service, webview_label, base_url FROM persistent_webviews")
            .map_err(|e| format!("Failed to prepare query: {e}"))?;

        let rows: Vec<(String, String, String)> = stmt
            .query_map([], |row| {
                Ok((
                    row.get::<_, String>(0)?, // service
                    row.get::<_, String>(1)?, // webview_label
                    row.get::<_, String>(2)?, // base_url
                ))
            })
            .map_err(|e| format!("Failed to query persistent webviews: {e}"))?
            .collect::<Result<Vec<_>, _>>()
            .map_err(|e| format!("Failed to collect webviews: {e}"))?;

        rows
    };

    for (service, webview_label, base_url) in webviews_to_restore {
        tracing::info!(
            "Restoring persistent webview {} for service {} at {}",
            webview_label,
            service,
            base_url
        );

        // Get project name from integration config if available
        let project_name: Option<String> = {
            let db = app_state
                .db
                .lock()
                .map_err(|e| format!("Failed to lock database: {e}"))?;
            db.query_row(
                "SELECT project_name FROM integration_config WHERE service = ?1",
                [&service],
                |row| row.get(0),
            )
            .ok()
        };

        // Recreate the webview window
        match crate::integrations::webview_auth::authenticate_with_webview(
            app_handle.clone(),
            &service,
            &base_url,
            project_name.as_deref(),
        )
        .await
        {
            Ok(_) => {
                // Store in memory tracking
                app_state
                    .integration_webviews
                    .lock()
                    .map_err(|e| format!("Failed to lock webviews: {e}"))?
                    .insert(service.clone(), webview_label.clone());

                tracing::info!("Successfully restored webview for {}", service);
            }
            Err(e) => {
                tracing::warn!("Failed to restore webview for {}: {}", service, e);
                // Remove from database if restoration failed
                let db = app_state
                    .db
                    .lock()
                    .map_err(|e| format!("Failed to lock database: {e}"))?;

                db.execute(
                    "DELETE FROM persistent_webviews WHERE service = ?1",
                    rusqlite::params![service],
                )
                .map_err(|e| format!("Failed to remove failed webview: {e}"))?;
            }
        }
    }

    Ok(())
}

/// Remove persistent webview from database (called when window is closed).
pub async fn remove_persistent_webview(
    service: &str,
    app_state: &State<'_, AppState>,
) -> Result<(), String> {
    let db = app_state
        .db
        .lock()
        .map_err(|e| format!("Failed to lock database: {e}"))?;

    db.execute(
        "DELETE FROM persistent_webviews WHERE service = ?1",
        rusqlite::params![service],
    )
    .map_err(|e| format!("Failed to remove persistent webview: {e}"))?;

    tracing::info!("Removed persistent webview for service: {}", service);
    Ok(())
}

// ============================================================================
// Integration Configuration Persistence
// ============================================================================

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct IntegrationConfig {
    pub service: String,
    pub base_url: String,
    pub username: Option<String>,
    pub project_name: Option<String>,
    pub space_key: Option<String>,
}

/// Save or update integration configuration (base URL, username, project, etc.)
#[tauri::command]
pub async fn save_integration_config(
    config: IntegrationConfig,
    app_state: State<'_, AppState>,
) -> Result<(), String> {
    let db = app_state
        .db
        .lock()
        .map_err(|e| format!("Failed to lock database: {e}"))?;

    db.execute(
        "INSERT OR REPLACE INTO integration_config
         (id, service, base_url, username, project_name, space_key, updated_at)
         VALUES (?1, ?2, ?3, ?4, ?5, ?6, datetime('now'))",
        rusqlite::params![
            uuid::Uuid::now_v7().to_string(),
            config.service,
            config.base_url,
            config.username,
            config.project_name,
            config.space_key,
        ],
    )
    .map_err(|e| format!("Failed to save integration config: {e}"))?;

    Ok(())
}

/// Get integration configuration for a specific service
#[tauri::command]
pub async fn get_integration_config(
    service: String,
    app_state: State<'_, AppState>,
) -> Result<Option<IntegrationConfig>, String> {
    let db = app_state
        .db
        .lock()
        .map_err(|e| format!("Failed to lock database: {e}"))?;

    let mut stmt = db
        .prepare("SELECT service, base_url, username, project_name, space_key FROM integration_config WHERE service = ?1")
        .map_err(|e| format!("Failed to prepare query: {e}"))?;

    let config = stmt
        .query_row([&service], |row| {
            Ok(IntegrationConfig {
                service: row.get(0)?,
                base_url: row.get(1)?,
                username: row.get(2)?,
                project_name: row.get(3)?,
                space_key: row.get(4)?,
            })
        })
        .optional()
        .map_err(|e| format!("Failed to query integration config: {e}"))?;

    Ok(config)
}

/// Get all integration configurations
#[tauri::command]
pub async fn get_all_integration_configs(
    app_state: State<'_, AppState>,
) -> Result<Vec<IntegrationConfig>, String> {
    let db = app_state
        .db
        .lock()
        .map_err(|e| format!("Failed to lock database: {e}"))?;

    let mut stmt = db
        .prepare(
            "SELECT service, base_url, username, project_name, space_key FROM integration_config",
        )
        .map_err(|e| format!("Failed to prepare query: {e}"))?;

    let configs = stmt
        .query_map([], |row| {
            Ok(IntegrationConfig {
                service: row.get(0)?,
                base_url: row.get(1)?,
                username: row.get(2)?,
                project_name: row.get(3)?,
                space_key: row.get(4)?,
            })
        })
        .map_err(|e| format!("Failed to query integration configs: {e}"))?
        .collect::<Result<Vec<_>, _>>()
        .map_err(|e| format!("Failed to collect integration configs: {e}"))?;

    Ok(configs)
}

/// Add a comment to an Azure DevOps work item
#[tauri::command]
pub async fn add_ado_comment(
    work_item_id: i64,
    comment_text: String,
    app_handle: tauri::AppHandle,
    app_state: State<'_, AppState>,
) -> Result<String, String> {
    // Get ADO configuration
    let (org_url, _project_name) = {
        let db = app_state
            .db
            .lock()
            .map_err(|e| format!("Failed to lock database: {e}"))?;
        let mut stmt = db.prepare(
            "SELECT base_url, project_name FROM integration_config WHERE service = 'azuredevops'"
        ).map_err(|e| format!("Failed to prepare query: {e}"))?;

        stmt.query_row([], |row| {
            Ok((row.get::<_, String>(0)?, row.get::<_, Option<String>>(1)?))
        })
        .map_err(|e| format!("Azure DevOps not configured: {e}"))?
    };

    // Get webview window
    let webview_label = {
        let webviews = app_state
            .integration_webviews
            .lock()
            .map_err(|e| format!("Failed to lock webviews: {e}"))?;
        webviews.get("azuredevops").cloned()
            .ok_or_else(|| "Azure DevOps browser window not open. Please open it in Settings → Integrations first.".to_string())?
    };

    let webview_window = app_handle
        .get_webview_window(&webview_label)
        .ok_or_else(|| "Azure DevOps browser window not found".to_string())?;

    // Add the comment
    crate::integrations::webview_fetch::add_azuredevops_comment_webview(
        &webview_window,
        &org_url,
        work_item_id,
        &comment_text,
    )
    .await
}