src/developer/ffx/plugins/assembly/src/operations/product.rs

// Copyright 2021 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 crate::operations::product::assembly_builder::ImageAssemblyConfigBuilder;
use crate::util;
use anyhow::{Context, Result};
use assembly_config::product_config::ProductAssemblyConfig;
use ffx_assembly_args::ProductArgs;
use log::info;

mod assembly_builder;

pub fn assemble(args: ProductArgs) -> Result<()> {
    let ProductArgs {
        product,
        outdir,
        gendir: _,
        input_bundles_dir,
        legacy_bundle_dir,
        additional_packages_path,
    } = args;

    info!("Loading configuration files.");
    info!("  product: {}", product.display());

    let config: ProductAssemblyConfig =
        util::read_config(&product).context("Loading product configuration")?;

    let mut builder = ImageAssemblyConfigBuilder::default();

    let legacy_bundle_path = legacy_bundle_dir.join("legacy").join("assembly_config.json");
    let emulator_bundle_path =
        input_bundles_dir.join("emulator_support").join("assembly_config.json");

    for bundle_path in vec![legacy_bundle_path, emulator_bundle_path] {
        builder
            .add_bundle(&bundle_path)
            .context(format!("Adding input bundle: {}", bundle_path.display()))?;
    }

    // Set structured configuration
    builder.set_bootfs_structured_config(config.define_bootfs_config()?);
    for (package, config) in config.define_repackaging()? {
        builder.set_structured_config(package, config)?;
    }

    builder
        .add_product_packages(&config.product.packages)
        .context("Adding product-provided packages")?;

    if let Some(package_config_path) = additional_packages_path {
        let additional_packages =
            util::read_config(&package_config_path).context("Loading additional package config")?;
        builder.add_product_packages(&additional_packages).context("Adding additional packages")?;
    }

    let image_assembly = builder.build(&outdir).context("Building Image Assembly config")?;
    assembly_validate_product::validate_product(&image_assembly)?;

    let image_assembly_path = outdir.join("image_assembly.json");
    let image_assembly_file = std::fs::File::create(&image_assembly_path).context(format!(
        "Failed to create image assembly config file: {}",
        image_assembly_path.display()
    ))?;
    serde_json::to_writer_pretty(image_assembly_file, &image_assembly)?;

    Ok(())
}