fix: reward per year for pcs clone (#22)

* feat: add pcs reward per year

* refactor: move lp class to file

* fix: ignore failed transactions

* fix: reward per year calculation

Author: Coac

src/providers/BSCDeFi.js

@@ -21,18 +21,18 @@ class BSCDeFi {
 
     this.defiPlatforms = {
       acryptos: new PancakeSwapCloneChef(web3, '0x96c8390BA28eB083A784280227C37b853bc408B7', 'pendingSushi', 'ACS'),
-      alpaca: new PancakeSwapCloneChef(web3, '0xA625AB01B08ce023B2a342Dbb12a16f2C8489A8F', 'pendingAlpaca'),
+      alpaca: new PancakeSwapCloneChef(web3, '0xA625AB01B08ce023B2a342Dbb12a16f2C8489A8F', 'pendingAlpaca', 'ALPACA'),
       apeSwap: new PancakeSwapCloneChef(web3, '0x5c8D727b265DBAfaba67E050f2f739cAeEB4A6F9', 'pendingCake', 'BANANA'),
       autoFarm: new AutoFarmChef(web3),
       bVault: new BVaultChef(web3),
       cafeSwap: new PancakeSwapCloneChef(web3, '0xc772955c33088a97D56d0BBf473d05267bC4feBB', 'pendingCake', 'BREW'),
-      coralFarm: new PancakeSwapCloneChef(web3, '0x713e34640ef300a0B178a9688458BbA8b1FA35A7', 'pendingCrl'),
-      goose: new PancakeSwapCloneChef(web3, '0xe70E9185F5ea7Ba3C5d63705784D8563017f2E57', 'pendingEgg'),
+      coralFarm: new PancakeSwapCloneChef(web3, '0x713e34640ef300a0B178a9688458BbA8b1FA35A7', 'pendingCrl', 'CRL'),
+      goose: new PancakeSwapCloneChef(web3, '0xe70E9185F5ea7Ba3C5d63705784D8563017f2E57', 'pendingEgg', 'EGG'),
       kebab: new PancakeSwapCloneChef(web3, '0x76FCeffFcf5325c6156cA89639b17464ea833ECd', 'pendingCake', 'KEBAB'),
       midasGold: new MidasGoldChef(web3),
       pancakeSwap: new PancakeSwapChef(web3),
       ramen: new PancakeSwapCloneChef(web3, '0x97dd424b4628c8d3bd7fcf3a4e974cebba011651', 'pendingCake', 'RAMEN'),
-      saltSwap: new PancakeSwapCloneChef(web3, '0xB4405445fFAcF2B86BC2bD7D1C874AC739265658', 'pendingSalt'),
+      saltSwap: new PancakeSwapCloneChef(web3, '0xB4405445fFAcF2B86BC2bD7D1C874AC739265658', 'pendingSalt', 'SALT'),
       slime: new PancakeSwapCloneChef(web3, '0x4B0073A79f2b46Ff5a62fA1458AAc86Ed918C80C', 'pendingReward', 'SLIME')
     }
   }

src/providers/bsc/AutoFarmChef.js

@@ -1,4 +1,3 @@
-const Web3 = require('web3')
 const MasterChef = require('./MasterChef')
 
 // Auto
@@ -7,21 +6,13 @@ const autoFarmAddress = '0x0895196562C7868C5Be92459FaE7f877ED450452'
 
 class AutoFarmChef extends MasterChef {
   constructor (web3) {
-    super(web3, autoFarmChefABI, autoFarmAddress)
+    super(web3, autoFarmChefABI, autoFarmAddress, 'pendingAUTO')
     this.contract = new web3.eth.Contract(autoFarmChefABI, autoFarmAddress)
   }
 
   async getStakedLPAmount (poolID, walletAddress) {
     return await this.contract.methods.stakedWantTokens(poolID, walletAddress).call()
   }
-
-  async getPendingReward (poolID, walletAddress) {
-    const pendingReward = await this.contract.methods.pendingAUTO(poolID, walletAddress).call()
-
-    return {
-      AUTO: parseFloat(Web3.utils.fromWei(pendingReward))
-    }
-  }
 }
 
 module.exports = AutoFarmChef

src/providers/bsc/MasterChef.js

@@ -7,13 +7,17 @@ const LiquidityPool = require('./LiquidityPool')
 const InputDataDecoder = require('ethereum-input-data-decoder')
 
 class MasterChef {
-  constructor (web3, chefABI, address, pendingMethodName, pendingSymbol = '') {
+  constructor (web3, chefABI, address, pendingMethodName = 'pendingCake', pendingSymbol = '') {
     this.web3 = web3
     this.address = address
     this.contract = new web3.eth.Contract(chefABI, address)
     this.decoder = new InputDataDecoder(chefABI)
     this.pendingMethodName = pendingMethodName
+
     this.pendingSymbol = pendingSymbol
+    if (this.pendingSymbol === '') {
+      this.pendingSymbol = this.pendingMethodName.replace('pending', '').toUpperCase()
+    }
   }
 
   lpTransactions (walletTx, poolID) {
@@ -52,14 +56,8 @@ class MasterChef {
 
   async getPendingReward (poolID, walletAddress) {
     const pendingReward = await this.contract.methods[this.pendingMethodName](poolID, walletAddress).call()
-
-    let tokenName = this.pendingSymbol
-    if (tokenName === '') {
-      tokenName = this.pendingMethodName.replace('pending', '').toUpperCase()
-    }
-
     const rewards = {}
-    rewards[tokenName] = parseFloat(Web3.utils.fromWei(pendingReward))
+    rewards[this.pendingSymbol] = parseFloat(Web3.utils.fromWei(pendingReward))
     return rewards
   }
 
@@ -86,7 +84,8 @@ class MasterChef {
 
     const poolRewardPerBlock = Web3.utils.fromWei(BigNumber(allocPoint).dividedBy(this.totalAllocPoint).multipliedBy(this.rewardPerBlock).integerValue().toFixed())
     const userPoolRewardPerBlock = poolRewardPerBlock * pool.share(lpTokenAmount)
-    const rewardPerYear = userPoolRewardPerBlock * 3600 * 24 * 365 / 3
+    const rewardPerYear = {}
+    rewardPerYear[this.pendingSymbol] = userPoolRewardPerBlock * 3600 * 24 * 365 / 3
 
     return {
       poolID, lpTokenAmount, totalDeposited, pendingReward, tokens, rewardPerYear, lpAddress, lpTransactions
@@ -102,9 +101,18 @@ class MasterChef {
     this.totalAllocPoint = await this.contract.methods.totalAllocPoint().call()
 
     try {
-      this.rewardPerBlock = await this.contract.methods.cakePerBlock().call() // TODO change reward per block
+      this.rewardPerBlock = await this.contract.methods[this.pendingSymbol.toLowerCase() + 'PerBlock']().call()
     } catch (e) {
-      this.rewardPerBlock = 0
+      try {
+        this.rewardPerBlock = await this.contract.methods[this.pendingSymbol + 'PerBlock']().call()
+      } catch (e) {
+        try {
+          const ABISymbol = this.pendingMethodName.replace('pending', '').toLowerCase()
+          this.rewardPerBlock = await this.contract.methods[ABISymbol + 'PerBlock']().call()
+        } catch (e) {
+          this.rewardPerBlock = 0
+        }
+      }
     }
 
     let pools = []

src/providers/bsc/PancakeSwapCloneChef.js

@@ -4,16 +4,24 @@ const MasterChef = require('./MasterChef')
 const pancakeSwapABI = require('../resources/abis/pancakeSwap.json')
 
 class PancakeSwapCloneChef extends MasterChef {
-  constructor (web3, address, pendingMethodName, tokenSymbol = '') {
-    const newChefABI = pancakeSwapABI.map(method => {
-      if (method.name !== 'pendingCake') {
-        return method
-      }
-      const newMethod = { ...method }
-      newMethod.name = pendingMethodName
-      return newMethod
-    })
-    super(web3, newChefABI, address, pendingMethodName)
+  constructor (web3, address, pendingMethodName, tokenSymbol) {
+    const ABISymbol = pendingMethodName.replace('pending', '').toLowerCase()
+
+    const newChefABI = pancakeSwapABI
+      .map(replaceABIMap('pendingCake', pendingMethodName))
+      .map(replaceABIMap('cakePerBlock', ABISymbol + 'PerBlock'))
+    super(web3, newChefABI, address, pendingMethodName, tokenSymbol)
+  }
+}
+
+function replaceABIMap (methodToReplace, newMethodName) {
+  return method => {
+    if (method.name !== methodToReplace) {
+      return method
+    }
+    const newMethod = { ...method }
+    newMethod.name = newMethodName
+    return newMethod
   }
 }
 

src/providers/bsc/Token.js

@@ -1,4 +1,5 @@
 const tokenABI = require('../resources/abis/token.json')
+const Web3 = require('web3')
 
 class Token {
   constructor (web3, address) {
@@ -9,6 +10,7 @@ class Token {
   async init () {
     this.symbol = await this.contract.methods.symbol().call()
     this.decimals = await this.contract.methods.decimals().call()
+    this.totalSupply = Web3.utils.fromWei(await this.contract.methods.totalSupply().call())
     return this
   }
 
@@ -19,7 +21,7 @@ class Token {
   }
 
   share (lpTokenAmount) {
-    return 0 // TODO implement this
+    return lpTokenAmount / this.totalSupply // TODO might not be correct
   }
 }