Initial Anrok integration

Author: fetchfern

apps/labrinth/migrations/20250910145542_tax-charges.sql

@@ -0,0 +1,10 @@
+ALTER TABLE charges ADD COLUMN tax_amount BIGINT NOT NULL DEFAULT 0;
+ALTER TABLE charges ADD COLUMN tax_platform_id TEXT;
+
+ALTER TABLE users_subscriptions ADD COLUMN user_aware_of_tax_changes BOOLEAN NOT NULL DEFAULT FALSE;
+
+CREATE TABLE products_tax_identifiers (
+    id SERIAL PRIMARY KEY,
+    tax_processor_id TEXT NOT NULL,
+    product_id BIGINT REFERENCES products(id) NOT NULL
+);

apps/labrinth/src/database/models/charge_item.rs

@@ -26,6 +26,9 @@ pub struct DBCharge {
 
     pub parent_charge_id: Option<DBChargeId>,
 
+    pub tax_amount: i64,
+    pub tax_platform_id: Option<String>,
+
     // Net is always in USD
     pub net: Option<i64>,
 }
@@ -45,6 +48,8 @@ struct ChargeQueryResult {
     payment_platform: String,
     payment_platform_id: Option<String>,
     parent_charge_id: Option<i64>,
+    tax_amount: i64,
+    tax_platform_id: Option<String>,
     net: Option<i64>,
 }
 
@@ -69,6 +74,8 @@ impl TryFrom<ChargeQueryResult> for DBCharge {
             payment_platform: PaymentPlatform::from_string(&r.payment_platform),
             payment_platform_id: r.payment_platform_id,
             parent_charge_id: r.parent_charge_id.map(DBChargeId),
+            tax_amount: r.tax_amount,
+            tax_platform_id: r.tax_platform_id,
             net: r.net,
         })
     }
@@ -81,7 +88,7 @@ macro_rules! select_charges_with_predicate {
             r#"
             SELECT
                 id, user_id, price_id, amount, currency_code, status, due, last_attempt,
-                charge_type, subscription_id,
+                charge_type, subscription_id, tax_amount, tax_platform_id,
                 -- Workaround for https://github.com/launchbadge/sqlx/issues/3336
                 subscription_interval AS "subscription_interval?",
                 payment_platform,
@@ -103,8 +110,8 @@ impl DBCharge {
     ) -> Result<DBChargeId, DatabaseError> {
         sqlx::query!(
             r#"
-            INSERT INTO charges (id, user_id, price_id, amount, currency_code, charge_type, status, due, last_attempt, subscription_id, subscription_interval, payment_platform, payment_platform_id, parent_charge_id, net)
-            VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15)
+            INSERT INTO charges (id, user_id, price_id, amount, currency_code, charge_type, status, due, last_attempt, subscription_id, subscription_interval, payment_platform, payment_platform_id, parent_charge_id, net, tax_amount, tax_platform_id)
+            VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17)
             ON CONFLICT (id)
             DO UPDATE
                 SET status = EXCLUDED.status,
@@ -116,6 +123,8 @@ impl DBCharge {
                     payment_platform_id = EXCLUDED.payment_platform_id,
                     parent_charge_id = EXCLUDED.parent_charge_id,
                     net = EXCLUDED.net,
+                    tax_amount = EXCLUDED.tax_amount,
+                    tax_platform_id = EXCLUDED.tax_platform_id,
                     price_id = EXCLUDED.price_id,
                     amount = EXCLUDED.amount,
                     currency_code = EXCLUDED.currency_code,
@@ -136,6 +145,8 @@ impl DBCharge {
             self.payment_platform_id.as_deref(),
             self.parent_charge_id.map(|x| x.0),
             self.net,
+            self.tax_amount,
+            self.tax_platform_id.as_deref(),
         )
             .execute(&mut **transaction)
         .await?;

apps/labrinth/src/database/models/mod.rs

@@ -17,6 +17,7 @@ pub mod organization_item;
 pub mod pat_item;
 pub mod payout_item;
 pub mod product_item;
+pub mod products_tax_identifier_item;
 pub mod project_item;
 pub mod report_item;
 pub mod session_item;

apps/labrinth/src/database/models/products_tax_identifier_item.rs

@@ -0,0 +1,66 @@
+use crate::database::models::ids::{DBProductId, DBProductPriceId};
+use crate::models::billing::ProductMetadata;
+use crate::routes::ApiError;
+
+pub struct DBProductsTaxIdentifier {
+    pub id: i32,
+    pub tax_processor_id: String,
+    pub product_id: DBProductId,
+}
+
+pub struct ProductInfo {
+    pub tax_identifier: DBProductsTaxIdentifier,
+    pub product_metadata: ProductMetadata,
+}
+
+impl DBProductsTaxIdentifier {
+    pub async fn get_product(
+        product_id: DBProductId,
+        exec: impl sqlx::Executor<'_, Database = sqlx::Postgres>,
+    ) -> Result<Option<Self>, ApiError> {
+        let maybe_row = sqlx::query!(
+            "SELECT * FROM products_tax_identifiers WHERE product_id = $1",
+            product_id.0,
+        )
+        .fetch_optional(exec)
+        .await?;
+
+        Ok(maybe_row.map(|row| DBProductsTaxIdentifier {
+            id: row.id,
+            tax_processor_id: row.tax_processor_id,
+            product_id: DBProductId(row.product_id),
+        }))
+    }
+}
+
+pub async fn product_info_by_product_price_id(
+    product_price_id: DBProductPriceId,
+    exec: impl sqlx::Executor<'_, Database = sqlx::Postgres>,
+) -> Result<Option<ProductInfo>, ApiError> {
+    let maybe_row = sqlx::query!(
+        r#"
+        SELECT
+          products_tax_identifiers.*,
+          products.metadata product_metadata
+        FROM products_prices
+        LEFT JOIN products ON products.id = products_prices.product_id
+        LEFT JOIN products_tax_identifiers ON products_tax_identifiers.product_id = products.id
+        WHERE products_prices.id = $1
+        "#,
+        product_price_id.0 as i64,
+    )
+    .fetch_optional(exec)
+    .await?;
+
+    match maybe_row {
+        None => Ok(None),
+        Some(row) => Ok(Some(ProductInfo {
+            tax_identifier: DBProductsTaxIdentifier {
+                id: row.id,
+                tax_processor_id: row.tax_processor_id,
+                product_id: DBProductId(row.product_id),
+            },
+            product_metadata: serde_json::from_value(row.product_metadata)?,
+        })),
+    }
+}

apps/labrinth/src/lib.rs

@@ -487,6 +487,9 @@ pub fn check_env_vars() -> bool {
     failed |= check_var::<String>("AVALARA_1099_API_TEAM_ID");
     failed |= check_var::<String>("AVALARA_1099_COMPANY_ID");
 
+    failed |= check_var::<String>("ANROK_API_URL");
+    failed |= check_var::<String>("ANROK_API_KEY");
+
     failed |= check_var::<String>("COMPLIANCE_PAYOUT_THRESHOLD");
 
     failed |= check_var::<String>("ARCHON_URL");