Data structure for source file

Author: dtolnay

gen/src/file.rs

@@ -0,0 +1,20 @@
+use syn::parse::{Parse, ParseStream, Result};
+use syn::{Attribute, Item};
+
+pub struct File {
+    pub attrs: Vec<Attribute>,
+    pub items: Vec<Item>,
+}
+
+impl Parse for File {
+    fn parse(input: ParseStream) -> Result<Self> {
+        let attrs = input.call(Attribute::parse_inner)?;
+
+        let mut items = Vec::new();
+        while !input.is_empty() {
+            items.push(input.parse()?);
+        }
+
+        Ok(File { attrs, items })
+    }
+}

gen/src/find.rs

@@ -1,6 +1,6 @@
-use crate::gen::{Error, Input, Result};
+use crate::gen::{Error, File, Input, Result};
 use crate::syntax::namespace::Namespace;
-use syn::{Attribute, File, Item};
+use syn::{Attribute, Item};
 
 pub(super) fn find_bridge_mod(syntax: File) -> Result<Input> {
     match scan(syntax.items)? {

gen/src/mod.rs

@@ -2,6 +2,7 @@
 // the cxxbridge CLI command.
 
 mod error;
+mod file;
 mod find;
 pub(super) mod include;
 pub(super) mod out;
@@ -11,6 +12,7 @@ mod write;
 mod tests;
 
 use self::error::{format_err, Error, Result};
+use self::file::File;
 use crate::syntax::namespace::Namespace;
 use crate::syntax::report::Errors;
 use crate::syntax::{self, check, Types};
@@ -56,7 +58,7 @@ fn generate_from_path(path: &Path, opt: Opt, header: bool) -> Vec<u8> {
 fn generate(source: &str, opt: Opt, header: bool) -> Result<Vec<u8>> {
     proc_macro2::fallback::force();
     let ref mut errors = Errors::new();
-    let syntax = syn::parse_file(&source)?;
+    let syntax: File = syn::parse_str(source)?;
     let bridge = find::find_bridge_mod(syntax)?;
     let ref namespace = bridge.namespace;
     let ref apis = syntax::parse_items(errors, bridge.module);