From 20c46b67d697a6fe40fb257f69c192a3f1204cd9 Mon Sep 17 00:00:00 2001
From: Brian Clifton <brian@clifton.me>
Date: Thu, 12 Oct 2017 15:51:06 -0700
Subject: [PATCH] Fix persistence issues with ledger client - client is now
 persisted after sync - file is now cleaned up after transition is complete
 (when new ledger file is being written)

Fixes https://github.com/brave/browser-laptop/issues/11494

Auditors: @evq
---
 app/browser/api/ledger.js | 25 ++++++++++++++++++++-----
 1 file changed, 20 insertions(+), 5 deletions(-)

diff --git a/app/browser/api/ledger.js b/app/browser/api/ledger.js
index 89d2bb19e12..3c821ac8e93 100644
--- a/app/browser/api/ledger.js
+++ b/app/browser/api/ledger.js
@@ -74,6 +74,7 @@ const v2RulesetPath = 'ledger-rulesV2.leveldb'
 let v2PublishersDB
 const v2PublishersPath = 'ledger-publishersV2.leveldb'
 const statePath = 'ledger-state.json'
+const newClientPath = 'ledger-newstate.json'
 
 // Definitions
 const miliseconds = {
@@ -1885,7 +1886,22 @@ const onCallback = (state, result, delayTime) => {
     })
   }
 
+  // persist the new ledger state
   muonWriter(statePath, regularResults)
+
+  // delete the temp file used during transition (if it still exists)
+  if (client && client.options && client.options.version === 'v2') {
+    const fs = require('fs')
+    fs.access(pathName(newClientPath), fs.FF_OK, (err) => {
+      if (err) {
+        return
+      }
+      fs.unlink(pathName(newClientPath), (err) => {
+        if (err) console.error('unlink error: ' + err.toString())
+      })
+    })
+  }
+
   run(state, delayTime)
 
   return state
@@ -2246,9 +2262,9 @@ const deleteSynopsis = () => {
 let newClient = null
 const transitionWalletToBat = () => {
   let newPaymentId, result
-  const newClientPath = 'ledger-newstate.json'
 
   if (newClient === true) return
+
   // Restore newClient from the file
   if (!newClient) {
     const fs = require('fs')
@@ -2291,6 +2307,8 @@ const transitionWalletToBat = () => {
 
       if (typeof delayTime === 'undefined') delayTime = random.randomInt({ min: 1, max: 500 })
 
+      muonWriter(newClientPath, newClient)
+
       setTimeout(() => transitionWalletToBat(), delayTime)
     })
     return
@@ -2311,13 +2329,10 @@ const transitionWalletToBat = () => {
         result = newClient.transitioned(properties)
         client = newClient
         newClient = true
+        // NOTE: onLedgerCallback will save latest client to disk as ledger-state.json
         appActions.onLedgerCallback(result, random.randomInt({ min: miliseconds.minute, max: 10 * miliseconds.minute }))
         appActions.onBitcoinToBatTransitioned()
         notifications.showBraveWalletUpdated()
-        const fs = require('fs')
-        fs.unlink(pathName(newClientPath), (err) => {
-          if (err) console.error('unlink error: ' + err.toString())
-        })
       }
     })
   } catch (ex) {