diff options
| author | 魏曹先生 <1992414357@qq.com> | 2026-07-14 01:36:27 +0800 |
|---|---|---|
| committer | 魏曹先生 <1992414357@qq.com> | 2026-07-14 01:36:27 +0800 |
| commit | 4665fc94ab9508d115298dd988e2381354f46c01 (patch) | |
| tree | a2b14f5684c68d88a0a77570d0ac3ac8d863867b /mingling_picker_macros/src/internal_repeat.rs | |
| parent | 5ac0f0e98d05e95f7e3c58995f8d79670cbac772 (diff) | |
feat(picker): add core trait, types, and builder macro for argument
parsing
Implement the foundational `mingling_picker` library along with its
companion
`mingling_picker_macros` crate. The picker provides:
- `Pickable` trait for parsing types from raw strings
- `PickerResult` enum modeling parse outcomes
- `Picker` struct for storing and indexing command-line arguments
- `PickerRequirement` struct for declarative parameter definitions
- `PickerPattern` family (1..32) of typed pattern structs via the
`internal_repeat` proc macro
- `req!` proc macro as a succinct builder for `PickerRequirement`
Re-export `mingling_picker::macros::*` from the `mingling` crate when
the
`picker` feature is enabled, replacing the previous wildcard re-export
of
`mingling_macros`.
Diffstat (limited to 'mingling_picker_macros/src/internal_repeat.rs')
| -rw-r--r-- | mingling_picker_macros/src/internal_repeat.rs | 204 |
1 files changed, 204 insertions, 0 deletions
diff --git a/mingling_picker_macros/src/internal_repeat.rs b/mingling_picker_macros/src/internal_repeat.rs new file mode 100644 index 0000000..9feedfd --- /dev/null +++ b/mingling_picker_macros/src/internal_repeat.rs @@ -0,0 +1,204 @@ +use proc_macro::{Delimiter, Group, Ident, Literal, TokenStream, TokenTree}; + +pub(crate) fn internal_repeat(input: TokenStream) -> TokenStream { + let tokens: Vec<TokenTree> = input.into_iter().collect(); + let (range_start, range_end, body_start) = parse_range(&tokens); + + let mut body: Vec<TokenTree> = tokens[body_start..].to_vec(); + if body.len() == 1 { + if let TokenTree::Group(g) = &body[0] { + if g.delimiter() == Delimiter::Brace { + body = g.stream().into_iter().collect(); + } + } + } + + let mut result = Vec::new(); + for i in range_start..=range_end { + result.extend(expand_body(&body, i)); + } + result.into_iter().collect() +} + +/// Parse `start .. end =>` or `start ..= end =>` or `count =>` (backward compat). +/// Returns `(start, end_inclusive, body_start_index)`. +fn parse_range(tokens: &[TokenTree]) -> (usize, usize, usize) { + // Find => separator + let arrow_pos = tokens.windows(2).position(|w| { + matches!(&w[0], TokenTree::Punct(p) if p.as_char() == '=') + && matches!(&w[1], TokenTree::Punct(p) if p.as_char() == '>') + }); + + let (arrow_pos, body_start) = match arrow_pos { + Some(p) => (p, p + 2), + None => return (1, 12, 0), // fallback + }; + + let before: Vec<&TokenTree> = tokens[..arrow_pos].iter().collect(); + + // Try to find `..` or `..=` pattern + // `..` is two Punct('.') tokens + let dotdot = before.windows(2).position(|w| { + matches!(w[0], TokenTree::Punct(p) if p.as_char() == '.') + && matches!(w[1], TokenTree::Punct(p) if p.as_char() == '.') + }); + + if let Some(dd) = dotdot { + // Start value: tokens before `..` + let start = parse_usize_tokens(&before[..dd]); + let after_dd = &before[dd + 2..]; + + // Check for `..=` (inclusive range) + let (inclusive, end_tokens) = if after_dd.first().map_or( + false, + |t| matches!(t, TokenTree::Punct(p) if p.as_char() == '='), + ) { + (true, &after_dd[1..]) + } else { + (false, after_dd) + }; + + let end = parse_usize_tokens(end_tokens); + + if inclusive { + (start, end, body_start) + } else { + // Exclusive end: if end >= start, iterate start..end, so end_inclusive = end - 1 + if end > start { + (start, end - 1, body_start) + } else { + (1, 12, body_start) // fallback + } + } + } else { + // No `..` found — fallback to simple count + let count = parse_usize_tokens(&before); + (1, count, body_start) + } +} + +/// Parse a sequence of tokens as a single usize value. +fn parse_usize_tokens(tokens: &[&TokenTree]) -> usize { + let s: String = tokens + .iter() + .map(|t| match t { + TokenTree::Literal(l) => l.to_string(), + TokenTree::Ident(id) => id.to_string(), + _ => String::new(), + }) + .collect::<Vec<_>>() + .join("") + .replace(' ', ""); + + s.parse().unwrap_or(12) +} + +/// Walk tokens, replacing `$` in identifier tails, and expanding +/// `( … )+` / `( … ,)+` / `( … ;)+` groups. +fn expand_body(tokens: &[TokenTree], outer: usize) -> Vec<TokenTree> { + let mut out = Vec::new(); + let mut i = 0; + while i < tokens.len() { + // Check for a parenthesized repetition group: ( ... ) sep? + + if let Some(exp) = try_expand_paren_group(tokens, i, outer) { + let (items, consumed) = exp; + out.extend(items); + i += consumed; + continue; + } + + // Identifier followed by `$` → combined ident + number + if let TokenTree::Ident(id) = &tokens[i] { + if i + 1 < tokens.len() { + if let TokenTree::Punct(p) = &tokens[i + 1] { + if p.as_char() == '$' { + let name = format!("{}{outer}", id.to_string()); + out.push(TokenTree::Ident(Ident::new(&name, id.span()))); + i += 2; + continue; + } + } + } + out.push(tokens[i].clone()); + i += 1; + continue; + } + + match &tokens[i] { + TokenTree::Punct(p) if p.as_char() == '$' => { + out.push(TokenTree::Literal(Literal::usize_suffixed(outer))); + } + TokenTree::Group(g) => { + let inner = expand_body_vec(&g.stream(), outer); + out.push(TokenTree::Group(Group::new( + g.delimiter(), + inner.into_iter().collect(), + ))); + } + other => out.push(other.clone()), + } + i += 1; + } + out +} + +fn expand_body_vec(stream: &TokenStream, outer: usize) -> Vec<TokenTree> { + let v: Vec<TokenTree> = stream.clone().into_iter().collect(); + expand_body(&v, outer) +} + +/// Try to expand `( … )+` / `( … ,)+` / `( … ;)+` at position `i`. +/// +/// The `+` is AFTER the closing paren. An optional separator (`,` or `;`) +/// may appear between `)` and `+`. +/// Returns `(expanded_tokens, consumed_count)` or `None`. +fn try_expand_paren_group( + tokens: &[TokenTree], + i: usize, + outer: usize, +) -> Option<(Vec<TokenTree>, usize)> { + let group = match tokens.get(i)? { + TokenTree::Group(g) if g.delimiter() == Delimiter::Parenthesis => g, + _ => return None, + }; + + // Check tokens AFTER the group for `+` (optionally preceded by `,` or `;`) + let rest = &tokens[i + 1..]; + let sep: Option<&str> = match rest.first() { + Some(TokenTree::Punct(p)) if p.as_char() == ',' => { + if matches!(rest.get(1), Some(TokenTree::Punct(p)) if p.as_char() == '+') { + Some(",") + } else { + return None; + } + } + Some(TokenTree::Punct(p)) if p.as_char() == ';' => { + if matches!(rest.get(1), Some(TokenTree::Punct(p)) if p.as_char() == '+') { + Some(";") + } else { + return None; + } + } + Some(TokenTree::Punct(p)) if p.as_char() == '+' => None, + _ => return None, + }; + + let consumed = if sep.is_some() { 3 } else { 2 }; // group + sep? + + + + let inner: Vec<TokenTree> = group.stream().into_iter().collect(); + + let mut out = Vec::new(); + for n in 1..=outer { + if n > 1 { + if let Some(s) = sep { + out.push(TokenTree::Punct(proc_macro::Punct::new( + s.chars().next().unwrap(), + proc_macro::Spacing::Alone, + ))); + } + } + out.extend(expand_body(&inner, n)); + } + + Some((out, consumed)) +} |
