var.rs

use std::fmt;

use proc_macro2::{Ident, TokenStream};

use quote::{quote, ToTokens};
use syn::Type;

pub enum Var {
    Primitive {
        id: VarId,
        name: Ident,
        ty: Type,
        ty_str: String,
    },
    Array {
        id: VarId,
        name: Ident,
        ty: Type,
        ty_str: String,
        length: u32,
    },
}

impl fmt::Debug for Var {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        match self {
            Var::Primitive {
                id, name, ty_str, ..
            } => writeln!(f, "Var #{} - {}: {}", id.0, name, ty_str),
            Var::Array {
                id,
                name,
                ty_str,
                length,
                ..
            } => writeln!(f, "Var #{} - {}: [{}; {}]", id.0, name, ty_str, length),
        }
    }
}

#[repr(transparent)]
#[derive(Debug, PartialEq, Hash, Copy, Clone)]
pub struct VarId(pub u32);

impl ToTokens for &VarId {
    fn to_tokens(&self, tokens: &mut TokenStream) {
        let id = self.0;
        tokens.extend(quote! { #id });
    }
}