增加了一项可以添加Keep注解的功能。 (#177)

* 增加了一项可以添加Keep注解的功能。

* Add unit test of keep annotation

Author: smilechenbiao

src/main/kotlin/extensions/ExtensionsCollector.kt

@@ -2,7 +2,7 @@ package extensions
 
 import extensions.wu.seal.PropertyPrefixSupport
 import extensions.wu.seal.PropertySuffixSupport
-
+import extensions.chen.biao.KeepAnnotationSupport
 /**
  * extension collect, all extensions will be hold by this class's extensions property
  */
@@ -12,6 +12,7 @@ object ExtensionsCollector {
      */
     val extensions = listOf(
             PropertyPrefixSupport,
-            PropertySuffixSupport
+            PropertySuffixSupport,
+            KeepAnnotationSupport
     )
 }

src/main/kotlin/extensions/chen/biao/KeepAnnotationSupport.kt

@@ -0,0 +1,61 @@
+package extensions.chen.biao
+
+import com.intellij.ui.layout.panel
+import extensions.Extension
+import wu.seal.jsontokotlin.classscodestruct.Annotation
+import wu.seal.jsontokotlin.classscodestruct.KotlinDataClass
+import javax.swing.JCheckBox
+import javax.swing.JPanel
+
+/**
+ * @author chenbiao
+ * create at 2019/5/16
+ * description:
+ */
+object KeepAnnotationSupport : Extension() {
+
+
+    val configKey = "chen.biao.add_keep_annotation_enable"
+
+    override fun createUI(): JPanel {
+
+        val checkBox = JCheckBox("Enable Add a Keep Annotation To The Class").apply {
+            isSelected = getConfig(configKey).toBoolean()
+            addActionListener {
+                setConfig(configKey, isSelected.toString())
+            }
+        }
+
+        return panel {
+            row {
+                checkBox()
+            }
+        }
+    }
+
+    override fun intercept(kotlinDataClass: KotlinDataClass): KotlinDataClass {
+
+        return if (getConfig(configKey).toBoolean()) {
+
+            val classAnnotationString = "@Keep"
+
+            val classAnnotation = Annotation.fromAnnotationString(classAnnotationString)
+
+            return kotlinDataClass.copy(annotations = listOf(classAnnotation))
+        } else {
+            kotlinDataClass
+        }
+
+    }
+
+    override fun intercept(originClassImportDeclaration: String): String {
+
+        val classAnnotationImportClassString = "import android.support.annotation.Keep"
+
+        return if (getConfig(configKey).toBoolean()) {
+            originClassImportDeclaration.append(classAnnotationImportClassString)
+        } else {
+            originClassImportDeclaration
+        }
+    }
+}

src/test/kotlin/extensions/chen.biao/KeepAnnotationSupportTest.kt

@@ -0,0 +1,39 @@
+package extensions.chen.biao
+
+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 chenbiao
+ * create at 2019/5/17
+ * description:
+ */
+class KeepAnnotationSupportTest {
+
+    val json = """{"name":"chenbiao"}"""
+
+
+    var expectResult = """@Keep
+data class Test(
+    val name: String // chenbiao
+)"""
+
+    @Before
+    fun setUp() {
+        TestConfig.setToTestInitState()
+    }
+
+    @Test
+    fun interceptTest(){
+        val kotlinDataClass = json.generateKotlinDataClass()
+        KeepAnnotationSupport.getTestHelper().setConfig("chen.biao.add_keep_annotation_enable","true")
+        val generatedCode = kotlinDataClass.applyInterceptors(listOf(KeepAnnotationSupport)).getCode()
+        print(generatedCode)
+        generatedCode.trimMargin().should.equal(expectResult.trimMargin())
+
+    }
+
+}