hanchenye-llvm-project/lldb/gtest/do-gtest.py

#!/usr/bin/env python

from __future__ import print_function

import os
import re
import select
import subprocess
import sys

def find_makefile_dirs():
    makefile_dirs = []
    for root, dirs, files in os.walk("."):
        for file in files:
            if file == "Makefile":
                makefile_dirs.append(root)
    return makefile_dirs

_TESTDIR_RELATIVE_REGEX = re.compile(r"^([^/:]+:\d+:)")
_COMBINER_TERMINATION_REGEX = re.compile(r"^.+FAILED.+$")

def filter_run_line(sub_expr, line):
    return _TESTDIR_RELATIVE_REGEX.subn(sub_expr, line)

def line_combine_printer(file, previous_data, new_line_subn_result):
    (accumulated_line, combine_lines_left) = previous_data
    (incoming_line, sub_match_count) = new_line_subn_result

    if sub_match_count > 0:
        # New line was a match.  Don't print yet, start an accumulation.
        if len(accumulated_line) > 0:
            # Flush anything previously there.
            print(accumulated_line, file=file)
        return (incoming_line + ": ", 3)
    else:
        # If we're combining and incoming is a "[  FAILED ]" line, we've gone too far on a combine.
        if (len(accumulated_line) > 0) and _COMBINER_TERMINATION_REGEX.match(incoming_line):
            # Stop the combine.
            print(accumulated_line, file=file)
            print(incoming_line, file=file)
            return ("", 0)

        if len(accumulated_line) > 0:
            new_line = accumulated_line + ", " + incoming_line
        else:
            new_line = incoming_line

        remaining_count = combine_lines_left - 1
        if remaining_count > 0:
            return (new_line, remaining_count)
        else:
            # Time to write it out.
            if len(new_line) > 0:
                print(new_line, file=file)
            return ("", 0)

def call_make(makefile_dir, extra_args=None):
    command = ["make", "-C", makefile_dir]
    if extra_args:
        command.extend(extra_args)

    # Replace the matched no-directory filename with one where the makefile directory is prepended.
    sub_expr = makefile_dir + r"/\1";

    proc = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)

    stdout_data = ("", 0)
    stderr_data = ("", 0)

    while True:
        reads = [proc.stdout.fileno(), proc.stderr.fileno()]
        select_result = select.select(reads, [], [])

        for fd in select_result[0]:
            if fd == proc.stdout.fileno():
                line = proc.stdout.readline()
                stdout_data = line_combine_printer(sys.stdout, stdout_data, filter_run_line(sub_expr, line.rstrip()))
            elif fd == proc.stderr.fileno():
                line = proc.stderr.readline()
                stderr_data = line_combine_printer(sys.stderr, stderr_data, filter_run_line(sub_expr, line.rstrip()))

        proc_retval = proc.poll()
        if proc_retval != None:
            # Process stopped.  Drain output before finishing up.

            # Drain stdout.
            while True:
                line = proc.stdout.readline().rstrip()
                if line and len(line) > 0:
                    stdout_data = line_combine_printer(sys.stdout, stdout_data, filter_run_line(sub_expr, line))
                else:
                    break

            # Drain stderr.
            while True:
                line = proc.stderr.readline().rstrip()
                if line and len(line) > 0:
                    stderr_data = line_combine_printer(sys.stderr, stderr_data, filter_run_line(sub_expr, line))
                else:
                    break

            return proc_retval


global_retval = 0
extra_args = None

if len(sys.argv) > 1:
    if sys.argv[1] == 'clean':
        extra_args = ['clean']

for makefile_dir in find_makefile_dirs():
    print("found makefile dir: {}".format(makefile_dir))
    retval = call_make(makefile_dir, extra_args)
    if retval != 0:
        global_retval = retval
        
sys.exit(global_retval)