#[cfg(feature = "local_fs")]
use chrono::{TimeZone, Utc};

use super::*;
#[cfg(feature = "local_fs")]
use crate::server::server_api::ai::{
    ArtifactDownloadCommonFields, FileArtifactResponseData, ScreenshotArtifactResponseData,
};

#[cfg(feature = "local_fs")]
fn sample_file_download_response(filename: &str, filepath: &str) -> ArtifactDownloadResponse {
    ArtifactDownloadResponse::File {
        common: ArtifactDownloadCommonFields {
            artifact_uid: "artifact-123".to_string(),
            created_at: Utc.with_ymd_and_hms(2024, 1, 15, 10, 30, 0).unwrap(),
        },
        data: FileArtifactResponseData {
            download_url: "https://storage.example.com/report.txt".to_string(),
            expires_at: Utc.with_ymd_and_hms(2024, 1, 15, 11, 30, 0).unwrap(),
            content_type: "text/plain".to_string(),
            filepath: filepath.to_string(),
            filename: filename.to_string(),
            description: Some("daily summary".to_string()),
            size_bytes: Some(42),
        },
    }
}

#[test]
fn sanitized_basename_accepts_plain_filename() {
    assert_eq!(
        sanitized_basename("report.txt"),
        Some("report.txt".to_string())
    );
}

#[test]
fn sanitized_basename_extracts_from_path() {
    assert_eq!(
        sanitized_basename("outputs/report.txt"),
        Some("report.txt".to_string())
    );
}

#[test]
#[cfg(feature = "local_fs")]
fn extension_for_content_type_recognizes_image_jpg_alias() {
    assert_eq!(extension_for_content_type("image/jpg"), Some("jpg"));
    assert_eq!(extension_for_content_type("image/jpeg"), Some("jpg"));
}

#[test]
#[cfg(feature = "local_fs")]
fn default_download_filename_prefers_server_filename() {
    assert_eq!(
        default_download_filename(&sample_file_download_response(
            "report.txt",
            "outputs/ignored.txt"
        )),
        "report.txt"
    );
}

#[test]
#[cfg(feature = "local_fs")]
fn default_download_filename_uses_content_type_extension_when_filename_missing() {
    assert_eq!(
        default_download_filename(&sample_file_download_response("", "outputs/report.txt")),
        "artifact-artifact-123.txt"
    );
}

#[test]
#[cfg(feature = "local_fs")]
fn default_download_filename_omits_extension_when_content_type_unknown() {
    let artifact = ArtifactDownloadResponse::Screenshot {
        common: ArtifactDownloadCommonFields {
            artifact_uid: "screenshot-123".to_string(),
            created_at: Utc.with_ymd_and_hms(2024, 1, 15, 10, 30, 0).unwrap(),
        },
        data: ScreenshotArtifactResponseData {
            download_url: "https://storage.example.com/screenshot".to_string(),
            expires_at: Utc.with_ymd_and_hms(2024, 1, 15, 11, 30, 0).unwrap(),
            content_type: "application/octet-stream".to_string(),
            description: Some("dashboard screenshot".to_string()),
        },
    };

    assert_eq!(
        default_download_filename(&artifact),
        "artifact-screenshot-123"
    );
}

#[test]
#[cfg(feature = "local_fs")]
fn download_destination_uses_explicit_path() {
    assert_eq!(
        download_destination(
            &sample_file_download_response("report.txt", "outputs/report.txt"),
            Some(PathBuf::from("downloads/report.txt"))
        ),
        PathBuf::from("downloads/report.txt")
    );
}