import sys
import os
import subprocess
import platform

JAVA10_EXPORTS = [
    '--add-exports=jdk.compiler/com.sun.tools.javac.api=ALL-UNNAMED',
    '--add-exports=jdk.compiler/com.sun.tools.javac.util=ALL-UNNAMED',
    '--add-exports=jdk.compiler/com.sun.tools.javac.tree=ALL-UNNAMED',
    '--add-exports=jdk.compiler/com.sun.tools.javac.main=ALL-UNNAMED',
    '--add-exports=jdk.compiler/com.sun.tools.javac.code=ALL-UNNAMED',
    '--add-exports=jdk.compiler/com.sun.tools.javac.processing=ALL-UNNAMED',
    '--add-exports=jdk.compiler/com.sun.tools.javac.parser=ALL-UNNAMED',
    '--add-exports=jdk.compiler/com.sun.tools.javac.comp=ALL-UNNAMED',
]


def get_classpath(cmd):
    for i, part in enumerate(cmd):
        if part == '-classpath':
            i += 1
            if i < len(cmd):
                return cmd[i]
            else:
                return None
    return None


def parse_args(argv):
    parsed = []
    for i in range(len(argv)):
        if not argv[i].startswith('-'):
            parsed.append(argv[i])
            if len(parsed) >= 3:
                break
    return parsed + [argv[i + 1 :]]


def add_error_prone_to_cmd(error_prone_tool, cmd):
    if not error_prone_tool:
        return cmd
    error_prone_flags = []
    cmd_wo_epf = []
    for f in cmd:  # divide list to error_prone flags and others
        if f.startswith('-Xep'):
            error_prone_flags.append(f)
        elif '-proc:none' != f:  # Remove disabling AP when error_prone present
            cmd_wo_epf.append(f)
    cmd = cmd_wo_epf
    if '-processor' in cmd:
        classpath = get_classpath(cmd)
        if classpath:
            error_prone_tool = error_prone_tool + os.pathsep + classpath
    return (
        cmd
        + JAVA10_EXPORTS
        + ['-processorpath', error_prone_tool, '-XDcompilePolicy=byfile']
        + [' '.join(['-Xplugin:ErrorProne'] + error_prone_flags)]
    )


# NOTE: legacy, only for "devtools/ya/jbuild"
def just_do_it(argv):
    java, javac, error_prone_tool, javac_cmd = parse_args(argv)
    error_prone_flags = []
    for f in javac_cmd:
        if f.startswith('-Xep'):
            error_prone_flags.append(f)
    for f in error_prone_flags:
        if f in javac_cmd:
            javac_cmd.remove(f)
    if '-processor' in javac_cmd:
        classpath = get_classpath(javac_cmd)
        if classpath:
            error_prone_tool = error_prone_tool + os.pathsep + classpath
    cmd = (
        [javac]
        + JAVA10_EXPORTS
        + ['-processorpath', error_prone_tool, '-XDcompilePolicy=byfile']
        + [' '.join(['-Xplugin:ErrorProne'] + error_prone_flags)]
        + javac_cmd
    )
    if platform.system() == 'Windows':
        sys.exit(subprocess.Popen(cmd).wait())
    else:
        os.execv(cmd[0], cmd)


if __name__ == '__main__':
    just_do_it(sys.argv[1:])