From 24ef322b57d29d26224e7f9e772b3691143b2da2 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Gustav=20S=C3=B6rn=C3=A4s?= Date: Fri, 19 Feb 2021 19:19:44 +0100 Subject: begin work on automatic link --- Cargo.lock | 1 + sylt_macro/Cargo.toml | 1 + sylt_macro/src/lib.rs | 21 +++++++++++++++++++++ 3 files changed, 23 insertions(+) diff --git a/Cargo.lock b/Cargo.lock index f7d2ec1..44e16d0 100644 --- a/Cargo.lock +++ b/Cargo.lock @@ -531,6 +531,7 @@ dependencies = [ name = "sylt_macro" version = "0.1.0" dependencies = [ + "lazy_static", "quote", "syn", ] diff --git a/sylt_macro/Cargo.toml b/sylt_macro/Cargo.toml index 9ac045e..04ae39d 100644 --- a/sylt_macro/Cargo.toml +++ b/sylt_macro/Cargo.toml @@ -12,3 +12,4 @@ proc-macro = true [dependencies] syn = { version = "1.0", features=["full"] } quote = "1.0" +lazy_static = "1.4" diff --git a/sylt_macro/src/lib.rs b/sylt_macro/src/lib.rs index 1cafde8..9bfef65 100644 --- a/sylt_macro/src/lib.rs +++ b/sylt_macro/src/lib.rs @@ -1,5 +1,7 @@ +use lazy_static::lazy_static; use proc_macro::TokenStream; use quote::quote; +use std::borrow::Cow; use syn::{Expr, Pat, Token, parse::{Parse, ParseStream, Result}, parse_macro_input}; struct ExternBlock { @@ -91,3 +93,22 @@ pub fn extern_function(tokens: TokenStream) -> TokenStream { }; TokenStream::from(tokens) } + +#[proc_macro_attribute] +pub fn extern_link(attr: TokenStream, tokens: TokenStream) -> TokenStream { + let parsed: syn::ItemFn = parse_macro_input!(tokens); + + let name = if attr.is_empty() { + Cow::Borrowed(&parsed.sig.ident) + } else { + let attr: syn::Ident = parse_macro_input!(attr); + Cow::Owned(attr) + }; + + let tokens = quote! { + #parsed + + + }; + TokenStream::from(tokens) +} -- cgit v1.2.1