#!/usr/bin/env python3
# Copyright 2021 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""A utility for generating the structured metrics validator.

Takes as input a structured.xml file describing all events and produces a C++
header and implementation file exposing validator function calls.
"""

import argparse
import sys

import setup_modules  # pylint: disable=unused-import

import chromium_src.tools.metrics.structured.code_generator_cpp as code_generator_cpp
from chromium_src.tools.metrics.structured.sync import model

parser = argparse.ArgumentParser(
  description='Generate structured metrics validator'
)
parser.add_argument('--input', help='Path to structured.xml')
parser.add_argument('--cros_input', help='Path to structured_chromiumos.xml')
parser.add_argument('--output', help='Path to generated files.')


def main():
  args = parser.parse_args()
  structured = model.Model(open(args.input, encoding='utf-8').read(), 'chrome')

  if args.cros_input is not None:
    cros_structured = model.Model(
      open(args.cros_input, encoding='utf-8').read(), 'cros'
    )
    structured = model.merge_models(structured, cros_structured)

  code_generator_cpp.ValidatorHeaderTemplate(
    args.output, 'structured_metrics_validator.h'
  ).write_file()

  code_generator_cpp.ValidatorImplTemplate(
    structured, args.output, 'structured_metrics_validator.cc'
  ).write_file()

  return 0


if __name__ == '__main__':
  sys.exit(main())