fix: ignore public/weak tokens in imports

parse.js

@@ -511,11 +511,20 @@ function onoptionMap (tokens) {
 
 function onimport (tokens) {
   tokens.shift()
-  const file = tokens.shift().replace(/^"+|"+$/gm, '')
+  let token = tokens.shift()
 
-  if (tokens[0] !== ';') throw new Error('Unexpected token: ' + tokens[0] + '. Expected ";"')
+  if (token === 'public' || token === 'weak') {
+    token = tokens.shift()
+  }
+  if (!/^".*"$/.test(token)) {
+    throw new Error('Unexpected import <' + token + '>. Expecting a string literal.')
+  }
+  const file = token.replace(/^"+|"+$/gm, '')
+
+  token = tokens.shift()
+
+  if (token !== ';') throw new Error('Unexpected token: ' + token + '. Expected ";"')
 
-  tokens.shift()
   return file
 }
 

test/fixtures/import.json

@@ -1,7 +1,12 @@
 {
   "syntax": 3,
   "package": null,
-  "imports": ["./result.proto", "./other_result.proto"],
+  "imports": [
+    "./result.proto",
+    "./other_result.proto",
+    "./public.proto",
+    "./weak.proto"
+  ],
   "enums": [],
   "extends": [],
   "messages": [

test/fixtures/import.proto

@@ -1,5 +1,7 @@
 import "./result.proto";
 import "./other_result.proto";
+import public "./public.proto";
+import weak "./weak.proto";
 
 message SearchResponse {
   repeated Result result = 1;

test/index.js

@@ -53,6 +53,13 @@ tape('schema with imports', function (t) {
   t.end()
 })
 
+tape('schema without quotes', function (t) {
+  t.throws(function () {
+    schema.parse('import foo;')
+  }, /Unexpected import <foo>. Expecting a string literal./)
+  t.end()
+})
+
 tape('schema with imports loaded by path', function (t) {
   t.same(schema.parse(fixture('search.proto')), require('./fixtures/search.json'))
   t.end()