added iearn adapter

Author: sobolev-igor

contracts/adapters/IEarnAdapter.sol

@@ -0,0 +1,58 @@
+pragma solidity 0.6.2;
+pragma experimental ABIEncoderV2;
+
+import { Adapter } from "./Adapter.sol";
+import { Component } from "../Structs.sol";
+import { ERC20 } from "../ERC20.sol";
+
+
+/**
+ * @dev YToken contract interface.
+ * Only the functions required for IearnAdapter contract are added.
+ * The YToken contracts is available here
+ * https://github.com/iearn-finance/itoken/tree/master/contracts.
+ */
+interface YToken {
+    function token() external view returns (address);
+    function getPricePerFullShare() external view returns (uint256);
+}
+
+
+/**
+ * @title Adapter for iearn.finance protocol.
+ * @dev Implementation of Adapter interface.
+ */
+contract IearnAdapter is Adapter {
+
+    /**
+     * @return Name of the protocol.
+     * @dev Implementation of Adapter function.
+     */
+    function getProtocolName() external pure override returns (string memory) {
+        return("iearn.finance");
+    }
+
+    /**
+     * @return Amount of yToken held by the given user.
+     * @dev Implementation of Adapter function.
+     */
+    function getAssetAmount(address asset, address user) external view override returns (int256) {
+        return int256(ERC20(asset).balanceOf(user));
+    }
+
+    /**
+     * @return Struct with underlying assets rates for the given asset.
+     * @dev Implementation of Adapter function.
+     * Repeats calculations made in stableswap contract.
+     */
+    function getUnderlyingRates(address asset) external view override returns (Component[] memory) {
+        Component[] memory components = new Component[](1);
+
+        components[0] = Component({
+            underlying: YToken(asset).token(),
+            rate: YToken(asset).getPricePerFullShare()
+        });
+
+        return components;
+    }
+}

migrations_scripts/2_use_registry_and_add_new_adapter.js

@@ -0,0 +1,28 @@
+const IearnAdapter = artifacts.require('IearnAdapter');
+
+module.exports = (deployer, network, accounts) => {
+  const yDAIAddress = '0x16de59092dAE5CcF4A1E6439D611fd0653f0Bd01';
+  const yUSDCAddress = '0xd6aD7a6750A7593E092a9B218d66C0A814a3436e';
+  const yUSDTAddress = '0x83f798e925BcD4017Eb265844FDDAbb448f1707D';
+  const ySUSDAddress = '0xF61718057901F84C4eEC4339EF8f0D86D2B45600';
+  const yTUSDAddress = '0x73a052500105205d34Daf004eAb301916DA8190f';
+  const yWBTCAddress = '0x04Aa51bbcB46541455cCF1B8bef2ebc5d3787EC9';
+
+  const iearnAdapterAssets = [
+    yDAIAddress,
+    yUSDCAddress,
+    yUSDTAddress,
+    ySUSDAddress,
+    yTUSDAddress,
+    yWBTCAddress,
+  ];
+
+  let adapterRegistry;
+
+  deployer.deploy(IearnAdapter, { from: accounts[0] })
+    .then(() => AdapterRegistry.at('0xaF51e57D3c78CE8495219Ceb6d559B85E62F680E')
+      .then((registry) => {
+        adapterRegistry = registry;
+        return adapterRegistry.methods['addAdapter(address,address[])'](IearnAdapter.address, iearnAdapterAssets, { from: accounts[0] });
+      }));
+};

test/IearnAdapter.js

@@ -0,0 +1,67 @@
+const { BN } = web3.utils;
+const AdapterRegistry = artifacts.require('./AdapterRegistry');
+const IearnAdapter = artifacts.require('./IearnAdapter');
+
+contract('IearnAdapter', () => {
+  const daiAddress = '0x6B175474E89094C44Da98b954EedeAC495271d0F';
+  const yDAIAddress = '0x16de59092dAE5CcF4A1E6439D611fd0653f0Bd01';
+  const yUSDCAddress = '0xd6aD7a6750A7593E092a9B218d66C0A814a3436e';
+  const yUSDTAddress = '0x83f798e925BcD4017Eb265844FDDAbb448f1707D';
+  const ySUSDAddress = '0xF61718057901F84C4eEC4339EF8f0D86D2B45600';
+  const yTUSDAddress = '0x73a052500105205d34Daf004eAb301916DA8190f';
+  const yWBTCAddress = '0x04Aa51bbcB46541455cCF1B8bef2ebc5d3787EC9';
+  const testAddress = '0x42b9dF65B219B3dD36FF330A4dD8f327A6Ada990';
+
+  let accounts;
+  let adapterRegistry;
+  let iearnAdapter;
+
+  beforeEach(async () => {
+    accounts = await web3.eth.getAccounts();
+    await IearnAdapter.new({ from: accounts[0] })
+      .then((result) => {
+        iearnAdapter = result.contract;
+      });
+    await AdapterRegistry.new(
+      [iearnAdapter.options.address],
+      [[yDAIAddress,
+        yUSDCAddress,
+        yUSDTAddress,
+        ySUSDAddress,
+        yTUSDAddress,
+        yWBTCAddress,
+      ]],
+      { from: accounts[0] },
+    )
+      .then((result) => {
+        adapterRegistry = result.contract;
+      });
+  });
+
+  it('should be correct balances and rates', async () => {
+    await adapterRegistry.methods['getBalancesAndRates(address)'](testAddress)
+      .call()
+      .then((result) => {
+        const base = new BN(10).pow(new BN(34));
+        const yDAIAmount = new BN(result[0].balances[0].amount);
+        const yDAIRate = new BN(result[0].rates[0].components[0].rate);
+        const daiAmount = yDAIRate.mul(yDAIAmount).div(base).toNumber() / 100;
+        // eslint-disable-next-line no-console
+        console.log(`Deposited yDAI amount: ${yDAIAmount.toString()}`);
+        // eslint-disable-next-line no-console
+        console.log(`yDAI rate: ${yDAIRate.toString()}`);
+        // eslint-disable-next-line no-console
+        console.log(`Means its: ${daiAmount} DAI locked`);
+        // eslint-disable-next-line no-console
+        console.log(`Deposited yUSDC amount: ${result[0].balances[1].amount.toString()}`);
+        // eslint-disable-next-line no-console
+        console.log(`Deposited yUSDT amount: ${result[0].balances[2].amount.toString()}`);
+
+        assert.equal(result[0].balances[0].decimals, 18);
+        assert.equal(result[0].balances[0].asset, yDAIAddress);
+        assert.equal(result[0].rates[0].asset, yDAIAddress);
+        assert.equal(result[0].rates[0].components[0].underlying, daiAddress);
+        assert.equal(result[0].name, 'iearn.finance');
+      });
+  });
+});