use proc_macro::TokenStream;
use quote::quote;
use std::{
fs::File,
io::{BufRead, BufReader},
};
use syn::{parse_macro_input, LitStr};
#[proc_macro]
pub fn include_lines(input: TokenStream) -> TokenStream {
let lines = get_lines(parse_macro_input!(input as LitStr).value());
TokenStream::from(quote! {
[#(#lines),*]
})
}
#[proc_macro]
pub fn include_lines_s(input: TokenStream) -> TokenStream {
let lines = get_lines(parse_macro_input!(input as LitStr).value());
TokenStream::from(quote! {
[#(String::from(#lines)),*]
})
}
#[proc_macro]
pub fn count_lines(input: TokenStream) -> TokenStream {
let num_lines = get_lines(parse_macro_input!(input as LitStr).value()).count();
TokenStream::from(quote! {
#num_lines
})
}
fn get_lines(filename: String) -> impl Iterator<Item = String> {
let file = File::open(filename).expect("Error opening the file: ");
BufReader::new(file).lines().filter_map(|l| l.ok())
}