fixed path term

wttech · Nov 23, 2022 · 2f086b8 · 2f086b8
1 parent d0799c9
commit 2f086b8
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/app/aem/core/src/main/kotlin/com/cognifide/apm/core/grammar/ReferenceFinder.kt b/app/aem/core/src/main/kotlin/com/cognifide/apm/core/grammar/ReferenceFinder.kt
@@ -24,7 +24,7 @@ import com.cognifide.apm.api.scripts.LaunchEnvironment
 import com.cognifide.apm.api.scripts.LaunchMode
 import com.cognifide.apm.api.scripts.Script
 import com.cognifide.apm.api.services.ScriptFinder
-import com.cognifide.apm.core.grammar.argument.toPlainString
+import com.cognifide.apm.core.grammar.common.getPath
 import com.cognifide.apm.core.grammar.executioncontext.ExecutionContext
 import com.cognifide.apm.core.grammar.parsedscript.ParsedScript
 import com.cognifide.apm.core.progress.ProgressImpl
@@ -104,13 +104,13 @@ class ReferenceFinder(
     inner class InternalVisitor(private val executionContext: ExecutionContext, val refGraph: ReferenceGraph, val currentNode: ReferenceGraph.TreeNode) : com.cognifide.apm.core.grammar.antlr.ApmLangBaseVisitor<Unit>() {
         val scripts = mutableSetOf<Script>()
 
-        override fun visitImportScript(ctx: com.cognifide.apm.core.grammar.antlr.ApmLangParser.ImportScriptContext?) {
-            val foundPath = ctx?.path()?.STRING_LITERAL()?.toPlainString()
+        override fun visitImportScript(ctx: com.cognifide.apm.core.grammar.antlr.ApmLangParser.ImportScriptContext) {
+            val foundPath = getPath(ctx.path())
             createTransition(foundPath, ReferenceGraph.TransitionType.IMPORT)
         }
 
-        override fun visitRunScript(ctx: com.cognifide.apm.core.grammar.antlr.ApmLangParser.RunScriptContext?) {
-            val foundPath = ctx?.path()?.STRING_LITERAL()?.toPlainString()
+        override fun visitRunScript(ctx: com.cognifide.apm.core.grammar.antlr.ApmLangParser.RunScriptContext) {
+            val foundPath = getPath(ctx.path())
             createTransition(foundPath, ReferenceGraph.TransitionType.RUN_SCRIPT)
         }
 

diff --git a/app/aem/core/src/main/kotlin/com/cognifide/apm/core/grammar/ScriptRunner.kt b/app/aem/core/src/main/kotlin/com/cognifide/apm/core/grammar/ScriptRunner.kt
@@ -26,8 +26,8 @@ import com.cognifide.apm.api.status.Status
 import com.cognifide.apm.core.grammar.antlr.ApmLangParser.*
 import com.cognifide.apm.core.grammar.argument.ArgumentResolverException
 import com.cognifide.apm.core.grammar.argument.Arguments
-import com.cognifide.apm.core.grammar.argument.toPlainString
 import com.cognifide.apm.core.grammar.common.getIdentifier
+import com.cognifide.apm.core.grammar.common.getPath
 import com.cognifide.apm.core.grammar.executioncontext.ExecutionContext
 import com.cognifide.apm.core.grammar.parsedscript.InvalidSyntaxException
 import com.cognifide.apm.core.grammar.parsedscript.InvalidSyntaxMessageFactory
@@ -99,7 +99,7 @@ class ScriptRunner(
         }
 
         override fun visitRunScript(ctx: RunScriptContext) {
-            val path = ctx.path().STRING_LITERAL().toPlainString()
+            val path = getPath(ctx.path())
             val arguments = executionContext.resolveArguments(ctx.namedArguments())
             val loadScript = executionContext.loadScript(path)
             if (executionContext.scriptIsOnStack(loadScript))