// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.

//! Download a copy of `protoc` for the current platform or use a local copy.

use flowey::node::prelude::*;

#[derive(Serialize, Deserialize)]
pub struct ProtocPackage {
    pub protoc_bin: PathBuf,
    pub include_dir: PathBuf,
}

/// Resolve protoc paths from a base directory and validate that they exist.
/// If make_executable is true, this function will attempt to make the protoc binary executable.
fn resolve_protoc_from_dir(
    rt: &mut RustRuntimeServices<'_>,
    base_dir: &Path,
    make_executable: bool,
) -> anyhow::Result<ProtocPackage> {
    let protoc_bin = base_dir
        .join("bin")
        .join(rt.platform().binary("protoc"))
        .absolute()?;

    if !protoc_bin.exists() {
        anyhow::bail!("protoc binary not found at {}", protoc_bin.display())
    }

    let protoc_bin_executable = protoc_bin.is_executable()?;
    if !protoc_bin_executable && !make_executable {
        anyhow::bail!(
            "protoc binary at {} is not executable",
            protoc_bin.display()
        );
    }

    if make_executable {
        protoc_bin.make_executable()?;
    }

    let include_dir = base_dir.join("include").absolute()?;
    if !include_dir.exists() {
        anyhow::bail!(
            "protoc include directory not found at {}",
            include_dir.display()
        )
    }

    Ok(ProtocPackage {
        protoc_bin,
        include_dir,
    })
}

flowey_config! {
    /// Config for the resolve_protoc node.
    pub struct Config {
        /// What version to download (e.g: 27.1)
        pub version: Option<String>,
        /// Use a locally downloaded protoc
        pub local_path: Option<ConfigVar<PathBuf>>,
    }
}

flowey_request! {
    pub enum Request {
        /// Return paths to items in the protoc package
        Get(WriteVar<ProtocPackage>),
    }
}

new_flow_node_with_config!(struct Node);

impl FlowNodeWithConfig for Node {
    type Request = Request;
    type Config = Config;

    fn imports(ctx: &mut ImportCtx<'_>) {
        ctx.import::<crate::install_dist_pkg::Node>();
        ctx.import::<crate::download_gh_release::Node>();
        ctx.import::<crate::cache::Node>();
    }

    fn emit(
        config: Config,
        requests: Vec<Self::Request>,
        ctx: &mut NodeCtx<'_>,
    ) -> anyhow::Result<()> {
        let version = config.version;
        let local_path = config.local_path;
        let mut get_reqs = Vec::new();

        for req in requests {
            match req {
                Request::Get(v) => get_reqs.push(v),
            }
        }

        if version.is_some() && local_path.is_some() {
            anyhow::bail!("Cannot specify both version and local_path config");
        }

        if version.is_none() && local_path.is_none() {
            anyhow::bail!("Must specify a version or local_path config");
        }

        // -- end of req processing -- //

        if get_reqs.is_empty() {
            return Ok(());
        }

        if let Some(local_path) = local_path {
            ctx.emit_rust_step("use local protoc", |ctx| {
                let get_reqs = get_reqs.claim(ctx);
                let local_path = local_path.claim(ctx);
                move |rt| {
                    let local_path = rt.read(local_path);
                    log::info!("using protoc from base path {}", local_path.display());

                    // If a local path is specified, assume protoc is already executable. This is necessary because a
                    // nix-shell is unable to change file permissions but the file will be executable.
                    let pkg = resolve_protoc_from_dir(rt, &local_path, false)?;
                    rt.write_all(get_reqs, &pkg);

                    Ok(())
                }
            });

            return Ok(());
        }

        let version = version.expect("local requests handled above");

        let tag = format!("v{version}");
        let file_name = format!(
            "protoc-{}-{}.zip",
            version,
            match (ctx.platform(), ctx.arch()) {
                // protoc is not currently available for windows aarch64,
                // so emulate the x64 version
                (FlowPlatform::Windows, _) => "win64",
                (FlowPlatform::Linux(_), FlowArch::X86_64) => "linux-x86_64",
                (FlowPlatform::Linux(_), FlowArch::Aarch64) => "linux-aarch_64",
                (FlowPlatform::MacOs, FlowArch::X86_64) => "osx-x86_64",
                (FlowPlatform::MacOs, FlowArch::Aarch64) => "osx-aarch_64",
                (platform, arch) => anyhow::bail!("unsupported platform {platform} {arch}"),
            }
        );

        let protoc_zip = ctx.reqv(|v| crate::download_gh_release::Request {
            repo_owner: "protocolbuffers".into(),
            repo_name: "protobuf".into(),
            needs_auth: false,
            tag: tag.clone(),
            file_name: file_name.clone(),
            path: v,
        });

        let extract_zip_deps = crate::_util::extract::extract_zip_if_new_deps(ctx);
        ctx.emit_rust_step("unpack protoc", |ctx| {
            let extract_zip_deps = extract_zip_deps.clone().claim(ctx);
            let get_reqs = get_reqs.claim(ctx);
            let protoc_zip = protoc_zip.claim(ctx);
            move |rt| {
                let protoc_zip = rt.read(protoc_zip);

                let extract_dir = crate::_util::extract::extract_zip_if_new(
                    rt,
                    extract_zip_deps,
                    &protoc_zip,
                    &tag,
                )?;

                let pkg = resolve_protoc_from_dir(rt, &extract_dir, true)?;
                rt.write_all(get_reqs, &pkg);

                Ok(())
            }
        });

        Ok(())
    }
}
