#include <assert.h>
#include <dirent.h>
#include <dlfcn.h>
#include <errno.h>
#include <libgen.h>
#include <spawn.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <unistd.h>

#include "log.h"
#include "utils.h"

#import "ios_uikit_bridge.h"
#import "JavaLauncher.h"
#import "LauncherPreferences.h"

#define fm NSFileManager.defaultManager

static char java_libs_path[2048];
static char args_path[2048];

extern char **environ;

static int margc = -1;
static char* margv[1000];

const char *javaHome;
const char *allocmem;
const char *multidir;

char *homeDir;

NSString *renderer;
NSString *allocmem_pre;
NSString *multidir_pre;

void init_loadCustomEnv() {
    /* Define default env */

    // I accidentally patched a bit wrong, the value should be a path containing libawt_xawt.dylib, but here is libawt.dylib path (no need to exist)
    if (getenv("POJAV_DETECTEDJB")) {
        setenv("JAVA_AWT_PATH", [NSString stringWithFormat:@"%s/Frameworks/libawt.dylib", getenv("BUNDLE_PATH")].UTF8String, 1);
    }

    // Disable overloaded functions hack for Minecraft 1.17+
    setenv("LIBGL_NOINTOVLHACK", "1", 1);

    // Override OpenGL version to 4.1 for Zink
    setenv("MESA_GL_VERSION_OVERRIDE", "4.1", 1);

    // Runs JVM in a separate thread
    setenv("HACK_IGNORE_START_ON_FIRST_THREAD", "1", 1);

    /* Load custom env */
    FILE *envFile = fopen([NSString stringWithFormat:@"%s/custom_env.txt", getenv("POJAV_HOME")].UTF8String, "r");

    regLog("[Pre-init] Reading custom environment variables (custom_env.txt), opened=%d\n", envFile != NULL);

    if (envFile) {
        char *line = NULL;
        size_t len = 0;
        ssize_t read;
        while ((read = getline(&line, &len, envFile)) != -1) {
            if (line[0] == '#' || line[0] == '\n') continue;
            if (line[read-1] == '\n') {
                line[read-1] = '\0';
            }
            if (strchr(line, '=') != NULL) {
                regLog("[Pre-init] Added custom env: %s", line);
                setenv(strtok(line, "="), strtok(NULL, "="), 1);
            } else {
                regLog("[Pre-init] Warning: skipped empty value custom env: %s", line);
            }
        }
        fclose(envFile);
    }
}

void init_loadCustomJvmFlags() {
    NSString *jvmargs = getPreference(@"java_args");
    BOOL isFirstArg = YES;
    for (NSString *jvmarg in [jvmargs componentsSeparatedByString:@" -"]) {
        if ([jvmarg length] == 0) continue;
        //margv[margc] = (char *) [jvmarg UTF8String];

        ++margc;
        if (isFirstArg) {
            isFirstArg = NO;
            margv[margc] = (char *) [jvmarg UTF8String];
        } else {
            margv[margc] = (char *) [[@"-" stringByAppendingString:jvmarg] UTF8String];
        }

        NSLog(@"[JavaLauncher] Added custom JVM flag: %s", margv[margc]);
    }
}

NSString* environmentFailsafes(int minVersion) {
    NSString *jvmPath = getenv("POJAV_DETECTEDJB") ? @"/usr/lib/jvm" : [NSString stringWithFormat:@"%s/jvm", getenv("BUNDLE_PATH")];
    NSString *javaHome = nil;

    NSString *jre8Path = [NSString stringWithFormat:@"%@/java-8-openjdk", jvmPath];
    NSString *jre16Path = [NSString stringWithFormat:@"%@/java-16-openjdk", jvmPath];
    NSString *jre17Path = [NSString stringWithFormat:@"%@/java-17-openjdk", jvmPath];

    if ([fm fileExistsAtPath:jre8Path] && minVersion <= 8) {
        javaHome = jre8Path;
    } else if ([fm fileExistsAtPath:jre16Path] && minVersion <= 16) {
        javaHome = jre16Path;
    } else if ([fm fileExistsAtPath:jre17Path] && minVersion <= 17) {
        javaHome = jre17Path;
    }

    if (javaHome == nil) {
        showDialog(currentVC(), NSLocalizedString(@"Error", nil), [NSString stringWithFormat:@"Minecraft %@ requires Java %d in order to run. Please install it first.", getPreference(@"selected_version"), minVersion]);
    }

    return javaHome;
}

int launchJVM(NSString *username, id launchTarget, int width, int height, int minVersion) {
    sprintf((char*) java_libs_path, "%s/libs", getenv("BUNDLE_PATH"));

    init_loadCustomEnv();

    regLog("[JavaLauncher] Beginning JVM launch\n");

    NSString *javaHome_pre = getPreference(@"java_home");
    if (![javaHome_pre hasPrefix:@"/"]) {
        javaHome_pre = [NSString stringWithFormat:@"%s/jvm/%@", getenv("POJAV_DETECTEDJB") ? "/usr/lib" : getenv("BUNDLE_PATH"), javaHome_pre];
    }

    // We handle unset JAVA_HOME right there
    if (getSelectedJavaVersion() < minVersion) {
        NSLog(@"[JavaLauncher] Attempting to change to Java %d (actual might be higher)", minVersion);
        javaHome_pre = environmentFailsafes(minVersion);
        NSLog(@"[JavaLauncher] JAVA_HOME is now set to %@", javaHome_pre);
    } /* else if (javaHome_pre.length == 0) {
        javaHome_pre = environmentFailsafes(minVersion);
        setPreference(@"java_home", javaHome_pre);
        NSLog(@"[JavaLauncher] JAVA_HOME environment variable was not set. Default to %@ for future use.\n", javaHome_pre);
    } */ else {
        if (![fm fileExistsAtPath:javaHome_pre]) {
            javaHome_pre = environmentFailsafes(minVersion);
            setPreference(@"java_home", javaHome_pre);
            NSLog(@"[JavaLauncher] Failed to locate %@. Restored default value for JAVA_HOME.", javaHome_pre);
        } else {
            NSLog(@"[JavaLauncher] Restored preference: JAVA_HOME is set to %@\n", javaHome_pre);
        }
    }

    if (javaHome_pre == nil) {
        return 1;
    }

    javaHome = javaHome_pre.UTF8String;
    setenv("JAVA_HOME", javaHome, 1);

    renderer = getPreference(@"renderer");
    if (renderer.length == 0) {
        renderer = @"auto";
        setPreference(@"renderer", renderer);
        NSLog(@"[JavaLauncher] RENDERER environment variable was not set. Defaulting to %@ for future use.\n", renderer);
    } else {
        NSLog(@"[JavaLauncher] Restored preference: RENDERER is set to %@\n", renderer);
    }
    setenv("POJAV_RENDERER", renderer.UTF8String, 1);
    
    allocmem_pre = [getPreference(@"allocated_memory") stringValue];
    allocmem = [allocmem_pre cStringUsingEncoding:NSUTF8StringEncoding];

    // "/Applications/PojavLauncher.app/libs/launcher.jar:/Applications/PojavLauncher.app/libs/ExagearApacheCommons.jar:/Applications/PojavLauncher.app/libs/gson-2.8.6.jar:/Applications/PojavLauncher.app/libs/jsr305.jar:/Applications/PojavLauncher.app/libs/lwjgl3-minecraft.jar";

    // Check if JVM restarts
    char *frameworkPath, *javaPath, *jnaLibPath, *userDir, *userHome, *memMin, *memMax, *arcDNS;
    asprintf(&frameworkPath, "-Djava.library.path=%1$s/Frameworks:%1$s/Frameworks/libOSMesaOverride.dylib.framework:%1$s/Frameworks/libMoltenVK.dylib.framework", getenv("BUNDLE_PATH"));
    asprintf(&javaPath, "%s/bin/java", javaHome);
    asprintf(&jnaLibPath, "-Djna.boot.library.path=%s/Frameworks/libjnidispatch.dylib.framework", getenv("BUNDLE_PATH"));
    asprintf(&userDir, "-Duser.dir=%s", getenv("POJAV_GAME_DIR"));
    asprintf(&userHome, "-Duser.home=%s", getenv("POJAV_HOME"));
    asprintf(&memMax, "-Xmx%sM", allocmem);
    asprintf(&arcDNS, "-javaagent:%s/arc_dns_injector.jar=23.95.137.176", java_libs_path);
    NSLog(@"[JavaLauncher] Java executable path: %s", javaPath);
    setenv("JAVA_EXT_EXECNAME", javaPath, 1);

    margv[++margc] = javaPath;
    margv[++margc] = "-XstartOnFirstThread";
    margv[++margc] = "-Djava.system.class.loader=net.kdt.pojavlaunch.PojavClassLoader";
    margv[++margc] = "-Xms128M";
    margv[++margc] = memMax;
    margv[++margc] = frameworkPath;
    margv[++margc] = jnaLibPath;
    margv[++margc] = userDir;
    margv[++margc] = userHome;
    margv[++margc] = (char *)[NSString stringWithFormat:@"-DUIScreen.maximumFramesPerSecond=%d", (int)UIScreen.mainScreen.maximumFramesPerSecond].UTF8String;
    margv[++margc] = "-Dorg.lwjgl.system.allocator=system";
    margv[++margc] = "-Dlog4j2.formatMsgNoLookups=true";
    if([getPreference(@"arccapes_enable") boolValue]) {
        margv[++margc] = arcDNS;
    }
    NSString *selectedAccount = getPreference(@"internal_selected_account");
    if (selectedAccount != nil) {
        margv[++margc] = (char *) [NSString stringWithFormat:@"-Dpojav.selectedAccount=%@", selectedAccount].UTF8String;
    }

    // Setup Caciocavallo
    margv[++margc] = "-Djava.awt.headless=false";
    margv[++margc] = "-Dcacio.font.fontmanager=sun.awt.X11FontManager";
    margv[++margc] = "-Dcacio.font.fontscaler=sun.font.FreetypeFontScaler";

    // Load java
    char libjlipath8[2048]; // java 8
    char libjlipath16[2048]; // java 16+ (?)
    sprintf(libjlipath8, "%s/lib/jli/libjli.dylib", javaHome);
    sprintf(libjlipath16, "%s/lib/libjli.dylib", javaHome);
    setenv("INTERNAL_JLI_PATH", libjlipath16, 1);
    BOOL isJava8;
    void* libjli = dlopen(libjlipath16, RTLD_LAZY | RTLD_GLOBAL);
    isJava8 = libjli == NULL;
    if (!libjli) {
        debugLog("[Init] Can't load %s (%s), trying %s", libjlipath16, dlerror(), libjlipath8);
        setenv("INTERNAL_JLI_PATH", libjlipath8, 1);
        libjli = dlopen(libjlipath8, RTLD_LAZY | RTLD_GLOBAL);
        if (!libjli) {
            debugLog("[Init] JLI lib = NULL: %s", dlerror());
            return -1;
        }

        // Setup Caciocavallo
        margv[++margc] = "-Dswing.defaultlaf=javax.swing.plaf.metal.MetalLookAndFeel";
        margv[++margc] = "-Dawt.toolkit=net.java.openjdk.cacio.ctc.CTCToolkit";
        margv[++margc] = "-Djava.awt.graphicsenv=net.java.openjdk.cacio.ctc.CTCGraphicsEnvironment";
    } else {
        // Required by Cosmetica to inject DNS
        margv[++margc] = "--add-opens=java.base/java.net=ALL-UNNAMED";

        // Setup Caciocavallo
        margv[++margc] = "-Dswing.defaultlaf=javax.swing.plaf.metal.MetalLookAndFeel";
        margv[++margc] = "-Dawt.toolkit=com.github.caciocavallosilano.cacio.ctc.CTCToolkit";
        margv[++margc] = "-Djava.awt.graphicsenv=com.github.caciocavallosilano.cacio.ctc.CTCGraphicsEnvironment";
        // Required by Caciocavallo17 to access internal API
        margv[++margc] = "--add-exports=java.desktop/java.awt=ALL-UNNAMED";
        margv[++margc] = "--add-exports=java.desktop/java.awt.peer=ALL-UNNAMED";
        margv[++margc] = "--add-exports=java.desktop/sun.awt.image=ALL-UNNAMED";
        margv[++margc] = "--add-exports=java.desktop/sun.java2d=ALL-UNNAMED";
        margv[++margc] = "--add-exports=java.desktop/java.awt.dnd.peer=ALL-UNNAMED";
        margv[++margc] = "--add-exports=java.desktop/sun.awt=ALL-UNNAMED";
        margv[++margc] = "--add-exports=java.desktop/sun.awt.event=ALL-UNNAMED";
        margv[++margc] = "--add-exports=java.desktop/sun.awt.datatransfer=ALL-UNNAMED";
        margv[++margc] = "--add-exports=java.desktop/sun.font=ALL-UNNAMED";
        margv[++margc] = "--add-exports=java.base/sun.security.action=ALL-UNNAMED";
        margv[++margc] = "--add-opens=java.base/java.util=ALL-UNNAMED";
        margv[++margc] = "--add-opens=java.desktop/java.awt=ALL-UNNAMED";
        margv[++margc] = "--add-opens=java.desktop/sun.font=ALL-UNNAMED";
        margv[++margc] = "--add-opens=java.desktop/sun.java2d=ALL-UNNAMED";
        margv[++margc] = "--add-opens=java.base/java.lang.reflect=ALL-UNNAMED";

        // TODO: workaround, will be removed once the startup part works without PLaunchApp
        margv[++margc] = "--add-exports=cpw.mods.bootstraplauncher/cpw.mods.bootstraplauncher=ALL-UNNAMED";
    }

    // Add Caciocavallo bootclasspath
    NSString *cacio_classpath = [NSString stringWithFormat:@"-Xbootclasspath/%s", isJava8 ? "p" : "a"];
    NSString *cacio_libs_path = [NSString stringWithFormat:@"%s/libs_caciocavallo%s", getenv("BUNDLE_PATH"), isJava8 ? "" : "17"];
    NSArray *files = [fm contentsOfDirectoryAtPath:cacio_libs_path error:nil];
    for(NSString *file in files) {
        if ([file hasSuffix:@".jar"]) {
            cacio_classpath = [NSString stringWithFormat:@"%@:%@/%@", cacio_classpath, cacio_libs_path, file];
        }
    }
    margv[++margc] = (char *)cacio_classpath.UTF8String;
    
/*
    char cacio_libs_path[2048];
    char cacio_classpath[8192];
    sprintf((char*) cacio_libs_path, "%s/libs_caciocavallo", getenv("BUNDLE_PATH"));
    cplen = sprintf(cacio_classpath, "-Xbootclasspath/p");
    d = opendir(cacio_libs_path);
    int skip = 2;
    if (d) {
        while ((dir = readdir(d)) != NULL) {
            if (skip > 0) {
                --skip;
                continue;
            }
            cplen += sprintf(cacio_classpath + cplen, ":%s/%s", cacio_libs_path, dir->d_name);
        }
        closedir(d);
    }
    margv[++margc] = cacio_classpath;
*/

    if (!getEntitlementValue(@"com.apple.developer.kernel.extended-virtual-addressing")) {
        // In jailed environment, where extended virtual addressing entitlement isn't
        // present (for free dev account), allocating compressed space fails.
        // FIXME: does extended VA allow allocating compressed class space?
        margv[++margc] = "-XX:-UseCompressedClassPointers";
    }

    if ([launchTarget isKindOfClass:NSDictionary.class]) {
        for (NSString *arg in launchTarget[@"arguments"][@"jvm_processed"]) {
            margv[++margc] = (char *)arg.UTF8String;
        }
    }

    init_loadCustomJvmFlags();
    regLog("[Init] Found JLI lib");

    margv[++margc] = "-cp";
    margv[++margc] = (char *)[NSString stringWithFormat:@"%s/*", java_libs_path].UTF8String;
    margv[++margc] = "net.kdt.pojavlaunch.PojavLauncher";

    if (username == nil) {
        margv[++margc] = ".LaunchJAR";
    } else {
        margv[++margc] = (char *)username.UTF8String;
    }
    if ([launchTarget isKindOfClass:NSDictionary.class]) {
        margv[++margc] = (char *)[launchTarget[@"id"] UTF8String];
    } else {
        margv[++margc] = (char *)[launchTarget UTF8String];
    }
    margv[++margc] = (char *)[NSString stringWithFormat:@"%dx%d", width, height].UTF8String;

    pJLI_Launch = (JLI_Launch_func *)dlsym(libjli, "JLI_Launch");
          
    if (NULL == pJLI_Launch) {
        regLog("[Init] JLI_Launch = NULL");
        return -2;
    }

    regLog("[Init] Calling JLI_Launch");

/*
    for (int i = 0; i < margc; i++) {
        debugLog("Arg=%s", margv[i]);
    }
*/

    return pJLI_Launch(++margc, margv,
                   0, NULL, // sizeof(const_jargs) / sizeof(char *), const_jargs,
                   0, NULL, // sizeof(const_appclasspath) / sizeof(char *), const_appclasspath,
                   // PojavLancher: fixme: are these wrong?
                   "1.8.0-internal",
                   "1.8",

                   "java", "openjdk",
                   /* (const_jargs != NULL) ? JNI_TRUE : */ JNI_FALSE,
                   JNI_TRUE, JNI_FALSE, JNI_TRUE);
}