@@ -24,7 +24,7 @@ import com.cognifide.apm.api.scripts.LaunchEnvironment
2424import com.cognifide.apm.api.scripts.LaunchMode
2525import com.cognifide.apm.api.scripts.Script
2626import com.cognifide.apm.api.services.ScriptFinder
27- import com.cognifide.apm.core.grammar.argument.toPlainString
27+ import com.cognifide.apm.core.grammar.common.getPath
2828import com.cognifide.apm.core.grammar.executioncontext.ExecutionContext
2929import com.cognifide.apm.core.grammar.parsedscript.ParsedScript
3030import com.cognifide.apm.core.progress.ProgressImpl
@@ -104,13 +104,13 @@ class ReferenceFinder(
104104 inner class InternalVisitor (private val executionContext : ExecutionContext , val refGraph : ReferenceGraph , val currentNode : ReferenceGraph .TreeNode ) : com.cognifide.apm.core.grammar.antlr.ApmLangBaseVisitor<Unit>() {
105105 val scripts = mutableSetOf<Script >()
106106
107- override fun visitImportScript (ctx : com.cognifide.apm.core.grammar.antlr.ApmLangParser .ImportScriptContext ? ) {
108- val foundPath = ctx? .path()?. STRING_LITERAL ()?.toPlainString( )
107+ override fun visitImportScript (ctx : com.cognifide.apm.core.grammar.antlr.ApmLangParser .ImportScriptContext ) {
108+ val foundPath = getPath( ctx.path())
109109 createTransition(foundPath, ReferenceGraph .TransitionType .IMPORT )
110110 }
111111
112- override fun visitRunScript (ctx : com.cognifide.apm.core.grammar.antlr.ApmLangParser .RunScriptContext ? ) {
113- val foundPath = ctx? .path()?. STRING_LITERAL ()?.toPlainString( )
112+ override fun visitRunScript (ctx : com.cognifide.apm.core.grammar.antlr.ApmLangParser .RunScriptContext ) {
113+ val foundPath = getPath( ctx.path())
114114 createTransition(foundPath, ReferenceGraph .TransitionType .RUN_SCRIPT )
115115 }
116116
0 commit comments