use super::CxxQtItem;
use proc_macro2::TokenStream;
use quote::{ToTokens, TokenStreamExt};
use syn::parse::{Parse, ParseStream};
use syn::{AttrStyle, Attribute, Result};
pub struct CxxQtFile {
pub attrs: Vec<Attribute>,
pub items: Vec<CxxQtItem>,
}
impl Parse for CxxQtFile {
fn parse(input: ParseStream) -> Result<Self> {
Ok(CxxQtFile {
attrs: input.call(Attribute::parse_inner)?,
items: {
let mut items = Vec::new();
while !input.is_empty() {
items.push(input.parse()?);
}
items
},
})
}
}
impl ToTokens for CxxQtFile {
fn to_tokens(&self, tokens: &mut TokenStream) {
tokens.append_all(
self.attrs
.iter()
.filter(|attr| matches!(attr.style, AttrStyle::Inner(_))),
);
tokens.append_all(&self.items);
}
}
pub fn parse_qt_file(path: impl AsRef<std::path::Path>) -> Result<CxxQtFile> {
let source = std::fs::read_to_string(path.as_ref()).expect("Could not read path {} to string");
if source.starts_with("#!") && !source.starts_with("#![") {
let shebang_end = source.find('\n').unwrap_or(source.len());
syn::parse_str(&source[shebang_end..])
} else {
syn::parse_str(&source)
}
}