Fix "un-rolling" a list type.

IntersectMBO · Jun 3, 2024 · c892262 · c892262
1 parent f34e999
commit c892262
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/plutus-tx/src/PlutusTx/Blueprint/Definition/Unroll.hs b/plutus-tx/src/PlutusTx/Blueprint/Definition/Unroll.hs
@@ -91,6 +91,7 @@ type family Unroll (p :: Type) :: [Type] where
   Unroll BuiltinString = '[BuiltinString]
   Unroll (BuiltinList a) = Prepend (BuiltinList a) (GUnroll (Rep a))
   Unroll BuiltinByteString = '[BuiltinByteString]
+  Unroll [a] = Unroll a
   Unroll p = Prepend p (GUnroll (Break (NoGeneric p) (Rep p)))
 
 -- | Detect stuck type family: https://blog.csongor.co.uk/report-stuck-families/#custom-type-errors

diff --git a/plutus-tx/test/Blueprint/Spec.hs b/plutus-tx/test/Blueprint/Spec.hs
@@ -71,9 +71,18 @@ $( PlutusTx.asData
 testUnrollNop :: Unroll Nop :~: '[Nop]
 testUnrollNop = Refl
 
+testUnrollListNop :: Unroll [Nop] :~: '[Nop]
+testUnrollListNop = Refl
+
+testUnrollListsNop :: Unroll [[[Nop]]] :~: '[Nop]
+testUnrollListsNop = Refl
+
 testUnrollBaz :: Unroll Baz :~: [Baz, Integer]
 testUnrollBaz = Refl
 
+testUnrollListBaz :: Unroll [Baz] :~: [Baz, Integer]
+testUnrollListBaz = Refl
+
 testUnrollZap :: Unroll Zap :~: [Zap, Nop, Integer, Bool]
 testUnrollZap = Refl