Implement "if null" / "if not null" live templates

idea/resources/liveTemplates/Kotlin.xml

@@ -48,7 +48,22 @@
             <option name="KOTLIN_STATEMENT" value="true" />
         </context>
     </template>
-
+    <template resource-bundle="org.jetbrains.jet.plugin.JetBundle" key="livetemplate.description.ifn"
+              name="ifn" toReformat="true" toShortenFQNames="true"
+              value="if ($VAR$ == null) {&#10;$END$&#10;}">
+      <variable alwaysStopAt="true" defaultValue="&quot;i&quot;" expression="kotlinSuggestVariableName()" name="VAR" />
+      <context>
+        <option name="KOTLIN_EXPRESSION" value="true" />
+      </context>
+    </template>
+    <template resource-bundle="org.jetbrains.jet.plugin.JetBundle" key="livetemplate.description.inn"
+              name="inn" toReformat="true" toShortenFQNames="true"
+              value="if ($VAR$ != null) {&#10;$END$&#10;}">
+      <variable alwaysStopAt="true" defaultValue="&quot;i&quot;" expression="kotlinSuggestVariableName()" name="VAR" />
+      <context>
+        <option name="KOTLIN_EXPRESSION" value="true" />
+      </context>
+    </template>
     <template resource-bundle="org.jetbrains.jet.plugin.JetBundle" key="livetemplate.description.void"
               name="void" toReformat="true" toShortenFQNames="true" value="fun $NAME$($PARAMS$) {&#10;$END$&#10;}">
         <variable alwaysStopAt="true" defaultValue="" expression="" name="NAME" />

idea/src/org/jetbrains/jet/plugin/JetBundle.properties

@@ -89,6 +89,8 @@ goto.super.class.chooser.title=Choose super class or interface
 livetemplate.description.main=main() function
 livetemplate.description.soutp=Prints function parameter names and values to System.out
 livetemplate.description.iter=Iterate over elements of iterable
+livetemplate.description.ifn=Inserts ''if null'' expression
+livetemplate.description.inn=Inserts ''if not null'' expression
 livetemplate.description.void=Function returning nothing
 livetemplate.description.fun0=Function with no parameters
 livetemplate.description.fun1=Function with one parameter

idea/src/org/jetbrains/jet/plugin/liveTemplates/macro/JetSuggestVariableNameMacro.java

@@ -16,14 +16,14 @@
 
 package org.jetbrains.jet.plugin.liveTemplates.macro;
 
-import com.intellij.codeInsight.template.Expression;
+import com.intellij.openapi.project.Project;
-import com.intellij.codeInsight.template.ExpressionContext;
-import com.intellij.codeInsight.template.Macro;
-import com.intellij.codeInsight.template.Result;
 import org.jetbrains.annotations.NotNull;
+import org.jetbrains.jet.lang.descriptors.VariableDescriptor;
+import org.jetbrains.jet.lang.resolve.scopes.JetScope;
+import org.jetbrains.jet.lang.types.expressions.ExpressionTypingServices;
 import org.jetbrains.jet.plugin.JetBundle;
 
-public class JetSuggestVariableNameMacro extends Macro {
+public class JetSuggestVariableNameMacro extends BaseJetVariableMacro {
     @Override
     public String getName() {
         return "kotlinSuggestVariableName";
@@ -35,7 +35,12 @@ public class JetSuggestVariableNameMacro extends Macro {
     }
 
     @Override
-    public Result calculateResult(@NotNull Expression[] params, ExpressionContext context) {
+    protected boolean isSuitable(
-        return null;  //TODO
+            @NotNull VariableDescriptor variableDescriptor,
+            @NotNull JetScope scope,
+            @NotNull Project project,
+            ExpressionTypingServices callResolverContext
+    ) {
+        return variableDescriptor.getType().isNullable();
     }
 }

idea/testData/templates/ifn.exp.kt

@@ -0,0 +1,16 @@
+val a: String = ""
+val b: String? = ""
+
+class MyClass {
+    val x: String = ""
+    val y: String? = ""
+
+    fun foo() {
+        val s: String = ""
+        val t: String? = ""
+
+        if (b == null) {
+            <caret>
+        }
+    }
+}

idea/testData/templates/ifn.kt

@@ -0,0 +1,14 @@
+val a: String = ""
+val b: String? = ""
+
+class MyClass {
+    val x: String = ""
+    val y: String? = ""
+
+    fun foo() {
+        val s: String = ""
+        val t: String? = ""
+
+        <caret>
+    }
+}

idea/testData/templates/inn.exp.kt

@@ -0,0 +1,16 @@
+val a: String = ""
+val b: String? = ""
+
+class MyClass {
+    val x: String = ""
+    val y: String? = ""
+
+    fun foo() {
+        val s: String = ""
+        val t: String? = ""
+
+        if (b != null) {
+            <caret>
+        }
+    }
+}

idea/testData/templates/inn.kt

@@ -0,0 +1,14 @@
+val a: String = ""
+val b: String? = ""
+
+class MyClass {
+    val x: String = ""
+    val y: String? = ""
+
+    fun foo() {
+        val s: String = ""
+        val t: String? = ""
+
+        <caret>
+    }
+}

idea/tests/org/jetbrains/jet/plugin/codeInsight/LiveTemplatesTest.java

@@ -198,7 +198,22 @@ public class LiveTemplatesTest extends LightCodeInsightFixtureTestCase {
         checkAfter();
     }
 
+    private void doTestIfnInn() {
+        start();
 
+        assertStringItems("b", "t", "y");
+        typeAndNextTab("b");
+
+        checkAfter();
+    }
+
+    public void testIfn() {
+        doTestIfnInn();
+    }
+
+    public void testInn() {
+        doTestIfnInn();
+    }
 
     private void paremeterless() {
         start();