data class generation added

RomanTcv · Dec 13, 2020 · 7af95fb · 7af95fb
1 parent eadb673
commit 7af95fb
Show file tree

Hide file tree

Showing 4 changed files with 31 additions and 8 deletions.
diff --git a/.idea/codeStyles/Project.xml b/.idea/codeStyles/Project.xml
diff --git a/annotations/src/main/java/com/r12/mutable/annotations/Mutable.kt b/annotations/src/main/java/com/r12/mutable/annotations/Mutable.kt
@@ -5,7 +5,12 @@ package com.r12.mutable.annotations
  *
  *  className - Your own custom class name
  *  mutableSuffix - If className is blank will generate default name, for instance - MutableInspection or InspectionMutable.
+ *  dataClass - generate data class or not
  */
 @Retention(AnnotationRetention.SOURCE)
 @Target(allowedTargets = [AnnotationTarget.CLASS])
-annotation class Mutable(val className: String = "", val mutableSuffix: Boolean = true)
+annotation class Mutable(
+    val className: String = "",
+    val mutableSuffix: Boolean = true,
+    val dataClass: Boolean = false
+)
diff --git a/processor/src/main/java/com/r12/mutable/processor/extensions/ElementsExt.kt b/processor/src/main/java/com/r12/mutable/processor/extensions/ElementsExt.kt
@@ -18,4 +18,6 @@ internal fun Element.fileName(): String {
 
 internal fun Element.getPackage(elements: Elements): String = elements.packageOf(this)
 
-internal fun Elements.packageOf(element: Element) = getPackageOf(element).toString()
+internal fun Elements.packageOf(element: Element) = getPackageOf(element).toString()
+
+internal fun Element.isDataClass(): Boolean = getAnnotation(Mutable::class.java).dataClass
diff --git a/processor/src/main/java/com/r12/mutable/processor/factories/InterfaceGeneratorFactory.kt b/processor/src/main/java/com/r12/mutable/processor/factories/InterfaceGeneratorFactory.kt
@@ -4,6 +4,7 @@ import com.r12.mutable.processor.entities.Field
 import com.r12.mutable.processor.extensions.fileName
 import com.r12.mutable.processor.extensions.formatType
 import com.r12.mutable.processor.extensions.getPackage
+import com.r12.mutable.processor.extensions.isDataClass
 import com.squareup.kotlinpoet.*
 import com.squareup.kotlinpoet.metadata.ImmutableKmClass
 import com.squareup.kotlinpoet.metadata.KotlinPoetMetadataPreview
@@ -23,13 +24,14 @@ internal class InterfaceGeneratorFactory(
 
         val fields = element.getFields()
         val isAbstract = element.isAbstract()
+        val isDataClass = !isAbstract && element.isDataClass()
 
         val typeSpec = TypeSpec
             .classBuilder(fileName)
             .addSuperinterface(kmClass.superInterface())
             .addProperties(fields.map(Field::toPropertySpec))
             .primaryConstructor(fields.generateConstructor())
-            .addModifiers(kmClass.getModifiers(isAbstract))
+            .addModifiers(kmClass.getModifiers(isAbstract, isDataClass))
             .addExtension(element, fullPath, isAbstract)
             .build()
 
@@ -87,10 +89,15 @@ internal class InterfaceGeneratorFactory(
         return this
     }
 
-    private fun ImmutableKmClass.getModifiers(isAbstract: Boolean): List<KModifier> {
+    private fun ImmutableKmClass.getModifiers(isAbstract: Boolean, isDataClass: Boolean): List<KModifier> {
         val list = mutableListOf<KModifier>()
 
-        if (isAbstract) list.add(KModifier.ABSTRACT) else list.add(KModifier.OPEN)
+        val kModifier = if (isAbstract) {
+            KModifier.ABSTRACT
+        } else {
+            if (isDataClass) KModifier.DATA else KModifier.OPEN
+        }
+        list.add(kModifier)
         if (isInternal) list.add(KModifier.INTERNAL)
 
         return list