From 96a146e7eb5566de37a9d5c30d3a87f918bd6073 Mon Sep 17 00:00:00 2001
From: Joseph Ferano <joseph@ferano.io>
Date: Tue, 24 May 2022 00:21:43 +0700
Subject: [PATCH] Check if user already submitted their wallet when granting WL
 role

---
 Bot/Whitelist.fs | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/Bot/Whitelist.fs b/Bot/Whitelist.fs
index 6d98e51..4442603 100644
--- a/Bot/Whitelist.fs
+++ b/Bot/Whitelist.fs
@@ -44,7 +44,13 @@ You need to **BUY** Whitelist with 💰 $GBT...
 
 let grantWhitelistRole isOg (ctx : IDiscordContext) =
     task {
-        let roleId = if isOg then GuildEnvironment.roleWhiteOGPending else GuildEnvironment.roleWhitelistPending
+        let! wallet = InviteTracker.getWalletAddress (ctx.GetDiscordMember().Id)
+        let roleId =
+            match isOg , wallet with
+            | true  , Some _ -> GuildEnvironment.roleWhiteOG
+            | false , Some _ -> GuildEnvironment.roleWhitelist
+            | true  , None   -> GuildEnvironment.roleWhiteOGPending
+            | false , None   -> GuildEnvironment.roleWhitelistPending
         let role = ctx.GetGuild().GetRole(roleId)
         let user = ctx.GetDiscordMember()
         do! user.GrantRoleAsync(role)