Separate Module for Lemmy HTTP requests

src/lemmy/mod.rs

@@ -0,0 +1,346 @@
+use std::collections::HashMap;
+use std::env;
+use std::env::VarError;
+use std::error::Error;
+use chrono::NaiveDateTime;
+use lemmy_api_common::community::{ListCommunities, ListCommunitiesResponse};
+use lemmy_api_common::lemmy_db_views::structs::PostView;
+use lemmy_api_common::lemmy_db_views_actor::structs::CommunityView;
+use lemmy_api_common::person::{Login, LoginResponse};
+use lemmy_api_common::post::{CreatePost, FeaturePost, GetPosts, GetPostsResponse};
+use lemmy_api_common::sensitive::Sensitive;
+use lemmy_db_schema::newtypes::{CommunityId, DbUrl, InstanceId, LanguageId, PersonId, PostId};
+use lemmy_db_schema::{ListingType, PostFeatureType, SubscribedType};
+use reqwest::StatusCode;
+use serde_derive::{Deserialize, Serialize};
+use serde_json::json;
+use url::Url;
+use crate::HTTP_CLIENT;
+
+pub(crate) struct Credentials {
+    username: String,
+    password: String
+}
+
+impl Credentials {
+    pub(crate) fn get_username(&self) -> Sensitive<String> {
+        return Sensitive::new(self.username.clone());
+    }
+
+    pub(crate) fn get_password(&self) -> Sensitive<String> {
+        return Sensitive::new(self.password.clone());
+    }
+
+    pub(crate) fn set_credentials() -> Result<Self, VarError> {
+        let username = env::var("LEMMY_USERNAME")?;
+        let password = env::var("LEMMY_PASSWORD")?;
+        Ok(Credentials {
+            username,
+            password,
+        })
+    }
+}
+
+pub(crate) struct Lemmy {
+    jwt_token: Sensitive<String>,
+    instance: String,
+}
+
+pub(crate) async fn login(credentials: &Credentials, instance: &str) -> Result<Lemmy, Box<dyn Error>> {
+    let login_params = Login {
+        username_or_email: credentials.get_username(),
+        password: credentials.get_password(),
+        totp_2fa_token: None,
+    };
+
+    let response = HTTP_CLIENT
+        .post(instance.to_string() + "/api/v3/user/login")
+        .json(&login_params)
+        .send()
+        .await?;
+
+    match response.status() {
+        StatusCode::OK => {
+            let data: LoginResponse = response.json().await.expect("Successful Login Request should return JSON");
+            match data.jwt {
+                Some(token) => Ok(Lemmy {
+                    jwt_token: token.clone(),
+                    instance: instance.to_string(),
+                }),
+                None => Err(panic!("Login did not return JWT token. Are the credentials valid?"))
+            }
+        },
+        status => Err(panic!("Unexpected HTTP Status '{}' during Login", status.to_string()))
+    }
+}
+
+impl Lemmy {
+    pub(crate) async fn post(&self, post: CustomCreatePost) -> Result<PostId, Box<dyn Error>> {
+        let response = HTTP_CLIENT
+            .post(format!("{}/api/v3/post", &self.instance))
+            .bearer_auth(&self.jwt_token.to_string())
+            .json(&post)
+            .send()
+            .await?
+            .text()
+            .await?;
+
+        let json_data = serde_json::from_str::<HashMap<&str, CustomPostView>>(&response)?
+            .remove("post_view")
+            .expect("Element should be present");
+
+        Ok(json_data.post.id)
+    }
+
+    async fn feature(&self, params: CustomFeaturePost) -> Result<CustomPostView, Box<dyn Error>> {
+        let response = HTTP_CLIENT
+            .post(format!("{}/api/v3/post/feature", &self.instance))
+            .bearer_auth(&self.jwt_token.to_string())
+            .json(&params)
+            .send()
+            .await?
+            .text()
+            .await?;
+
+        let json_data = serde_json::from_str::<HashMap<&str, CustomPostView>>(&response)?
+            .remove("post_view")
+            .expect("Element should be present");
+        Ok(json_data)
+    }
+
+    pub(crate) async fn unpin(&self, post_id: PostId, location: PostFeatureType) -> Result<CustomPostView, Box<dyn Error>> {
+        let pin_params = CustomFeaturePost {
+            post_id,
+            featured: false,
+            feature_type: location,
+        };
+        self.feature(pin_params).await
+    }
+
+    pub(crate) async fn pin(&self, post_id: PostId, location: PostFeatureType) -> Result<CustomPostView, Box<dyn Error>> {
+        let pin_params = CustomFeaturePost {
+            post_id,
+            featured: true,
+            feature_type: location,
+        };
+        self.feature(pin_params).await
+    }
+
+    pub(crate) async fn get_community_pinned(&self, community: CommunityId) -> Result<Vec<CustomPostView>, Box<dyn Error>> {
+        let list_params = GetPosts {
+            community_id: Some(community),
+            type_: Some(ListingType::Local),
+            ..Default::default()
+        };
+
+        let response = HTTP_CLIENT
+            .get(format!("{}/api/v3/post/list", &self.instance))
+            .bearer_auth(&self.jwt_token.to_string())
+            .query(&list_params)
+            .send()
+            .await?
+            .text()
+            .await?;
+
+        let json_data: CustomGetPostsResponse = serde_json::from_str(&response)?;
+
+        Ok(json_data.posts.iter().filter(|post| {
+                post.post.featured_community
+            })
+            .cloned()
+            .collect()
+        )
+    }
+
+    pub(crate) async fn get_local_pinned(&self) -> Result<Vec<CustomPostView>, Box<dyn Error>> {
+        let list_params = GetPosts {
+            type_: Some(ListingType::Local),
+            ..Default::default()
+        };
+
+        let response = HTTP_CLIENT
+            .get(format!("{}/api/v3/post/list", &self.instance))
+            .bearer_auth(&self.jwt_token.to_string())
+            .query(&list_params)
+            .send()
+            .await?
+            .text()
+            .await?;
+
+        let json_data: CustomGetPostsResponse = serde_json::from_str(&response)?;
+
+        Ok(json_data.posts.iter().filter(|post| {
+            post.post.featured_local
+        })
+            .cloned()
+            .collect()
+        )
+    }
+
+    pub(crate) async fn get_communities(&self) -> Result<HashMap<String, CommunityId>, Box<dyn Error>> {
+        let list_params = ListCommunities {
+            type_: Some(ListingType::Local),
+            ..Default::default()
+        };
+
+        let response = HTTP_CLIENT
+            .get(format!("{}/api/v3/community/list", &self.instance))
+            .bearer_auth(&self.jwt_token.to_string())
+            .query(&list_params)
+            .send()
+            .await?
+            .text()
+            .await?;
+
+        let json_data: CustomListCommunitiesResponse = serde_json::from_str(&response)?;
+
+        let mut communities: HashMap<String, CommunityId> = HashMap::new();
+        for community_view in json_data.communities {
+            let community = community_view.community;
+            communities.insert(community.name, community.id);
+        }
+
+        return Ok(communities)
+    }
+}
+
+#[derive(Clone, Debug, Default, Deserialize, Serialize)]
+pub(crate) struct CustomCreatePost {
+    pub(crate) name: String,
+    pub(crate) community_id: CommunityId,
+    pub(crate) url: Option<Url>,
+    pub(crate) body: Option<String>,
+    pub(crate) honeypot: Option<String>,
+    pub(crate) nsfw: Option<bool>,
+    pub(crate) language_id: Option<LanguageId>,
+}
+
+#[derive(Clone, Debug, Default, Deserialize, Serialize)]
+pub(crate) struct CustomFeaturePost {
+    pub(crate) post_id: PostId,
+    pub(crate) featured: bool,
+    pub(crate) feature_type: PostFeatureType,
+}
+
+#[derive(Clone, Debug, Default, Deserialize, Serialize)]
+pub(crate) struct CustomListCommunitiesResponse {
+    pub(crate) communities: Vec<CustomCommunityView>
+}
+
+#[derive(Clone, Debug, Deserialize, Serialize)]
+pub(crate) struct CustomCommunityView {
+    pub(crate) community: CustomCommunity,
+    pub(crate) subscribed: SubscribedType,
+    pub(crate) blocked: bool,
+    pub(crate) counts: CustomCommunityAggregates
+}
+
+#[derive(Clone, Debug, Deserialize, Serialize)]
+pub(crate) struct CustomCommunity {
+    pub(crate) actor_id: String,
+    pub(crate) banner: Option<String>,
+    pub(crate) deleted: bool,
+    pub(crate) description: Option<String>,
+    pub(crate) hidden: bool,
+    pub(crate) icon: Option<String>,
+    pub(crate) id: CommunityId,
+    pub(crate) instance_id: i32,
+    pub(crate) local: bool,
+    pub(crate) name: String,
+    pub(crate) nsfw: bool,
+    pub(crate) posting_restricted_to_mods: bool,
+    pub(crate) published: String,
+    pub(crate) removed: bool,
+    pub(crate) title: String,
+    pub(crate) updated: Option<String>,
+}
+
+#[derive(Clone, Debug, Deserialize, Serialize)]
+pub(crate) struct CustomCommunityAggregates {
+    pub(crate) comments: i64,
+    pub(crate) community_id: i32,
+    pub(crate) posts: i64,
+    pub(crate) published: String,
+    pub(crate) subscribers: i64,
+    pub(crate) users_active_day: i64,
+    pub(crate) users_active_half_year: i64,
+    pub(crate) users_active_month: i64,
+    pub(crate) users_active_week: i64,
+}
+
+#[derive(Clone, Debug, Deserialize, Serialize)]
+pub(crate) struct CustomPostView {
+    pub(crate) community: CustomCommunity,
+    pub(crate) counts: CustomPostAggregates,
+    pub(crate) creator: CustomPerson,
+    pub(crate) creator_banned_from_community: bool,
+    pub(crate) creator_blocked: bool,
+    pub(crate) creator_is_admin: bool,
+    pub(crate) creator_is_moderator: bool,
+    pub(crate) my_vote: Option<i16>,
+    pub(crate) post: CustomPost,
+    pub(crate) read: bool,
+    pub(crate) saved: bool,
+    pub(crate) subscribed: SubscribedType,
+    pub(crate) unread_comments: i64,
+}
+
+#[derive(Clone, Debug, Deserialize, Serialize)]
+pub(crate) struct CustomPostAggregates {
+    pub(crate) comments: i64,
+    pub(crate) downvotes: i64,
+    pub(crate) post_id: PostId,
+    pub(crate) published: String,
+    pub(crate) score: i64,
+    pub(crate) upvotes: i64,
+}
+
+#[derive(Clone, Debug, Deserialize, Serialize)]
+pub(crate) struct CustomPerson {
+    pub(crate) actor_id: String,
+    pub(crate) avatar: Option<String>,
+    pub(crate) ban_expires: Option<String>,
+    pub(crate) banned: bool,
+    pub(crate) banner: Option<String>,
+    pub(crate) bio: Option<String>,
+    pub(crate) bot_account: bool,
+    pub(crate) deleted: bool,
+    pub(crate) display_name: Option<String>,
+    pub(crate) id: PersonId,
+    pub(crate) instance_id : InstanceId,
+    pub(crate) local: bool,
+    pub(crate) matrix_user_id: Option<String>,
+    pub(crate) name: String,
+    pub(crate) published: String,
+    pub(crate) updated: Option<String>,
+}
+
+#[derive(Clone, Debug, Deserialize, Serialize)]
+pub(crate) struct CustomPost {
+    pub(crate) id: PostId,
+    pub(crate) name: String,
+    pub(crate) url: Option<String>,
+    pub(crate) body: Option<String>,
+    pub(crate) creator_id: PersonId,
+    pub(crate) community_id: CommunityId,
+    pub(crate) removed: bool,
+    pub(crate) locked: bool,
+    pub(crate) published: String,
+    pub(crate) updated: Option<String>,
+    pub(crate) deleted: bool,
+    pub(crate) nsfw: bool,
+    pub(crate) embed_title: Option<String>,
+    pub(crate) embed_description: Option<String>,
+    pub(crate) thumbnail_url: Option<String>,
+    pub(crate) ap_id: String,
+    pub(crate) local: bool,
+    pub(crate) embed_video_url: Option<String>,
+    pub(crate) language_id: LanguageId,
+    pub(crate) featured_community: bool,
+    pub(crate) featured_local: bool,
+}
+
+#[derive(Clone, Debug, Deserialize, Serialize)]
+pub(crate) struct CustomGetPostsResponse {
+    pub(crate) posts: Vec<CustomPostView>,
+}