-
Notifications
You must be signed in to change notification settings - Fork 315
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
chore: kickoff LLD integration of wallet sync
- Loading branch information
Showing
20 changed files
with
452 additions
and
149 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
17 changes: 17 additions & 0 deletions
17
apps/ledger-live-desktop/src/newArch/WalletSync/components/WalletSyncContext.tsx
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,17 @@ | ||
import React from "react"; | ||
import { useWatchWalletSync, WalletSyncUserState } from "../hooks/useWatchWalletSync"; | ||
|
||
export const WalletSyncContext = React.createContext<WalletSyncUserState>({ | ||
visualPending: false, | ||
walletSyncError: null, | ||
onUserRefresh: () => {}, | ||
}); | ||
|
||
export const useWalletSyncUserState = () => React.useContext(WalletSyncContext); | ||
|
||
export function WalletSyncProvider({ children }: { children: React.ReactNode }) { | ||
const walletSyncState = useWatchWalletSync(); | ||
return ( | ||
<WalletSyncContext.Provider value={walletSyncState}>{children}</WalletSyncContext.Provider> | ||
); | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
191 changes: 191 additions & 0 deletions
191
apps/ledger-live-desktop/src/newArch/WalletSync/hooks/useWatchWalletSync.ts
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,191 @@ | ||
import { useCallback, useEffect, useMemo, useState } from "react"; | ||
import { useDispatch, useSelector, useStore } from "react-redux"; | ||
import { CloudSyncSDK, UpdateEvent } from "@ledgerhq/live-wallet/cloudsync/index"; | ||
import walletsync, { | ||
liveSlug, | ||
DistantState, | ||
walletSyncWatchLoop, | ||
LocalState, | ||
Schema, | ||
} from "@ledgerhq/live-wallet/walletsync/index"; | ||
import { getAccountBridge } from "@ledgerhq/live-common/bridge/index"; | ||
import { useTrustchainSdk } from "./useTrustchainSdk"; | ||
import { walletSelector } from "~/renderer/reducers/wallet"; | ||
import { MemberCredentials, Trustchain, TrustchainSDK } from "@ledgerhq/trustchain/types"; | ||
import { | ||
memberCredentialsSelector, | ||
trustchainSelector, | ||
setTrustchain, | ||
} from "@ledgerhq/trustchain/store"; | ||
import { TrustchainEjected } from "@ledgerhq/trustchain/errors"; | ||
import { State } from "~/renderer/reducers"; | ||
import { cache as bridgeCache } from "~/renderer/bridge/cache"; | ||
import { setAccountNames, walletSyncUpdate } from "@ledgerhq/live-wallet/store"; | ||
import { replaceAccounts } from "~/renderer/actions/accounts"; | ||
import noop from "lodash/noop"; | ||
import { log } from "@ledgerhq/logs"; | ||
|
||
function localStateSelector(state: State): LocalState { | ||
return { | ||
accounts: { | ||
list: state.accounts, | ||
}, | ||
accountNames: state.wallet.accountNames, | ||
}; | ||
} | ||
|
||
function latestDistantStateSelector(state: State): DistantState | null { | ||
return walletSelector(state).wsState.data; | ||
} | ||
|
||
export type WalletSyncUserState = { | ||
visualPending: boolean; | ||
walletSyncError: Error | null; | ||
onUserRefresh: () => void; | ||
}; | ||
|
||
export function useCloudSyncSDK(): CloudSyncSDK<Schema> { | ||
const trustchainSdk = useTrustchainSdk(); | ||
|
||
const store = useStore(); | ||
const dispatch = useDispatch(); | ||
|
||
const getCurrentVersion = useCallback(() => { | ||
const state = walletSelector(store.getState()); | ||
return state.wsState.version; | ||
}, [store]); | ||
|
||
const saveNewUpdate = useCallback( | ||
async (event: UpdateEvent<DistantState>) => { | ||
log("walletsync", "saveNewUpdate", { event }); | ||
switch (event.type) { | ||
case "new-data": { | ||
const state = store.getState(); | ||
const version = event.version; | ||
const data = event.data; | ||
const wsState = walletSelector(state).wsState; | ||
const localState = localStateSelector(state); | ||
const ctx = { getAccountBridge, bridgeCache, blacklistedTokenIds: [] }; | ||
|
||
const resolved = await walletsync.resolveIncomingDistantState( | ||
ctx, | ||
localState, | ||
wsState.data, | ||
data, | ||
); | ||
|
||
if (resolved.hasChanges) { | ||
const s = store.getState(); | ||
const localState = localStateSelector(s); | ||
const newLocalState = walletsync.applyUpdate(localState, resolved.update); | ||
dispatch(setAccountNames(newLocalState.accountNames)); | ||
dispatch(walletSyncUpdate(data, version)); | ||
dispatch(replaceAccounts(newLocalState.accounts.list)); | ||
log("walletsync", "resolved. changes applied."); | ||
} else { | ||
log("walletsync", "resolved. no changes to apply."); | ||
} | ||
break; | ||
} | ||
case "pushed-data": { | ||
dispatch(walletSyncUpdate(event.data, event.version)); | ||
break; | ||
} | ||
case "deleted-data": { | ||
dispatch(walletSyncUpdate(null, 0)); | ||
break; | ||
} | ||
} | ||
}, | ||
[store, dispatch], | ||
); | ||
|
||
const cloudSyncSDK = useMemo( | ||
() => | ||
new CloudSyncSDK({ | ||
slug: liveSlug, | ||
schema: walletsync.schema, | ||
trustchainSdk, | ||
getCurrentVersion, | ||
saveNewUpdate, | ||
}), | ||
[trustchainSdk, getCurrentVersion, saveNewUpdate], | ||
); | ||
|
||
return cloudSyncSDK; | ||
} | ||
|
||
function useOnTrustchainRefreshNeeded( | ||
trustchainSdk: TrustchainSDK, | ||
memberCredentials: MemberCredentials | null, | ||
): (trustchain: Trustchain) => Promise<void> { | ||
const onTrustchainRefreshNeeded = useCallback( | ||
async (trustchain: Trustchain) => { | ||
try { | ||
if (!memberCredentials) return; | ||
log("walletsync", "onTrustchainRefreshNeeded " + trustchain.rootId); | ||
const newTrustchain = await trustchainSdk.restoreTrustchain(trustchain, memberCredentials); | ||
setTrustchain(newTrustchain); | ||
} catch (e) { | ||
if (e instanceof TrustchainEjected) { | ||
// TODO: eject the trustchain | ||
console.error("Not Implemented Yet", e); | ||
// dispatch(resetTrustchainStore()); | ||
} | ||
} | ||
}, | ||
[trustchainSdk, memberCredentials], | ||
); | ||
return onTrustchainRefreshNeeded; | ||
} | ||
|
||
export function useWatchWalletSync(): WalletSyncUserState { | ||
const store = useStore(); | ||
const memberCredentials = useSelector(memberCredentialsSelector); | ||
const trustchain = useSelector(trustchainSelector); | ||
const trustchainSdk = useTrustchainSdk(); | ||
const walletSyncSdk = useCloudSyncSDK(); | ||
const onTrustchainRefreshNeeded = useOnTrustchainRefreshNeeded(trustchainSdk, memberCredentials); | ||
|
||
const [visualPending, setVisualPending] = useState(true); | ||
const [walletSyncError, setWalletSyncError] = useState<Error | null>(null); | ||
const [onUserRefresh, setOnUserRefresh] = useState<() => void>(() => noop); | ||
const state = useMemo( | ||
() => ({ visualPending, walletSyncError, onUserRefresh }), | ||
[visualPending, walletSyncError, onUserRefresh], | ||
); | ||
|
||
// pull and push wallet sync loop | ||
useEffect(() => { | ||
if (!trustchain || !memberCredentials) { | ||
setOnUserRefresh(() => noop); | ||
return; | ||
} | ||
|
||
const { unsubscribe, onUserRefreshIntent } = walletSyncWatchLoop({ | ||
walletSyncSdk, | ||
trustchain, | ||
memberCredentials, | ||
setVisualPending, | ||
getState: () => store.getState(), | ||
localStateSelector, | ||
latestDistantStateSelector, | ||
onError: e => setWalletSyncError(e && e instanceof Error ? e : new Error(String(e))), | ||
onStartPolling: () => setWalletSyncError(null), | ||
onTrustchainRefreshNeeded, | ||
}); | ||
|
||
setOnUserRefresh(() => onUserRefreshIntent); | ||
|
||
return unsubscribe; | ||
}, [ | ||
store, | ||
trustchainSdk, | ||
walletSyncSdk, | ||
trustchain, | ||
memberCredentials, | ||
onTrustchainRefreshNeeded, | ||
]); | ||
|
||
return state; | ||
} |
Oops, something went wrong.