Merges pull request #5

wallet views

2 files changed, 122 insertions, 14 deletions

diff --git a/jb/managers/email_manager.py b/jb/managers/email_manager.py
index 7298c60..08ac0cf 100644
--- a/jb/managers/email_manager.py
+++ b/jb/managers/email_manager.py
@@ -1,10 +1,19 @@
 import requests
 from generalresearch.config import is_debug
+from generalresearch.currency import USDCent
+from generalresearch.models.thl.wallet.cashout_method import (
+    CashoutMethodOut,
+    CashoutRequestInfo,
+)
 
 from jb.config import settings
 
 MAUTIC_BASE_URL = "https://mail.jamesbillings67.com"
-EMAIL_TEMPLATE_ID = 1
+LOGIN_EMAIL_TEMPLATE_ID = 1
+# Todo: We need a new template for the cashout confirmation
+CASHOUT_REQUEST_EMAIL_TEMPLATE_ID = 3
+# Todo: We need a new template for the cashout status / sent
+CASHOUT_STATUS_EMAIL_TEMPLATE_ID = 4
 
 assert settings.mautic_api_key
 auth_headers = {"Authorization": f"Basic {settings.mautic_api_key.get_secret_value()}"}
@@ -26,11 +35,14 @@ def send_login_email_from_url(mautic_url: str, magic_link: str) -> None:
     if is_debug():
         print("MAGIC_LINK: ", magic_link)
         return
-
     email_tokens = {
         "magic_link": magic_link,
     }
-    body = {"tokens": email_tokens}
+    send_email_with_tokens(email_tokens, mautic_url)
+
+
+def send_email_with_tokens(tokens: dict[str, str], mautic_url: str) -> None:
+    body = {"tokens": tokens}
     response = requests.post(url=mautic_url, json=body, headers=auth_headers)
 
     try:
@@ -47,9 +59,7 @@ def send_login_email_from_url(mautic_url: str, magic_link: str) -> None:
 
 def send_login_email(email: str, magic_token: str) -> None:
     contact_id = get_or_create_contact(email=email)
-    mautic_url = (
-        f"{MAUTIC_BASE_URL}/api/emails/{EMAIL_TEMPLATE_ID}/contact/{contact_id}/send"
-    )
+    mautic_url = f"{MAUTIC_BASE_URL}/api/emails/{LOGIN_EMAIL_TEMPLATE_ID}/contact/{contact_id}/send"
     magic_link = f"{settings.base_url}auth/magic-link/?token={magic_token}"
     send_login_email_from_url(mautic_url, magic_link)
 
@@ -58,8 +68,45 @@ def send_amt_link_email(email: str, magic_token: str) -> None:
     # Don't actually associate the email with the worker ID
     #  until they click the link
     contact_id = get_or_create_contact(email=email)
-    mautic_url = (
-        f"{MAUTIC_BASE_URL}/api/emails/{EMAIL_TEMPLATE_ID}/contact/{contact_id}/send"
-    )
+    mautic_url = f"{MAUTIC_BASE_URL}/api/emails/{LOGIN_EMAIL_TEMPLATE_ID}/contact/{contact_id}/send"
     magic_link = f"{settings.base_url}auth/link-amt/?token={magic_token}"
     send_login_email_from_url(mautic_url, magic_link)
+
+
+def send_cashout_confirmation_email(
+    email: str, token: str, cashout_method: CashoutMethodOut, amount: USDCent
+) -> None:
+    contact_id = get_or_create_contact(email=email)
+    mautic_url = f"{MAUTIC_BASE_URL}/api/emails/{CASHOUT_REQUEST_EMAIL_TEMPLATE_ID}/contact/{contact_id}/send"
+    magic_link = f"{settings.base_url}wallet/cashout/confirm/?token={token}"
+    if is_debug():
+        print("MAGIC_LINK: ", magic_link)
+        return
+    email_tokens = {
+        "magic_link": magic_link,
+        "cashout_method_name": cashout_method.name,
+        "cashout_method_description": cashout_method.description,
+        "cashout_method_type": cashout_method.type.value,
+        "amount": amount.to_usd_str(),
+    }
+    if cashout_method.image_url:
+        email_tokens["image_url"] = cashout_method.image_url
+    send_email_with_tokens(email_tokens, mautic_url)
+
+
+def send_cashout_status_email(email: str, cashout: CashoutRequestInfo) -> None:
+    assert cashout.status is not None
+    email_tokens = {
+        "cashout_id": str(cashout.id),
+        "cashout_status": cashout.status.value,
+    }
+    # todo: (if tango) format credentials and redemption instructions
+    # "credentials": "",
+    # "redemption_instructions": "",
+
+    if is_debug():
+        print("CASHOUT_STATUS_EMAIL: ", email_tokens)
+        return
+    contact_id = get_or_create_contact(email=email)
+    mautic_url = f"{MAUTIC_BASE_URL}/api/emails/{CASHOUT_STATUS_EMAIL_TEMPLATE_ID}/contact/{contact_id}/send"
+    send_email_with_tokens(email_tokens, mautic_url)
diff --git a/jb/managers/thl.py b/jb/managers/thl.py
index 85e0697..96f624e 100644
--- a/jb/managers/thl.py
+++ b/jb/managers/thl.py
@@ -1,15 +1,16 @@
 import requests
+from fastapi import HTTPException, status
 from generalresearch.currency import USDCent
 from generalresearch.models.thl.definitions import PayoutStatus
 from generalresearch.models.thl.payout import UserPayoutEvent
 from generalresearch.models.thl.task_status import TaskStatusResponse
 from generalresearch.models.thl.wallet.cashout_method import (
+    CashoutMethodOut,
     CashoutRequestInfo,
     CashoutRequestResponse,
 )
 
 from jb.config import settings
-from jb.models.auth import User
 
 
 def get_task_status(tsid: str) -> TaskStatusResponse | None:
@@ -22,20 +23,30 @@ def get_task_status(tsid: str) -> TaskStatusResponse | None:
 
 
 def user_cashout_request(
-    user: User, amount: USDCent, cashout_method_id: str
+    product_user_id: str, amount: USDCent, cashout_method_id: str
 ) -> CashoutRequestInfo:
     assert isinstance(amount, USDCent)
-    assert USDCent(0) < amount < USDCent(10_00)
+    assert USDCent(0) < amount <= USDCent(100_00)
     url = f"{settings.fsb_host}{settings.product_id}/cashout/"
 
     body: dict[str, str | int] = {
-        "bpuid": user.product_user_id,
+        "bpuid": product_user_id,
         "amount": int(amount),
         "cashout_method_id": cashout_method_id,
     }
     res = requests.post(url, json=body)
-    d = res.json()
+    if res.status_code == status.HTTP_400_BAD_REQUEST:
+        try:
+            message = res.json().get("msg")
+        except (ValueError, AttributeError):
+            message = None
+        raise HTTPException(
+            status_code=status.HTTP_400_BAD_REQUEST,
+            detail=message or "Cashout request failed",
+        )
+    res.raise_for_status()
 
+    d = res.json()
     return CashoutRequestResponse.model_validate(d).cashout
 
 
@@ -53,6 +64,49 @@ def manage_pending_cashout(
     return UserPayoutEvent.model_validate(d)
 
 
+def get_paypal_cashout_method_if_exists(
+    product_user_id: str, client_ip: str
+) -> str | None:
+    """
+    Make sure a paypal cashout method exists for this user. Use their login email.
+    We check if it exists first, so that once a user can change their paypal email,
+        we do not overwrite it.
+    """
+    url = f"{settings.fsb_host}{settings.product_id}/cashout_methods/"
+    params = {"bpuid": product_user_id, "ip": client_ip}
+    res = requests.get(url, params=params)
+    assert res.status_code == status.HTTP_200_OK, res.text
+    cms = res.json()["cashout_methods"]
+    res = next(filter(lambda x: x["type"] == "PAYPAL", cms), None)
+    if res:
+        return res["id"]
+    return None
+
+
+def create_paypal_cashout_method(product_user_id: str, email: str) -> dict:
+    url = f"{settings.fsb_host}{settings.product_id}/cashout_methods/"
+    body = {"bpuid": product_user_id, "type": "PAYPAL", "email": email}
+    res = requests.post(url, json=body)
+    assert res.status_code == status.HTTP_200_OK, res.text
+    return res.json()["cashout_method"]
+
+
+def create_paypal_cashout_method_if_not_exists(
+    product_user_id: str, email: str, client_ip: str
+) -> None:
+    if not get_paypal_cashout_method_if_exists(product_user_id, client_ip=client_ip):
+        create_paypal_cashout_method(product_user_id, email)
+
+
+def get_cashout_method(cashout_method_id: str) -> CashoutMethodOut:
+    url = (
+        f"{settings.fsb_host}{settings.product_id}/cashout_methods/{cashout_method_id}/"
+    )
+    res = requests.get(url)
+    assert res.status_code == status.HTTP_200_OK, res.text
+    return CashoutMethodOut.model_validate(res.json()["cashout_method"])
+
+
 def get_wallet_balance(amt_worker_id: str) -> USDCent:
     # This will raise an Exception if wallet balance is negative
     url = f"{settings.fsb_host}{settings.product_id}/wallet/"
@@ -67,3 +121,10 @@ def get_wallet_balance_if_non_negative(amt_worker_id: str) -> USDCent | None:
     if amt >= 0:
         return USDCent(amt)
     return None
+
+
+def get_cashout_detail(cashout_id: str) -> CashoutRequestInfo:
+    url = f"{settings.fsb_host}{settings.product_id}/cashout/{cashout_id}/"
+    res = requests.get(url)
+    assert res.status_code == status.HTTP_200_OK, res.text
+    return CashoutRequestInfo.model_validate(res.json()["cashout"])