Exemplo n.º 1
0
def parse_schema_descriptor(proto_descriptor):
    """Parse schema descriptor and return schema object.

  Args:
    proto_descriptor: Path to a protobuf file set descriptor binary
      containing schema files.

  Returns:
    A schema.Schema object built from the provided schema_files.

  Raises:
    Exception: No schema files found.
  """

    schema_obj = schema.Schema()
    schema_parser = nwv_parser.Parser(schema_obj)

    logging.debug('Parse schema descriptor.')
    schema_parser.add_file_set(proto_descriptor.file)

    return schema_obj
Exemplo n.º 2
0
def codegen(request, response):
    """Generates wdl c code for devices from Jinja templates.

  Args:
    request: CodeGeneratorRequest, see google/protobuf/compiler/plugin.proto
    response: CodeGeneratorResponse, see google/protobuf/compiler/plugin.proto.
      Output files filled with generated files.

  Raises:
    Exception: Valid Jinja template directory required for wdl plugin.
  """

    args = {}
    for argument in request.parameter.split(','):
        (key, value) = argument.split('=', 1)
        values = value.split(':')
        if len(values) == 1:
            args[key] = values[0]
        else:
            args[key] = values

    legacy_mode_enabled = ('legacy_mode_enabled' in args
                           and args['legacy_mode_enabled'].lower() == 'true')

    gen_dependencies = ('gen_dependencies' in args
                        and args['gen_dependencies'].lower() == 'true')

    codegen_reference_mode = ('codegen_reference_mode' in args and
                              args['codegen_reference_mode'].lower() == 'true')

    if args['templates'] is None or not args['templates']:
        raise Exception('wdl_plugin: \'templates\' argument is empty')

    if isinstance(args['templates'], list):
        template_files = args['templates']
    else:
        template_files = [
            args['templates'],
        ]

    template_languages = {
        'c': template_set.TemplateLanguage.C,
        'cpp': template_set.TemplateLanguage.CPLUSPLUS,
        'java': template_set.TemplateLanguage.JAVA,
        'js': template_set.TemplateLanguage.JAVASCRIPT,
        'md': template_set.TemplateLanguage.MARKDOWN,
        'objc': template_set.TemplateLanguage.OBJECTIVEC,
    }
    if 'language' not in args or args['language'] not in template_languages:
        language = template_set.TemplateLanguage.BASE
    else:
        language = template_languages[args['language']]

    schema_obj = schema.Schema()
    schema_parser = nwv_parser.Parser(schema_obj)

    file_descs_to_gen = [
        proto_file for proto_file in request.proto_file
        if (('semantic' not in proto_file.name) and (
            'retention' not in proto_file.name))
    ]

    dependency_set = []

    # This file needs to get added to the dependency list if we're in
    # codegen mode, since this file doesn't show up as a dependency by
    # default, but is still necessary for some code-generated targets.
    if (codegen_reference_mode):
        dependency_set.append('google/protobuf/field_mask.proto')

    for proto_file in file_descs_to_gen:
        dependency_set.append(proto_file.name)

    schema_parser.add_file_set(file_descs_to_gen)

    gwvc.check(schema_obj)

    # Add two spaces to each log messages to make it line up with our output.
    template_set.log = lambda *a: print(' ', *a)

    templates = template_set.TemplateSet(
        template_files,
        legacy_mode_enabled=legacy_mode_enabled,
        language=language)

    if gen_dependencies:
        templates.codegen(schema_obj, None, dependency_set)
    else:
        templates.codegen(schema_obj, None, request.file_to_generate)

    for filename, content in templates.output_files:
        out_file = response.file.add()
        out_file.name = filename
        # The newline was added in the legacy template_set file writer,
        # so it's included here to preserve compatibility.
        out_file.content = content.encode('utf-8') + '\n'