Beispiel #1
0
def main(argv):
    if FLAGS.output_dir and FLAGS.output_file:
        sys.stderr.write(
            'Only one of --output_dir or --output_file can be set.')
        sys.exit(1)

    if not FLAGS.output_dir:
        FLAGS.output_dir = DEFAULT_OUTPUT_DIR
    if not FLAGS.output_file:
        FLAGS.output_file = DEFAULT_OUTPUT_FILE

    bzl_files = argv[1:]
    try:
        strip_prefix = common.validate_strip_prefix(FLAGS.strip_prefix,
                                                    bzl_files)
    except common.InputError as err:
        print(err.message)
        sys.exit(1)

    rulesets = []
    load_sym_extractor = load_extractor.LoadExtractor()
    for bzl_file in bzl_files:
        load_symbols = []
        try:
            load_symbols = load_sym_extractor.extract(bzl_file)
        except load_extractor.LoadExtractorError as e:
            print("ERROR: Error extracting loaded symbols from %s: %s" %
                  (bzl_file, str(e)))
            sys.exit(2)

        # TODO(dzc): Make MacroDocExtractor and RuleDocExtractor stateless.
        macro_doc_extractor = macro_extractor.MacroDocExtractor()
        rule_doc_extractor = rule_extractor.RuleDocExtractor()
        macro_doc_extractor.parse_bzl(bzl_file)
        rule_doc_extractor.parse_bzl(bzl_file, load_symbols)
        merged_language = merge_languages(macro_doc_extractor.proto(),
                                          rule_doc_extractor.proto())
        rulesets.append(
            rule.RuleSet(bzl_file, merged_language, macro_doc_extractor.title,
                         macro_doc_extractor.description, strip_prefix,
                         FLAGS.format))

    writer_options = WriterOptions(FLAGS.output_dir, FLAGS.output_file,
                                   FLAGS.zip, FLAGS.overview,
                                   FLAGS.overview_filename, FLAGS.link_ext)
    if FLAGS.format == "markdown":
        markdown_writer = MarkdownWriter(writer_options)
        markdown_writer.write(rulesets)
    elif FLAGS.format == "html":
        html_writer = HtmlWriter(writer_options)
        html_writer.write(rulesets)
    else:
        sys.stderr.write(
            'Invalid output format: %s. Possible values are markdown and html'
            % FLAGS.format)
Beispiel #2
0
def main(argv):
    if FLAGS.output_dir and FLAGS.output_file:
        sys.stderr.write(
            'Only one of --output_dir or --output_file can be set.')
        sys.exit(1)

    if not FLAGS.output_dir:
        FLAGS.output_dir = DEFAULT_OUTPUT_DIR
    if not FLAGS.output_file:
        FLAGS.output_file = DEFAULT_OUTPUT_FILE

    bzl_files = argv[1:]
    try:
        strip_prefix = common.validate_strip_prefix(FLAGS.strip_prefix,
                                                    bzl_files)
    except common.InputError as err:
        print(err.message)
        sys.exit(1)

    runfiles = runfiles_lib.Create()
    if not runfiles:
        # TODO(laszlocsomor): fix https://github.com/bazelbuild/bazel/issues/6212
        # and remove this if-block once Bazel is released with that bugfix.
        if (not os.environ.get("RUNFILES_DIR")
                and not os.environ.get("RUNFILES_MANIFEST_FILE")):
            argv0 = sys.argv[0]
            pos = argv0.rfind('%s%s%s' % (os.sep, WORKSPACE_DIR, os.sep))
            if pos > -1:
                dirpath = argv0[0:pos]
                if not os.path.isdir(dirpath):
                    print("ERROR: Cannot access runfiles directory (%s)" %
                          dirpath)
                    sys.exit(1)
                runfiles = runfiles_lib.CreateDirectoryBased(dirpath)

    if not runfiles:
        print("ERROR: Cannot load runfiles")
        sys.exit(1)

    rulesets = []
    load_sym_extractor = load_extractor.LoadExtractor()
    for bzl_file in bzl_files:
        load_symbols = []
        try:
            load_symbols = load_sym_extractor.extract(bzl_file)
        except load_extractor.LoadExtractorError as e:
            print("ERROR: Error extracting loaded symbols from %s: %s" %
                  (bzl_file, str(e)))
            sys.exit(2)

        # TODO(dzc): Make MacroDocExtractor and RuleDocExtractor stateless.
        macro_doc_extractor = macro_extractor.MacroDocExtractor()
        rule_doc_extractor = rule_extractor.RuleDocExtractor()
        macro_doc_extractor.parse_bzl(bzl_file)
        rule_doc_extractor.parse_bzl(bzl_file, load_symbols)
        merged_language = merge_languages(macro_doc_extractor.proto(),
                                          rule_doc_extractor.proto())
        rulesets.append(
            rule.RuleSet(bzl_file, merged_language, macro_doc_extractor.title,
                         macro_doc_extractor.description, strip_prefix,
                         FLAGS.format))
    writer_options = WriterOptions(FLAGS.output_dir, FLAGS.output_file,
                                   FLAGS.zip, FLAGS.overview,
                                   FLAGS.overview_filename, FLAGS.link_ext,
                                   FLAGS.site_root)
    if FLAGS.format == "markdown":
        markdown_writer = MarkdownWriter(writer_options, runfiles)
        markdown_writer.write(rulesets)
    elif FLAGS.format == "html":
        html_writer = HtmlWriter(writer_options, runfiles)
        html_writer.write(rulesets)
    else:
        sys.stderr.write(
            'Invalid output format: %s. Possible values are markdown and html'
            % FLAGS.format)
Beispiel #3
0
def main(argv):
    parser = OptionParser()
    parser.add_option(
        '--output_dir',
        default='',
        help=
        'The directory to write the output generated documentation to if --zip=false'
    )
    parser.add_option(
        '--output_file',
        default='',
        help='The output zip archive file to write if --zip=true.')
    parser.add_option(
        '--format',
        default='markdown',
        help='The output format. Possible values are markdown and html')
    parser.add_option(
        '--zip',
        action='store_true',
        default=True,
        help='Whether to generate a ZIP arhive containing the output files. If '
        '--zip is true, then skydoc will generate a zip file, skydoc.zip by '
        'default or as specified by --output_file. If --zip is false, then '
        'skydoc will generate documentation, either in Markdown or HTML as '
        'specifed by --format, in the current directory or --output_dir if set.'
    )
    parser.add_option(
        '--strip_prefix',
        default='',
        help='The directory prefix to strip from all generated docs, which are '
        'generated in subdirectories that match the package structure of the '
        'input .bzl files. The prefix to strip must be common to all .bzl files; '
        'otherwise, skydoc will raise an error.')
    parser.add_option('--overview',
                      action='store_true',
                      help='Whether to generate an overview page')
    parser.add_option('--overview_filename',
                      default='index',
                      help='The file name to use for the overview page.')
    parser.add_option(
        '--link_ext',
        default='html',
        help='The file extension used for links in the generated documentation'
    )
    parser.add_option(
        '--site_root',
        default='',
        help=
        'The site root to be prepended to all URLs in the generated documentation'
    )
    (options, args) = parser.parse_args(argv)

    if options.output_dir and options.output_file:
        sys.stderr.write(
            'Only one of --output_dir or --output_file can be set.')
        sys.exit(1)

    if not options.output_dir:
        options.output_dir = DEFAULT_OUTPUT_DIR
    if not options.output_file:
        options.output_file = DEFAULT_OUTPUT_FILE

    bzl_files = args[1:]

    try:
        strip_prefix = common.validate_strip_prefix(options.strip_prefix,
                                                    bzl_files)
    except common.InputError as err:
        print(err.message)
        sys.exit(1)

    runfiles = runfiles_lib.Create()
    if not runfiles:
        # TODO(laszlocsomor): fix https://github.com/bazelbuild/bazel/issues/6212
        # and remove this if-block once Bazel is released with that bugfix.
        if (not os.environ.get("RUNFILES_DIR")
                and not os.environ.get("RUNFILES_MANIFEST_FILE")):
            argv0 = sys.argv[0]
            pos = argv0.rfind('%s%s%s' % (os.sep, WORKSPACE_DIR, os.sep))
            if pos > -1:
                dirpath = argv0[0:pos]
                if not os.path.isdir(dirpath):
                    print("ERROR: Cannot access runfiles directory (%s)" %
                          dirpath)
                    sys.exit(1)
                runfiles = runfiles_lib.CreateDirectoryBased(dirpath)

    if not runfiles:
        print("ERROR: Cannot load runfiles")
        sys.exit(1)

    rulesets = []
    load_sym_extractor = load_extractor.LoadExtractor()
    for bzl_file in bzl_files:
        load_symbols = []
        try:
            load_symbols = load_sym_extractor.extract(bzl_file)
        except load_extractor.LoadExtractorError as e:
            print("ERROR: Error extracting loaded symbols from %s: %s" %
                  (bzl_file, str(e)))
            sys.exit(2)

        # TODO(dzc): Make MacroDocExtractor and RuleDocExtractor stateless.
        macro_doc_extractor = macro_extractor.MacroDocExtractor()
        rule_doc_extractor = rule_extractor.RuleDocExtractor()
        macro_doc_extractor.parse_bzl(bzl_file)
        rule_doc_extractor.parse_bzl(bzl_file, load_symbols)
        merged_language = merge_languages(macro_doc_extractor.proto(),
                                          rule_doc_extractor.proto())
        rulesets.append(
            rule.RuleSet(bzl_file, merged_language, macro_doc_extractor.title,
                         macro_doc_extractor.description, strip_prefix,
                         options.format))
    writer_options = WriterOptions(options.output_dir, options.output_file,
                                   options.zip, options.overview,
                                   options.overview_filename, options.link_ext,
                                   options.site_root)
    if options.format == "markdown":
        markdown_writer = MarkdownWriter(writer_options, runfiles)
        markdown_writer.write(rulesets)
    elif options.format == "html":
        html_writer = HtmlWriter(writer_options, runfiles)
        html_writer.write(rulesets)
    else:
        sys.stderr.write(
            'Invalid output format: %s. Possible values are markdown and html'
            % options.format)