#[smart_beaver::extension]
pub mod psp34_metadata_extension {
    use crate::PSP34Metadata;

    #[smart_beaver::storage]
    pub struct Token {
        #[smart_beaver::init(Data::default())]
        metadata: Data,
    }

    impl PSP34Metadata for Token {
        #[ink(message)]
        fn get_attribute(&self, id: Id, key: Vec<u8>) -> Option<Vec<u8>> {
            self.metadata.get_attribute(id, key)
        }
    }
}