Add update support for kotlin

Author: Layoric

src/ServiceStackIDEA/src/main/java/net/servicestack/idea/AddServiceStackRefHandler.java

@@ -41,7 +41,7 @@ public static void handleOk(String addressUrl, String qualifiedPackageName,
                                 Module module, StringBuilder errorMessage) {
         List<String> javaCodeLines = getDtoLines(addressUrl, qualifiedPackageName, fileName, errorMessage);
         if (javaCodeLines == null) return;
-        defaultNativeTypesHandler = getDefaultNativeTypesHandler(module);
+        defaultNativeTypesHandler = IDEAUtils.getDefaultNativeTypesHandler(module);
         boolean showDto = true;
         final MavenProjectsManager mavenProjectsManager = MavenProjectsManager.getInstance(module.getProject());
 
@@ -243,15 +243,4 @@ public static String getDtoNameWithoutExtension(String name) {
             return name.substring(0, p);
         }
     }
-
-    public static INativeTypesHandler getDefaultNativeTypesHandler(Module module) {
-        if(GradleBuildFileHelper.isGradleModule(module) && GradleBuildFileHelper.isUsingKotlin(module)) {
-            return new KotlinNativeTypesHandler();
-        }
-
-        if(IDEAPomFileHelper.isMavenProjectWithKotlin(module)) {
-            return new KotlinNativeTypesHandler();
-        }
-        return new JavaNativeTypesHandler();
-    }
 }

src/ServiceStackIDEA/src/main/java/net/servicestack/idea/IDEAUtils.java

@@ -61,4 +61,21 @@ public static void closeFile(Module module, String filePath) {
         }
         FileEditorManager.getInstance(module.getProject()).closeFile(fileByUrl);
     }
+
+    public static INativeTypesHandler getDefaultNativeTypesHandler(Module module) {
+        if(GradleBuildFileHelper.isGradleModule(module) && GradleBuildFileHelper.isUsingKotlin(module)) {
+            return new KotlinNativeTypesHandler();
+        }
+
+        if(IDEAPomFileHelper.isMavenProjectWithKotlin(module)) {
+            return new KotlinNativeTypesHandler();
+        }
+        return new JavaNativeTypesHandler();
+    }
+
+    public static INativeTypesHandler getNativeTypesHandler(Module module,String fileName) {
+        if(fileName.endsWith(".kt")) return new KotlinNativeTypesHandler();
+        if(fileName.endsWith(".java")) return new JavaNativeTypesHandler();
+        return getDefaultNativeTypesHandler(module);
+    }
 }

src/ServiceStackIDEA/src/main/java/net/servicestack/idea/UpdateServiceStackReferenceIntention.java

@@ -2,6 +2,8 @@
 
 import com.intellij.codeInsight.intention.impl.QuickEditAction;
 import com.intellij.openapi.editor.Editor;
+import com.intellij.openapi.module.Module;
+import com.intellij.openapi.module.ModuleUtil;
 import com.intellij.openapi.project.Project;
 import com.intellij.openapi.util.Iconable;
 import com.intellij.psi.PsiFile;
@@ -29,11 +31,13 @@ public String getFamilyName() {
     @Override
     public boolean isAvailable(@NotNull Project project, Editor editor, PsiFile psiFile) {
         try {
-            if(psiFile == null || !(psiFile instanceof PsiJavaFile)) {
+            if(psiFile == null) {
                 return false;
             }
-            PsiJavaFile classFile = (PsiJavaFile)psiFile;
-            if(UpdateServiceStackUtils.containsOptionsHeader(classFile)) {
+            if(!(psiFile instanceof PsiJavaFile || psiFile.getFileType().getDefaultExtension().equals("kt"))) {
+                return false;
+            }
+            if(UpdateServiceStackUtils.containsOptionsHeader(psiFile)) {
                 return true;
             }
         } catch (Exception e) {

src/ServiceStackIDEA/src/main/java/net/servicestack/idea/UpdateServiceStackUtils.java

@@ -5,6 +5,8 @@
 import com.intellij.notification.Notifications;
 import com.intellij.openapi.editor.Document;
 import com.intellij.openapi.fileEditor.FileDocumentManager;
+import com.intellij.openapi.module.Module;
+import com.intellij.openapi.module.ModuleUtil;
 import com.intellij.openapi.util.TextRange;
 import com.intellij.psi.PsiFile;
 import com.intellij.psi.PsiJavaFile;
@@ -63,11 +65,14 @@ public static void updateServiceStackReference(PsiFile psiFile) {
             return;
         }
 
+        Module module = ModuleUtil.findModuleForPsiElement(psiFile);
+        INativeTypesHandler nativeTypesHandler = IDEAUtils.getNativeTypesHandler(module,psiFile.getName());
+
         String existingPath = builder.getPath();
         if(existingPath == null || existingPath.equals("/")) {
-            builder.setPath("/types/java");
+            builder.setPath("/" + nativeTypesHandler.getRelativeTypesUrl());
         } else {
-            builder.setPath(existingPath + "/types/java");
+            builder.setPath(existingPath + "/" + nativeTypesHandler.getRelativeTypesUrl());
         }
 
         Map<String,String> options = new HashMap<String,String>();
@@ -131,7 +136,7 @@ public static void updateServiceStackReference(PsiFile psiFile) {
         }
     }
 
-    public static boolean containsOptionsHeader(PsiJavaFile psiJavaFile) {
+    public static boolean containsOptionsHeader(PsiFile psiJavaFile) {
         Document dtoDocument = FileDocumentManager.getInstance().getDocument(psiJavaFile.getVirtualFile());
         if(dtoDocument == null) {
             return false;