Account for titlecase in casing lints

Author: Jules-Bertholet

compiler/rustc_lint/src/lib.rs

@@ -24,6 +24,7 @@
 #![feature(box_patterns)]
 #![feature(iter_order_by)]
 #![feature(rustc_attrs)]
+#![feature(titlecase)]
 #![feature(try_blocks)]
 // tidy-alphabetical-end
 

compiler/rustc_lint/src/nonstandard_style.rs

@@ -51,34 +51,41 @@ declare_lint_pass!(NonCamelCaseTypes => [NON_CAMEL_CASE_TYPES]);
 /// be upper cased or lower cased. For the purposes of the lint suggestion, we care about being able
 /// to change the char's case.
 fn char_has_case(c: char) -> bool {
-    let mut l = c.to_lowercase();
-    let mut u = c.to_uppercase();
-    while let Some(l) = l.next() {
-        match u.next() {
-            Some(u) if l != u => return true,
-            _ => {}
+    let (mut l, mut t) = (c.to_lowercase(), c.to_titlecase());
+    loop {
+        match (l.next(), t.next()) {
+            (lc, tc) if lc != tc => return true,
+            (None, None) => return false,
+            (_, _) => {}
         }
     }
-    u.next().is_some()
 }
 
-fn is_camel_case(name: &str) -> bool {
+/// FIXME: we should add a more efficient version
+/// in the stdlib for this
+fn changes_when_titlecased(c: char) -> bool {
+    !c.to_titlecase().eq([c])
+}
+
+fn is_upper_camel_case(name: &str) -> bool {
     let name = name.trim_matches('_');
-    if name.is_empty() {
+    let Some(fst) = name.chars().next() else {
         return true;
-    }
+    };
 
-    // start with a non-lowercase letter rather than non-uppercase
-    // ones (some scripts don't have a concept of upper/lowercase)
-    !name.chars().next().unwrap().is_lowercase()
+    // some scripts don't have a concept of upper/lowercase
+    !changes_when_titlecased(fst)
         && !name.contains("__")
         && !name.chars().collect::<Vec<_>>().array_windows().any(|&[fst, snd]| {
             // contains a capitalisable character followed by, or preceded by, an underscore
-            char_has_case(fst) && snd == '_' || char_has_case(snd) && fst == '_'
+            char_has_case(fst) && snd == '_'
+                || char_has_case(snd) && fst == '_'
+                // uses uppercase when it should use titlecase
+                || snd.is_uppercase() && changes_when_titlecased(snd)
         })
 }
 
-fn to_camel_case(s: &str) -> String {
+fn to_upper_camel_case(s: &str) -> String {
     s.trim_matches('_')
         .split('_')
         .filter(|component| !component.is_empty())
@@ -87,24 +94,31 @@ fn to_camel_case(s: &str) -> String {
 
             let mut new_word = true;
             let mut prev_is_lower_case = true;
+            let mut prev_is_lowercased_sigma = false;
 
             for c in component.chars() {
                 // Preserve the case if an uppercase letter follows a lowercase letter, so that
                 // `camelCase` is converted to `CamelCase`.
-                if prev_is_lower_case && c.is_uppercase() {
+                if prev_is_lower_case && (c.is_uppercase() | c.is_titlecase()) {
                     new_word = true;
                 }
 
                 if new_word {
-                    camel_cased_component.extend(c.to_uppercase());
+                    camel_cased_component.extend(c.to_titlecase());
                 } else {
                     camel_cased_component.extend(c.to_lowercase());
                 }
 
-                prev_is_lower_case = c.is_lowercase();
+                prev_is_lower_case = c.is_lowercase() || c.is_titlecase();
+                prev_is_lowercased_sigma = !new_word && c == 'Σ';
                 new_word = false;
             }
 
+            if prev_is_lowercased_sigma {
+                camel_cased_component.pop();
+                camel_cased_component.push('ς');
+            }
+
             camel_cased_component
         })
         .fold((String::new(), None), |(acc, prev): (String, Option<String>), next| {
@@ -126,8 +140,8 @@ impl NonCamelCaseTypes {
     fn check_case(&self, cx: &EarlyContext<'_>, sort: &str, ident: &Ident) {
         let name = ident.name.as_str();
 
-        if !is_camel_case(name) {
-            let cc = to_camel_case(name);
+        if !is_upper_camel_case(name) {
+            let cc = to_upper_camel_case(name);
             let sub = if *name != cc {
                 NonCamelCaseTypeSub::Suggestion { span: ident.span, replace: cc }
             } else {
@@ -239,14 +253,20 @@ impl NonSnakeCase {
                 continue;
             }
             for ch in s.chars() {
-                if !buf.is_empty() && buf != "'" && ch.is_uppercase() && !last_upper {
-                    words.push(buf);
+                if !buf.is_empty()
+                    && buf != "'"
+                    && (ch.is_uppercase() || ch.is_titlecase())
+                    && !last_upper
+                {
+                    // We lowercase only at the end, to handle final sigma correctly
+                    words.push(buf.to_lowercase());
                     buf = String::new();
                 }
-                last_upper = ch.is_uppercase();
-                buf.extend(ch.to_lowercase());
+                last_upper = ch.is_uppercase() || ch.is_titlecase();
+                buf.push(ch);
             }
-            words.push(buf);
+            // We lowercase only at the end, to handle final sigma correctly
+            words.push(buf.to_lowercase());
         }
         words.join("_")
     }
@@ -266,7 +286,8 @@ impl NonSnakeCase {
 
             // This correctly handles letters in languages with and without
             // cases, as well as numbers and underscores.
-            !ident.chars().any(char::is_uppercase)
+            // FIXME: we should add a standard library impl of `c.to_lowercase().eq([c])`
+            ident.chars().all(|c| c.to_lowercase().eq([c]))
         }
 
         let name = ident.name.as_str();
@@ -478,10 +499,12 @@ impl<'a, 'b, F: FnOnce() -> NonUpperCaseGlobal<'b>> Diagnostic<'a, ()>
 impl NonUpperCaseGlobals {
     fn check_upper_case(cx: &LateContext<'_>, sort: &str, did: Option<LocalDefId>, ident: &Ident) {
         let name = ident.name.as_str();
-        if name.chars().any(|c| c.is_lowercase()) {
+        // FIXME: we should add a more efficient version
+        // in the stdlib for `c.to_uppercase().eq([c])`
+        if !name.chars().all(|c| c.to_uppercase().eq([c])) {
             let uc = NonSnakeCase::to_snake_case(name).to_uppercase();
 
-            // If the item is exported, suggesting changing it's name would be breaking-change
+            // If the item is exported, suggesting changing its name would be a breaking change
             // and could break users without a "nice" applicable fix, so let's avoid it.
             let can_change_usages = if let Some(did) = did {
                 !cx.tcx.effective_visibilities(()).is_exported(did)

compiler/rustc_lint/src/nonstandard_style/tests.rs

@@ -1,21 +1,37 @@
-use super::{is_camel_case, to_camel_case};
+use super::{is_upper_camel_case, to_upper_camel_case};
 
 #[test]
 fn camel_case() {
-    assert!(!is_camel_case("userData"));
-    assert_eq!(to_camel_case("userData"), "UserData");
+    assert!(!is_upper_camel_case("userData"));
+    assert_eq!(to_upper_camel_case("userData"), "UserData");
 
-    assert!(is_camel_case("X86_64"));
+    assert!(is_upper_camel_case("X86_64"));
 
-    assert!(!is_camel_case("X86__64"));
-    assert_eq!(to_camel_case("X86__64"), "X86_64");
+    assert!(!is_upper_camel_case("X86__64"));
+    assert_eq!(to_upper_camel_case("X86__64"), "X86_64");
 
-    assert!(!is_camel_case("Abc_123"));
-    assert_eq!(to_camel_case("Abc_123"), "Abc123");
+    assert!(!is_upper_camel_case("Abc_123"));
+    assert_eq!(to_upper_camel_case("Abc_123"), "Abc123");
 
-    assert!(!is_camel_case("A1_b2_c3"));
-    assert_eq!(to_camel_case("A1_b2_c3"), "A1B2C3");
+    assert!(!is_upper_camel_case("A1_b2_c3"));
+    assert_eq!(to_upper_camel_case("A1_b2_c3"), "A1B2C3");
 
-    assert!(!is_camel_case("ONE_TWO_THREE"));
-    assert_eq!(to_camel_case("ONE_TWO_THREE"), "OneTwoThree");
+    assert!(!is_upper_camel_case("ONE_TWO_THREE"));
+    assert_eq!(to_upper_camel_case("ONE_TWO_THREE"), "OneTwoThree");
+
+    // FIXME(@Jules-Bertholet): This test doesn't work due to what I believe
+    // is a Unicode spec bug - uppercase Georgian letters have
+    // incorrect titlecase mappings.
+    // I've reported it to Unicode.
+    // Georgian mtavruli is only used in all-caps
+    //assert!(!is_upper_camel_case("ᲫალაᲔრთობაშია"));
+    //assert_eq!(to_upper_camel_case("ᲫალაᲔრთობაშია"), "ძალა_ერთობაშია");
+
+    assert!(!is_upper_camel_case("LJNJaaaDŽooo"));
+    assert_eq!(to_upper_camel_case("LJNJaaLjNJaDŽooo"), "LjnjaaLjNjaDžooo");
+
+    // Final sigma
+    assert!(!is_upper_camel_case("ΦΙΛΟΣ_ΦΙΛΟΣ"));
+    assert_eq!(to_upper_camel_case("ΦΙΛΟΣ_ΦΙΛΟΣ"), "ΦιλοςΦιλος");
+    assert!(is_upper_camel_case("ΦιλοσΦιλοσ"));
 }

tests/ui/lint/lint-nonstandard-style-unicode-1.rs

@@ -42,8 +42,18 @@ struct Hello_World;
 struct 你_ӟ;
 //~^ ERROR type `你_ӟ` should have an upper camel case name
 
-// and this is ok:
+struct ΦΙΛΟΣ_Σ;
+//~^ ERROR type `ΦΙΛΟΣ_Σ` should have an upper camel case name
+
+struct Σ_ΦΙΛΟΣ;
+//~^ ERROR type `Σ_ΦΙΛΟΣ` should have an upper camel case name
+
+// these are ok:
 
 struct 你_好;
 
+struct ძალა_ერთობაშია;
+
+struct Σ;
+
 fn main() {}

tests/ui/lint/lint-nonstandard-style-unicode-1.stderr

@@ -46,5 +46,17 @@ error: type `你_ӟ` should have an upper camel case name
 LL | struct 你_ӟ;
    |        ^^^^ help: convert the identifier to upper camel case: `你Ӟ`
 
-error: aborting due to 7 previous errors
+error: type `ΦΙΛΟΣ_Σ` should have an upper camel case name
+  --> $DIR/lint-nonstandard-style-unicode-1.rs:45:8
+   |
+LL | struct ΦΙΛΟΣ_Σ;
+   |        ^^^^^^^ help: convert the identifier to upper camel case: `ΦιλοςΣ`
+
+error: type `Σ_ΦΙΛΟΣ` should have an upper camel case name
+  --> $DIR/lint-nonstandard-style-unicode-1.rs:48:8
+   |
+LL | struct Σ_ΦΙΛΟΣ;
+   |        ^^^^^^^ help: convert the identifier to upper camel case: `ΣΦιλος`
+
+error: aborting due to 9 previous errors
 

tests/ui/lint/lint-nonstandard-style-unicode-2.rs

@@ -0,0 +1,31 @@
+#![allow(dead_code)]
+#![forbid(non_snake_case)]
+
+// 2. non_snake_case
+
+
+fn LJNJaaLjNJaDŽooo() {}
+//~^ ERROR function `LJNJaaLjNJaDŽooo` should have a snake case name
+//~| WARN identifier contains 5 non normalized (NFKC) characters
+
+fn LjnjaaLjNjaDžooo() {}
+//~^ ERROR function `LjnjaaLjNjaDžooo` should have a snake case name
+//~| WARN identifier contains 5 non normalized (NFKC) characters
+
+// test final sigma casing
+fn ΦΙΛΟΣ_ΦΙΛΟΣ() {}
+//~^ ERROR function `ΦΙΛΟΣ_ΦΙΛΟΣ` should have a snake case name
+
+fn Σ() {}
+//~^ ERROR function `Σ` should have a snake case name
+
+fn ΦΙΛΟΣ_Σ() {}
+//~^ ERROR function `ΦΙΛΟΣ_Σ` should have a snake case name
+
+fn Σ_ΦΙΛΟΣ() {}
+//~^ ERROR function `Σ_ΦΙΛΟΣ` should have a snake case name
+
+// this is ok
+fn φιλοσ_φιλοσ() {}
+
+fn main() {}

tests/ui/lint/lint-nonstandard-style-unicode-2.stderr

@@ -0,0 +1,61 @@
+warning: identifier contains 5 non normalized (NFKC) characters: 'LJ', 'NJ', 'Lj', 'NJ', and 'DŽ'
+  --> $DIR/lint-nonstandard-style-unicode-2.rs:7:4
+   |
+LL | fn LJNJaaLjNJaDŽooo() {}
+   |    ^^^^^^^^^^^
+   |
+   = note: these characters are included in the Not_NFKC Unicode general security profile
+   = note: `#[warn(uncommon_codepoints)]` on by default
+
+warning: identifier contains 5 non normalized (NFKC) characters: 'Lj', 'nj', 'Lj', 'Nj', and 'Dž'
+  --> $DIR/lint-nonstandard-style-unicode-2.rs:11:4
+   |
+LL | fn LjnjaaLjNjaDžooo() {}
+   |    ^^^^^^^^^^^
+   |
+   = note: these characters are included in the Not_NFKC Unicode general security profile
+
+error: function `LJNJaaLjNJaDŽooo` should have a snake case name
+  --> $DIR/lint-nonstandard-style-unicode-2.rs:7:4
+   |
+LL | fn LJNJaaLjNJaDŽooo() {}
+   |    ^^^^^^^^^^^ help: convert the identifier to snake case: `ljnjaa_ljnja_džooo`
+   |
+note: the lint level is defined here
+  --> $DIR/lint-nonstandard-style-unicode-2.rs:2:11
+   |
+LL | #![forbid(non_snake_case)]
+   |           ^^^^^^^^^^^^^^
+
+error: function `LjnjaaLjNjaDžooo` should have a snake case name
+  --> $DIR/lint-nonstandard-style-unicode-2.rs:11:4
+   |
+LL | fn LjnjaaLjNjaDžooo() {}
+   |    ^^^^^^^^^^^ help: convert the identifier to snake case: `ljnjaa_ljnja_džooo`
+
+error: function `ΦΙΛΟΣ_ΦΙΛΟΣ` should have a snake case name
+  --> $DIR/lint-nonstandard-style-unicode-2.rs:16:4
+   |
+LL | fn ΦΙΛΟΣ_ΦΙΛΟΣ() {}
+   |    ^^^^^^^^^^^ help: convert the identifier to snake case: `φιλος_φιλος`
+
+error: function `Σ` should have a snake case name
+  --> $DIR/lint-nonstandard-style-unicode-2.rs:19:4
+   |
+LL | fn Σ() {}
+   |    ^ help: convert the identifier to snake case: `σ`
+
+error: function `ΦΙΛΟΣ_Σ` should have a snake case name
+  --> $DIR/lint-nonstandard-style-unicode-2.rs:22:4
+   |
+LL | fn ΦΙΛΟΣ_Σ() {}
+   |    ^^^^^^^ help: convert the identifier to snake case: `φιλος_σ`
+
+error: function `Σ_ΦΙΛΟΣ` should have a snake case name
+  --> $DIR/lint-nonstandard-style-unicode-2.rs:25:4
+   |
+LL | fn Σ_ΦΙΛΟΣ() {}
+   |    ^^^^^^^ help: convert the identifier to snake case: `σ_φιλος`
+
+error: aborting due to 6 previous errors; 2 warnings emitted
+

tests/ui/lint/lint-nonstandard-style-unicode-3.rs

@@ -21,4 +21,11 @@ static τεχ: f32 = 3.14159265;
 
 static __密__封__线__内__禁__止__答__题__: bool = true;
 
+static ძალა_ერთობაშია: () = ();
+//~^ ERROR static variable `ძალა_ერთობაშია` should have an upper case name
+
+static Nj: () = ();
+//~^ ERROR static variable `Nj` should have an upper case name
+//~| WARN identifier contains a non normalized (NFKC) character
+
 fn main() {}

tests/ui/lint/lint-nonstandard-style-unicode-3.stderr

@@ -1,3 +1,12 @@
+warning: identifier contains a non normalized (NFKC) character: 'Nj'
+  --> $DIR/lint-nonstandard-style-unicode-3.rs:27:8
+   |
+LL | static Nj: () = ();
+   |        ^
+   |
+   = note: this character is included in the Not_NFKC Unicode general security profile
+   = note: `#[warn(uncommon_codepoints)]` on by default
+
 error: static variable `τεχ` should have an upper case name
   --> $DIR/lint-nonstandard-style-unicode-3.rs:17:8
    |
@@ -10,5 +19,17 @@ note: the lint level is defined here
 LL | #![forbid(non_upper_case_globals)]
    |           ^^^^^^^^^^^^^^^^^^^^^^
 
-error: aborting due to 1 previous error
+error: static variable `ძალა_ერთობაშია` should have an upper case name
+  --> $DIR/lint-nonstandard-style-unicode-3.rs:24:8
+   |
+LL | static ძალა_ერთობაშია: () = ();
+   |        ^^^^^^^^^^^^^^ help: convert the identifier to upper case: `ᲫᲐᲚᲐ_ᲔᲠᲗᲝᲑᲐᲨᲘᲐ`
+
+error: static variable `Nj` should have an upper case name
+  --> $DIR/lint-nonstandard-style-unicode-3.rs:27:8
+   |
+LL | static Nj: () = ();
+   |        ^ help: convert the identifier to upper case: `NJ`
+
+error: aborting due to 3 previous errors; 1 warning emitted