"""Functions related to flat config payloads

See proto definitions for descriptions of arguments.
"""

load(
    "@proto//chromiumos/config/payload/flat_config.proto",
    flat_config_pb = "chromiumos.config.payload",
)

def _read(file_path):
    return proto.from_jsonpb(
        flat_config_pb.FlatConfigList,
        io.read_file(file_path),
    )

def _get_designs_by_overlay(design_configs):
    """Returns dict of {overlay1: [design1, design2], overlay2...}"""
    overlay_to_designs = {}
    for design_config in design_configs.values:
        design_name = design_config.hw_design.name
        build_target = design_config.sw_config.system_build_target
        program_id = design_config.hw_design.program_id.value
        program_name = design_config.program.name.lower() or program_id.lower()
        overlay = build_target.portage_build_target.overlay_name or program_name
        designs = overlay_to_designs.setdefault(overlay, [])
        designs.append(design_name)

        # Add an explicit entry for all earlier projects with explicit overlays
        overlay_to_designs.setdefault(design_name, []).append(design_name)

    overlay_to_unique_designs = {}
    for overlay in overlay_to_designs:
        overlay_to_unique_designs[overlay] = set(overlay_to_designs[overlay])
    return overlay_to_unique_designs

def _get_configs_by_design(design_configs):
    """Returns all config that is common to a given design

     Args:
       design_configs: A FlatConfigList proto. Required.

    Returns: dict of {design1:
       (program, hw_design, odm, hw_components), design2...}
    """
    designs_to_configs = {}
    for design_config in design_configs.values:
        design_name = design_config.hw_design.name

        # Components are common across all designs, so just pull the first
        if design_name not in designs_to_configs:
            designs_to_configs[design_name] = struct(
                program = design_config.program,
                odm = design_config.odm,
                hw_design = design_config.hw_design,
                hw_components = design_config.hw_components,
            )
    return designs_to_configs

def _create(
        program,
        hw_design,
        odm,
        hw_design_config,
        hw_components = None,
        device_brand = None,
        oem = None,
        sw_config = None,
        brand_sw_config = None):
    return flat_config_pb.FlatConfig(
        program = program,
        hw_design = hw_design,
        odm = odm,
        hw_design_config = hw_design_config,
        hw_components = hw_components,
        device_brand = device_brand,
        oem = oem,
        sw_config = sw_config,
        brand_sw_config = brand_sw_config,
    )

def _create_list(flat_configs):
    return flat_config_pb.FlatConfigList(values = flat_configs)

flat_config = struct(
    read = _read,
    create = _create,
    create_list = _create_list,
    get_designs_by_overlay = _get_designs_by_overlay,
    get_configs_by_design = _get_configs_by_design,
)