asterinas/kernel/libs/comp-sys/component-macro/src/init_comp.rs

// SPDX-License-Identifier: MPL-2.0

use proc_macro2::{TokenStream, TokenTree};
use quote::{ToTokens, TokenStreamExt};
use syn::parse::Parse;

/// The content inside typeflag macro
pub struct ComponentInitFunction {
    function: Vec<TokenTree>,
    pub function_name: TokenTree,
}

impl Parse for ComponentInitFunction {
    fn parse(input: syn::parse::ParseStream) -> syn::Result<Self> {
        let mut vec: Vec<TokenTree> = Vec::new();
        vec.push(input.parse().unwrap());
        let function_name: TokenTree = input.parse().unwrap();
        vec.push(function_name.clone());
        while !input.is_empty() {
            vec.push(input.parse().unwrap())
        }
        Ok(ComponentInitFunction {
            function: vec,
            function_name,
        })
    }
}

impl ToTokens for ComponentInitFunction {
    fn to_tokens(&self, tokens: &mut TokenStream) {
        for token in &self.function {
            tokens.append(token.clone());
        }
    }
}