use aim_core::integration::install::{DesktopIntegrationRequest, InstallRequest, execute_install};
use aim_core::platform::DesktopHelpers;
use std::fs;
use std::os::unix::fs::PermissionsExt;
use tempfile::tempdir;

#[test]
fn install_writes_desktop_entry_and_reports_refresh_warning_only() {
    let root = tempdir().unwrap();
    let staging_root = root.path().join("staging");
    let payload_root = root.path().join("payloads");
    let desktop_root = root.path().join("applications");

    fs::create_dir(&staging_root).unwrap();
    fs::create_dir(&payload_root).unwrap();
    fs::create_dir(&desktop_root).unwrap();

    let outcome = execute_install(&InstallRequest {
        staging_root: &staging_root,
        final_payload_path: &payload_root.join("bat.AppImage"),
        artifact_bytes: b"\x7fELFAppImage",
        desktop: Some(DesktopIntegrationRequest {
            desktop_entry_path: &desktop_root.join("aim-bat.desktop"),
            desktop_entry_contents: "[Desktop Entry]\nName=bat\nExec=bat.AppImage\nType=Application\n",
            icon_path: None,
            icon_bytes: None,
        }),
        helpers: DesktopHelpers::default(),
    })
    .unwrap();

    assert!(outcome.desktop_entry_path.unwrap().exists());
    assert!(!outcome.warnings.is_empty());
}

#[test]
fn install_executes_refresh_helpers_when_available() {
    let root = tempdir().unwrap();
    let staging_root = root.path().join("staging");
    let payload_root = root.path().join("payloads");
    let desktop_root = root.path().join("applications");
    let helper_root = root.path().join("helpers");
    let log_path = root.path().join("helpers.log");

    fs::create_dir(&staging_root).unwrap();
    fs::create_dir(&payload_root).unwrap();
    fs::create_dir(&desktop_root).unwrap();
    fs::create_dir(&helper_root).unwrap();

    let update_helper = helper_root.join("update-desktop-database");
    let icon_helper = helper_root.join("gtk-update-icon-cache");
    fs::write(
        &update_helper,
        format!("#!/bin/sh\necho desktop:$1 >> {}\n", log_path.display()),
    )
    .unwrap();
    fs::write(
        &icon_helper,
        format!("#!/bin/sh\necho icon:$3 >> {}\n", log_path.display()),
    )
    .unwrap();
    fs::set_permissions(&update_helper, fs::Permissions::from_mode(0o755)).unwrap();
    fs::set_permissions(&icon_helper, fs::Permissions::from_mode(0o755)).unwrap();

    let icon_root = root.path().join("icons/hicolor/256x256/apps");
    fs::create_dir_all(&icon_root).unwrap();

    let outcome = execute_install(&InstallRequest {
        staging_root: &staging_root,
        final_payload_path: &payload_root.join("bat.AppImage"),
        artifact_bytes: b"\x7fELFAppImage\x89PNG\r\n\x1a\nicondataIEND\xaeB`\x82",
        desktop: Some(DesktopIntegrationRequest {
            desktop_entry_path: &desktop_root.join("aim-bat.desktop"),
            desktop_entry_contents: "[Desktop Entry]\nName=bat\nExec=bat.AppImage\nType=Application\n",
            icon_path: Some(&icon_root.join("bat.png")),
            icon_bytes: None,
        }),
        helpers: DesktopHelpers {
            update_desktop_database: true,
            gtk_update_icon_cache: true,
            update_desktop_database_path: Some(update_helper),
            gtk_update_icon_cache_path: Some(icon_helper),
        },
    })
    .unwrap();

    assert!(outcome.warnings.is_empty());
    let log = fs::read_to_string(&log_path).unwrap();
    assert!(log.contains("desktop:"));
    assert!(log.contains("icon:"));
}

#[test]
fn install_extracts_icon_from_appimage_payload_when_icon_path_is_requested() {
    let root = tempdir().unwrap();
    let staging_root = root.path().join("staging");
    let payload_root = root.path().join("payloads");
    let desktop_root = root.path().join("applications");
    let icon_root = root.path().join("icons/hicolor/256x256/apps");

    fs::create_dir(&staging_root).unwrap();
    fs::create_dir(&payload_root).unwrap();
    fs::create_dir(&desktop_root).unwrap();
    fs::create_dir_all(&icon_root).unwrap();

    let outcome = execute_install(&InstallRequest {
        staging_root: &staging_root,
        final_payload_path: &payload_root.join("bat.AppImage"),
        artifact_bytes: b"\x7fELFAppImage\x89PNG\r\n\x1a\nicondataIEND\xaeB`\x82",
        desktop: Some(DesktopIntegrationRequest {
            desktop_entry_path: &desktop_root.join("aim-bat.desktop"),
            desktop_entry_contents: "[Desktop Entry]\nName=bat\nExec=bat.AppImage\nType=Application\n",
            icon_path: Some(&icon_root.join("bat.png")),
            icon_bytes: None,
        }),
        helpers: DesktopHelpers::default(),
    })
    .unwrap();

    let icon_path = outcome.icon_path.unwrap();
    assert!(icon_path.exists());
    assert!(
        fs::read(&icon_path)
            .unwrap()
            .starts_with(b"\x89PNG\r\n\x1a\n")
    );
}