Clear referral code if invalid/claimed/expired (#6072)

open-chat-labs · Jul 19, 2024 · 1787a2e · 1787a2e
1 parent 04660a2
commit 1787a2e
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/frontend/openchat-client/src/openchat.ts b/frontend/openchat-client/src/openchat.ts
@@ -4347,8 +4347,16 @@ export class OpenChat extends OpenChatAgentWorker {
                     if (this._referralCode !== undefined) {
                         gaTrack("registered_user_with_referral_code", "registration");
                     }
-                    this.clearReferralCode();
                 }
+
+                switch (res.kind) {
+                    case "success":
+                    case "referral_code_invalid":
+                    case "referral_code_already_claimed":
+                    case "referral_code_expired":
+                        this.clearReferralCode();
+                }
+
                 return res;
             })
             .catch(() => ({ kind: "internal_error" }));