From cdc8f9af4bea3ab17577eea67bf2ed7671f7c270 Mon Sep 17 00:00:00 2001
From: Harvey Tindall <hrfee@protonmail.ch>
Date: Wed, 6 Sep 2023 22:12:36 +0100
Subject: [PATCH] referrals: unlink/disable referrals for profile

---
 api-profiles.go        |  13 +++++
 ts/modules/profiles.ts | 106 +++++++++++++++++++++--------------------
 2 files changed, 68 insertions(+), 51 deletions(-)

diff --git a/api-profiles.go b/api-profiles.go
index 5220782..3623d33 100644
--- a/api-profiles.go
+++ b/api-profiles.go
@@ -181,5 +181,18 @@ func (app *appContext) EnableReferralForProfile(gc *gin.Context) {
 // @Security Bearer
 // @tags Profiles & Settings
 func (app *appContext) DisableReferralForProfile(gc *gin.Context) {
+	profileName := gc.Param("profile")
+	profile, ok := app.storage.GetProfileKey(profileName)
+	if !ok {
+		respondBool(200, true, gc)
+		return
+	}
+
+	app.storage.DeleteInvitesKey(profile.ReferralTemplateKey)
+
+	profile.ReferralTemplateKey = ""
+
+	app.storage.SetProfileKey(profileName, profile)
+
 	respondBool(200, true, gc)
 }
diff --git a/ts/modules/profiles.ts b/ts/modules/profiles.ts
index 7824ed3..8bf8c32 100644
--- a/ts/modules/profiles.ts
+++ b/ts/modules/profiles.ts
@@ -168,6 +168,61 @@ export class ProfileEditor {
         }
     }
 
+    load = () => _get("/profiles", null, (req: XMLHttpRequest) => {
+        if (req.readyState == 4) {
+            if (req.status == 200) {
+                let resp = req.response as profileResp;
+                if (Object.keys(resp.profiles).length == 0) {
+                    this.empty = true;
+                } else {
+                    this.empty = false;
+                    for (let name in resp.profiles) {
+                        if (name in this._profiles) {
+                            this._profiles[name].update(name, resp.profiles[name]);
+                        } else {
+                            this._profiles[name] = new profile(name, resp.profiles[name]);
+                            if (window.ombiEnabled)
+                                this._profiles[name].setOmbiFunc((ombi: boolean) => {
+                                    if (ombi) {
+                                        this._ombiProfiles.delete(name, (req: XMLHttpRequest) => {
+                                            if (req.readyState == 4) {
+                                                if (req.status != 204) {
+                                                    window.notifications.customError("errorDeleteOmbi", window.lang.notif("errorUnknown"));
+                                                    return;
+                                                }
+                                                this._profiles[name].ombi = false;
+                                            }
+                                        });
+                                    } else {
+                                        window.modals.profiles.close();
+                                        this._ombiProfiles.load(name);
+                                    }
+                                });
+                            if (window.referralsEnabled)
+                                this._profiles[name].setReferralFunc((enabled: boolean) => {
+                                    if (enabled) {
+                                        this.disableReferrals(name);
+                                    } else {
+                                        this.enableReferrals(name);
+                                    }
+                                });
+                            this._table.appendChild(this._profiles[name].asElement());
+                        }
+                    }
+                }
+                this.default = resp.default_profile;
+                window.modals.profiles.show();
+            } else {
+                window.notifications.customError("profileEditor", window.lang.notif("errorLoadProfiles"));
+            }
+        }
+    })
+
+    disableReferrals = (name: string) => _delete("/profiles/referral/" + name, null, (req: XMLHttpRequest) => {
+        if (req.readyState != 4) return;
+        this.load();
+    });
+
     enableReferrals = (name: string) => {
         const referralsInviteSelect = document.getElementById("enable-referrals-profile-invites") as HTMLSelectElement;
         _get("/invites", null, (req: XMLHttpRequest) => {
@@ -219,57 +274,6 @@ export class ProfileEditor {
         window.modals.enableReferralsProfile.show();
     };
 
-    load = () => _get("/profiles", null, (req: XMLHttpRequest) => {
-        if (req.readyState == 4) {
-            if (req.status == 200) {
-                let resp = req.response as profileResp;
-                if (Object.keys(resp.profiles).length == 0) {
-                    this.empty = true;
-                } else {
-                    this.empty = false;
-                    for (let name in resp.profiles) {
-                        if (name in this._profiles) {
-                            this._profiles[name].update(name, resp.profiles[name]);
-                        } else {
-                            this._profiles[name] = new profile(name, resp.profiles[name]);
-                            if (window.ombiEnabled)
-                                this._profiles[name].setOmbiFunc((ombi: boolean) => {
-                                    if (ombi) {
-                                        this._ombiProfiles.delete(name, (req: XMLHttpRequest) => {
-                                            if (req.readyState == 4) {
-                                                if (req.status != 204) {
-                                                    window.notifications.customError("errorDeleteOmbi", window.lang.notif("errorUnknown"));
-                                                    return;
-                                                }
-                                                this._profiles[name].ombi = false;
-                                            }
-                                        });
-                                    } else {
-                                        window.modals.profiles.close();
-                                        this._ombiProfiles.load(name);
-                                    }
-                                });
-                            if (window.referralsEnabled)
-                                this._profiles[name].setReferralFunc((enabled: boolean) => {
-                                    if (enabled) {
-                                        // FIXME: Unlink template
-                                        console.log("FIXME");
-                                    } else {
-                                        this.enableReferrals(name);
-                                    }
-                                });
-                            this._table.appendChild(this._profiles[name].asElement());
-                        }
-                    }
-                }
-                this.default = resp.default_profile;
-                window.modals.profiles.show();
-            } else {
-                window.notifications.customError("profileEditor", window.lang.notif("errorLoadProfiles"));
-            }
-        }
-    })
-
     constructor() {
         (document.getElementById('setting-profiles') as HTMLSpanElement).onclick = this.load;
         document.addEventListener("profiles-default", (event: CustomEvent) => {