Payouts code (#765)

* push to rebase * finish most * finish most * Finish impl * Finish paypal * run prep * Fix comp err
2023-11-29 11:00:08 -07:00
parent f731c1080d
commit d4f9c97cca
56 changed files with 2210 additions and 1420 deletions
--- a/src/database/models/payout_item.rs
+++ b/src/database/models/payout_item.rs
@@ -0,0 +1,117 @@
+use crate::models::payouts::{PayoutMethodType, PayoutStatus};
+use chrono::{DateTime, Utc};
+use rust_decimal::Decimal;
+use serde::{Deserialize, Serialize};
+
+use super::{DatabaseError, PayoutId, UserId};
+
+#[derive(Deserialize, Serialize, Clone, Debug)]
+pub struct Payout {
+    pub id: PayoutId,
+    pub user_id: UserId,
+    pub created: DateTime<Utc>,
+    pub status: PayoutStatus,
+    pub amount: Decimal,
+
+    pub fee: Option<Decimal>,
+    pub method: Option<PayoutMethodType>,
+    pub method_address: Option<String>,
+    pub platform_id: Option<String>,
+}
+
+impl Payout {
+    pub async fn insert(
+        &self,
+        transaction: &mut sqlx::Transaction<'_, sqlx::Postgres>,
+    ) -> Result<(), DatabaseError> {
+        sqlx::query!(
+            "
+            INSERT INTO payouts (
+                id, amount, fee, user_id, status, method, method_address, platform_id
+            )
+            VALUES (
+                $1, $2, $3, $4, $5, $6, $7, $8
+            )
+            ",
+            self.id.0,
+            self.amount,
+            self.fee,
+            self.user_id.0,
+            self.status.as_str(),
+            self.method.map(|x| x.as_str()),
+            self.method_address,
+            self.platform_id,
+        )
+        .execute(&mut **transaction)
+        .await?;
+
+        Ok(())
+    }
+
+    pub async fn get<'a, 'b, E>(id: PayoutId, executor: E) -> Result<Option<Payout>, DatabaseError>
+    where
+        E: sqlx::Executor<'a, Database = sqlx::Postgres>,
+    {
+        Payout::get_many(&[id], executor)
+            .await
+            .map(|x| x.into_iter().next())
+    }
+
+    pub async fn get_many<'a, E>(
+        payout_ids: &[PayoutId],
+        exec: E,
+    ) -> Result<Vec<Payout>, DatabaseError>
+    where
+        E: sqlx::Executor<'a, Database = sqlx::Postgres>,
+    {
+        use futures::TryStreamExt;
+
+        let results = sqlx::query!(
+            "
+            SELECT id, user_id, created, amount, status, method, method_address, platform_id, fee
+            FROM payouts
+            WHERE id = ANY($1)
+            ",
+            &payout_ids.into_iter().map(|x| x.0).collect::<Vec<_>>()
+        )
+        .fetch_many(exec)
+        .try_filter_map(|e| async {
+            Ok(e.right().map(|r| Payout {
+                id: PayoutId(r.id),
+                user_id: UserId(r.user_id),
+                created: r.created,
+                status: PayoutStatus::from_string(&r.status),
+                amount: r.amount,
+                method: r.method.map(|x| PayoutMethodType::from_string(&x)),
+                method_address: r.method_address,
+                platform_id: r.platform_id,
+                fee: r.fee,
+            }))
+        })
+        .try_collect::<Vec<Payout>>()
+        .await?;
+
+        Ok(results)
+    }
+
+    pub async fn get_all_for_user(
+        user_id: UserId,
+        exec: impl sqlx::Executor<'_, Database = sqlx::Postgres>,
+    ) -> Result<Vec<PayoutId>, DatabaseError> {
+        let results = sqlx::query!(
+            "
+            SELECT id
+            FROM payouts
+            WHERE user_id = $1
+            ",
+            user_id.0
+        )
+        .fetch_all(exec)
+        .await?;
+
+        Ok(results
+            .into_iter()
+            .map(|r| PayoutId(r.id))
+            .collect::<Vec<_>>())
+    }
+}