fix: scan package name

cssxsh · Jul 12, 2022 · 0ad3dc9 · 0ad3dc9
1 parent 664f03c
commit 0ad3dc9
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/main/kotlin/xyz/cssxsh/mirai/hibernate/MiraiHibernateLoader.kt b/src/main/kotlin/xyz/cssxsh/mirai/hibernate/MiraiHibernateLoader.kt
@@ -62,9 +62,12 @@ public interface MiraiHibernateLoader {
         public constructor(plugin: JvmPlugin) : this(
             autoScan = true,
             packageName = with(plugin::class.java) {
-                classLoader.getDefinedPackage("$packageName.entry")?.name
-                    ?: classLoader.getDefinedPackage("$packageName.model")?.name
-                    ?: packageName
+                val packagePath = packageName.replace('.', '/')
+                for(name in listOf("entry", "entity", "entities", "model", "models", "bean", "beans", "dto")) {
+                    classLoader.getResource("$packagePath/$name") ?: continue
+                    return@with "$packageName.$name"
+                }
+                packageName
             },
             classLoader = plugin::class.java.classLoader,
             configuration = plugin.configFolder.resolve("hibernate.properties"),