From a762eb2fbf06126920db01e9e6424890670b47a2 Mon Sep 17 00:00:00 2001
From: Michal Nazarewicz <mina86@mina86.com>
Date: Thu, 14 Dec 2023 04:57:38 +0100
Subject: [PATCH] Introduce AsciiChar::from_digit constructor

Somewhat mimicking char::from_digit, introduce AsciiChar::from_digit
method which returns an ASCII digit character corresponding to given
digit.
---
 src/ascii_char.rs | 40 ++++++++++++++++++++++++++++++++++++++--
 1 file changed, 38 insertions(+), 2 deletions(-)

diff --git a/src/ascii_char.rs b/src/ascii_char.rs
index 39f2ceb..18d40fb 100644
--- a/src/ascii_char.rs
+++ b/src/ascii_char.rs
@@ -371,6 +371,42 @@ impl AsciiChar {
         ALL[ch as usize]
     }
 
+    /// Converts a digit in base 36 to an `AsciiChar`.
+    ///
+    /// This is ASCII version of `char::from_digit(digit, 36)` call.
+    ///
+    /// `from_digit()` will return `None` if the input is greater or equal 36.
+    ///
+    /// # Examples
+    ///
+    /// ```
+    /// # use ascii::AsciiChar;
+    /// assert_eq!(AsciiChar::from_digit(5), Some(AsciiChar::_5));
+    /// assert_eq!(AsciiChar::from_digit(15), Some(AsciiChar::f));
+    /// assert_eq!(AsciiChar::from_digit(25), Some(AsciiChar::p));
+    /// assert_eq!(AsciiChar::from_digit(37), None);
+    /// ```
+    #[inline]
+    #[must_use]
+    pub const fn from_digit(digit: u32) -> Option<Self> {
+        // This `use` is restricted to this function, and without it we'd need
+        // to specify `AsciiChar::` or `Self::` 36 times.
+        #[allow(clippy::enum_glob_use)]
+        use AsciiChar::*;
+
+        #[rustfmt::skip]
+        const ALL: [AsciiChar; 36] = [
+            _0, _1, _2, _3, _4, _5, _6, _7, _8, _9, a, b, c, d, e, f,
+            g, h, i, j, k, l, m, n, o, p, q, r, s, t, u, v, w, x, y, z,
+        ];
+
+        if digit < 36 {
+            Some(ALL[digit as usize])
+        } else {
+            None
+        }
+    }
+
     /// Create an `AsciiChar` from a `char`, in a `const fn` way.
     ///
     /// Within non-`const fn` functions the more general
@@ -794,8 +830,8 @@ macro_rules! impl_into_partial_eq_ord {
     ($wider:ty, $to_wider:expr) => {
         impl From<AsciiChar> for $wider {
             #[inline]
-            fn from(a: AsciiChar) -> $wider {
-                $to_wider(a)
+            fn from(ch: AsciiChar) -> $wider {
+                $to_wider(ch)
             }
         }
         impl PartialEq<$wider> for AsciiChar {