Add PAW provider. (#23)

* Added PAW.

* Implemented unique.

Author: Ameliaaaaaaa

Cargo.toml

@@ -38,14 +38,15 @@ tracing = "0.1"
 tracing-subscriber = { version = "0.3", features = ["env-filter", "time"] }
 
 [features]
-default = ["cache", "avtrdb", "vrcdb", "vrcds", "vrcwb", "title"]
+default = ["cache", "avtrdb", "vrcdb", "vrcds", "vrcwb", "paw", "title"]
 
 # VRChat Avatar Database Providers
 cache = ["dep:tokio-rusqlite-new"]
 avtrdb = ["dep:reqwest", "discord"]
 vrcdb = ["dep:reqwest", "discord"]
 vrcds = ["dep:reqwest", "discord"]
 vrcwb = ["dep:reqwest", "discord"]
+paw = ["dep:reqwest"]
 
 discord = ["dep:discord-presence", "dep:cached"]
 title = ["dep:crossterm"]

README.md

@@ -44,6 +44,7 @@ Place the file in the VRChat directory or `PATH` and set your launch options
 - [VRCDB - Avatar Search] - [Discord](https://discord.gg/q427ecnUvj) / [VRCX](https://vrcx.vrcdb.com/avatars/Avatar/VRCX) / [Web](https://vrcdb.com) / [World](https://vrchat.com/home/world/wrld_1146f625-5d42-40f5-bfe7-06a7664e2796)
 - [VRCDS - Project Dark Star] - [Discord](https://discord.gg/QT4uatfU8h) / [VRCX](https://avtr.nekosunevr.co.uk/vrcx_search) / [Web](https://avtr.nekosunevr.co.uk)
 - [VRCWB - World Balancer] - [Discord](https://discord.gg/Uw7aAShdsp) / [VRCX](https://avatar.worldbalancer.com/vrcx_search.php) / [Web](https://avatar.worldbalancer.com/search.php)
+- [PAW - Puppy's Avatar World] - [Discord](https://discord.gg/zHhs4nQYxX) / [VRCX](https://paw-api.amelia.fun/vrcx_search) / [Web](https://paw.amelia.fun)
 
 #### Unsupported Avatar Database Providers
 
@@ -66,3 +67,5 @@ Additional contributions welcome, please open an issue, pull request, or join Di
 [VRCWB - World Balancer]: https://avatar.worldbalancer.com
 
 [VRCX]: https://github.com/vrcx-team/VRCX?tab=readme-ov-file#--vrcx
+
+[PAW - Puppy's Avatar World]: https://paw.amelia.fun

src/lib.rs

@@ -127,6 +127,9 @@ pub async fn process_avatars((_tx, rx): (Sender<PathBuf>, Receiver<PathBuf>)) ->
     #[cfg(feature = "vrcwb")]
     let vrcwb = VrcWB::default();
 
+    #[cfg(feature = "paw")]
+    let paw = Paw::default();
+
     let providers = VecDeque::from([
         #[cfg(feature = "avtrdb")]
         Type::AVTRDB(&avtrdb),
@@ -136,6 +139,8 @@ pub async fn process_avatars((_tx, rx): (Sender<PathBuf>, Receiver<PathBuf>)) ->
         Type::VRCDS(&vrcds),
         #[cfg(feature = "vrcwb")]
         Type::VRCWB(&vrcwb),
+        #[cfg(feature = "paw")]
+        Type::PAW(&paw),
     ]);
 
     while let Ok(path) = rx.recv() {

src/provider/mod.rs

@@ -14,6 +14,8 @@ pub mod vrcdb;
 pub mod vrcds;
 #[cfg(feature = "vrcwb")]
 pub mod vrcwb;
+#[cfg(feature = "paw")]
+pub mod paw;
 
 pub mod prelude;
 
@@ -33,6 +35,9 @@ pub enum Type<'a> {
     #[cfg(feature = "vrcwb")]
     #[strum(to_string = "VRCWB - World Balancer")]
     VRCWB(&'a VrcWB),
+    #[cfg(feature = "paw")]
+    #[strum(to_string = "PAW - Puppy's Avatar World")]
+    PAW(&'a Paw),
 }
 
 #[async_trait]
@@ -74,6 +79,8 @@ impl Provider for Type<'_> {
             Type::VRCDS(p) => p.send_avatar_id(avatar_id).await,
             #[cfg(feature = "vrcwb")]
             Type::VRCWB(p) => p.send_avatar_id(avatar_id).await,
+            #[cfg(feature = "paw")]
+            Type::PAW(p) => p.send_avatar_id(avatar_id).await,
         }
     }
 }

src/provider/paw.rs

@@ -0,0 +1,93 @@
+use std::{time::Duration};
+
+use anyhow::{bail, Result};
+use async_trait::async_trait;
+use reqwest::{Client, StatusCode};
+use serde::Deserialize;
+
+use crate::{
+    provider::{Provider, Type},
+    USER_AGENT,
+};
+
+const URL: &str = "https://paw-api.amelia.fun/update";
+const AVATAR_URL: &str = "https://paw-api.amelia.fun/avatar";
+
+pub struct Paw {
+    client: Client,
+}
+
+impl Default for Paw {
+    fn default() -> Self {
+        Self {
+            client: Client::default(),
+        }
+    }
+}
+
+#[derive(Debug, Deserialize)]
+struct PawResponse {
+    success: bool,
+    code: u16,
+    result: Option<serde_json::Value>,
+    avatar: Option<serde_json::Value>,
+}
+
+#[async_trait]
+impl Provider for Paw {
+    async fn check_avatar_id(&self, _avatar_id: &str) -> Result<bool> {
+        let name = Type::PAW(self);
+        let response = self
+            .client
+            .get(AVATAR_URL)
+            .header("User-Agent", USER_AGENT)
+            .query(&[("avatarId", _avatar_id)])
+            .timeout(Duration::from_secs(3))
+            .send()
+            .await?;
+
+        let status = response.status();
+        let text = response.text().await?;
+        debug!("[{name}] {status} | {text}");
+
+        if status != StatusCode::OK {
+            bail!("[{name}] Failed to check avatar: {status} | {text}");
+        }
+
+        let data = serde_json::from_str::<PawResponse>(&text)?;
+
+        Ok(data.success && data.code == 200 && data.result.is_some())
+    }
+
+    async fn send_avatar_id(&self, avatar_id: &str) -> Result<bool> {
+        let name = Type::PAW(self);
+        let response = self
+            .client
+            .post(URL)
+            .header("User-Agent", USER_AGENT)
+            .query(&[("avatarId", avatar_id)])
+            .timeout(Duration::from_secs(3))
+            .send()
+            .await?;
+
+        let status = response.status();
+        let text = response.text().await?;
+        debug!("[{name}] {status} | {text}");
+
+        let unique = match status {
+            StatusCode::OK => {
+                let data = serde_json::from_str::<PawResponse>(&text)?;
+
+                !matches!(data.avatar.as_ref(), Some(avatar) if !avatar.is_null() && !(avatar.is_array() && avatar.as_array().unwrap().is_empty()))
+            },
+            StatusCode::TOO_MANY_REQUESTS => {
+                warn!("[{name}] 429 Rate Limit, Please Wait 10 seconds...");
+                tokio::time::sleep(Duration::from_secs(10)).await;
+                Box::pin(self.send_avatar_id(avatar_id)).await?
+            }
+            _ => bail!("[{name}] {status} | {text}"),
+        };
+
+        Ok(unique)
+    }
+}

src/provider/prelude.rs

@@ -8,3 +8,5 @@ pub use super::vrcdb::VrcDB;
 pub use super::vrcds::VrcDS;
 #[cfg(feature = "vrcwb")]
 pub use super::vrcwb::VrcWB;
+#[cfg(feature = "paw")]
+pub use super::paw::Paw;