mod api;
mod data;
mod generator;

use dotenv::dotenv;

use proc_macro::TokenStream;
use syn::{ItemFn, __private::ToTokens};

use crate::generator::generate_body_function_from_head;

#[proc_macro]
pub fn implement(_item: TokenStream) -> TokenStream {
    // TODO: Evaluate the use of dotenv in this crate
    dotenv().ok();

    let implemented_fn = generate_body_function_from_head(_item.to_string()).unwrap();

    println!("{}", implemented_fn);

    implemented_fn.parse().unwrap()
}

#[proc_macro_attribute]
pub fn auto_implement(_args: TokenStream, input: TokenStream) -> TokenStream {
    let ast: ItemFn = syn::parse(input).expect("Failed to parse input as a function");

    // Search for the information within the attributes.

    let mut prompt_input = String::new();

    let fn_header = ast.sig.to_token_stream().to_string();

    // println!("Function header: {}", fn_header);

    for attr in ast.attrs {
        let data = attr.to_token_stream().to_string();

        prompt_input.push_str(&data);
        prompt_input.push('\n');
    }

    prompt_input.push_str(&fn_header);

    // println!("Information extracted: {:?}", prompt_input);

    dotenv().ok();

    let implemented_fn = generate_body_function_from_head(prompt_input).unwrap();

    // println!("\n{}\n", implemented_fn);

    // #[allow(long_running_const_eval)]
    // loop {
    // let mut line = String::new();
    // let b1 = std::io::stdin().read_line(&mut line).unwrap();
    // println!("no of bytes read , {}", b1);
    // }

    // static line = String::new();
    // println!("Enter your name :");
    //
    // println!("Hello , {}", line);
    // println!("no of bytes read , {}", b1);

    implemented_fn.parse().unwrap()
}