From 3f9648eed48cd8b6cd35d0ae2ee5bbe25fa735ac Mon Sep 17 00:00:00 2001
From: Owen Jacobson <owen@grimoire.ca>
Date: Mon, 21 Oct 2024 00:36:44 -0400
Subject: Canonicalize login and channel names.

Canonicalization does two things:

* It prevents duplicate names that differ only by case or only by normalization/encoding sequence; and
* It makes certain name-based comparisons "case-insensitive" (generalizing via Unicode's case-folding rules).

This change is complicated, as it means that every name now needs to be stored in two forms. Unfortunately, this is _very likely_ a breaking schema change. The migrations in this commit perform a best-effort attempt to canonicalize existing channel or login names, but it's likely any existing channels or logins with non-ASCII characters will not be canonicalize correctly. Since clients look at all channel names and all login names on boot, and since the code in this commit verifies canonicalization when reading from the database, this will effectively make the server un-usuable until any incorrectly-canonicalized values are either manually canonicalized, or removed

It might be possible to do better with [the `icu` sqlite3 extension][icu], but (a) I'm not convinced of that and (b) this commit is already huge; adding database extension support would make it far larger.

[icu]: https://sqlite.org/src/dir/ext/icu

For some references on why it's worth storing usernames this way, see <https://www.b-list.org/weblog/2018/nov/26/case/> and the refernced talk, as well as <https://www.b-list.org/weblog/2018/feb/11/usernames/>. Bennett's treatment of this issue is, to my eye, much more readable than the referenced Unicode technical reports, and I'm inclined to trust his opinion given that he maintains a widely-used, internet-facing user registration library for Django.
---
 src/name.rs | 85 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 85 insertions(+)
 create mode 100644 src/name.rs

(limited to 'src/name.rs')

diff --git a/src/name.rs b/src/name.rs
new file mode 100644
index 0000000..9187d33
--- /dev/null
+++ b/src/name.rs
@@ -0,0 +1,85 @@
+use std::fmt;
+
+use crate::normalize::{ident, nfc};
+
+#[derive(Clone, Debug, Eq, PartialEq, serde::Deserialize, serde::Serialize, sqlx::Type)]
+#[serde(from = "String", into = "String")]
+pub struct Name {
+    display: nfc::String,
+    canonical: ident::String,
+}
+
+impl Name {
+    pub fn new<D, C>(display: D, canonical: C) -> Result<Self, Error>
+    where
+        D: AsRef<str>,
+        C: AsRef<str>,
+    {
+        let name = Self::from(display);
+
+        if name.canonical.as_str() == canonical.as_ref() {
+            Ok(name)
+        } else {
+            Err(Error::CanonicalMismatch(
+                canonical.as_ref().into(),
+                name.canonical,
+                name.display,
+            ))
+        }
+    }
+
+    pub fn optional<D, C>(display: Option<D>, canonical: Option<C>) -> Result<Option<Self>, Error>
+    where
+        D: AsRef<str>,
+        C: AsRef<str>,
+    {
+        display
+            .zip(canonical)
+            .map(|(display, canonical)| Self::new(display, canonical))
+            .transpose()
+    }
+
+    pub fn display(&self) -> &nfc::String {
+        &self.display
+    }
+
+    pub fn canonical(&self) -> &ident::String {
+        &self.canonical
+    }
+}
+
+#[derive(Debug, thiserror::Error)]
+pub enum Error {
+    #[error("stored canonical form {0:#?} does not match computed canonical form {:#?} for name {:#?}", .1.as_str(), .2.as_str())]
+    CanonicalMismatch(String, ident::String, nfc::String),
+}
+
+impl Default for Name {
+    fn default() -> Self {
+        Self::from(String::default())
+    }
+}
+
+impl fmt::Display for Name {
+    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
+        self.display.fmt(f)
+    }
+}
+
+impl<S> From<S> for Name
+where
+    S: AsRef<str>,
+{
+    fn from(name: S) -> Self {
+        let display = nfc::String::from(&name);
+        let canonical = ident::String::from(&name);
+
+        Self { display, canonical }
+    }
+}
+
+impl From<Name> for String {
+    fn from(name: Name) -> Self {
+        name.display.into()
+    }
+}
-- 
cgit v1.2.3