urid-derive 0.1.1

use proc_macro::TokenStream;
use proc_macro2::Literal;
use quote::quote;
use syn::{parse, Ident, Item};

/// Get the identity of the item we have to implement `UriBound` for.
///
/// This function also checks that the item has no generics, since this macro isn't smart enough to
/// implement `UriBound` for all arguments of the generic type.
fn get_type_ident(item: TokenStream) -> Ident {
    const PARSING_ERROR: &str = "Only structs, enums, types, and unions may have a URI";

    let (ident, generics) = match parse::<Item>(item).expect(PARSING_ERROR) {
        Item::Enum(definition) => (definition.ident, definition.generics),
        Item::Struct(definition) => (definition.ident, definition.generics),
        Item::Type(definition) => (definition.ident, definition.generics),
        Item::Union(definition) => (definition.ident, definition.generics),
        _ => panic!(PARSING_ERROR),
    };

    if !generics.params.is_empty() {
        panic!("The uri attribute does not support generic types");
    }
    ident
}

/// Parse the attribute argument and create the URI literal from it.
///
/// This includes multiple checks to assure that the literal is formatted correctly.
fn get_uri(attr: TokenStream) -> Literal {
    const PARSING_ERROR: &str = "A URI has to be a string literal";

    if parse::<Literal>(attr.clone()).is_err() {
        panic!(PARSING_ERROR);
    }
    //check if it's a litteral string
    let attr = attr.to_string();
    if !attr.starts_with('"') || !attr.ends_with('"') {
        panic!(PARSING_ERROR);
    }
    //Remove the enclosing "" to get the uri
    let uri = String::from(attr.get(1..attr.len() - 1).expect(PARSING_ERROR));
    if uri.contains("\\0") {
        panic!("Unexpected Null terminator");
    }

    let mut uri_vec: Vec<u8> = Vec::with_capacity(uri.len() + 1);
    uri_vec.extend(uri.as_bytes());
    uri_vec.push(0);

    Literal::byte_string(uri_vec.as_ref())
}

/// Implement `UriBound` for a given item.
pub fn impl_uri_bound(attr: TokenStream, mut item: TokenStream) -> TokenStream {
    let ident = get_type_ident(item.clone());
    let uri = get_uri(attr);

    let implementation: TokenStream = quote! {
        unsafe impl UriBound for #ident {
            const URI: &'static [u8] = #uri;
        }
    }
    .into();
    item.extend(implementation);
    item
}