Serialize Patio compilation unit to Protobuf

1 files changed, 13 insertions, 0 deletions

diff --git a/src/main/java/org/shedlang/compiler/patio/protobuf/PatioToProtobuf.java b/src/main/java/org/shedlang/compiler/patio/protobuf/PatioToProtobuf.java
index 6a849beb..334c998a 100644
--- a/src/main/java/org/shedlang/compiler/patio/protobuf/PatioToProtobuf.java
+++ b/src/main/java/org/shedlang/compiler/patio/protobuf/PatioToProtobuf.java
@@ -30,6 +30,19 @@ public class PatioToProtobuf {
         return builder.build();
     }
 
+    public static PatioCompilationUnit deserializeCompilationUnit(
+        PatioProtobuf.PatioCompilationUnit compilationUnitProtobuf
+    ) {
+        var source = deserializeSource(compilationUnitProtobuf.getSource(), Optional.empty());
+
+        return new PatioCompilationUnit(
+            compilationUnitProtobuf.getElementsList().stream()
+                .map(elementProtobuf -> deserializeCompilationUnitElement(elementProtobuf, source))
+                .collect(Collectors.toList()),
+            source
+        );
+    }
+
     // == Compilation unit elements ==
 
     public static PatioProtobuf.PatioCompilationUnitElement serializeCompilationUnitElement(