risingwave_expr_macro/
utils.rs

1// Copyright 2025 RisingWave Labs
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7//     http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15use proc_macro2::Ident;
16use syn::spanned::Spanned;
17use syn::{Token, VisRestricted, Visibility};
18
19/// Convert a string from `snake_case` to `CamelCase`.
20pub fn to_camel_case(input: &str) -> String {
21    input
22        .split('_')
23        .map(|word| {
24            let mut chars = word.chars();
25            match chars.next() {
26                None => String::new(),
27                Some(first_char) => {
28                    format!("{}{}", first_char.to_uppercase(), chars.as_str())
29                }
30            }
31        })
32        .collect()
33}
34
35pub(crate) fn extend_vis_with_super(vis: Visibility) -> Visibility {
36    let Visibility::Restricted(vis) = vis else {
37        return vis;
38    };
39    let VisRestricted {
40        pub_token,
41        paren_token,
42        mut in_token,
43        mut path,
44    } = vis;
45    let first_segment = path.segments.first_mut().unwrap();
46    if first_segment.ident == "self" {
47        *first_segment = Ident::new("super", first_segment.span()).into();
48    } else if first_segment.ident == "super" {
49        let span = first_segment.span();
50        path.segments.insert(0, Ident::new("super", span).into());
51        in_token.get_or_insert(Token![in](in_token.span()));
52    }
53    Visibility::Restricted(VisRestricted {
54        pub_token,
55        paren_token,
56        in_token,
57        path,
58    })
59}
60
61#[cfg(test)]
62mod tests {
63    use quote::ToTokens;
64    use syn::Visibility;
65
66    use crate::utils::extend_vis_with_super;
67
68    #[test]
69    fn test_extend_vis_with_super() {
70        let cases = [
71            ("pub", "pub"),
72            ("pub(crate)", "pub(crate)"),
73            ("pub(self)", "pub(super)"),
74            ("pub(super)", "pub(in super::super)"),
75            ("pub(in self)", "pub(in super)"),
76            (
77                "pub(in self::context::data)",
78                "pub(in super::context::data)",
79            ),
80            (
81                "pub(in super::context::data)",
82                "pub(in super::super::context::data)",
83            ),
84            ("pub(in crate::func::impl_)", "pub(in crate::func::impl_)"),
85            (
86                "pub(in ::risingwave_expr::func::impl_)",
87                "pub(in ::risingwave_expr::func::impl_)",
88            ),
89        ];
90        for (input, expected) in cases {
91            let input: Visibility = syn::parse_str(input).unwrap();
92            let expected: Visibility = syn::parse_str(expected).unwrap();
93            let output = extend_vis_with_super(input);
94            let expected = expected.into_token_stream().to_string();
95            let output = output.into_token_stream().to_string();
96            assert_eq!(expected, output);
97        }
98    }
99}
risingwave_expr_macro/utils.rs

risingwave_expr_macro/
utils.rs
