From 852e233c98493758a68dc8721159546fb8864b48 Mon Sep 17 00:00:00 2001
From: Iru21 <33987046+Iru21@users.noreply.github.com>
Date: Wed, 23 Aug 2023 12:20:54 +0200
Subject: [PATCH] Generate two shadowed jars, one with db drivers

---
 build.gradle.kts                              | 100 ++++++++++--------
 .../kotlin/me/iru/data/DatabaseConnection.kt  |   8 +-
 2 files changed, 61 insertions(+), 47 deletions(-)

diff --git a/build.gradle.kts b/build.gradle.kts
index c8f6201..7864a95 100644
--- a/build.gradle.kts
+++ b/build.gradle.kts
@@ -1,3 +1,4 @@
+import com.github.jengelman.gradle.plugins.shadow.tasks.ShadowJar
 import org.jetbrains.kotlin.gradle.tasks.KotlinCompile
 import org.ajoberstar.grgit.*
 
@@ -30,9 +31,11 @@ val pluginName: String by project
 val pluginVersion: String by project
 val minecraftVersion: String by project
 
+val withDrivers: Configuration by configurations.creating
+
 tasks {
 
-    withType<KotlinCompile>() {
+    withType<KotlinCompile> {
         kotlinOptions.jvmTarget = "1.8"
     }
 
@@ -61,56 +64,64 @@ tasks {
     shadowJar {
         dependsOn(processResources)
         dependsOn(makeDefaults)
-        archiveFileName.set("${pluginName}-${pluginVersion}.jar")
         relocate("org.bstats", group)
+        archiveFileName.set("${pluginName}-${pluginVersion}.jar")
     }
-}
 
-tasks.register("createReleases") {
-    dependsOn(tasks["createGitHubRelease"])
-    dependsOn(tasks.modrinth)
-    dependsOn(tasks.modrinthSyncBody)
-}
+    val shadowJarDrivers by registering(ShadowJar::class) {
+        dependsOn(processResources)
+        dependsOn(makeDefaults)
+        relocate("org.bstats", group)
+        archiveFileName.set("${pluginName}-${pluginVersion}-drv.jar")
+        configurations = listOf(project.configurations["runtimeClasspath"], withDrivers)
 
-modrinth {
-    val version = pluginVersion
-
-    token.set(System.getenv("MODRINTH_TOKEN"))
-    projectId.set("authy")
-    versionNumber.set(version)
-    versionType.set("release")
-    versionName.set("$pluginName $version")
-    uploadFile.set(tasks.shadowJar as Any)
-    gameVersions.addAll("1.17", "1.18", "1.19", "1.20")
-    loaders.addAll("spigot", "paper", "purpur")
-    changelog.set(rootProject.file("changelog.md").readText())
-    syncBodyFrom.set(rootProject.file("README.md").readText())
-}
+    }
 
-tasks.register("createGitHubRelease") {
-    dependsOn(tasks.shadowJar)
-    doLast {
-        val git = Grgit.open {
-            dir = projectDir
-        }
-        val tagName = "v${pluginVersion}"
-        git.tag.add {
-            name = tagName
-            message = "Release $tagName"
-            force = true
-        }
-        git.push {
-            tags = true
+    modrinth {
+        val version = pluginVersion
+
+        token.set(System.getenv("MODRINTH_TOKEN"))
+        projectId.set("authy")
+        versionNumber.set(version)
+        versionType.set("release")
+        versionName.set("$pluginName $version")
+        uploadFile.set(shadowJar as Any)
+        gameVersions.addAll("1.17", "1.18", "1.19", "1.20")
+        loaders.addAll("spigot", "paper", "purpur")
+        changelog.set(rootProject.file("changelog.md").readText())
+        syncBodyFrom.set(rootProject.file("README.md").readText())
+    }
+
+    val createGitHubRelease by registering {
+        dependsOn(shadowJar)
+        doLast {
+            val git = Grgit.open {
+                dir = projectDir
+            }
+            val tagName = "v${pluginVersion}"
+            git.tag.add {
+                name = tagName
+                message = "Release $tagName"
+                force = true
+            }
+            git.push {
+                tags = true
+            }
+            git.close()
+
+            val command = "gh release create $tagName -F changelog.md -t \"$tagName\" \"build/libs/${pluginName}-${pluginVersion}.jar\""
+            val process = ProcessBuilder(command.split(" ")).start()
+            process.waitFor()
         }
-        git.close()
+    }
 
-        val command = "gh release create $tagName -F changelog.md -t \"$tagName\" \"build/libs/${pluginName}-${pluginVersion}.jar\""
-        val process = ProcessBuilder(command.split(" ")).start()
-        process.waitFor()
+    register("createReleases") {
+        dependsOn(createGitHubRelease)
+        dependsOn(modrinth)
+        dependsOn(modrinthSyncBody)
     }
 }
 
-
 repositories {
     mavenCentral()
     mavenLocal()
@@ -122,7 +133,8 @@ dependencies {
     compileOnly("org.apache.logging.log4j:log4j-api:2.20.0")
     compileOnly("org.apache.logging.log4j:log4j-core:2.20.0")
 
-    runtimeOnly("mysql:mysql-connector-java:8.0.33")
+    withDrivers("mysql:mysql-connector-java:8.0.33")
+    withDrivers("org.xerial:sqlite-jdbc:3.42.0.0")
 
     implementation("org.spigotmc:spigot-api:${minecraftVersion}-R0.1-SNAPSHOT")
     implementation("org.bstats:bstats-bukkit:3.0.2")
@@ -132,8 +144,4 @@ dependencies {
 val compileKotlin: KotlinCompile by tasks
 compileKotlin.kotlinOptions {
     jvmTarget = "1.8"
-}
-val compileTestKotlin: KotlinCompile by tasks
-compileTestKotlin.kotlinOptions {
-    jvmTarget = "1.8"
 }
\ No newline at end of file
diff --git a/src/main/kotlin/me/iru/data/DatabaseConnection.kt b/src/main/kotlin/me/iru/data/DatabaseConnection.kt
index aaea925..8285e95 100644
--- a/src/main/kotlin/me/iru/data/DatabaseConnection.kt
+++ b/src/main/kotlin/me/iru/data/DatabaseConnection.kt
@@ -33,6 +33,12 @@ class DatabaseConnection(val type: DatabaseType) {
     }
 
     private fun connectSQLite(): Connection {
+        Class.forName("org.sqlite.JDBC");
+
+        if(!authy.dataFolder.exists()) {
+            authy.dataFolder.mkdir()
+        }
+
         return DriverManager.getConnection("jdbc:sqlite:${authy.dataFolder}${File.separator}data.db")
     }
 
@@ -42,7 +48,7 @@ class DatabaseConnection(val type: DatabaseType) {
         val password = authy.config.getString("database.credentials.password")
         val databaseName = authy.config.getString("database.credentials.database")
 
-        Class.forName("com.mysql.jdbc.Driver")
+        Class.forName("com.mysql.cj.jdbc.Driver")
 
         val t = DriverManager
             .getConnection("jdbc:mysql://$host", user, password)