Fixes after review: get last statement in codeFragmetn using psi instead of findElementAt

2016-08-03 17:01:11 +03:00
parent ed204e9bac
commit 7087f10660
1 changed files with 1 additions and 1 deletions
@@ -412,6 +412,6 @@ class KotlinCodeFragmentFactory: CodeFragmentFactory() {
            }
        })

-        return getContextElement(codeFragment.findElementAt(codeFragment.text.length - 5)) as? KtElement
+        return codeFragment.getContentElement().statements.lastOrNull()
    }
 }