chromium/tools/metrics/structured/gen_validator.py

#!/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 code_generator_cpp
from 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())