Merge pull request #252 from wuseal/feature#84

fix #84

Author: wuseal

src/main/kotlin/extensions/ExtensionsCollector.kt

@@ -5,6 +5,7 @@ import extensions.wu.seal.KeepAnnotationSupportForAndroidX
 import extensions.jose.han.ParcelableAnnotationSupport
 import extensions.ted.zeng.PropertyAnnotationLineSupport
 import extensions.wu.seal.ClassNameSuffixSupport
+import extensions.wu.seal.ForceInitDefaultValueWithOriginJsonValueSupport
 import extensions.wu.seal.PropertyPrefixSupport
 import extensions.wu.seal.PropertySuffixSupport
 import extensions.xu.rui.PrimitiveTypeNonNullableSupport
@@ -24,6 +25,7 @@ object ExtensionsCollector {
         PropertyPrefixSupport,
         PropertySuffixSupport,
         ClassNameSuffixSupport,
-        PrimitiveTypeNonNullableSupport
+        PrimitiveTypeNonNullableSupport,
+        ForceInitDefaultValueWithOriginJsonValueSupport
     )
 }

src/main/kotlin/extensions/wu/seal/ForceInitDefaultValueWithOriginJsonValueSupport.kt

@@ -0,0 +1,49 @@
+package extensions.wu.seal
+
+import extensions.Extension
+import wu.seal.jsontokotlin.classscodestruct.KotlinDataClass
+import wu.seal.jsontokotlin.codeelements.getDefaultValue
+import wu.seal.jsontokotlin.ui.checkBox
+import wu.seal.jsontokotlin.ui.horizontalLinearLayout
+import wu.seal.jsontokotlin.utils.TYPE_STRING
+import javax.swing.JPanel
+
+/**
+ * This extension make the default value of data class property to be json value
+ * This extension should be put at last
+ * Created by Seal.Wu on 2019-11-09
+ */
+object ForceInitDefaultValueWithOriginJsonValueSupport : Extension() {
+
+    private const val configKey = "wu.seal.force_init_default_value_with_origin_json_value"
+
+    override fun createUI(): JPanel {
+        return horizontalLinearLayout {
+            (checkBox(
+                "Force init Default Value With Origin Json Value",
+                getConfig(configKey).toBoolean()
+            ) { isSelectedAfterClick ->
+                setConfig(configKey, isSelectedAfterClick.toString())
+            })()
+            fillSpace()
+        }
+    }
+
+    override fun intercept(kotlinDataClass: KotlinDataClass): KotlinDataClass {
+        return if (getConfig(configKey).toBoolean()) {
+            val newP = kotlinDataClass.properties.map {
+                val newV = if (it.originJsonValue.isNullOrBlank()) getDefaultValue(it.type) else {
+                    if (it.type == TYPE_STRING) {
+                        """"${it.originJsonValue}""""
+                    } else {
+                        it.originJsonValue
+                    }
+                }
+                it.copy(value = newV)
+            }
+            kotlinDataClass.copy(properties = newP)
+        } else {
+            kotlinDataClass
+        }
+    }
+}

src/test/kotlin/extensions/wu/seal/ForceInitDefaultValueWithOriginJsonValueSupportTest.kt

@@ -0,0 +1,32 @@
+package extensions.wu.seal
+
+import com.winterbe.expekt.should
+import org.junit.Before
+import org.junit.Test
+import wu.seal.jsontokotlin.applyInterceptor
+import wu.seal.jsontokotlin.generateKotlinDataClass
+import wu.seal.jsontokotlin.test.TestConfig
+
+class ForceInitDefaultValueWithOriginJsonValueSupportTest {
+    val json = """{"a":1}"""
+
+    val expectResult = """data class Test(
+    val a: Int = 1 // 1
+)"""
+    private  val configKey = "wu.seal.force_init_default_value_with_origin_json_value"
+
+
+    @Before
+    fun setUp() {
+        TestConfig.setToTestInitState()
+    }
+
+    @Test
+    fun intercept() {
+        ForceInitDefaultValueWithOriginJsonValueSupport.getTestHelper().setConfig(configKey,true.toString())
+        val dataClass =
+            json.generateKotlinDataClass("Test").applyInterceptor(ForceInitDefaultValueWithOriginJsonValueSupport)
+        val resultCode = dataClass.getCode()
+        resultCode.should.be.equal(expectResult)
+    }
+}