#!/usr/bin/env vpython3
# 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.
"""Convert a ConfigBundle instance to a FlatConfig instance.

ConfigBundle is a fully normalized format, where eg: Partners are referred to by
id. By denormalizing into a flattened format, we can more easily query projects."""

import argparse
import logging

from common import config_bundle_utils
from checker import io_utils


def flatten(infile, outfile):
  """Flatten a ConfigBundle .jsonproto file.

    Take a ConfigBundle in .jsonproto format, read it, flatten, and write the
    output FlatConfig to the given output file.

    Args:
      infile  (str): input file containing ConfigBundle in .jsonproto format
      outfile (str): filename to write flattened config to
  pars"""

  config = io_utils.read_config(infile)
  flatlist = config_bundle_utils.flatten_config(config)
  io_utils.write_message_json(flatlist, outfile)
  print(str(len(flatlist.values)))


if __name__ == "__main__":
  parser = argparse.ArgumentParser(description=__doc__)
  parser.add_argument(
      '-i',
      '--input',
      type=str,
      required=True,
      help="""ConfigBundle to flatten in jsonpb format.""")
  parser.add_argument(
      '-o',
      '--output',
      type=str,
      required=True,
      help='output file to write FlatConfigList jsonproto to')
  parser.add_argument("-l", "--log", type=str, help='set logging level')
  parser.add_argument(
      "-v", "--verbose", help="increase output verbosity", action="store_true")

  args = parser.parse_args()
  # pylint: disable=invalid-name
  loglevel = logging.INFO if args.verbose else logging.WARNING
  if args.log:
    loglevel = {
        "critical": logging.CRITICAL,
        "error": logging.ERROR,
        "warning": logging.WARNING,
        "info": logging.INFO,
        "debug": logging.DEBUG,
    }.get(args.log.lower())

    if not loglevel:
      logging.error("invalid value for -l/--log '%s'", args.log)

  logging.basicConfig(level=loglevel)
  flatten(args.input, args.output)