Fix bug when chain was not found from nested lambda or anonymous

This commit is contained in:
Vitaliy.Bibaev
2017-10-26 15:19:12 +03:00
committed by Yan Zhulanow
parent bb90c5ad08
commit 6ce6301613
@@ -34,7 +34,7 @@ abstract class KotlinChainBuilderBase(private val transformer: ChainTransformer<
var element = getLatestElementInScope(start)
while (element != null) {
element.accept(visitor)
element = getLatestElementInScope(toUpperLevel(element))
element = toUpperLevel(element)
}
return visitor.chains().map { transformer.transform(it, startElement) }
@@ -47,7 +47,7 @@ abstract class KotlinChainBuilderBase(private val transformer: ChainTransformer<
current = current.parent
}
return current
return getLatestElementInScope(current)
}
protected abstract fun createChainsBuilder(): ChainBuilder