#!/usr/bin/env python3
#
# Copyright 2018 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

"""A script to parse nasm's file lists out of Makefile.in."""

import os
import sys

def ParseFileLists(path):
    ret = {}
    with open(path) as f:
        in_file_list = False
        split_line = ""
        for line in f:
            line = line.rstrip()
            if not in_file_list:
                if "-- Begin File Lists --" in line:
                    in_file_list = True
                continue
            if "-- End File Lists --" in line:
                if split_line:
                    raise ValueError("End comment was preceded by split line")
                break
            line = split_line + line
            split_line = ""
            if line.endswith('\\'):
                split_line = line[:-1]
                continue
            line = line.strip()
            if not line or line.startswith('#'):
                continue
            name, value = line.split('=')
            name = name.strip()
            value = value.replace("$(O)", "c")
            files = value.split()
            files.sort()
            files = [file for file in files if file.endswith(('.c', '.h'))]
            ret[name] = files
    return ret

def PrintFileList(out, name, files):
    if len(files) == 0:
        print("%s = []" % (name, ), file=out)
    elif len(files) == 1:
        print("%s = [ \"%s\" ]" % (name, files[0]), file=out)
    else:
        print("%s = [" % (name, ), file=out)
        for f in files:
            print("  \"%s\"," % (f, ), file=out)
        print("]", file=out)


def main():
    file_lists = ParseFileLists("Makefile.in")
    with open("nasm_sources.gni", "w") as out:
        print(
            """# Copyright (c) 2018 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

# This file is created by generate_nasm_sources.py. Do not edit manually.
""",
            file=out)
        # Results in duplicated symbols in nasm.c
        file_lists['LIBOBJ_NW'].remove('nasmlib/errfile.c')

        # We don't use the disassembler and these files have compiler warnings.
        file_lists['LIBOBJ_NW'].remove('disasm/disasm.c')
        file_lists['LIBOBJ_NW'].remove('disasm/sync.c')

        PrintFileList(out, "nasmlib_sources",
                      file_lists['LIBOBJ_NW'] + file_lists['WARNOBJ'] +
                      file_lists['WARNFILES'])
        PrintFileList(out, "nasm_sources", file_lists['NASM'])

if __name__ == "__main__":
    main()
