use proc_macro::{Delimiter, Group, Ident, Literal, Punct, Spacing, Span, TokenStream, TokenTree};
use std::iter::FromIterator;

pub struct Error {
    span: Span,
    msg: String,
}

impl Error {
    pub fn new(span: Span, msg: impl Into<String>) -> Self {
        Error {
            span,
            msg: msg.into(),
        }
    }
}

pub fn compile_error(err: Error) -> TokenStream {
    // compile_error!($msg)
    TokenStream::from_iter(vec![
        TokenTree::Ident(Ident::new("compile_error", err.span)),
        TokenTree::Punct({
            let mut punct = Punct::new('!', Spacing::Alone);
            punct.set_span(err.span);
            punct
        }),
        TokenTree::Group({
            let mut group = Group::new(Delimiter::Brace, {
                TokenStream::from_iter(vec![TokenTree::Literal({
                    let mut string = Literal::string(&err.msg);
                    string.set_span(err.span);
                    string
                })])
            });
            group.set_span(err.span);
            group
        }),
    ])
}