Fix bug #83

wuseal · Dec 23, 2018 · 3e69d41 · 3e69d41
1 parent 7933b4f
commit 3e69d41
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 4 deletions.
diff --git a/src/main/kotlin/wu/seal/jsontokotlin/ImportClassWriter.kt b/src/main/kotlin/wu/seal/jsontokotlin/ImportClassWriter.kt
@@ -78,11 +78,19 @@ object ImportClassWriter : IImportClassWriter {
 
             if (importClassLineString !in text) {
 
-                val packageIndex = text.indexOf("package ")
-                val importIndex = Math.max(text.lastIndexOf("import"), packageIndex)
+                val packageIndex =  try {
+                    "^[\\s]*package".toRegex(RegexOption.MULTILINE).find(text)!!.range.endInclusive
+                } catch (e: Exception) {
+                    -1
+                }
+                val lastImportKeywordIndex = try {
+                    "^[\\s]*import".toRegex(RegexOption.MULTILINE).findAll(text).last().range.endInclusive
+                } catch (e: Exception) {
+                    -1
+                }
+                val index = Math.max(lastImportKeywordIndex, packageIndex)
                 val insertIndex =
-                    if (importIndex == -1) 0 else editFile.getLineEndOffset(editFile.getLineNumber(importIndex))
-
+                    if (index == -1) 0 else editFile.getLineEndOffset(editFile.getLineNumber(index))
 
                 executeCouldRollBackAction(project) {
                     editFile.insertString(insertIndex, "\n" + importClassLineString + "\n")

diff --git a/src/main/kotlin/wu/seal/jsontokotlin/MakeKotlinClassAction.kt b/src/main/kotlin/wu/seal/jsontokotlin/MakeKotlinClassAction.kt
@@ -16,6 +16,7 @@ import wu.seal.jsontokotlin.feedback.sendActionInfo
 import wu.seal.jsontokotlin.ui.JsonInputDialog
 import wu.seal.jsontokotlin.utils.ClassCodeFilter
 import wu.seal.jsontokotlin.utils.executeCouldRollBackAction
+import java.awt.SystemColor.text
 import java.net.URL
 import java.util.*
 
@@ -144,6 +145,23 @@ class MakeKotlinClassAction : AnAction("MakeKotlinClass") {
                 if (offset == 0) {
                     offset = document.textLength
                 }
+                val lastPackageKeywordLineEndIndex = try {
+                    "^[\\s]*package.+\n$".toRegex(RegexOption.MULTILINE).findAll(document.text).last().range.endInclusive
+                } catch (e: Exception) {
+                    -1
+                }
+                val lastImportKeywordLineEndIndex = try {
+                    "^[\\s]*import.+\n$".toRegex(RegexOption.MULTILINE).findAll(document.text).last().range.endInclusive
+                } catch (e: Exception) {
+                    -1
+                }
+                if (offset < lastPackageKeywordLineEndIndex) {
+                    offset = lastPackageKeywordLineEndIndex + 1
+                }
+                if (offset < lastImportKeywordLineEndIndex) {
+                    offset = lastImportKeywordLineEndIndex + 1
+                }
+
             } else {
                 offset = document.textLength
             }