#!/usr/bin/env python3

import os
import re
import sys

RE_MULTILINE_COMMENT_START = re.compile(r"^( *)/\*\*?$")
RE_MULTILINE_COMMENT_END = re.compile(r"^ *\*/$")
RE_TRAILING_COMMENT = re.compile(r"^( *)([^ ].*;) +// (.*)$")

spaces = ""
in_comment = False
for fname in sys.argv[1:]:
    result = []
    with open(fname, "r") as file:
        for line in file:
            line = line.rstrip("\n")

            if m := RE_MULTILINE_COMMENT_START.match(line):
                assert not in_comment
                spaces = m[1]
                in_comment = True
            elif RE_MULTILINE_COMMENT_END.match(line):
                assert in_comment
                in_comment = False
            elif m := RE_TRAILING_COMMENT.match(line):
                assert not in_comment
                spaces, code, comment = m.groups()
                result.append(f"{spaces}// {comment}\n")
                result.append(f"{spaces}{code}\n")
            else:
                if in_comment:
                    line = line.lstrip()
                    result.append(f"{spaces}//{" " + line if line else ""}\n")
                else:
                    result.append(f"{line}\n")
    with open(fname + ".new", "w") as file:
        file.writelines(result)
    os.replace(fname + ".new", fname)