[OTE-438] Allow users to read data if they are in FIRST_STRIKE_CLOSE_…

…ONLY
dydxprotocol · Jun 18, 2024 · c596696 · c596696
1 parent 2bafa9a
commit c596696
Show file tree

Hide file tree

Showing 4 changed files with 25 additions and 1 deletion.
diff --git a/indexer/services/comlink/__tests__/lib/compliance-and-geo-check.test.ts b/indexer/services/comlink/__tests__/lib/compliance-and-geo-check.test.ts
@@ -189,6 +189,26 @@ describe('compliance-check', () => {
     });
   });
 
+  it.each([
+    ['query', `/v4/check-compliance-query?address=${testConstants.defaultAddress}`],
+    ['param', `/v4/check-compliance-param/${testConstants.defaultAddress}`],
+  ])('does not return 403 if address in request is in FIRST_STRIKE_CLOSE_ONLY and from restricted country (%s)', async (
+    _name: string,
+    path: string,
+  ) => {
+    isRestrictedCountrySpy.mockReturnValueOnce(true);
+    await ComplianceStatusTable.create({
+      ...testConstants.compliantStatusData,
+      status: ComplianceStatus.FIRST_STRIKE_CLOSE_ONLY,
+    });
+    await sendRequestToApp({
+      type: RequestMethod.GET,
+      path,
+      expressApp: complianceCheckApp,
+      expectedStatus: 200,
+    });
+  });
+
   it.each([
     ['query', `/v4/check-compliance-query?address=${testConstants.defaultAddress}`],
     ['param', `/v4/check-compliance-param/${testConstants.defaultAddress}`],

diff --git a/indexer/services/comlink/public/api-documentation.md b/indexer/services/comlink/public/api-documentation.md
@@ -3575,6 +3575,7 @@ This operation does not require authentication
 |*anonymous*|MANUAL|
 |*anonymous*|US_GEO|
 |*anonymous*|CA_GEO|
+|*anonymous*|GB_GEO|
 |*anonymous*|SANCTIONED_GEO|
 |*anonymous*|COMPLIANCE_PROVIDER|
 

diff --git a/indexer/services/comlink/public/swagger.json b/indexer/services/comlink/public/swagger.json
@@ -350,6 +350,7 @@
           "MANUAL",
           "US_GEO",
           "CA_GEO",
+          "GB_GEO",
           "SANCTIONED_GEO",
           "COMPLIANCE_PROVIDER"
         ],

diff --git a/indexer/services/comlink/src/lib/compliance-and-geo-check.ts b/indexer/services/comlink/src/lib/compliance-and-geo-check.ts
@@ -47,7 +47,9 @@ export async function complianceAndGeoCheck(
       { readReplica: true },
     );
     if (updatedStatus.length > 0) {
-      if (updatedStatus[0].status === ComplianceStatus.CLOSE_ONLY) {
+      if (updatedStatus[0].status === ComplianceStatus.CLOSE_ONLY ||
+        updatedStatus[0].status === ComplianceStatus.FIRST_STRIKE_CLOSE_ONLY
+      ) {
         return next();
       } else if (updatedStatus[0].status === ComplianceStatus.BLOCKED) {
         return create4xxResponse(