chromium/tools/style_variable_generator/main.py

#!/usr/bin/env python3
# Copyright 2019 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import argparse
import sys
import os

sys.path += [os.path.dirname(os.path.dirname(__file__))]

from style_variable_generator.css_generator import CSSStyleGenerator
from style_variable_generator.ts_generator import TSStyleGenerator
from style_variable_generator.proto_generator import ProtoStyleGenerator, ProtoJSONStyleGenerator
from style_variable_generator.json_generator import JSONStyleGenerator
from style_variable_generator.views_generator import ViewsCCStyleGenerator, ViewsHStyleGenerator
from style_variable_generator.base_generator import Modes
from style_variable_generator.color_mappings_generator import ColorMappingsCCStyleGenerator, ColorMappingsHStyleGenerator


def parseGeneratorOptionList(options):
    result = {}
    if options is None:
        return result
    for key_value_pair in options:
        key, value = key_value_pair.split('=', 1)
        key = key.strip()
        result[key] = value
    return result


def main():
    parser = argparse.ArgumentParser(
        description='Generate style variables from JSON5 color file.')

    generators = [
        CSSStyleGenerator, ViewsCCStyleGenerator, ViewsHStyleGenerator,
        ProtoStyleGenerator, ProtoJSONStyleGenerator, TSStyleGenerator,
        JSONStyleGenerator, ColorMappingsCCStyleGenerator,
        ColorMappingsHStyleGenerator
    ]

    parser.add_argument(
        '--generator',
        choices=[g.GetName() for g in generators],
        required=True,
        help='type of file to generate, provide this option multiple '
        'times to use multiple generators',
        action='append')
    parser.add_argument('--generate-single-mode',
                        choices=Modes.ALL,
                        help='generates output for a single mode')
    parser.add_argument(
        '--out-file',
        help='file to write output to, the number of out-files '
        'must match the number of generators if specified',
        action='append')
    parser.add_argument('--generator-option',
                        metavar='KEY=VALUE',
                        action='append',
                        help='Set a option specific to the selected generator '
                        'via a key value pair. See the README.md for a '
                        'full list of generator specific options.')
    parser.add_argument('targets', nargs='+', help='source json5 color files')

    args = parser.parse_args()

    if args.out_file and len(args.generator) != len(args.out_file):
        raise ValueError(
            'number of --out-files must match number of --generators')

    for i, g in enumerate(args.generator):
        gen_constructor = next(x for x in generators if g == x.GetName())
        style_generator = gen_constructor()

        style_generator.AddJSONFilesToModel(args.targets)

        style_generator.generate_single_mode = args.generate_single_mode
        style_generator.generator_options = parseGeneratorOptionList(
            args.generator_option)

        if args.out_file:
            style_generator.out_file_path = args.out_file[i]
            with open(args.out_file[i], 'w') as f:
                f.write(style_generator.Render())
        else:
            print('=========', style_generator.GetName(), '=========')
            print(style_generator.Render())

    return 0


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