fix: Accept imports after declarations (#2178)

dcodeIO · web-flow · commit 24651c9ce5b8 · 2026-04-28T01:28:35.000+02:00
diff --git a/src/parse.js b/src/parse.js
@@ -967,10 +967,6 @@ function parse(source, root, options) {
 
             case "import":
 
-                /* istanbul ignore if */
-                if (!head)
-                    throw illegal(token);
-
                 parseImport();
                 break;
 
diff --git a/tests/comp_import-after-decl.js b/tests/comp_import-after-decl.js
@@ -0,0 +1,14 @@
+var tape = require("tape");
+var protobuf = require("..");
+
+tape.test("imports after declarations", function(test) {
+    var parsed = protobuf.parse("syntax = \"proto3\";" +
+        "message Foo { string name = 1; }" +
+        "import \"bar.proto\";" +
+        "enum Status { UNKNOWN = 0; }" +
+        "import weak \"baz.proto\";");
+
+    test.deepEqual(parsed.imports, [ "bar.proto" ], "should parse imports after messages");
+    test.deepEqual(parsed.weakImports, [ "baz.proto" ], "should parse weak imports after enums");
+    test.end();
+});
