waterui-cli 0.1.3

//! Apple platform implementations for macOS, iOS, iOS Simulator, etc.

use std::path::PathBuf;
use std::{env, fmt::Write};

use color_eyre::eyre::{self, bail};
use smol::fs;
use target_lexicon::{
    Aarch64Architecture, Architecture, DefaultToHost, Environment, OperatingSystem, Triple, Vendor,
};

use crate::{
    apple::{
        backend::AppleBackend,
        device::{AppleDevice, AppleSimulator},
        toolchain::{AppleSdk, AppleToolchain, Xcode},
    },
    build::{BuildOptions, RustBuild},
    device::Artifact,
    platform::{PackageOptions, Platform},
    project::Project,
    utils::{copy_file, run_command},
};

// ============================================================================
// Common Apple Platform Trait
// ============================================================================

/// Trait for Apple-specific platform functionality.
///
/// This trait provides Apple-specific methods that are shared across all Apple platforms.
/// Each platform type (`MacOS`, Ios, `IosSimulator`, etc.) implements this trait.
pub trait ApplePlatformExt: Platform {
    /// Get the SDK name for xcodebuild (e.g., "macosx", "iphoneos", "iphonesimulator").
    fn sdk_name(&self) -> &'static str;

    /// Check if this platform is a simulator.
    fn is_simulator(&self) -> bool;

    /// Get the architecture for this platform.
    fn arch(&self) -> Architecture;
}

// ============================================================================
// Shared Implementation Helpers
// ============================================================================

/// Build Rust library for an Apple platform.
async fn build_rust_lib(
    project: &Project,
    triple: Triple,
    options: BuildOptions,
) -> eyre::Result<PathBuf> {
    let build = RustBuild::new(project.root(), triple, options.is_hot_reload());
    let lib_dir = build.build_lib(options.is_release()).await?;

    // If output_dir is specified, copy the library there
    if let Some(output_dir) = options.output_dir() {
        let lib_name = project.crate_name().replace('-', "_");
        let source_lib = lib_dir.join(format!("lib{lib_name}.a"));

        if source_lib.exists() {
            fs::create_dir_all(output_dir).await?;
            let dest_lib = output_dir.join("libwaterui_app.a");
            copy_file(&source_lib, &dest_lib).await?;
        }
    }

    Ok(lib_dir)
}

async fn validate_local_apple_backend(project: &Project) -> eyre::Result<()> {
    let Some(waterui_path) = project.manifest().waterui_path.as_deref() else {
        return Ok(());
    };

    let waterui_root = {
        let candidate = PathBuf::from(waterui_path);
        if candidate.is_absolute() {
            candidate
        } else {
            project.root().join(candidate)
        }
    };

    let package_manifest = waterui_root.join("backends/apple/Package.swift");
    if package_manifest.exists() {
        return Ok(());
    }

    let gitmodules_path = waterui_root.join(".gitmodules");
    let submodule_hint = if gitmodules_path.exists() {
        fs::read_to_string(&gitmodules_path)
            .await
            .ok()
            .filter(|c| c.contains("backends/apple"))
            .map(|_| {
                format!(
                    "It looks like `backends/apple` is a git submodule; run `git submodule update --init --recursive` in `{}`.",
                    waterui_root.display()
                )
            })
    } else {
        None
    };

    let mut message = format!(
        "Local Apple backend Swift package manifest not found at `{}`.\n\
         This is typically caused by an incomplete local WaterUI checkout (e.g. missing submodules) or an incorrect `waterui_path` in `Water.toml`.\n",
        package_manifest.display()
    );

    if let Some(hint) = submodule_hint {
        writeln!(&mut message, "{hint}\n").unwrap();
    } else {
        writeln!(
            &mut message,
            "If you're using a local WaterUI checkout, ensure `backends/apple/` exists and contains `Package.swift`."
        ).unwrap();
    }

    bail!("{message}");
}

/// Clean Xcode build artifacts for an Apple platform.
async fn clean_apple(project: &Project) -> eyre::Result<()> {
    let Some(backend) = project.apple_backend() else {
        return Ok(()); // Nothing to clean if no backend configured
    };

    let project_path = project.backend_path::<AppleBackend>();
    let xcodeproj = project_path.join(format!("{}.xcodeproj", backend.scheme));

    if !xcodeproj.exists() {
        return Ok(());
    }

    run_command(
        "xcodebuild",
        [
            "-project",
            xcodeproj.to_str().unwrap_or_default(),
            "-scheme",
            &backend.scheme,
            "clean",
        ],
    )
    .await?;

    let build_dir = project_path.join("build");
    if build_dir.exists() {
        fs::remove_dir_all(&build_dir).await?;
    }

    Ok(())
}

/// Package an Apple app using xcodebuild.
async fn package_apple<P: ApplePlatformExt>(
    platform: &P,
    project: &Project,
    options: PackageOptions,
) -> eyre::Result<Artifact> {
    let backend = project
        .apple_backend()
        .ok_or_else(|| eyre::eyre!("Apple backend must be configured"))?;

    let project_path = project.backend_path::<AppleBackend>();
    let xcodeproj = project_path.join(format!("{}.xcodeproj", backend.scheme));

    if !xcodeproj.exists() {
        bail!(
            "Xcode project not found at {}. Did you run 'water create'?",
            xcodeproj.display()
        );
    }

    validate_local_apple_backend(project).await?;

    // Tell Xcode not to call `water build` again (we already built)
    // SAFETY: CLI runs on main thread before spawning build processes
    unsafe {
        env::set_var("WATERUI_SKIP_RUST_BUILD", "1");
    }

    let configuration = if options.is_debug() {
        "Debug"
    } else {
        "Release"
    };

    let derived_data = project_path.join(".water/DerivedData");

    let target_dir = project.target_dir();

    // Copy the built Rust library to where Xcode expects it
    let profile = if options.is_debug() {
        "debug"
    } else {
        "release"
    };
    let lib_dir = target_dir.join(platform.triple().to_string()).join(profile);
    let lib_name = project.crate_name().replace('-', "_");
    let source_lib = lib_dir.join(format!("lib{lib_name}.a"));

    // Xcode uses "Debug-iphonesimulator" for simulators, "Debug" for macOS
    let products_config = if platform.sdk_name() == "macosx" {
        configuration.to_string()
    } else {
        format!("{configuration}-{}", platform.sdk_name())
    };
    let products_dir = derived_data.join("Build/Products").join(&products_config);
    fs::create_dir_all(&products_dir).await?;
    let dest_lib = products_dir.join("libwaterui_app.a");
    copy_file(&source_lib, &dest_lib).await?;

    // Build with xcodebuild
    // Determine the Xcode arch name from the platform architecture
    let arch_name = match platform.arch() {
        Architecture::Aarch64(_) => "arm64",
        Architecture::X86_64 => "x86_64",
        _ => unimplemented!(),
    };
    let archs_arg = format!("ARCHS={arch_name}");

    let mut args = vec![
        "-project",
        xcodeproj.to_str().unwrap_or_default(),
        "-scheme",
        &backend.scheme,
        "-configuration",
        configuration,
        "-sdk",
        platform.sdk_name(),
        "-derivedDataPath",
        derived_data.to_str().unwrap_or_default(),
        &archs_arg,
        "ONLY_ACTIVE_ARCHITECTURE=YES",
        "build",
    ];

    if platform.is_simulator() || options.is_debug() {
        args.extend([
            "CODE_SIGNING_ALLOWED=NO",
            "CODE_SIGNING_REQUIRED=NO",
            "CODE_SIGN_IDENTITY=-",
        ]);
    }

    run_command("xcodebuild", args.iter().copied()).await?;

    // Reset the environment variable
    unsafe {
        env::set_var("WATERUI_SKIP_RUST_BUILD", "0");
    }

    let app_path = products_dir.join(format!("{}.app", backend.scheme));

    if !app_path.exists() {
        bail!(
            "Built app not found at {}. Check xcodebuild output for errors.",
            app_path.display()
        );
    }

    Ok(Artifact::new(project.bundle_identifier(), app_path))
}

// ============================================================================
// macOS Platform
// ============================================================================

/// macOS platform for building and running on the current Mac.
#[derive(Debug, Clone, Copy, Default)]
pub struct MacOS;

impl MacOS {
    /// Create a new macOS platform instance.
    #[must_use]
    pub const fn new() -> Self {
        Self
    }
}

impl Platform for MacOS {
    type Device = AppleDevice;
    type Toolchain = AppleToolchain;

    async fn scan(&self) -> eyre::Result<Vec<Self::Device>> {
        // macOS doesn't have simulators to scan
        Ok(vec![])
    }

    async fn build(&self, project: &Project, options: BuildOptions) -> eyre::Result<PathBuf> {
        build_rust_lib(project, self.triple(), options).await
    }

    fn toolchain(&self) -> Self::Toolchain {
        (Xcode, AppleSdk::Macos)
    }

    fn triple(&self) -> Triple {
        Triple {
            architecture: self.arch(),
            vendor: Vendor::Apple,
            operating_system: OperatingSystem::Darwin(None),
            environment: Environment::Unknown,
            binary_format: target_lexicon::BinaryFormat::Macho,
        }
    }

    async fn clean(&self, project: &Project) -> eyre::Result<()> {
        clean_apple(project).await
    }

    async fn package(&self, project: &Project, options: PackageOptions) -> eyre::Result<Artifact> {
        package_apple(self, project, options).await
    }
}

impl ApplePlatformExt for MacOS {
    fn sdk_name(&self) -> &'static str {
        "macosx"
    }

    fn is_simulator(&self) -> bool {
        false
    }

    fn arch(&self) -> Architecture {
        DefaultToHost::default().0.architecture
    }
}

// ============================================================================
// iOS Platform (Physical Device)
// ============================================================================

/// iOS platform for building and running on physical iOS devices.
#[derive(Debug, Clone, Copy, Default)]
pub struct Ios;

impl Ios {
    /// Create a new iOS platform instance.
    #[must_use]
    pub const fn new() -> Self {
        Self
    }
}

impl Platform for Ios {
    type Device = AppleDevice;
    type Toolchain = AppleToolchain;

    async fn scan(&self) -> eyre::Result<Vec<Self::Device>> {
        // TODO: Scan for physical iOS devices
        Ok(vec![])
    }

    async fn build(&self, project: &Project, options: BuildOptions) -> eyre::Result<PathBuf> {
        build_rust_lib(project, self.triple(), options).await
    }

    fn toolchain(&self) -> Self::Toolchain {
        (Xcode, AppleSdk::Ios)
    }

    fn triple(&self) -> Triple {
        Triple {
            architecture: Architecture::Aarch64(Aarch64Architecture::Aarch64),
            vendor: Vendor::Apple,
            operating_system: OperatingSystem::IOS(None),
            environment: Environment::Unknown,
            binary_format: target_lexicon::BinaryFormat::Macho,
        }
    }

    async fn clean(&self, project: &Project) -> eyre::Result<()> {
        clean_apple(project).await
    }

    async fn package(&self, project: &Project, options: PackageOptions) -> eyre::Result<Artifact> {
        package_apple(self, project, options).await
    }
}

impl ApplePlatformExt for Ios {
    fn sdk_name(&self) -> &'static str {
        "iphoneos"
    }

    fn is_simulator(&self) -> bool {
        false
    }

    fn arch(&self) -> Architecture {
        Architecture::Aarch64(Aarch64Architecture::Aarch64)
    }
}

// ============================================================================
// iOS Simulator Platform
// ============================================================================

/// iOS Simulator platform for building and running on the iOS Simulator.
#[derive(Debug, Clone, Copy, Default)]
pub struct IosSimulator;

impl IosSimulator {
    /// Create a new iOS Simulator platform instance.
    #[must_use]
    pub const fn new() -> Self {
        Self
    }
}

impl Platform for IosSimulator {
    type Device = AppleDevice;
    type Toolchain = AppleToolchain;

    async fn scan(&self) -> eyre::Result<Vec<Self::Device>> {
        let simulators = AppleSimulator::scan().await?;

        let filtered: Vec<AppleDevice> = simulators
            .into_iter()
            .filter(|sim| {
                // Filter to only iOS simulators (iPhone, iPad)
                !sim.device_type_identifier.contains("Apple-TV")
                    && !sim.device_type_identifier.contains("Apple-Watch")
                    && !sim.device_type_identifier.contains("Apple-Vision")
            })
            .map(AppleDevice::Simulator)
            .collect();

        Ok(filtered)
    }

    async fn build(&self, project: &Project, options: BuildOptions) -> eyre::Result<PathBuf> {
        build_rust_lib(project, self.triple(), options).await
    }

    fn toolchain(&self) -> Self::Toolchain {
        (Xcode, AppleSdk::Ios)
    }

    fn triple(&self) -> Triple {
        let arch = self.arch();
        let env = match arch {
            Architecture::X86_64 => Environment::Unknown,
            _ => Environment::Sim,
        };

        Triple {
            architecture: arch,
            vendor: Vendor::Apple,
            operating_system: OperatingSystem::IOS(None),
            environment: env,
            binary_format: target_lexicon::BinaryFormat::Macho,
        }
    }

    async fn clean(&self, project: &Project) -> eyre::Result<()> {
        clean_apple(project).await
    }

    async fn package(&self, project: &Project, options: PackageOptions) -> eyre::Result<Artifact> {
        package_apple(self, project, options).await
    }
}

impl ApplePlatformExt for IosSimulator {
    fn sdk_name(&self) -> &'static str {
        "iphonesimulator"
    }

    fn is_simulator(&self) -> bool {
        true
    }

    fn arch(&self) -> Architecture {
        DefaultToHost::default().0.architecture
    }
}

// ============================================================================
// Legacy ApplePlatform (for backwards compatibility)
// ============================================================================

/// Legacy Apple platform enum for backwards compatibility.
///
/// This struct wraps the platform kind and delegates to the appropriate
/// platform implementation. New code should use `MacOS`, `Ios`, or `IosSimulator` directly.
#[derive(Debug, Clone)]
pub struct ApplePlatform {
    arch: Architecture,
    kind: ApplePlatformKind,
}

/// Apple platform kinds.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum ApplePlatformKind {
    /// macOS
    MacOS,
    /// iOS (physical device)
    Ios,
    /// iOS Simulator
    IosSimulator,
    /// tvOS (physical device)
    TvOs,
    /// tvOS Simulator
    TvOsSimulator,
    /// watchOS (physical device)
    WatchOs,
    /// watchOS Simulator
    WatchOsSimulator,
    /// visionOS (physical device)
    VisionOs,
    /// visionOS Simulator
    VisionOsSimulator,
}

impl ApplePlatform {
    /// Create a new Apple platform with the specified architecture and kind.
    #[must_use]
    pub const fn new(arch: Architecture, kind: ApplePlatformKind) -> Self {
        Self { arch, kind }
    }

    /// Create an Apple platform for the current macOS host.
    #[must_use]
    pub fn macos() -> Self {
        Self {
            arch: DefaultToHost::default().0.architecture,
            kind: ApplePlatformKind::MacOS,
        }
    }

    /// Create an Apple platform for iOS (physical device).
    #[must_use]
    pub const fn ios() -> Self {
        Self {
            arch: Architecture::Aarch64(Aarch64Architecture::Aarch64),
            kind: ApplePlatformKind::Ios,
        }
    }

    /// Create an Apple platform for iOS Simulator (native architecture).
    #[must_use]
    pub fn ios_simulator() -> Self {
        Self {
            arch: DefaultToHost::default().0.architecture,
            kind: ApplePlatformKind::IosSimulator,
        }
    }

    /// Create an Apple platform for iOS Simulator on ARM64 (Apple Silicon).
    #[must_use]
    pub const fn ios_simulator_arm64() -> Self {
        Self {
            arch: Architecture::Aarch64(Aarch64Architecture::Aarch64),
            kind: ApplePlatformKind::IosSimulator,
        }
    }

    /// Create an Apple platform for iOS Simulator on `x86_64` (Intel).
    #[must_use]
    pub const fn ios_simulator_x86_64() -> Self {
        Self {
            arch: Architecture::X86_64,
            kind: ApplePlatformKind::IosSimulator,
        }
    }

    /// Create an Apple platform for macOS on ARM64 (Apple Silicon).
    #[must_use]
    pub const fn macos_arm64() -> Self {
        Self {
            arch: Architecture::Aarch64(Aarch64Architecture::Aarch64),
            kind: ApplePlatformKind::MacOS,
        }
    }

    /// Create an Apple platform for macOS on `x86_64` (Intel).
    #[must_use]
    pub const fn macos_x86_64() -> Self {
        Self {
            arch: Architecture::X86_64,
            kind: ApplePlatformKind::MacOS,
        }
    }

    /// Parse a device type identifier to determine the platform.
    #[must_use]
    pub fn from_device_type_identifier(id: &str) -> Self {
        let is_simulator = id.contains("CoreSimulator");
        let arch = if is_simulator {
            DefaultToHost::default().0.architecture
        } else {
            Architecture::Aarch64(Aarch64Architecture::Aarch64)
        };

        let kind = if id.contains("Apple-TV") {
            if is_simulator {
                ApplePlatformKind::TvOsSimulator
            } else {
                ApplePlatformKind::TvOs
            }
        } else if id.contains("Apple-Watch") {
            if is_simulator {
                ApplePlatformKind::WatchOsSimulator
            } else {
                ApplePlatformKind::WatchOs
            }
        } else if id.contains("Apple-Vision") {
            if is_simulator {
                ApplePlatformKind::VisionOsSimulator
            } else {
                ApplePlatformKind::VisionOs
            }
        } else if id.contains("Mac") {
            ApplePlatformKind::MacOS
        } else if is_simulator {
            ApplePlatformKind::IosSimulator
        } else {
            ApplePlatformKind::Ios
        };

        Self { arch, kind }
    }

    /// Get the platform kind.
    #[must_use]
    pub const fn kind(&self) -> &ApplePlatformKind {
        &self.kind
    }

    /// Get the architecture.
    #[must_use]
    pub const fn arch(&self) -> &Architecture {
        &self.arch
    }

    /// Get the SDK name for xcodebuild.
    #[must_use]
    pub const fn sdk_name(&self) -> &'static str {
        match self.kind {
            ApplePlatformKind::MacOS => "macosx",
            ApplePlatformKind::Ios => "iphoneos",
            ApplePlatformKind::IosSimulator => "iphonesimulator",
            ApplePlatformKind::TvOs => "appletvos",
            ApplePlatformKind::TvOsSimulator => "appletvsimulator",
            ApplePlatformKind::WatchOs => "watchos",
            ApplePlatformKind::WatchOsSimulator => "watchsimulator",
            ApplePlatformKind::VisionOs => "xros",
            ApplePlatformKind::VisionOsSimulator => "xrsimulator",
        }
    }

    /// Check if this platform is a simulator.
    #[must_use]
    pub const fn is_simulator(&self) -> bool {
        matches!(
            self.kind,
            ApplePlatformKind::IosSimulator
                | ApplePlatformKind::TvOsSimulator
                | ApplePlatformKind::WatchOsSimulator
                | ApplePlatformKind::VisionOsSimulator
        )
    }
}

impl Platform for ApplePlatform {
    type Device = AppleDevice;
    type Toolchain = AppleToolchain;

    async fn scan(&self) -> eyre::Result<Vec<Self::Device>> {
        let simulators = AppleSimulator::scan().await?;

        let filtered: Vec<AppleDevice> = simulators
            .into_iter()
            .filter(|sim| {
                let sim_platform = Self::from_device_type_identifier(&sim.device_type_identifier);
                matches!(
                    (&self.kind, &sim_platform.kind),
                    (
                        ApplePlatformKind::IosSimulator,
                        ApplePlatformKind::IosSimulator
                    ) | (
                        ApplePlatformKind::TvOsSimulator,
                        ApplePlatformKind::TvOsSimulator
                    ) | (
                        ApplePlatformKind::WatchOsSimulator,
                        ApplePlatformKind::WatchOsSimulator
                    ) | (
                        ApplePlatformKind::VisionOsSimulator,
                        ApplePlatformKind::VisionOsSimulator,
                    )
                )
            })
            .map(AppleDevice::Simulator)
            .collect();

        Ok(filtered)
    }

    async fn build(&self, project: &Project, options: BuildOptions) -> eyre::Result<PathBuf> {
        build_rust_lib(project, self.triple(), options).await
    }

    fn toolchain(&self) -> Self::Toolchain {
        let sdk = match self.kind {
            ApplePlatformKind::MacOS => AppleSdk::Macos,
            ApplePlatformKind::Ios | ApplePlatformKind::IosSimulator => AppleSdk::Ios,
            ApplePlatformKind::TvOs | ApplePlatformKind::TvOsSimulator => AppleSdk::TvOs,
            ApplePlatformKind::WatchOs | ApplePlatformKind::WatchOsSimulator => AppleSdk::WatchOs,
            ApplePlatformKind::VisionOs | ApplePlatformKind::VisionOsSimulator => {
                AppleSdk::VisionOs
            }
        };
        (Xcode, sdk)
    }

    fn triple(&self) -> Triple {
        let (os, env) = match self.kind {
            ApplePlatformKind::MacOS => (OperatingSystem::Darwin(None), Environment::Unknown),
            ApplePlatformKind::Ios => (OperatingSystem::IOS(None), Environment::Unknown),
            ApplePlatformKind::IosSimulator => match self.arch {
                Architecture::X86_64 => (OperatingSystem::IOS(None), Environment::Unknown),
                _ => (OperatingSystem::IOS(None), Environment::Sim),
            },
            ApplePlatformKind::TvOs => (OperatingSystem::TvOS(None), Environment::Unknown),
            ApplePlatformKind::TvOsSimulator => (OperatingSystem::TvOS(None), Environment::Sim),
            ApplePlatformKind::WatchOs => (OperatingSystem::WatchOS(None), Environment::Unknown),
            ApplePlatformKind::WatchOsSimulator => {
                (OperatingSystem::WatchOS(None), Environment::Sim)
            }
            ApplePlatformKind::VisionOs => (OperatingSystem::VisionOS(None), Environment::Unknown),
            ApplePlatformKind::VisionOsSimulator => {
                (OperatingSystem::VisionOS(None), Environment::Sim)
            }
        };

        Triple {
            architecture: self.arch,
            vendor: Vendor::Apple,
            operating_system: os,
            environment: env,
            binary_format: target_lexicon::BinaryFormat::Macho,
        }
    }

    async fn clean(&self, project: &Project) -> eyre::Result<()> {
        clean_apple(project).await
    }

    async fn package(&self, project: &Project, options: PackageOptions) -> eyre::Result<Artifact> {
        package_apple(self, project, options).await
    }
}

impl ApplePlatformExt for ApplePlatform {
    fn sdk_name(&self) -> &'static str {
        self.sdk_name()
    }

    fn is_simulator(&self) -> bool {
        self.is_simulator()
    }

    fn arch(&self) -> Architecture {
        self.arch
    }
}