status-repos.py

#!/usr/bin/env python

import argparse
import os
import sys

import build_tools


def parse_args(args):
    parser = argparse.ArgumentParser(
        description='Gets various info about each source in the toolchain.',
        parents=[build_tools.get_parent_parser()])

    parser.add_argument(
        '-k',
        '--keep-going',
        action='store_true',
        help='Keep going even if errors are encountered.')
    parser.add_argument(
        '-i',
        '--ignored',
        action='store_true',
        help='Show ignored files.')
    parser.add_argument(
        '--long',
        action='store_true',
        help='Print the long form status.')
    parser.add_argument(
        '--no-fetch',
        action='store_true',
        help="don't fetch before checking status")
    parser.add_argument(
        '-l',
        '--list',
        action='store_true',
        help=(
            'Just print the name of every source. '
            'Ignores -k, -i, --no-fetch, and --long.'))
    parser.add_argument(
        '-b',
        '--branch-only',
        action='store_true',
        help=(
            'Just print the checked-out branch for each source. '
            'Ignores -k, -i, --no-fetch, and --long.'))
    parser.add_argument(
        '--export',
        action='store_true',
        help=(
            'export a json blob of branches'))
    parser.add_argument(
        '--export-for-windows',
        action='store_true',
        help=(
            'export a json blob of branches for windows fancy builder'))
    return parser.parse_args(args)


def status_repos(args):
    """
    Prints out the status of each repo in the toolchain.

    TODO(ted): Not sure how I feel about the fact that the behavior of this is
    so tied to Source type, but I'd rather have it do something useful than
    keeping everything 'pure'.

    Actually, I think I hate everything about this.
    """
    toolchain = build_tools.get_toolchain(args)
    if args.export or args.export_for_windows:
        if args.export_for_windows:
            sep = '\\\"'
        else:
            sep = '\"'
        jsonblob = '['
        for source in toolchain:
            if jsonblob.endswith('}'):
                jsonblob += ','
            jsonblob += '{' + sep + 'source' + sep + ':' + sep + source.name
            jsonblob += sep + ',' + sep + 'version' + sep + ':'
            jsonblob += sep + source.version + sep + '}'
        jsonblob += ']'
        print(jsonblob)
    else:
        if args.list:
            def status(source):
                print(source.name)
        elif args.branch_only:
            def status(source):
                if isinstance(source, build_tools.source.Repo):
                    print(source.name + ': ' + source.current_branch())
                else:
                    print(source.name + ': ' + source.version)

        else:
            def status(source):
                if source.use_artifacts:
                    source.artifact_status()
                elif isinstance(source, build_tools.source.Repo):
                    if not args.no_fetch:
                        rv = source.cmd(['git', 'fetch', '--all'], quiet=True)
                        if rv != 0:
                            print('could not fetch ' + source.name)
                            if not args.keep_going:
                                return rv
                    cmd = ['git', 'status']
                    if not args.long:
                        cmd.append('-sb')
                    if args.ignored:
                        cmd.append('--ignored')
                    rv = source.cmd(cmd)
                    if rv != 0:
                        print('could not get status of ' + source.name)
                        if not args.keep_going:
                            return rv

        return toolchain.for_each_source(status, args.keep_going)


if '__main__' == __name__:
    parsed_args = parse_args(sys.argv[1:])
    rv = status_repos(parsed_args)
    sys.exit(rv)