Don't add canEqual and equals if users already defined them

alexarchambault · Jan 16, 2025 · d5c4241 · d5c4241
1 parent 8bdb2bb
commit d5c4241
Showing 1 changed file with 30 additions and 26 deletions.
diff --git a/src/main/scala/dataclass/Macros.scala b/src/main/scala/dataclass/Macros.scala
@@ -124,33 +124,37 @@ private[dataclass] class Macros(val c: Context) extends ImplTransformers {
               tq"({type L[..$tparams0]=$WildcardType})#L"
           }
 
-          val canEqualMethod = {
-            val hashCheck =
-              if (cachedHashCode) q"obj.hashCode == hashCode" else q"true"
-            Seq(
-              q"""
-                override def canEqual(obj: Any): _root_.scala.Boolean =
-                  obj != null && obj.isInstanceOf[$tpname[..$wildcardedTparams]] && $hashCheck
-               """
-            )
-          }
+          val canEqualMethod =
+            if (hasCanEqual) Nil
+            else {
+              val hashCheck =
+                if (cachedHashCode) q"obj.hashCode == hashCode" else q"true"
+              Seq(
+                q"""
+                  override def canEqual(obj: Any): _root_.scala.Boolean =
+                    obj != null && obj.isInstanceOf[$tpname[..$wildcardedTparams]] && $hashCheck
+                 """
+              )
+            }
 
-          val equalsMethod = {
-            val fldChecks = paramss.flatten
-              .map { param =>
-                q"this.${param.name} == other.${param.name}"
-              }
-              .foldLeft[Tree](q"true")((a, b) => q"$a && $b")
-            Seq(
-              q"""
-                override def equals(obj: Any): _root_.scala.Boolean =
-                  this.eq(obj.asInstanceOf[AnyRef]) || canEqual(obj) && {
-                    val other = obj.asInstanceOf[$tpname[..$wildcardedTparams]]
-                    $fldChecks
-                  }
-               """
-            )
-          }
+          val equalsMethod =
+            if (hasEquals) Nil
+            else {
+              val fldChecks = paramss.flatten
+                .map { param =>
+                  q"this.${param.name} == other.${param.name}"
+                }
+                .foldLeft[Tree](q"true")((a, b) => q"$a && $b")
+              Seq(
+                q"""
+                  override def equals(obj: Any): _root_.scala.Boolean =
+                    this.eq(obj.asInstanceOf[AnyRef]) || canEqual(obj) && {
+                      val other = obj.asInstanceOf[$tpname[..$wildcardedTparams]]
+                      $fldChecks
+                    }
+                 """
+              )
+            }
 
           val hashCodeMethod =
             if (hasHashCode) Nil