Use ArrayString in more places via aformat! (#2902)

This also bumps `to-arraystring` up to 0.2, required for `aformat`.

Author: GnomedDev

.github/workflows/ci.yml

@@ -3,8 +3,8 @@ name: CI
 on: [push, pull_request]
 
 env:
-  rust_min: 1.74.0
-  rust_nightly: nightly-2024-02-08
+  rust_min: 1.79.0
+  rust_nightly: nightly-2024-06-21
 
 jobs:
   test:

Cargo.toml

@@ -10,7 +10,7 @@ readme = "README.md"
 repository = "https://github.com/serenity-rs/serenity.git"
 version = "0.12.2"
 edition = "2021"
-rust-version = "1.74"
+rust-version = "1.79"
 include = ["src/**/*", "LICENSE.md", "README.md", "CHANGELOG.md", "build.rs"]
 
 [workspace]
@@ -36,8 +36,9 @@ small-fixed-array = { version = "0.4", features = ["serde"] }
 bool_to_bitflags = { version = "0.1.2" }
 nonmax = { version = "0.5.5", features = ["serde"] }
 strum = { version = "0.26", features = ["derive"] }
-to-arraystring = "0.1.0"
+to-arraystring = "0.2.0"
 extract_map = { version = "0.1.0", features = ["serde", "iter_mut"] }
+aformat = "0.1.3"
 # Optional dependencies
 fxhash = { version = "0.2.1", optional = true }
 chrono = { version = "0.4.31", default-features = false, features = ["clock", "serde"], optional = true }

src/builder/bot_auth_parameters.rs

@@ -1,7 +1,6 @@
 use std::borrow::Cow;
 
 use arrayvec::ArrayVec;
-use to_arraystring::ToArrayString;
 use url::Url;
 
 #[cfg(feature = "http")]

src/gateway/shard.rs

@@ -770,10 +770,9 @@ impl Shard {
 }
 
 async fn connect(base_url: &str) -> Result<WsClient> {
-    let url =
-        Url::parse(&format!("{base_url}?v={}", constants::GATEWAY_VERSION)).map_err(|why| {
-            warn!("Error building gateway URL with base `{}`: {:?}", base_url, why);
-
+    let url = Url::parse(&aformat!("{}?v={}", CapStr::<64>(base_url), constants::GATEWAY_VERSION))
+        .map_err(|why| {
+            warn!("Error building gateway URL with base `{base_url}`: {why:?}");
             Error::Gateway(GatewayError::BuildingUrl)
         })?;
 

src/http/client.rs

@@ -14,7 +14,6 @@ use reqwest::{Client, ClientBuilder, Response as ReqwestResponse, StatusCode};
 use secrecy::{ExposeSecret as _, Secret};
 use serde::de::DeserializeOwned;
 use serde_json::{from_value, json, to_string, to_vec};
-use to_arraystring::ToArrayString as _;
 use tracing::{debug, trace};
 
 use super::multipart::{Multipart, MultipartUpload};
@@ -219,7 +218,7 @@ fn parse_token(token: &str) -> Arc<str> {
     if token.starts_with("Bot ") || token.starts_with("Bearer ") {
         Arc::from(token)
     } else {
-        Arc::from(format!("Bot {token}"))
+        Arc::from(aformat!("Bot {}", CapStr::<128>(token)).as_str())
     }
 }
 

src/internal/prelude.rs

@@ -4,9 +4,11 @@
 
 pub use std::result::Result as StdResult;
 
+pub use aformat::{aformat, aformat_into, ArrayString, CapStr};
 pub use extract_map::{ExtractKey, ExtractMap, LendingIterator};
 pub use serde_json::Value;
 pub use small_fixed_array::{FixedArray, FixedString, TruncatingInto};
+pub use to_arraystring::ToArrayString;
 
 pub(crate) use super::utils::join_to_string;
 pub use crate::error::{Error, Result};

src/model/id.rs

@@ -3,9 +3,9 @@
 use std::fmt;
 
 use nonmax::NonMaxU64;
-use to_arraystring::ToArrayString;
 
 use super::Timestamp;
+use crate::internal::prelude::*;
 
 macro_rules! newtype_display_impl {
     ($name:ident) => {
@@ -116,6 +116,8 @@ macro_rules! id_u64 {
 
             impl ToArrayString for $name {
                 type ArrayString = <u64 as ToArrayString>::ArrayString;
+                const MAX_LENGTH: usize = <u64 as ToArrayString>::MAX_LENGTH;
+
                 fn to_arraystring(self) -> Self::ArrayString {
                     self.get().to_arraystring()
                 }

src/model/mention.rs

@@ -4,9 +4,8 @@ use std::fmt;
 #[cfg(all(feature = "model", feature = "utils"))]
 use std::str::FromStr;
 
-use to_arraystring::ToArrayString;
-
 use super::prelude::*;
+use crate::internal::prelude::*;
 #[cfg(all(feature = "model", feature = "utils"))]
 use crate::utils;
 
@@ -113,19 +112,16 @@ impl fmt::Display for Mention {
 }
 
 impl ToArrayString for Mention {
-    type ArrayString = arrayvec::ArrayString<{ 20 + 4 }>;
+    const MAX_LENGTH: usize = 20 + 4;
+    type ArrayString = ArrayString<{ 20 + 4 }>;
 
     fn to_arraystring(self) -> Self::ArrayString {
-        let (prefix, id) = match self {
-            Self::Channel(id) => ("<#", id.get()),
-            Self::Role(id) => ("<@&", id.get()),
-            Self::User(id) => ("<@", id.get()),
-        };
-
         let mut out = Self::ArrayString::new();
-        out.push_str(prefix);
-        out.push_str(&id.to_arraystring());
-        out.push('>');
+        match self {
+            Self::Channel(id) => aformat_into!(out, "<#{id}>"),
+            Self::Role(id) => aformat_into!(out, "<@&{id}>"),
+            Self::User(id) => aformat_into!(out, "<@{id}>"),
+        };
 
         out
     }

src/model/misc.rs

@@ -8,8 +8,6 @@ use std::fmt::Write;
 use std::result::Result as StdResult;
 use std::str::FromStr;
 
-use arrayvec::ArrayString;
-
 use super::prelude::*;
 use crate::internal::prelude::*;
 #[cfg(all(feature = "model", any(feature = "cache", feature = "utils")))]

src/utils/argument_convert/mod.rs

@@ -22,7 +22,7 @@ pub use role::*;
 mod emoji;
 pub use emoji::*;
 
-use super::DOMAINS;
+use super::{DOMAINS, MAX_DOMAIN_LEN};
 use crate::model::prelude::*;
 use crate::prelude::*;
 
@@ -124,8 +124,11 @@ pub fn parse_message_id_pair(s: &str) -> Option<(ChannelId, MessageId)> {
 /// ```
 #[must_use]
 pub fn parse_message_url(s: &str) -> Option<(GuildId, ChannelId, MessageId)> {
+    use aformat::{aformat, CapStr};
+
     for domain in DOMAINS {
-        if let Some(parts) = s.strip_prefix(&format!("https://{domain}/channels/")) {
+        let prefix = aformat!("https://{}/channels/", CapStr::<MAX_DOMAIN_LEN>(domain));
+        if let Some(parts) = s.strip_prefix(prefix.as_str()) {
             let mut parts = parts.splitn(3, '/');
 
             let guild_id = parts.next()?.parse().ok()?;