// Copyright 2019 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

use std::collections::HashMap;

use sdk_metadata::{CcPrebuiltLibrary, CpuArchitecture, JsonObject};

use crate::app::Result;
use crate::file_provider::{merge_files, FileProvider};
use crate::tarball::{InputTarball, OutputTarball, TarballContent};

impl FileProvider for CcPrebuiltLibrary {
    fn get_common_files(&self) -> Vec<String> {
        self.headers.clone()
    }

    fn get_arch_files(&self) -> HashMap<CpuArchitecture, Vec<String>> {
        let mut result = HashMap::new();
        for (arch, group) in &self.binaries {
            let mut files = Vec::new();
            files.push(group.link.clone());
            if let Some(dist) = group.dist.clone() {
                files.push(dist);
            }
            if let Some(debug) = group.debug.clone() {
                files.push(debug);
            }
            result.insert(arch.clone(), files);
        }
        result
    }
}

pub fn merge_cc_prebuilt_library<F: TarballContent>(
    meta_path: &str,
    base: &impl InputTarball<F>,
    complement: &impl InputTarball<F>,
    output: &mut impl OutputTarball<F>,
) -> Result<()> {
    let base_meta: CcPrebuiltLibrary = base.get_metadata(meta_path)?;
    let complement_meta: CcPrebuiltLibrary = complement.get_metadata(meta_path)?;
    merge_files(&base_meta, base, &complement_meta, complement, output)?;

    let mut meta = base_meta.clone();
    for (arch, group) in &complement_meta.binaries {
        if !meta.binaries.contains_key(arch) {
            meta.binaries.insert(arch.clone(), group.clone());
        }
    }
    meta.validate()?;
    output.write_json(meta_path, &meta)?;

    Ok(())
}

#[cfg(test)]
mod tests {
    use crate::testing::{MockInputTarball, MockOutputTarball};

    use super::*;

    #[test]
    fn test_merge() {
        let meta = "pkg/foobar/meta.json";
        let base_data = r#"
        {
            "name": "foobar",
            "type": "cc_prebuilt_library",
            "format": "shared",
            "root": "pkg/foobar",
            "headers": [
                "pkg/foobar/include/one.h",
                "pkg/foobar/include/two.h"
            ],
            "include_dir": "pkg/foobar/include",
            "deps": [
                "raboof"
            ],
            "binaries": {
                "x64": {
                    "link": "arch/x64/lib/libfoobar.so"
                }
            }
        }
        "#;
        let complement_data = r#"
        {
            "name": "foobar",
            "type": "cc_prebuilt_library",
            "format": "shared",
            "root": "pkg/foobar",
            "headers": [
                "pkg/foobar/include/one.h",
                "pkg/foobar/include/two.h"
            ],
            "include_dir": "pkg/foobar/include",
            "deps": [
                "raboof"
            ],
            "binaries": {
                "arm64": {
                    "link": "arch/arm64/lib/libfoobar.so"
                }
            }
        }
        "#;
        let base = MockInputTarball::new();
        base.add(meta, base_data);
        base.add("pkg/foobar/include/one.h", "one");
        base.add("pkg/foobar/include/two.h", "two");
        base.add("arch/x64/lib/libfoobar.so", "libx64");
        let complement = MockInputTarball::new();
        complement.add(meta, complement_data);
        complement.add("pkg/foobar/include/one.h", "one");
        complement.add("pkg/foobar/include/two.h", "two");
        complement.add("arch/arm64/lib/libfoobar.so", "libarm64");
        let mut output = MockOutputTarball::new();
        merge_cc_prebuilt_library(meta, &base, &complement, &mut output)
            .expect("Merge routine failed");
        output.assert_has_file(meta);
        output.assert_has_file("pkg/foobar/include/one.h");
        output.assert_has_file("pkg/foobar/include/two.h");
        output.assert_has_file("arch/x64/lib/libfoobar.so");
        output.assert_has_file("arch/arm64/lib/libfoobar.so");
        let data = CcPrebuiltLibrary::new(output.get_content(meta).as_bytes())
            .expect("Generated metadata is invalid");
        assert!(data.binaries.len() == 2, "Invalid number of architectures");
    }

    #[test]
    fn test_merge_failed() {
        let meta = "pkg/foobar/meta.json";
        let base_data = r#"
        {
            "name": "foobar",
            "type": "cc_prebuilt_library",
            "format": "shared",
            "root": "pkg/foobar",
            "headers": [
                "pkg/foobar/include/one.h"
            ],
            "include_dir": "pkg/foobar/include",
            "deps": [
                "raboof"
            ],
            "binaries": {
                "x64": {
                    "link": "arch/x64/lib/libfoobar.so"
                }
            }
        }
        "#;
        // This metadata has different headers.
        let complement_data = r#"
        {
            "name": "foobar",
            "type": "cc_prebuilt_library",
            "format": "shared",
            "root": "pkg/foobar",
            "headers": [
                "pkg/foobar/include/two.h"
            ],
            "include_dir": "pkg/foobar/include",
            "deps": [
                "raboof"
            ],
            "binaries": {
                "arm64": {
                    "link": "arch/arm64/lib/libfoobar.so"
                }
            }
        }
        "#;
        let base = MockInputTarball::new();
        base.add(meta, base_data);
        base.add("pkg/foobar/include/one.h", "one");
        base.add("arch/x64/lib/libfoobar.so", "libx64");
        let complement = MockInputTarball::new();
        complement.add(meta, complement_data);
        complement.add("pkg/foobar/include/two.h", "two");
        complement.add("arch/arm64/lib/libfoobar.so", "libarm64");
        let mut output = MockOutputTarball::new();
        let result = merge_cc_prebuilt_library(meta, &base, &complement, &mut output);
        assert!(result.is_err(), "Merge routine should have failed");
    }
}