commit 960cd4d635b98fc5daeeafee8b0a5601d45c70ad
author Wouter van Oortmerssen <aardappel@gmail.com> Fri Mar 08 21:31:55 2024 -0800
committer Wouter van Oortmerssen <aardappel@gmail.com> Fri Mar 08 21:31:55 2024 -0800
tree 010518da691022767cee6b3d5375232e14aab4ae
parent 6ff9e90e7e399f3977e99a315856b57c8afe5b4d

Lobster: Support required fields

src/idl_gen_lobster.cpp

diff --git a/src/idl_gen_lobster.cpp b/src/idl_gen_lobster.cpp
index c89e7bb..de4e152 100644
--- a/src/idl_gen_lobster.cpp
+++ b/src/idl_gen_lobster.cpp
@@ -145,11 +145,17 @@
           code += def + "() -> " + name + ":\n        ";
           code += "return " + name + "{ buf_, pos_ + " + offsets + " }\n";
         } else {
-          code += def + "() -> " + name + "?:\n        ";
+          code += def + "() -> " + name;
+          if (!field.IsRequired()) code += "?";
+          code += ":\n        ";
           code += std::string("let o = flatbuffers.field_") +
                   (field.value.type.struct_def->fixed ? "struct" : "table") +
-                  "(buf_, pos_, " + offsets + ")\n        return if o: " + name +
-                  " { buf_, o } else: nil\n";
+                  "(buf_, pos_, " + offsets + ")\n        return ";
+          if (field.IsRequired()) {
+            code += name + " { buf_, assert o }\n";
+          } else {
+            code += "if o: " + name + " { buf_, o } else: nil\n";
+          }
         }
         break;
       }