import os
import sys

from .lsp import LspServer, command, get_data_home

FILE_URI = "file://"

def locate_product_directory (jdt_ls_path):
    return os.path.join(jdt_ls_path, "org.eclipse.jdt.ls.product/target/repository")

def locate_launcher_jar (jdt_ls_path):
    plugins_directory = os.path.join(jdt_ls_path, "plugins")
    return os.path.join(plugins_directory, [jar for jar in os.listdir(plugins_directory) if jar.startswith("org.eclipse.equinox.launcher_")][0])

def locate_launcher_configuration (jdt_ls_path):
    return os.path.join(jdt_ls_path, "config_linux")

def download_jdt_ls (destination):
    raise Exception("download_jdt_ls not implemented")

# TODO
def locate_lombok ():
    return os.path.join(os.environ['HOME'], ".m2/repository/org/projectlombok/lombok/1.18.16/lombok-1.18.16.jar")

class Java(LspServer):
    def __init__ (self, arguments):
        self.path = arguments[0] if arguments else None

        if self.path is None or not os.path.exists(self.path):
            self.path = download_jdt_ls(os.path.join(get_data_home(), "java", "jdt.ls"))
        else:
            self.path = locate_product_directory(self.path)
        
        command = ["java"]
        if "--enable-lombok" in arguments:
            arguments.remove("--enable-lombok")
            command.append(f"-javaagent:{locate_lombok()}")
        
        data_directory = None
        if "--data" in arguments:
            data_directory = arguments.pop(arguments.index("--data") + 1)
            arguments.remove("--data")
        else:
            data_directory = os.path.join(get_data_home(), "java", "data")

        command = command + [
            "-agentlib:jdwp=transport=dt_socket,server=y,suspend=n,address=1044",
            "-Declipse.application=org.eclipse.jdt.ls.core.id1",
            "-Dosgi.bundles.defaultStartLevel=4",
            "-Declipse.product=org.eclipse.jdt.ls.core.product",
            "-noverify",
            "-Xmx1G",
            "-jar", locate_launcher_jar(self.path),
            "-configuration", locate_launcher_configuration(self.path),
            "-data", data_directory,
            "--add-modules=ALL-SYSTEM",
            "--add-opens", "java.base/java.util=ALL-UNNAMED",
            "--add-opens", "java.base/java.lang=ALL-UNNAMED"
        ] + arguments[1:]
        
        super().__init__(command)

    @command("java.apply.workspaceEdit")
    def workspaceEdit (self, arguments):
        for argument in arguments:
            self.send_to_client({
                "id": -1,
                "method": "workspace/applyEdit",
                "params": {
                    "edit": {
                        "changes": argument['changes']
                    }
                }
            })