Use enums

writeonly · May 7, 2024 · 82a91a6 · 82a91a6
1 parent 36faf3a
commit 82a91a6
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 13 deletions.
diff --git a/catculator-core/src/main/scala/pl/writeonly/catculator/core/adt/tree/BinaryTree.scala b/catculator-core/src/main/scala/pl/writeonly/catculator/core/adt/tree/BinaryTree.scala
@@ -1,11 +1,9 @@
 package pl.writeonly.catculator.core.adt.tree
 
-object BinaryTree {
-
-  case class Leaf[A](leaf: A) extends BinaryTree[A]
+enum BinaryTree[+A]:
+  case Leaf(leaf: A)
+  case Node(first: BinaryTree[A], follow: BinaryTree[A])
 
-  case class Node[A](first: BinaryTree[A], follow: BinaryTree[A])
-      extends BinaryTree[A]
+object BinaryTree {
 }
-
-sealed trait BinaryTree[+A]
+q
diff --git a/catculator-core/src/main/scala/pl/writeonly/catculator/core/adt/tree/Tree.scala b/catculator-core/src/main/scala/pl/writeonly/catculator/core/adt/tree/Tree.scala
@@ -2,13 +2,11 @@ package pl.writeonly.catculator.core.adt.tree
 
 import cats.data.NonEmptyList
 
+enum Tree[A]:
+  case Leaf(leaf: A)
+  case Node(children: NonEmptyList[Tree[A]])
+
 object Tree {
   def node[A](head: Tree[A], tail: Tree[A]*): Tree[A] =
     Node(NonEmptyList(head, tail.toList))
-
-  case class Leaf[A](leaf: A) extends Tree[A]
-
-  case class Node[A](children: NonEmptyList[Tree[A]]) extends Tree[A]
 }
-
-sealed trait Tree[+A]