# Copyright 2020 The ChromiumOS Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""IO-related helper functions."""

import json
import pathlib
from typing import Any, Dict

# Disable spurious no-name-in-module and import-error lints.
# pylint: disable=no-name-in-module, import-error
from chromiumos.config.payload import config_bundle_pb2
from chromiumos.config.payload import flat_config_pb2
from google.protobuf import json_format
from google.protobuf.message import Message


def write_message_json(
    message: Message,
    path: pathlib.Path,
    default_fields=False,
    use_integers_for_enums=False,
):
    """Take a Message and write it to a file as json.

    Args:
        message: protobuf message to write to file
        path: output file write json to
        default_fields: If true, include default values for fields
        use_integers_for_enums: If true, print integers instead of enum names.
    """
    with open(path, "w", encoding="utf-8") as outfile:
        outfile.write(
            json_format.MessageToJson(
                message,
                including_default_value_fields=default_fields,
                sort_keys=True,
                use_integers_for_enums=use_integers_for_enums,
            )
        )


def write_message_binary(message: Message, path: pathlib.Path):
    """Write a Message to a file as binary wire format.

    Args:
        message: protobuf message to write to file
        path: output file write to
    """
    with open(path, "wb") as outfile:
        outfile.write(message.SerializeToString())


def read_json_proto(message, path):
    """Read a jsonproto encoded message from a file.

    Args:
        message: an instance of the protobuffer message to read into
        path (str): path to the json encoded message on disk.

    Returns:
        reference to message
    """

    with open(path, "r", encoding="utf-8") as f:
        json_format.Parse(f.read(), message)
    return message


def read_config(path: str) -> config_bundle_pb2.ConfigBundle:
    """Reads a ConfigBundle message from a jsonpb file.

    Args:
        path: Path to the json proto.

    Returns:
        ConfigBundle parsed from file.
    """
    return read_json_proto(
        config_bundle_pb2.ConfigBundle(),
        path,
    )


def read_flat_config(path: str) -> flat_config_pb2.FlatConfigList:
    """Reads a FlatConfigList message from a jsonpb file.

    Args:
        path: Path to the json proto.

    Returns:
        FlatConfigList parsed from file.
    """
    return read_json_proto(
        flat_config_pb2.FlatConfigList(),
        path,
    )


def read_model_sku_json(factory_dir: pathlib.Path) -> Dict[str, Any]:
    """Reads and parses the model_sku.json file.

    Args:
        factory_dir: Path to a project's factory dir.

    Returns:
        Parsed model_sku.json as a dict
    """
    with open(
        factory_dir.joinpath("generated", "model_sku.json"), encoding="utf-8"
    ) as f:
        return json.load(f)