Merge pull request #207 from hanshengjian/3.4.0

Enable Parcelable Support

Author: wuseal

src/main/kotlin/extensions/ExtensionsCollector.kt

@@ -5,6 +5,7 @@ import extensions.ted.zeng.PropertyAnnotationLineSupport
 import extensions.wu.seal.ClassNameSuffixSupport
 import extensions.wu.seal.PropertyPrefixSupport
 import extensions.wu.seal.PropertySuffixSupport
+import extensions.jose.han.ParcelableAnnotationSupport
 
 /**
  * extension collect, all extensions will be hold by this class's extensions property
@@ -18,6 +19,7 @@ object ExtensionsCollector {
         PropertySuffixSupport,
         KeepAnnotationSupport,
         PropertyAnnotationLineSupport,
-        ClassNameSuffixSupport
+        ClassNameSuffixSupport,
+        ParcelableAnnotationSupport
     )
 }

src/main/kotlin/extensions/jose/han/ParcelableAnnotationSupport.kt

@@ -0,0 +1,79 @@
+package extensions.jose.han;
+
+import com.intellij.ui.layout.panel
+import com.intellij.util.ui.JBDimension
+import extensions.Extension
+import extensions.wu.seal.PropertySuffixSupport
+import wu.seal.jsontokotlin.classscodestruct.Annotation
+import wu.seal.jsontokotlin.classscodestruct.KotlinDataClass
+import java.awt.Color
+import java.awt.event.FocusEvent
+import java.awt.event.FocusListener
+import javax.swing.JCheckBox
+import javax.swing.JPanel
+import javax.swing.JTextArea
+import javax.swing.JTextField
+
+/**
+ *  @author jose.han
+ *  @Date 2019/7/27
+ */
+object ParcelableAnnotationSupport : Extension() {
+
+    val configKey = "jose.han.add_parcelable_annotatioin_enable"
+
+    override fun createUI(): JPanel {
+        val labelJField = JTextArea().apply {
+            text =  " android {\n" +
+                    "  ...\n" +
+                    "   androidExtensions {\n" +
+                    "    experimental = true\n" +
+                    "  }\n" +
+                    " }"
+            isEnabled = false
+            minimumSize = JBDimension(200, 200)
+            background = Color.BLACK
+        }
+
+
+        val checkBox = JCheckBox("Enable Parcelable Support").apply {
+            isSelected = ParcelableAnnotationSupport.getConfig(ParcelableAnnotationSupport.configKey).toBoolean()
+            addActionListener {
+                ParcelableAnnotationSupport.setConfig(ParcelableAnnotationSupport.configKey, isSelected.toString())
+            }
+        }
+
+        return panel {
+            row {
+                checkBox()
+                labelJField()
+            }
+        }
+    }
+
+    override fun intercept(kotlinDataClass: KotlinDataClass): KotlinDataClass {
+        return if (ParcelableAnnotationSupport.getConfig(ParcelableAnnotationSupport.configKey).toBoolean()) {
+
+            val classAnnotationString1 = "@SuppressLint(\"ParcelCreator\")"
+            val classAnnotationString2 = "@Parcelize"
+
+            val classAnnotation1 = Annotation.fromAnnotationString(classAnnotationString1)
+            val classAnnotation2 = Annotation.fromAnnotationString(classAnnotationString2)
+
+            return kotlinDataClass.copy(annotations = listOf(classAnnotation1,classAnnotation2),parentClassTemplate = "Parcelable")
+        } else {
+            kotlinDataClass
+        }
+    }
+
+    override fun intercept(originClassImportDeclaration: String): String {
+
+        val classAnnotationImportClassString = "import kotlinx.android.parcel.Parcelize".append("import android.os.Parcelable")
+
+        return if (ParcelableAnnotationSupport.getConfig(ParcelableAnnotationSupport.configKey).toBoolean()) {
+            originClassImportDeclaration.append(classAnnotationImportClassString)
+        } else {
+            originClassImportDeclaration
+        }
+    }
+}

src/test/kotlin/extensions/jose/han/ParcelableAnnotationSupportTest.kt

@@ -0,0 +1,43 @@
+package extensions.jose.han
+
+import com.winterbe.expekt.should
+import org.junit.Before
+import org.junit.Test
+import wu.seal.jsontokotlin.generateKotlinDataClass
+import wu.seal.jsontokotlin.test.TestConfig
+
+/**
+ * @author jose.han
+ * @Date 2019/04/27
+ */
+class ParcelableAnnotationSupportTest{
+    val json = """{"name":"jose.han","age":18,"height":18.7, "face":true}"""
+
+
+    var expectResult = """@SuppressLint("ParcelCreator")
+@Parcelize
+data class Test(
+    val name: String, // jose.han
+    val age: Int, // 18
+    val height: Double, // 18.7
+    val face: Boolean // true
+) : Parcelable
+"""
+
+    @Before
+    fun setUp() {
+        TestConfig.setToTestInitState()
+    }
+
+    @Test
+    fun interceptTest(){
+        val kotlinDataClass = json.generateKotlinDataClass()
+        ParcelableAnnotationSupport.getTestHelper().setConfig("jose.han.add_parcelable_annotatioin_enable","true")
+        val generatedCode = kotlinDataClass.applyInterceptors(listOf(ParcelableAnnotationSupport)).getCode()
+        print(generatedCode)
+        generatedCode.trimMargin().should.equal(expectResult.trimMargin())
+
+    }
+
+
+}