89 lines · 2.4 KB · Rust
RawDownload
#![cfg_attr(not(target_os = "windows"), allow(unused))] use proc_macro::TokenStream;use quote::quote;use syn::{LitStr, parse_macro_input}; /// A macro used in tests for cross-platform path string literals in tests. On Windows it replaces/// `/` with `\\` and adds `C:` to the beginning of absolute paths. On other platforms, the path is/// returned unmodified.////// # Example/// ```rust/// use util_macros::path;////// let path = path!("/Users/user/file.txt");/// #[cfg(target_os = "windows")]/// assert_eq!(path, "C:\\Users\\user\\file.txt");/// #[cfg(not(target_os = "windows"))]/// assert_eq!(path, "/Users/user/file.txt");/// ```#[proc_macro]pub fn path(input: TokenStream) -> TokenStream {    let path = parse_macro_input!(input as LitStr);    let mut path = path.value();     #[cfg(target_os = "windows")]    {        path = path.replace("/", "\\");        if path.starts_with("\\") {            path = format!("C:{}", path);        }    }     TokenStream::from(quote! {        #path    })} /// This macro replaces the path prefix `file:///` with `file:///C:/` for Windows./// But if the target OS is not Windows, the URI is returned as is.////// # Example/// ```rust/// use util_macros::uri;////// let uri = uri!("file:///path/to/file");/// #[cfg(target_os = "windows")]/// assert_eq!(uri, "file:///C:/path/to/file");/// #[cfg(not(target_os = "windows"))]/// assert_eq!(uri, "file:///path/to/file");/// ```#[proc_macro]pub fn uri(input: TokenStream) -> TokenStream {    let uri = parse_macro_input!(input as LitStr);    let uri = uri.value();     #[cfg(target_os = "windows")]    let uri = uri.replace("file:///", "file:///C:/");     TokenStream::from(quote! {        #uri    })} /// This macro replaces the line endings `\n` with `\r\n` for Windows./// But if the target OS is not Windows, the line endings are returned as is.////// # Example/// ```rust/// use util_macros::line_endings;////// let text = line_endings!("Hello\nWorld");/// #[cfg(target_os = "windows")]/// assert_eq!(text, "Hello\r\nWorld");/// #[cfg(not(target_os = "windows"))]/// assert_eq!(text, "Hello\nWorld");/// ```#[proc_macro]pub fn line_endings(input: TokenStream) -> TokenStream {    let text = parse_macro_input!(input as LitStr);    let text = text.value();     #[cfg(target_os = "windows")]    let text = text.replace("\n", "\r\n");     TokenStream::from(quote! {        #text    })}