summaryrefslogtreecommitdiff
path: root/src/lib.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/lib.rs')
-rw-r--r--src/lib.rs62
1 files changed, 62 insertions, 0 deletions
diff --git a/src/lib.rs b/src/lib.rs
new file mode 100644
index 0000000..5649972
--- /dev/null
+++ b/src/lib.rs
@@ -0,0 +1,62 @@
+#![deny(clippy::all)]
+#![deny(clippy::pedantic)]
+#![deny(missing_docs)]
+
+//! Utility macros that should be in std but is not.
+
+use std::iter::repeat;
+
+use proc_macro::TokenStream;
+use quote::quote;
+use syn::parse::Parse;
+use syn::{parse_macro_input, LitChar, LitInt, Token};
+
+/// Repeats a character N number of times resulting in a string literal.
+///
+/// # Arguments
+/// Seperated by commas.
+///
+/// - The character to repeat
+/// - Number of times to repeat it.
+///
+/// # Examples
+/// ```
+/// # use utility_macros::repeat_char;
+///
+/// let eight_o = repeat_char!('o', 8);
+///
+/// assert_eq!(eight_o, "oooooooo");
+/// ```
+#[proc_macro]
+pub fn repeat_char(input: TokenStream) -> TokenStream
+{
+ let RepeatCharMacroArgs { character, count } =
+ parse_macro_input!(input as RepeatCharMacroArgs);
+
+ let repeated = repeat(character.value()).take(count).collect::<String>();
+
+ quote! {
+ #repeated
+ }
+ .into()
+}
+
+struct RepeatCharMacroArgs
+{
+ character: LitChar,
+ count: usize,
+}
+
+impl Parse for RepeatCharMacroArgs
+{
+ fn parse(input: syn::parse::ParseStream) -> syn::Result<Self>
+ {
+ let character = input.parse::<LitChar>()?;
+
+ input.parse::<Token![,]>()?;
+
+ let count = input.parse::<LitInt>()?.base10_parse::<usize>()?;
+
+ Ok(Self { character, count })
+ }
+}