fix set_bit

Author: burrbull

src/util.rs

@@ -56,86 +56,38 @@ pub trait ToSanitizedSnakeCase {
 
 impl ToSanitizedSnakeCase for str {
     fn to_sanitized_not_keyword_snake_case(&self) -> Cow<str> {
+        const INTERNALS: [&str; 4] = ["set_bit", "clear_bit", "bit", "bits"];
+
         let s = self.replace(BLACKLIST_CHARS, "");
         match s.chars().next().unwrap_or('\0') {
             '0' | '1' | '2' | '3' | '4' | '5' | '6' | '7' | '8' | '9' => {
-                Cow::from(format!("_{}", s.to_snake_case()))
+                format!("_{}", s.to_snake_case()).into()
+            }
+            _ => {
+                let s = Cow::from(s.to_snake_case());
+                if INTERNALS.contains(&s.as_ref()) {
+                    s + "_"
+                } else {
+                    s
+                }
             }
-            _ => Cow::from(s.to_snake_case()),
         }
     }
 }
 
-pub fn sanitize_keyword<'a>(s: Cow<'a, str>) -> Cow<'a, str> {
-    macro_rules! keywords {
-        ($s:expr, $($kw:ident),+,) => {
-            Cow::from(match &$s.to_lowercase()[..] {
-                $(stringify!($kw) => concat!(stringify!($kw), "_")),+,
-                _ => return s,
-            })
-        }
-    }
-
-    keywords! {
-        s,
-        abstract,
-        alignof,
-        as,
-        async,
-        await,
-        become,
-        box,
-        break,
-        const,
-        continue,
-        crate,
-        do,
-        else,
-        enum,
-        extern,
-        false,
-        final,
-        fn,
-        for,
-        if,
-        impl,
-        in,
-        let,
-        loop,
-        macro,
-        match,
-        mod,
-        move,
-        mut,
-        offsetof,
-        override,
-        priv,
-        proc,
-        pub,
-        pure,
-        ref,
-        return,
-        self,
-        sizeof,
-        static,
-        struct,
-        super,
-        trait,
-        true,
-        try,
-        type,
-        typeof,
-        unsafe,
-        unsized,
-        use,
-        virtual,
-        where,
-        while,
-        yield,
-        set_bit,
-        clear_bit,
-        bit,
-        bits,
+pub fn sanitize_keyword<'a>(sc: Cow<'a, str>) -> Cow<'a, str> {
+    const KEYWORDS: [&str; 54] = [
+        "abstract", "alignof", "as", "async", "await", "become", "box", "break", "const",
+        "continue", "crate", "do", "else", "enum", "extern", "false", "final", "fn", "for", "if",
+        "impl", "in", "let", "loop", "macro", "match", "mod", "move", "mut", "offsetof",
+        "override", "priv", "proc", "pub", "pure", "ref", "return", "self", "sizeof", "static",
+        "struct", "super", "trait", "true", "try", "type", "typeof", "unsafe", "unsized", "use",
+        "virtual", "where", "while", "yield",
+    ];
+    if KEYWORDS.contains(&sc.as_ref()) {
+        sc + "_"
+    } else {
+        sc
     }
 }