From eaca66a450160a9b7166f59218bdb8982d332b61 Mon Sep 17 00:00:00 2001
From: SrGooglo <srgooglo@ragestudio.net>
Date: Sun, 28 Jan 2024 20:10:26 +0100
Subject: [PATCH] added `resolveJavaPath` util

---
 src/main/utils/resolveJavaPath.js | 178 ++++++++++++++++++++++++++++++
 1 file changed, 178 insertions(+)
 create mode 100644 src/main/utils/resolveJavaPath.js

diff --git a/src/main/utils/resolveJavaPath.js b/src/main/utils/resolveJavaPath.js
new file mode 100644
index 0000000..65d90af
--- /dev/null
+++ b/src/main/utils/resolveJavaPath.js
@@ -0,0 +1,178 @@
+/* Copyright 2013 Joseph Spencer.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *       http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import * as fs from "fs"
+import * as path from "path"
+import * as cp from "child_process"
+
+import {execSync} from "child_process"
+import which from "which"
+
+import WinReg from "winreg"
+
+const isWindows = process.platform.indexOf('win') === 0
+
+const jdkRegistryKeyPaths = [
+    "\\SOFTWARE\\JavaSoft\\JDK",
+    "\\SOFTWARE\\JavaSoft\\Java Development Kit"
+]
+
+const jreRegistryKeyPaths = [
+    "\\SOFTWARE\\JavaSoft\\Java Runtime Environment"
+]
+
+async function findJavaHome(options) {
+    const allowJre = !!(options && options.allowJre)
+
+    const JAVA_FILENAME = (allowJre ? 'java' : 'javac') + (isWindows ? '.exe' : '')
+
+    // Search both "x64" and "x86" registries for Java runtimes if not specified
+    const regs = (options && options.registry) ? [options.registry] : ["x64", "x86"]
+
+    // From env
+    if (process.env.JAVA_HOME && dirIsJavaHome(process.env.JAVA_HOME, JAVA_FILENAME)) {
+        return process.env.JAVA_HOME
+    }
+
+    // From registry (windows only)
+    if (isWindows) {
+        const possibleKeyPaths = allowJre ? jdkRegistryKeyPaths.concat(jreRegistryKeyPaths) : jdkRegistryKeyPaths
+        const javaHome = await findInRegistry(possibleKeyPaths, regs)
+
+        if (javaHome) {
+            return javaHome
+        }
+    }
+
+    // From PATH
+    return await findInPath(JAVA_FILENAME)
+}
+
+function findInPath(JAVA_FILENAME) {
+    return new Promise((resolve) => {
+        which(JAVA_FILENAME, (err, proposed) => {
+            if (err || !proposed) {
+                return resolve(null)
+            }
+
+            if (/\.jenv\/shims/.test(proposed)) {
+                try {
+                    proposed = execSync(`jenv which ${JAVA_FILENAME}`).toString().trim()
+                } catch (ex) {
+                    console.error(ex)
+                }
+            }
+
+            //resolve symlinks
+            proposed = fs.realpathSync(proposed)
+
+            //get the /bin directory
+            proposed = path.dirname(proposed)
+
+            //on mac, java install has a utility script called java_home that does the
+            //dirty work for us
+            const macUtility = path.resolve(proposed, 'java_home')
+
+            if (fs.existsSync(macUtility)) {
+                let buffer
+
+                try {
+                    buffer = cp.execSync(macUtility, { cwd: proposed })
+
+                    const javaHome = '' + buffer.toString().replace(/\n$/, '')
+
+                    return resolve(javaHome)
+                } catch (error) {
+                    return resolve(null)
+                }
+            }
+
+            //up one from /bin
+            resolve(path.dirname(proposed))
+        })
+    })
+}
+
+async function findInRegistry(keyPaths, regArchs) {
+    if (!keyPaths.length) return null
+
+    const promises = []
+
+    for (const arch of regArchs) {
+        for (const keyPath of keyPaths) {
+            promises.push(promisifyFindPossibleRegKey(keyPath, arch))
+        }
+    }
+
+    const keysFoundSegments = await Promise.all(promises)
+
+    const keysFound = Array.prototype.concat.apply([], keysFoundSegments)
+
+    if (!keysFound.length) {
+        return null
+    }
+
+    const sortedKeysFound = keysFound.sort(function (a, b) {
+        const aVer = parseFloat(a.key)
+        const bVer = parseFloat(b.key)
+
+        return bVer - aVer
+    })
+
+    for (const key of sortedKeysFound) {
+        const res = await promisifyFindJavaHomeInRegKey(key)
+
+        if (res) {
+            return res
+        }
+    }
+
+    return null
+}
+
+function promisifyFindPossibleRegKey(keyPath, regArch) {
+    return new Promise((resolve) => {
+        const winreg = new WinReg({
+            hive: WinReg.HKLM,
+            key: keyPath,
+            arch: regArch
+        })
+
+        winreg.keys((err, result) => {
+            if (err) {
+                return resolve([])
+            }
+            resolve(result)
+        })
+    })
+}
+
+function promisifyFindJavaHomeInRegKey(reg) {
+    return new Promise((resolve) => {
+        reg.get('JavaHome', function (err, home) {
+            if (err || !home) {
+                return resolve(null)
+            }
+
+            return resolve(home.value)
+        })
+    })
+}
+
+function dirIsJavaHome(dir, javaFilename) {
+    return fs.existsSync('' + dir) && fs.statSync(dir).isDirectory() && fs.existsSync(path.resolve(dir, 'bin', javaFilename))
+}
+
+export default findJavaHome
\ No newline at end of file