import os
import glob
import sys
import subprocess
import time

SUCCEEDED = "\033[32msucceeded\033[0m"
FAILED = "\033[31mfailed\033[0m"
SKIPPED = "\033[33mskipped\033[0m"

success_count = 0
fail_count = 0
skip_count = 0
exit_status = 0

total_time = time.monotonic()

build_format = '| {:29} | {:30} | {:18} | {:7} | {:6} | {:6} |'
build_separator = '-' * 106

# If examples are not specified in arguments, build all
all_examples = []

for entry in os.scandir("examples/device"):
    if entry.is_dir():
        all_examples.append("device/" + entry.name)

for entry in os.scandir("examples/host"):
    if entry.is_dir():
        all_examples.append("host/" + entry.name)

if len(sys.argv) > 1:
    input_examples = list(set(all_examples).intersection(sys.argv))
    if len(input_examples) > 0:
        all_examples = input_examples

all_examples.sort()

# If boards are not specified in arguments, build all
all_boards = []

for entry in os.scandir("hw/bsp"):
    if entry.is_dir():
        all_boards.append(entry.name)

if len(sys.argv) > 1:
    input_boards = list(set(all_boards).intersection(sys.argv))
    if len(input_boards) > 0:
        all_boards = input_boards

all_boards.sort()

def build_example(example, board):
    subprocess.run("make -C examples/{} BOARD={} clean".format(example, board), shell=True,
                   stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
    return subprocess.run("make -j -C examples/{} BOARD={} all".format(example, board), shell=True,
                          stdout=subprocess.PIPE, stderr=subprocess.STDOUT)

def build_size(example, board):
    #elf_file = 'examples/device/{}/_build/build-{}/{}-firmware.elf'.format(example, board, board)
    elf_file = 'examples/{}/_build/build-{}/*.elf'.format(example, board)
    size_output = subprocess.run('size {}'.format(elf_file), shell=True, stdout=subprocess.PIPE).stdout.decode("utf-8")
    size_list = size_output.split('\n')[1].split('\t')
    flash_size = int(size_list[0])
    sram_size = int(size_list[1]) + int(size_list[2])
    return (flash_size, sram_size)

def skip_example(example, board):
    ex_dir = 'examples/' + example
    board_mk = 'hw/bsp/{}/board.mk'.format(board)

    with open(board_mk) as mk:
        mk_contents = mk.read()

        # Skip all ESP32-S2 board for CI
        if 'CROSS_COMPILE = xtensa-esp32s2-elf-' in mk_contents:
            return 1

        # Skip all OPT_MCU_NONE these are WIP port
        if '-DCFG_TUSB_MCU=OPT_MCU_NONE' in mk_contents:
            return 1

        # Skip if CFG_TUSB_MCU in board.mk to match skip file
        for skip_file in glob.iglob(ex_dir + '/.skip.MCU_*'):
            mcu_cflag = '-DCFG_TUSB_MCU=OPT_' + os.path.basename(skip_file).split('.')[2]
            if mcu_cflag in mk_contents:
                return 1

        # Build only list, if exists only these MCU are built
        only_list = list(glob.iglob(ex_dir + '/.only.MCU_*'))
        if len(only_list) > 0:
            for only_file in only_list:
                mcu_cflag = '-DCFG_TUSB_MCU=OPT_' + os.path.basename(only_file).split('.')[2]
                if mcu_cflag in mk_contents:
                    return 0
            return 1

    return 0

print(build_separator)
print(build_format.format('Example', 'Board', '\033[39mResult\033[0m', 'Time', 'Flash', 'SRAM'))

for example in all_examples:
    print(build_separator)
    for board in all_boards:
        start_time = time.monotonic()

        flash_size = "-"
        sram_size = "-"

        # Check if board is skipped
        if skip_example(example, board):
            success = SKIPPED
            skip_count += 1
            print(build_format.format(example, board, success, '-', flash_size, sram_size))
        else:
            build_result = build_example(example, board)

            if build_result.returncode == 0:
                success = SUCCEEDED
                success_count += 1
                (flash_size, sram_size) = build_size(example, board)
            else:
                exit_status = build_result.returncode
                success = FAILED
                fail_count += 1

            build_duration = time.monotonic() - start_time
            print(build_format.format(example, board, success, "{:.2f}s".format(build_duration), flash_size, sram_size))

            if build_result.returncode != 0:
                print(build_result.stdout.decode("utf-8"))



total_time = time.monotonic() - total_time
print(build_separator)
print("Build Summary: {} {}, {} {}, {} {} and took {:.2f}s".format(success_count, SUCCEEDED, fail_count, FAILED, skip_count, SKIPPED, total_time))
print(build_separator)

sys.exit(exit_status)

@@ -966,34 +966,296 @@
 f entry.is_dir()
-:%0A
+ and entry.name != %22esp32s2%22:%0A        if os.path.isdir(entry.path + %22/boards%22):%0A            # family directory%0A            for subentry in os.scandir(entry.path + %22/boards%22):%0A                if subentry.is_dir(): all_boards.append(subentry.name)%0A        else:%0A    
         all_boar
@@ -2394,16 +2394,17 @@
 example%0A
+%0A
     boar
@@ -2449,22 +2449,33 @@
 rd)%0A
-%0A
     
-with open
+if not os.path.exists
 (boa
@@ -2484,14 +2484,8 @@
 _mk)
- as mk
 :%0A  
@@ -2494,159 +2494,154 @@
     
-mk_contents = mk.read()%0A%0A        # Skip all ESP32-S2 board for CI%0A        if 'CROSS_COMPILE = xtensa-esp32s2-elf-' in mk_contents:%0A            return 1
+board_mk = list(glob.iglob('hw/bsp/*/boards/%7B%7D/../../family.mk'.format(board)))%5B0%5D%0A%0A    with open(board_mk) as mk:%0A        mk_contents = mk.read()
 %0A%0A  

update build_all.py to support both single board and family group