small refactored

src/data.rs

@@ -0,0 +1,83 @@
+
+use reqwest::Client as HTTPClient;
+use serde::{Serialize, Deserialize};
+use serde_json::Value;
+use std::error::Error;
+use std::env;
+use uuid::Uuid;
+use chrono::Utc;
+
+
+#[derive(Debug, Serialize, Deserialize)]
+#[serde(rename_all = "snake_case")]
+enum PayloadKind {
+    NewMessage,
+    NewFollower,
+    NewShout,
+    NewApproval,
+    NewComment,
+    NewRate,
+}
+
+#[derive(Debug, Serialize, Deserialize)]
+struct Payload {
+    chat_id: Option<String>,
+    shout_id: Option<i32>,
+    author_id: Option<i32>,
+    topic_id: Option<i32>,
+    reaction_id: Option<i32>,
+    community_id: Option<i32>,
+    kind: PayloadKind,
+    body: String,
+}
+
+pub async fn get_auth_id(token: &str) -> Result<i32, Box<dyn Error>> {
+    let api_base = env::var("API_BASE")?;
+    let gql = match api_base.contains("v2") {
+        true => r#"mutation { getSession { user { id } } }"#, // v2
+        _ => r#"query { sessiom { user { id } } }"# // authorizer
+    };
+    let client = HTTPClient::new();
+    let response = client
+        .post(api_base)
+        .bearer_auth(token) // NOTE: auth token is here
+        .body(gql)
+        .send()
+        .await?;
+    let response_body: Value = response.json().await?;
+    let id = response_body["data"]["getSession"]["user"]["id"]
+        .as_i64()
+        .ok_or("Failed to get user id by token")? as i32;
+    Ok(id)
+}
+
+pub async fn create_first_chat(author_id: i32, con: &mut redis::aio::Connection) -> Result<Vec<String>, Box<dyn Error>> {
+    let chat_id = Uuid::new_v4().to_string();
+    let members = vec![author_id.to_string(), "1".to_string()];
+    let timestamp = Utc::now().timestamp();
+
+    let chat = serde_json::json!({
+        "id": chat_id.clone(),
+        "admins": members.clone(),
+        "members": members.clone(),
+        "title": "",
+        "createdBy": author_id,
+        "createdAt": timestamp,
+        "updatedAt": timestamp,
+    });
+
+    let _: () = redis::pipe()
+        .atomic()
+        .cmd("SADD")
+        .arg(format!("chats_by_author/{}", author_id))
+        .arg(&chat_id)
+        .ignore()
+        .set(format!("chats/{}", chat_id), chat.to_string())
+        .ignore()
+        .set(format!("chats/{}/next_message_id", chat_id), "0")
+        .ignore()
+        .query_async(con)
+        .await?;
+
+    Ok(vec![chat_id])
+}