// This is code from a real life usage of full-moon

use full_moon::{self, ast::*, node::Node, tokenizer::TokenKind, visitors::Visitor};

const CODE: &str = r#"

local Module = {}
Module.__index = Module

--[[
    Creates a new instance of Module.

    ```lua
        Module.new()
    ```

    @static
    @param name string - This is the name for this Module.
    @returns Module - Returns the new Module!
]]
function Module.new(name)

end

"#;

struct MemberVisitor {
    comments: Vec<String>,
}

impl Visitor for MemberVisitor {
    fn visit_function_declaration(&mut self, function: &FunctionDeclaration) {
        let (tokens, _) = function.surrounding_trivia();
        let mut tokens = tokens;
        tokens.retain(|t| t.token_kind() == TokenKind::MultiLineComment);
        self.comments.extend(
            tokens
                .into_iter()
                .map(|t| t.to_string())
                .collect::<Vec<String>>(),
        )
    }
}

fn generate() -> Result<(), Vec<full_moon::Error>> {
    let ast = full_moon::parse(CODE)?;

    let mut visitor = MemberVisitor {
        comments: Vec::new(),
    };

    visitor.visit_ast(&ast);

    println!("{:?}", visitor.comments);

    Ok(())
}

fn main() {
    generate().expect("Oh");
}