From 9467e9b2950979d7498ab55920cea748cbe44522 Mon Sep 17 00:00:00 2001
From: Paul Puey <paul@edge.app>
Date: Thu, 31 Oct 2024 22:23:41 -0700
Subject: [PATCH] Add affiliateAddress and xClientId params

- xClientId helps prevent timeouts
- snooze to also help prevent timeouts
- use affiliateAddress to filter txs not created in app
---
 src/partners/thorchain.ts | 26 ++++++++++++++++++++++----
 1 file changed, 22 insertions(+), 4 deletions(-)

diff --git a/src/partners/thorchain.ts b/src/partners/thorchain.ts
index 7572f5b..08aa590 100644
--- a/src/partners/thorchain.ts
+++ b/src/partners/thorchain.ts
@@ -52,7 +52,11 @@ const asThorchainResult = asObject({
 })
 
 const asThorchainPluginParams = asObject({
-  apiKeys: asObject({ thorchainAddress: asString }),
+  apiKeys: asObject({
+    thorchainAddress: asString,
+    affiliateAddress: asString,
+    xClientId: asOptional(asString)
+  }),
   settings: asObject({
     latestIsoDate: asOptional(asString, '1970-01-01T00:00:00.000Z')
   })
@@ -79,7 +83,7 @@ const makeThorchainPlugin = (info: ThorchainInfo): PartnerPlugin => {
     const ssFormatTxs: StandardTx[] = []
 
     const { settings, apiKeys } = asThorchainPluginParams(pluginParams)
-    const { thorchainAddress } = apiKeys
+    const { affiliateAddress, thorchainAddress, xClientId } = apiKeys
     let { latestIsoDate } = settings
 
     let previousTimestamp = new Date(latestIsoDate).getTime() - QUERY_LOOKBACK
@@ -91,12 +95,21 @@ const makeThorchainPlugin = (info: ThorchainInfo): PartnerPlugin => {
 
     let offset = 0
 
+    let headers: HeadersInit | undefined
+    if (xClientId != null) {
+      headers = {
+        'x-client-id': xClientId
+      }
+    }
+
     while (!done) {
       const url = `https://${midgardUrl}/v2/actions?address=${thorchainAddress}&type=swap,refund&affiliate=ej&offset=${offset}&limit=${LIMIT}`
       let jsonObj: ThorchainResult
       try {
-        const result = await fetch(url, {
-          method: 'GET'
+        await snooze(500)
+        const result = await retryFetch(url, {
+          method: 'GET',
+          headers
         })
         if (!result.ok) {
           const text = await result.text()
@@ -112,12 +125,17 @@ const makeThorchainPlugin = (info: ThorchainInfo): PartnerPlugin => {
       for (const rawTx of txs) {
         const {
           date,
+          metadata,
           in: txIns,
           out: txOuts,
           pools,
           status: txStatus
         } = asThorchainTx(rawTx) // Check RAW trasaction
 
+        const { swap } = metadata
+        if (swap?.affiliateAddress !== affiliateAddress) {
+          continue
+        }
         const status = 'complete'
         if (txStatus !== 'success') {
           continue