feat: add project & repeat_char macroHEADmaster

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 })
+    }
+}