Merge pull request #4 from jnorthrup/fix/classnotfound-exception

Fix: Resolve ClassNotFoundException for KTS scripts

Author: jnorthrup

src/main/kotlin/io/github/kscripting/kscript/code/Templates.kt

@@ -48,7 +48,7 @@ object Templates {
     fun createWrapperForScript(packageName: PackageName, className: String): String {
         val classReference = packageName.value + "." + className
 
-        return """
+        var wrapperTemplate = """
             |class Main_${className}{
             |    companion object {
             |        @JvmStatic
@@ -58,6 +58,12 @@ object Templates {
             |        }
             |    }
             |}""".trimStart().trimMargin()
+
+        if (packageName.value.isNotBlank()) {
+            wrapperTemplate = "package ${packageName.value};\n\n$wrapperTemplate"
+        }
+
+        return wrapperTemplate
     }
 
     fun createRunConfig(rootScriptName: String, rootScriptType: ScriptType, userArgs: List<String>): String {

src/main/kotlin/io/github/kscripting/kscript/creator/JarArtifactCreator.kt

@@ -43,17 +43,20 @@ class JarArtifactCreator(private val executor: Executor) {
             scriptContent = "package ${script.packageName.value}\n\n$scriptContent"
         }
 
-        FileUtils.createFile(scriptFile, scriptContent)
-
         val filesToCompile = mutableSetOf<OsPath>()
-        filesToCompile.add(scriptFile)
 
-        // create main-wrapper for kts scripts
         if (script.scriptLocation.scriptType == ScriptType.KTS) {
-            val wrapper = FileUtils.createFile(
-                basePath.resolve("$execClassName.kt"), Templates.createWrapperForScript(script.packageName, className)
-            )
-            filesToCompile.add(wrapper)
+            // For KTS scripts, combine script content and wrapper code into a single file.
+            // The package declaration is handled by Templates.createWrapperForScript.
+            val wrapperContent = Templates.createWrapperForScript(script.packageName, className)
+            scriptContent = "$scriptContent\n\n$wrapperContent"
+
+            FileUtils.createFile(scriptFile, scriptContent)
+            filesToCompile.add(scriptFile)
+        } else {
+            // For KT files, keep the existing logic.
+            FileUtils.createFile(scriptFile, scriptContent)
+            filesToCompile.add(scriptFile)
         }
 
         executor.compileKotlin(