naia-parse 0.13.0

//! Very limited rust parser
//!
//! https://doc.rust-lang.org/reference/expressions/struct-expr.html
//! https://docs.rs/syn/0.15.44/syn/enum.Type.html
//! https://ziglang.org/documentation/0.5.0/#toc-typeInfo

use proc_macro::{Delimiter, Group, TokenStream, TokenTree};

use std::iter::Peekable;

pub struct Attribute {
    pub name: String,
    pub tokens: Vec<String>,
}

#[allow(dead_code)]
pub enum Visibility {
    Public,
    Crate,
    Restricted,
    Private,
}

pub struct Field {
    pub vis: Visibility,
    pub field_name: Option<String>,
    pub ty: Type,
}

#[allow(dead_code)]
pub struct Type {
    pub is_option: bool,
    pub path: String,
}

pub struct Struct {
    pub name: String,
    pub tuple: bool,
    pub fields: Vec<Field>,
}

pub struct EnumVariant {
    pub name: String,
    pub tuple: bool,
    pub fields: Vec<Field>,
}

pub struct Enum {
    pub name: String,
    pub variants: Vec<EnumVariant>,
}

#[allow(dead_code)]
pub enum Data {
    Struct(Struct),
    Enum(Enum),
    Union(()),
}

pub fn next_visibility_modifier(
    source: &mut Peekable<impl Iterator<Item = TokenTree>>,
) -> Option<String> {
    if let Some(TokenTree::Ident(ident)) = source.peek() {
        if format!("{}", ident) == "pub" {
            source.next();

            // skip (crate) and alike
            if let Some(TokenTree::Group(group)) = source.peek() {
                if group.delimiter() == Delimiter::Parenthesis {
                    next_group(source);
                }
            }

            return Some("pub".to_string());
        }
    }

    None
}

pub fn next_punct(source: &mut Peekable<impl Iterator<Item = TokenTree>>) -> Option<String> {
    if let Some(TokenTree::Punct(punct)) = source.peek() {
        let punct = format!("{}", punct);
        source.next();
        return Some(punct);
    }

    None
}

pub fn next_exact_punct(
    source: &mut Peekable<impl Iterator<Item = TokenTree>>,
    pattern: &str,
) -> Option<String> {
    if let Some(TokenTree::Punct(punct)) = source.peek() {
        let punct = format!("{}", punct);
        if punct == pattern {
            source.next();
            return Some(punct);
        }
    }

    None
}

pub fn next_eof<T: Iterator>(source: &mut Peekable<T>) -> Option<()> {
    if source.peek().is_none() {
        Some(())
    } else {
        None
    }
}

pub fn next_ident(source: &mut Peekable<impl Iterator<Item = TokenTree>>) -> Option<String> {
    if let Some(TokenTree::Ident(ident)) = source.peek() {
        let ident = format!("{}", ident);
        source.next();
        Some(ident)
    } else {
        None
    }
}

pub fn next_group(source: &mut Peekable<impl Iterator<Item = TokenTree>>) -> Option<Group> {
    if let Some(TokenTree::Group(_)) = source.peek() {
        let group = match source.next().expect("expected to read another token") {
            TokenTree::Group(group) => group,
            _ => unreachable!("just checked with peek()!"),
        };
        Some(group)
    } else {
        None
    }
}

fn skip_doc_string(source: &mut Peekable<impl Iterator<Item = TokenTree>>) {
    while skip_doc_string_inner(source) {}
}

fn skip_doc_string_inner(source: &mut Peekable<impl Iterator<Item = TokenTree>>) -> bool {
    if next_exact_punct(source, "#").is_none() {
        return false;
    }
    if next_group(source).is_none() {
        return false;
    }
    true
}

#[allow(dead_code)]
pub fn debug_current_token(source: &mut Peekable<impl Iterator<Item = TokenTree>>) {
    println!("{:?}", source.peek());
}

fn next_type<T: Iterator<Item = TokenTree>>(source: &mut Peekable<T>) -> Option<Type> {
    let group = next_group(source);
    if let Some(group) = group {
        let mut tuple_type = Type {
            is_option: false,
            path: "".to_string(),
        };

        while let Some(next_ty) = next_type(&mut group.stream().into_iter().peekable()) {
            #[allow(clippy::format_push_string)]
            tuple_type.path.push_str(&format!("{}, ", next_ty.path));
        }

        return Some(tuple_type);
    }

    // read a path like a::b::c::d
    let mut ty = next_ident(source)?;
    while next_exact_punct(source, ":").is_some() {
        let _second_colon = next_exact_punct(source, ":").expect("Expecting second :");

        let next_ident = next_ident(source).expect("Expecting next path part after ::");
        #[allow(clippy::format_push_string)]
        ty.push_str(&format!("::{}", next_ident));
    }

    let angel_bracket = next_exact_punct(source, "<");
    if angel_bracket.is_some() {
        let mut generic_type = next_type(source).expect("Expecting generic argument");
        while let Some(_comma) = next_exact_punct(source, ",") {
            let next_ty = next_type(source).expect("Expecting generic argument");
            #[allow(clippy::format_push_string)]
            generic_type.path.push_str(&format!(", {}", next_ty.path));
        }

        let _closing_bracket =
            next_exact_punct(source, ">").expect("Expecting closing generic bracket");

        if ty == "Option" {
            Some(Type {
                path: generic_type.path,
                is_option: true,
            })
        } else {
            Some(Type {
                path: format!("{}<{}>", ty, generic_type.path),
                is_option: false,
            })
        }
    } else {
        Some(Type {
            path: ty,
            is_option: false,
        })
    }
}

fn next_fields(body: &mut Peekable<impl Iterator<Item = TokenTree>>, named: bool) -> Vec<Field> {
    let mut fields = vec![];

    loop {
        if next_eof(body).is_some() {
            break;
        }

        skip_doc_string(body);

        let _visibility = next_visibility_modifier(body);
        let field_name = if named {
            let field_name = next_ident(body).expect("Field name expected");

            let _ = next_exact_punct(body, ":").expect("Delimeter after field name expected");
            Some(field_name)
        } else {
            None
        };

        let ty = next_type(body).expect("Expected field type");
        let _punct = next_punct(body);

        fields.push(Field {
            vis: Visibility::Public,
            field_name,
            ty,
        });
    }
    fields
}

fn next_struct(source: &mut Peekable<impl Iterator<Item = TokenTree>>) -> Struct {
    let struct_name = next_ident(source).expect("Unnamed structs are not supported");

    let group = next_group(source);
    // unit struct
    if group.is_none() {};
    if let Some(group) = group {
        let delimiter = group.delimiter();
        let tuple = match delimiter {
            Delimiter::Parenthesis => true,
            Delimiter::Brace => false,

            _ => panic!("Struct with unsupported delimiter"),
        };

        let mut body = group.stream().into_iter().peekable();
        let fields = next_fields(&mut body, !tuple);

        if tuple {
            next_exact_punct(source, ";").expect("Expected ; on the end of tuple struct");
        }

        Struct {
            name: struct_name,
            tuple,
            fields,
        }
    } else {
        // skip ; at the end of struct like this: "struct Foo;"
        let _ = next_punct(source);

        Struct {
            name: struct_name,
            fields: vec![],
            tuple: true,
        }
    }
}

fn next_enum(source: &mut Peekable<impl Iterator<Item = TokenTree>>) -> Enum {
    let enum_name = next_ident(source).expect("Unnamed enums are not supported");

    let group = next_group(source);
    // unit enum
    if let Some(group) = group {
        let mut body = group.stream().into_iter().peekable();

        let mut variants = vec![];
        loop {
            if next_eof(&mut body).is_some() {
                break;
            }

            skip_doc_string(&mut body);

            let variant_name = next_ident(&mut body).expect("Unnamed variants are not supported");
            let group = next_group(&mut body);

            if let Some(group) = group {
                let delimiter = group.delimiter();
                let tuple = match delimiter {
                    Delimiter::Parenthesis => true,
                    Delimiter::Brace => false,

                    _ => panic!("Enum with unsupported delimiter"),
                };
                {
                    let mut body = group.stream().into_iter().peekable();
                    let fields = next_fields(&mut body, !tuple);
                    variants.push(EnumVariant {
                        name: variant_name,
                        tuple,
                        fields,
                    });
                }
                let _maybe_semicolon = next_exact_punct(&mut body, ";");
                let _maybe_coma = next_exact_punct(&mut body, ",");
            } else {
                variants.push(EnumVariant {
                    name: variant_name,
                    tuple: true,
                    fields: vec![],
                });
                let _maybe_comma = next_exact_punct(&mut body, ",");
                continue;
            }
        }

        Enum {
            name: enum_name,
            variants,
        }
    } else {
        Enum {
            name: enum_name,
            variants: vec![],
        }
    }
}

pub fn parse_data(input: TokenStream) -> Data {
    let mut source = input.into_iter().peekable();

    skip_doc_string(&mut source);

    let pub_or_type = next_ident(&mut source).expect("Not an ident");

    let type_keyword = if pub_or_type == "pub" {
        next_ident(&mut source).expect("pub(whatever) is not supported yet")
    } else {
        pub_or_type
    };

    let res = match type_keyword.as_str() {
        "struct" => Data::Struct(next_struct(&mut source)),
        "enum" => {
            let enum_ = next_enum(&mut source);
            Data::Enum(enum_)
        }
        "union" => unimplemented!("Unions are not supported"),
        unexpected => panic!("Unexpected keyword: {}", unexpected),
    };

    assert!(
        source.next().is_none(),
        "Unexpected data after end of the struct"
    );

    res
}