import logging
import shutil
import tempfile
from pathlib import Path
from os import pathsep
from typing import List

from jpype import JImplements, JOverride

logger = logging.getLogger(__name__)


COMPILER_OPTIONS = ["-target", "17", "-source", "17"]


def _to_jar_(jar_path: Path, root: Path):
    from java.io import ByteArrayOutputStream
    from java.util.jar import JarEntry, JarOutputStream

    out = ByteArrayOutputStream()
    with JarOutputStream(out) as jar:
        for p in root.glob("**/*.class"):
            p = p.resolve()
            jar.putNextEntry(JarEntry(str(p.relative_to(root).as_posix())))
            jar.write(p.read_bytes())
            jar.closeEntry()
    jar_path.write_bytes(out.toByteArray())


@JImplements("javax.tools.DiagnosticListener", deferred=True)
class _CompilerDiagnosticListener:

    def __init__(self):
        from javax.tools import Diagnostic
        self.errors: List[Diagnostic] = []

    @JOverride
    def report(self, diagnostic):
        from javax.tools import Diagnostic
        diagnostic: Diagnostic = diagnostic

        kind = diagnostic.getKind()

        if kind == Diagnostic.Kind.ERROR:
            self.errors.append(diagnostic)
        elif kind == Diagnostic.Kind.WARNING:
            logger.info(str(kind))


def java_compile(src_path: Path, jar_path: Path):
    """
    Compiles the provided Java source

    :param src_path: The path to the java file or the root directory of the java source files
    :param jar_path: The path to write the output jar to
    :raises ValueError: If an error occurs when compiling the Java source
    """

    from java.lang import System
    from java.io import Writer
    from java.nio.file import Path as JPath
    from javax.tools import StandardLocation, ToolProvider

    with tempfile.TemporaryDirectory() as out:
        outdir = Path(out).resolve()
        compiler = ToolProvider.getSystemJavaCompiler()
        fman = compiler.getStandardFileManager(None, None, None)
        cp = [JPath @ (Path(p)) for p in System.getProperty("java.class.path").split(pathsep)]
        fman.setLocationFromPaths(StandardLocation.CLASS_PATH, cp)
        if src_path.is_dir():
            fman.setLocationFromPaths(StandardLocation.SOURCE_PATH, [JPath @ (src_path.resolve())])
        fman.setLocationFromPaths(StandardLocation.CLASS_OUTPUT, [JPath @ (outdir)])
        sources = None
        if src_path.is_file():
            sources = fman.getJavaFileObjectsFromPaths([JPath @ (src_path)])
        else:
            glob = src_path.glob("**/*.java")
            sources = fman.getJavaFileObjectsFromPaths([JPath @ (p) for p in glob])

        diagnostics = _CompilerDiagnosticListener()
        task = compiler.getTask(Writer.nullWriter(), fman, diagnostics, COMPILER_OPTIONS, None, sources)

        if not task.call():
            msg = "\n".join([str(error) for error in diagnostics.errors])
            raise ValueError(msg)

        if jar_path.suffix == '.jar':
            jar_path.parent.mkdir(exist_ok=True, parents=True)
            _to_jar_(jar_path, outdir)
        else:
            shutil.copytree(outdir, jar_path, dirs_exist_ok=True)