Added support for shortened URLs (#187)

Author: egor-n

src/main/kotlin/kscript/app/AppHelpers.kt

@@ -138,11 +138,20 @@ fun createTmpScript(scriptText: String, extension: String = "kts"): File {
 
 fun fetchFromURL(scriptURL: String): File {
     val urlHash = md5(scriptURL)
-    val urlExtension = if (scriptURL.endsWith(".kt")) "kt" else "kts"
+    val scriptText = URL(scriptURL).readText()
+    val urlExtension = when {
+        scriptURL.endsWith(".kt") -> "kt"
+        scriptURL.endsWith(".kts") -> "kts"
+        else -> if (scriptText.contains("fun main")) {
+            "kt"
+        } else {
+            "kts"
+        }
+    }
     val urlCache = File(KSCRIPT_CACHE_DIR, "/url_cache_${urlHash}.$urlExtension")
 
     if (!urlCache.isFile) {
-        urlCache.writeText(URL(scriptURL).readText())
+        urlCache.writeText(scriptText)
     }
 
     return urlCache

test/test_suite.sh

@@ -77,6 +77,7 @@ assert "kscript i_do_not_exist.kts 2>&1" "[kscript] [ERROR] Could not read scrip
 
 ## make sure that it runs with remote URLs
 assert "kscript https://raw.githubusercontent.com/holgerbrandl/kscript/master/test/resources/url_test.kts" "I came from the internet"
+assert "kscript https://git.io/fxHBv" "main was called"
 
 ## there are some dependencies which are not jar, but maybe pom, aar, ..
 ## make sure they work, too