Add support for token transfers between addresses

src/vault-interaction.ts

@@ -45,22 +45,23 @@ export function handleVaultTransfer(event: TransferEvent): void {
     return
   }
 
-  if (!event.params.from.equals(SHARE_TOKEN_MINT_ADDRESS) || !event.params.to.equals(SHARE_TOKEN_MINT_ADDRESS)) {
-    updateVaultData(vault)
-
-    let investorAddress = ADDRESS_ZERO
-    if (!event.params.from.equals(SHARE_TOKEN_MINT_ADDRESS)) {
-      investorAddress = event.params.from
-    } else if (!event.params.to.equals(SHARE_TOKEN_MINT_ADDRESS)) {
-      investorAddress = event.params.to
-    }
+  updateVaultData(vault)
 
-    let investor = getInvestor(investorAddress)
+  if (!event.params.from.equals(SHARE_TOKEN_MINT_ADDRESS)) {
+    const investorAddress = event.params.from
+    const investor = getInvestor(investorAddress)
     investor.save()
+    updateInvestorVaultData(vault, investor)
+  }
 
+  if (!event.params.to.equals(SHARE_TOKEN_MINT_ADDRESS)) {
+    const investorAddress = event.params.to
+    const investor = getInvestor(investorAddress)
+    investor.save()
     updateInvestorVaultData(vault, investor)
-    updateVaultBreakDown(event.block, vault)
   }
+
+  updateVaultBreakDown(event.block, vault)
 }
 
 export function handleStrategyHarvest(event: ethereum.Event): void {