Support case classes with private constructors

Author: kciesielski

src/main/scala/com/softwaremill/macros/customize/CustomizeImpl.scala

@@ -20,6 +20,9 @@ class CustomizeImpl(val c: whitebox.Context) {
     def extractCaseClassesParts(classDecl: ClassDef) = classDecl match {
       case q"""case class $className(..$fields) extends ..$parents { ..$body }""" =>
         (className, fields, parents, body)
+      case q"""case class $className private(..$fields) extends ..$parents { ..$body }""" =>
+        (className, fields, parents, body)
+      case _ => c.abort(c.enclosingPosition, "Unsupported case class type. Cannot rewrite toString().")
     }
 
     def extractNewToString(typeName: TypeName, allFields: List[Tree]) = {

src/test/scala/ToStringMaskTest.scala

@@ -1,5 +1,4 @@
 import java.time.ZonedDateTime
-import java.util.Date
 
 import com.softwaremill.macros.customize.{customize, mask}
 import org.scalatest.{FlatSpec, Matchers}
@@ -26,11 +25,26 @@ class ToStringMaskTest extends FlatSpec with Matchers {
     // then
     u.toString should be("User30(2,***)")
   }
+
+  it should "work on case classes with private constructors" in {
+    // given
+    val u = UserPrivate("some@email2.com")
+
+    // then
+    u.toString should be("UserPrivate(1,***)")
+  }
 }
 
 @customize
 case class User30(id: Int, @mask email: String)
 
+@customize
+case class UserPrivate private(id: Int, @mask email: String)
+
+object UserPrivate {
+  def apply(email: String) = new UserPrivate(1, email)
+}
+
 object User30 {
   def apply(id: Int) = new User30(id, "default@email.com")
 }