feat: add small widgets

Author: null8626

src/lib.rs

@@ -29,6 +29,9 @@ cfg_if::cfg_if! {
     pub use client::Client;
     pub use error::{Error, Result};
     pub use snowflake::Snowflake; // for doc purposes
+
+    #[doc(inline)]
+    pub use widget::WidgetType;
   }
 }
 

src/widget.rs

@@ -1,10 +1,65 @@
 use crate::Snowflake;
 
+/// Widget type.
+#[non_exhaustive]
+pub enum WidgetType {
+  DiscordBot,
+  DiscordServer,
+}
+
+impl WidgetType {
+  const fn as_path(&self) -> &'static str {
+    match self {
+      Self::DiscordBot => "discord/bot",
+      Self::DiscordServer => "discord/server",
+    }
+  }
+}
+
 /// Generates a large widget URL.
 #[inline(always)]
-pub fn large<I>(id: I) -> String
+pub fn large<I>(ty: WidgetType, id: I) -> String
+where
+  I: Snowflake,
+{
+  crate::client::api!("/widgets/large/{}/{}", ty.as_path(), id.as_snowflake())
+}
+
+/// Generates a small widget URL for displaying votes.
+#[inline(always)]
+pub fn votes<I>(ty: WidgetType, id: I) -> String
+where
+  I: Snowflake,
+{
+  crate::client::api!(
+    "/widgets/small/votes/{}/{}",
+    ty.as_path(),
+    id.as_snowflake()
+  )
+}
+
+/// Generates a small widget URL for displaying an entity's owner.
+#[inline(always)]
+pub fn owner<I>(ty: WidgetType, id: I) -> String
+where
+  I: Snowflake,
+{
+  crate::client::api!(
+    "/widgets/small/owner/{}/{}",
+    ty.as_path(),
+    id.as_snowflake()
+  )
+}
+
+/// Generates a small widget URL for displaying social stats.
+#[inline(always)]
+pub fn social<I>(ty: WidgetType, id: I) -> String
 where
   I: Snowflake,
 {
-  crate::client::api!("/widgets/large/{}", id.as_snowflake())
+  crate::client::api!(
+    "/widgets/small/social/{}/{}",
+    ty.as_path(),
+    id.as_snowflake()
+  )
 }