update

Signed-off-by: George Lemon <[email protected]>
openpeeps · Jun 6, 2024 · 90ce6f7 · 90ce6f7
1 parent f87fc79
commit 90ce6f7
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/src/voodoo/extensibles.nim b/src/voodoo/extensibles.nim
@@ -25,7 +25,11 @@ macro extendEnum*(x: untyped, fields: untyped) =
       add otherFields, f
     else:
       error("Voodoo - Invalid enum extension. Expects either `nnkAsgn`, or `nnkIdent`")
-  ExtendableEnums[$x] = otherFields
+  if ExtendableEnums.hasKey($x):
+    for xfield in otherFields:
+      add ExtendableEnums[$x], xfield
+  else:
+    ExtendableEnums[$x] = otherFields
 
 template extendCase*(fieldNode: untyped, branchesNode: untyped) =
   ## Extend an object variant by adding new branches.

diff --git a/tests/test2.nim b/tests/test2.nim
@@ -5,7 +5,13 @@ extendEnum Cardinal:
   south
   east
 
+extendEnum Cardinal:
+  unknown
+
 import ./extensibleModule
 
+assert compiles Cardinal.north
+assert compiles Cardinal.west
 assert compiles Cardinal.south
-assert compiles Cardinal.east
+assert compiles Cardinal.east
+assert compiles Cardinal.unknown