From 68dba2e8628533f85407b64fefb3be1ab9938d2d Mon Sep 17 00:00:00 2001
From: Alexis Sellier <self@cloudhead.io>
Date: Wed, 16 Jun 2021 12:54:20 +0200
Subject: [PATCH] Refresh balance on page load

---
 src/App.svelte | 9 +++++++--
 src/session.ts | 2 +-
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/App.svelte b/src/App.svelte
index 14e975a5d..e2149ab15 100644
--- a/src/App.svelte
+++ b/src/App.svelte
@@ -1,7 +1,7 @@
 <script lang="typescript">
   import { Router, Route } from "svelte-routing";
   import { getConfig } from '@app/config';
-  import { session } from '@app/session';
+  import { state, session } from '@app/session';
 
   import Home from '@app/base/home/Index.svelte';
   import Vesting from '@app/base/vesting/Index.svelte';
@@ -13,6 +13,11 @@
   import Loading from '@app/Loading.svelte';
   import Modal from '@app/Modal.svelte';
 
+  const loadConfig = getConfig().then(cfg => {
+    state.refreshBalance(cfg);
+    return cfg;
+  });
+
   function handleKeydown(event: KeyboardEvent) {
     if (event.key === 'Enter') {
       let elems = document.querySelectorAll<HTMLElement>('button.primary');
@@ -39,7 +44,7 @@
 
 <svelte:window on:keydown={handleKeydown} />
 <div class="app">
-  {#await getConfig()}
+  {#await loadConfig}
     <!-- Loading wallet -->
     <div class="wrapper">
       <Loading center />
diff --git a/src/session.ts b/src/session.ts
index 4aae737ad..daefb7c1e 100644
--- a/src/session.ts
+++ b/src/session.ts
@@ -41,8 +41,8 @@ export interface Store extends Readable<State> {
 
 export const loadState = (initial: State): Store => {
   const store = writable<State>(initial);
-
   const session = window.localStorage.getItem("session");
+
   if (session) store.set({ connection: Connection.Connected, session: JSON.parse(session) });
 
   return {