Support case classes with private constructors

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("[email protected]")
+
+ // 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, "[email protected]")
 }