From cc75deb78bce0472aab0e717323efd99e9e24d87 Mon Sep 17 00:00:00 2001
From: Alexis Sellier <self@cloudhead.io>
Date: Mon, 8 Mar 2021 12:00:59 +0100
Subject: [PATCH] Implement vesting withdrawing

---
 src/base/vesting/Vesting.svelte | 42 ++++++++++++++++++++-------------
 src/base/vesting/state.js       |  5 ++--
 src/base/vesting/vesting.js     | 15 ++++++++++++
 3 files changed, 43 insertions(+), 19 deletions(-)

diff --git a/src/base/vesting/Vesting.svelte b/src/base/vesting/Vesting.svelte
index f8ad40f64..aab82a787 100644
--- a/src/base/vesting/Vesting.svelte
+++ b/src/base/vesting/Vesting.svelte
@@ -1,25 +1,23 @@
 <script lang="javascript">
   import { ethers } from 'ethers';
-  import { get, derived } from 'svelte/store';
+  import { get, derived, writable } from 'svelte/store';
   import { session } from '@app/session.js';
   import { STATE, state } from './state.js';
-  import { getInfo } from './vesting.js';
+  import { getInfo, withdrawVested } from './vesting.js';
 
   export let config = null;
 
   let contractAddress = "";
-  let info = null;
+  const info = writable(null);
 
   async function loadContract(config) {
     state.set(STATE.LOADING);
-    info = await getInfo(contractAddress, config);
+    info.set(await getInfo(contractAddress, config));
     state.set(STATE.IDLE);
   }
 
-  function withdrawVested() {}
-
-  let isBeneficiary = derived(session, (s) => {
-    return info && (info.beneficiary === s.address);
+  let isBeneficiary = derived([session, info], ([$s, $i]) => {
+    return $i && ($i.beneficiary === $s.address);
   });
 </script>
 
@@ -42,26 +40,36 @@
 </style>
 
 <main>
-  {#if info}
+  {#if $info}
     <div class="modal">
       <div class="modal-title">
         {contractAddress}
       </div>
       <div class="modal-body">
         <table>
-          <tr><td class="label">Beneficiary</td><td>{info.beneficiary}</td></tr>
-          <tr><td class="label">Allocation</td><td>{info.totalVesting} <strong>{info.symbol}</strong></td></tr>
-          <tr><td class="label">Withdrawn</td><td>{info.withdrawn} <strong>{info.symbol}</strong></td></tr>
-          <tr><td class="label">Withdrawable</td><td>{info.withdrawableBalance} <strong>{info.symbol}</strong></td></tr>
+          <tr><td class="label">Beneficiary</td><td>{$info.beneficiary}</td></tr>
+          <tr><td class="label">Allocation</td><td>{$info.totalVesting} <strong>{$info.symbol}</strong></td></tr>
+          <tr><td class="label">Withdrawn</td><td>{$info.withdrawn} <strong>{$info.symbol}</strong></td></tr>
+          <tr><td class="label">Withdrawable</td><td>{$info.withdrawableBalance} <strong>{$info.symbol}</strong></td></tr>
         </table>
       </div>
       <div class="modal-actions">
         {#if $isBeneficiary}
-          <button on:click={withdrawVested} class="primary small">
-            Withdraw
-          </button>
+          {#if $state === STATE.WITHDRAWING_SIGN}
+            <button disabled data-waiting class="primary small">
+              Waiting for signature...
+            </button>
+          {:else if $state === STATE.WITHDRAWING}
+            <button disabled data-waiting class="primary small">
+              Withdrawing...
+            </button>
+          {:else}
+            <button on:click={() => withdrawVested(contractAddress, config)} class="primary small">
+              Withdraw
+            </button>
+          {/if}
         {/if}
-        <button on:click={() => info = null} class="small">
+        <button on:click={() => $info = null} class="small">
           Back
         </button>
       </div>
diff --git a/src/base/vesting/state.js b/src/base/vesting/state.js
index 7552c3519..f7bcf5bc7 100644
--- a/src/base/vesting/state.js
+++ b/src/base/vesting/state.js
@@ -6,8 +6,9 @@ export const STATE = {
   LOADING: 1,
   FOUND: 2,
   NOT_FOUND: 3,
-  WITHDRAWING: 4,
-  WITHDRAWN: 5,
+  WITHDRAWING_SIGN: 4,
+  WITHDRAWING: 5,
+  WITHDRAWN: 6,
 };
 
 export const state = writable(STATE.IDLE);
diff --git a/src/base/vesting/vesting.js b/src/base/vesting/vesting.js
index 793d0e878..14cdcd6c4 100644
--- a/src/base/vesting/vesting.js
+++ b/src/base/vesting/vesting.js
@@ -1,4 +1,5 @@
 import { ethers } from "ethers";
+import { STATE, state } from "./state.js";
 
 const abi = [
   "function token() view returns (address)",
@@ -17,6 +18,20 @@ const tokenAbi = [
   "function symbol() view returns (string)",
 ];
 
+export async function withdrawVested(address, config) {
+  const contract = new ethers.Contract(address, abi, config.provider);
+  const signer = config.provider.getSigner();
+
+  state.set(STATE.WITHDRAWING_SIGN);
+
+  let tx = await contract.connect(signer).withdrawVested();
+
+  state.set(STATE.WITHDRAWING);
+  await tx.wait();
+  // TODO: Update balance.
+  state.set(STATE.WITHDRAWN);
+}
+
 export async function getInfo(address, config) {
   const contract = new ethers.Contract(address, abi, config.provider);
   const signer = config.provider.getSigner();