It's publishing time

Author: BluSpring

.github/workflows/ci.yml

@@ -32,8 +32,11 @@ jobs:
           key: ${{ runner.os }}-gradle-${{ hashFiles('**/gradle-wrapper.properties') }}
           restore-keys: |
             ${{ runner.os }}-gradle-
-      - name: Build artifacts
-        run: ./gradlew :native:jar
+      - name: Build and publish artifacts
+        run: ./gradlew publish
+        env:
+          MAVEN_USER: ${{ secrets.MAVEN_USER }}
+          MAVEN_PASS: ${{ secrets.MAVEN_PASS }}
       - name: Upload build artifacts
         uses: actions/upload-artifact@v4
         with:
@@ -64,8 +67,8 @@ jobs:
           key: ${{ runner.os }}-gradle-${{ hashFiles('**/gradle-wrapper.properties') }}
           restore-keys: |
             ${{ runner.os }}-gradle-
-      - name: Build artifacts
-        run: ./gradlew :native:jar
+      - name: Build and publish artifacts
+        run: ./gradlew :native:publish
       - name: Upload build artifacts
         uses: actions/upload-artifact@v4
         with:

build.gradle.kts

@@ -3,10 +3,32 @@ plugins {
 }
 
 subprojects {
+    apply(plugin = "java")
+    apply(plugin = "maven-publish")
+
     group = "xyz.bluspring"
-    version = "0.2.0"
+    version = "${rootProject.property("unitytranslate_version")}"
 
     repositories {
         mavenCentral()
     }
+
+    publishing {
+        repositories {
+            maven("https://mvn.devos.one/snapshots") {
+                credentials {
+                    username = System.getenv()["MAVEN_USER"]
+                    password = System.getenv()["MAVEN_PASS"]
+                }
+            }
+        }
+        publications {
+            register("maven", MavenPublication::class) {
+                groupId = "xyz.bluspring"
+                artifactId = "UnityTranslateLib"
+                version = "${rootProject.property("unitytranslate_version")}"
+                from(components.getByName("java"))
+            }
+        }
+    }
 }

gradle.properties

@@ -1,3 +1,5 @@
+unitytranslate_version = 0.2.0-SNAPSHOT
+
 # This isn't actually used in the download process, however it's used for the sake of caching the downloaded files.
 # https://github.com/OpenNMT/CTranslate2/blob/master/python/ctranslate2/version.py
 ct2_version = 4.5.0

library/build.gradle.kts

@@ -1,12 +1,8 @@
 plugins {
     kotlin("jvm") version "2.1.0"
     kotlin("plugin.serialization") version "2.1.0"
-    id("com.gradleup.shadow") version "8.3.5"
 }
 
-group = "xyz.bluspring"
-version = "0.2.0"
-
 repositories {
     mavenCentral()
 }

library/src/main/kotlin/xyz/bluspring/unitytranslate/library/UnityTranslateLib.kt

@@ -8,10 +8,13 @@ import xyz.bluspring.unitytranslate.library.models.argos.ArgosPackageIndex
 import xyz.bluspring.unitytranslate.library.translator.DummyTranslator
 import xyz.bluspring.unitytranslate.library.translator.ModelBasedTranslator
 import xyz.bluspring.unitytranslate.library.translator.Translator
+import java.nio.file.Files
 import java.nio.file.Path
+import java.nio.file.Paths
+import java.nio.file.StandardCopyOption
 import java.util.concurrent.ConcurrentHashMap
-import kotlin.io.path.Path
 import kotlin.io.path.absolutePathString
+import kotlin.io.path.exists
 
 class UnityTranslateLib(val path: Path) {
     val packageIndex = ModelPackageManager(this).apply {
@@ -58,8 +61,104 @@ class UnityTranslateLib(val path: Path) {
     companion object {
         val logger: Logger = LoggerFactory.getLogger(UnityTranslateLib::class.java)
 
+        // Modified from ImGui-java's library loading - https://github.com/SpaiR/imgui-java/blob/main/imgui-binding/src/main/java/imgui/ImGui.java
         init {
-            System.load(Path("UnityTranslateLibRsAgain.dll").toAbsolutePath().absolutePathString())
+            val libPath = System.getProperty("unitytranslate.library.path")
+            val libName = System.getProperty("unitytranslate.library.name", "UnityTranslateLib")
+            val fullLibName = resolveFullLibName()
+
+            if (libPath != null) {
+                System.load(Paths.get(libPath).resolve(fullLibName).absolutePathString())
+            } else {
+                try {
+                    System.loadLibrary(libName)
+                } catch (e: Throwable) {
+                    val extractedPath = try {
+                        tryLoadFromClassPath(fullLibName)
+                    } catch (e2: Exception) {
+                        val joined = RuntimeException("Failed to load natives for UnityTranslateLib!")
+                        joined.addSuppressed(e2)
+                        joined.addSuppressed(e)
+
+                        throw joined
+                    }
+
+                    System.load(extractedPath)
+                }
+            }
+        }
+
+        private fun resolveFullLibName(): String {
+            val osName = System.getProperty("os.name").lowercase()
+            val isWindows = osName.contains("win")
+            val isMac = osName.contains("mac")
+
+            val libPrefix = if (isWindows) "" else "lib"
+            val libSuffix = if (isWindows) ".dll" else if (isMac) ".dylib" else ".so"
+
+            return System.getProperty("unitytranslate.library.name", "${libPrefix}UnityTranslateLib${libSuffix}")
+        }
+
+        private val platformLibs: List<String>
+            get() {
+                val osName = System.getProperty("os.name").lowercase()
+                val osArch = System.getProperty("os.arch").lowercase()
+                val isWindows = osName.contains("win")
+                val isMac = osName.contains("mac")
+
+                val dir = "unitytranslate/${if (isWindows) "windows" else if (isMac) "osx" else "linux"}-${osArch}"
+
+                return if (osArch == "amd64") {
+                    if (isWindows)
+                        listOf(
+                            "$dir/UnityTranslateLib.dll",
+                            "$dir/ctranslate2.dll",
+                            "$dir/cudnn64_9.dll",
+                            "$dir/libiomp5md.dll"
+                        )
+                    else if (isMac)
+                        listOf()
+                    else
+                        listOf(
+                            "$dir/libctranslate2.so",
+                            "$dir/libcudnn.so",
+                            "$dir/libgomp.so",
+                            "$dir/libUnityTranslateLib.so"
+                        )
+                } else emptyList()
+            }
+
+        private fun tryLoadFromClassPath(fullLibName: String): String {
+            val classLoader = UnityTranslateLib::class.java.classLoader
+            val libs = platformLibs
+
+            if (libs.isEmpty())
+                throw Exception("Unsupported platform ${System.getProperty("os.name")} (${System.getProperty("os.arch")})!")
+
+            val tmpDir = Paths.get(System.getProperty("java.io.tmpdir")).resolve("unitytranslate-natives")
+
+            if (!tmpDir.exists())
+                tmpDir.toFile().mkdirs()
+
+            for (packedLibPath in libs) {
+                val libName = packedLibPath.split("/").last()
+
+                classLoader.getResourceAsStream(packedLibPath)?.use {
+                    val libPath = tmpDir.resolve(libName)
+                    try {
+                        Files.copy(it, libPath, StandardCopyOption.REPLACE_EXISTING)
+                    } catch (e: AccessDeniedException) {
+                        if (!libPath.exists())
+                            throw e
+                    }
+                }
+            }
+
+            val unityTranslatePath = tmpDir.resolve(fullLibName)
+            if (!unityTranslatePath.exists())
+                throw Exception("Failed to load library files for UnityTranslateLib!")
+
+            return unityTranslatePath.absolutePathString()
         }
     }
 }

native/build.gradle.kts

@@ -1,10 +1,6 @@
 import java.net.URI
 import java.util.zip.ZipFile
 
-plugins {
-    java
-}
-
 val CT2_URL = "https://nightly.link/OpenNMT/CTranslate2/workflows/ci/master/python-wheels-{dist}-{arch}.zip"
 val ct2Version = rootProject.property("ct2_version")!! as String