feat(logic): optimise logic (#10)

optimise logic
XdpCs · May 1, 2024 · 73c1da0 · 73c1da0
1 parent bdcd996
commit 73c1da0
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/index.html b/index.html
@@ -151,8 +151,13 @@ <h1>Withdraw FLT</h1>
                 const lockedBalances = await contractFltDrop.lockedBalances(account);
                 console.log(`Lock Time: ${lockedBalances.unlockTime}`);
                 showLockTime.textContent = blockchainTimestampToDateTime(lockedBalances.unlockTime.toString());
-                connectButton.style.display = 'none';
-                withdrawButton.style.display = '';
+                if (lockedBalance > 0) {
+                    connectButton.style.display = 'none';
+                    withdrawButton.style.display = '';
+                } else {
+                    alert("There are no FLT tokens available for withdrawal.");
+                }
+
             } catch (error) {
                 console.error("Error connecting to wallet:", error);
                 alert("Error connecting to wallet. Please make sure you have an Ethereum wallet installed and unlocked.");
@@ -162,7 +167,7 @@ <h1>Withdraw FLT</h1>
         async function withdrawOnClick() {
             console.log("Withdrawing FLT...");
             try {
-                if (lockedBalance === 0) {
+                if (lockedBalance == 0) {
                     throw new Error('There are no FLT tokens available for withdrawal.');
                 }
                 const provider = new ethers.BrowserProvider(window.ethereum);
@@ -175,7 +180,7 @@ <h1>Withdraw FLT</h1>
                 await tx.wait();
                 alert("FLT tokens withdrawn successfully.");
             } catch (error) {
-                console.error("Withdrawal error:", error);
+                console.error("Withdrawal error:", error.message);
                 alert("Withdrawal error: " + error.message);
             }
         }