aboutsummaryrefslogtreecommitdiffstats
path: root/packages/migrations/src/migration.ts
diff options
context:
space:
mode:
Diffstat (limited to 'packages/migrations/src/migration.ts')
-rw-r--r--packages/migrations/src/migration.ts192
1 files changed, 192 insertions, 0 deletions
diff --git a/packages/migrations/src/migration.ts b/packages/migrations/src/migration.ts
new file mode 100644
index 000000000..a22c34a23
--- /dev/null
+++ b/packages/migrations/src/migration.ts
@@ -0,0 +1,192 @@
+import { artifacts, wrappers } from '@0xproject/contracts';
+import { assetDataUtils } from '@0xproject/order-utils';
+import { ContractAddresses } from '@0xproject/types';
+import { BigNumber } from '@0xproject/utils';
+import { Web3Wrapper } from '@0xproject/web3-wrapper';
+import { Provider, TxData } from 'ethereum-types';
+import * as _ from 'lodash';
+
+import { erc20TokenInfo, erc721TokenInfo } from './utils/token_info';
+
+interface MigrationsResult {
+ erc20Proxy: wrappers.ERC20ProxyContract;
+ erc721Proxy: wrappers.ERC721ProxyContract;
+ zrxToken: wrappers.ZRXTokenContract;
+ etherToken: wrappers.WETH9Contract;
+ exchange: wrappers.ExchangeContract;
+ assetProxyOwner: wrappers.AssetProxyOwnerContract;
+ forwarder: wrappers.ForwarderContract;
+ orderValidator: wrappers.OrderValidatorContract;
+}
+
+let _cachedMigrationsResult: MigrationsResult | undefined;
+let _cachedContractAddresses: ContractAddresses | undefined;
+
+/**
+ * Custom migrations should be defined in this function. This will be called with the CLI 'migrate:v2' command.
+ * Migrations could be written to run in parallel, but if you want contract addresses to be created deterministically,
+ * the migration should be written to run synchronously.
+ * @param provider Web3 provider instance.
+ * @param txDefaults Default transaction values to use when deploying contracts.
+ */
+export async function runMigrationsAsync(provider: Provider, txDefaults: Partial<TxData>): Promise<MigrationsResult> {
+ const web3Wrapper = new Web3Wrapper(provider);
+
+ // Proxies
+ const erc20Proxy = await wrappers.ERC20ProxyContract.deployFrom0xArtifactAsync(
+ artifacts.ERC20Proxy,
+ provider,
+ txDefaults,
+ );
+ const erc721Proxy = await wrappers.ERC721ProxyContract.deployFrom0xArtifactAsync(
+ artifacts.ERC721Proxy,
+ provider,
+ txDefaults,
+ );
+
+ // ZRX
+ const zrxToken = await wrappers.ZRXTokenContract.deployFrom0xArtifactAsync(
+ artifacts.ZRXToken,
+ provider,
+ txDefaults,
+ );
+
+ // Ether token
+ const etherToken = await wrappers.WETH9Contract.deployFrom0xArtifactAsync(artifacts.WETH9, provider, txDefaults);
+
+ // Exchange
+ const zrxAssetData = assetDataUtils.encodeERC20AssetData(zrxToken.address);
+ const exchange = await wrappers.ExchangeContract.deployFrom0xArtifactAsync(
+ artifacts.Exchange,
+ provider,
+ txDefaults,
+ zrxAssetData,
+ );
+
+ // Multisigs
+ const accounts: string[] = await web3Wrapper.getAvailableAddressesAsync();
+ const owners = [accounts[0], accounts[1]];
+ const confirmationsRequired = new BigNumber(2);
+ const secondsRequired = new BigNumber(0);
+ const owner = accounts[0];
+
+ // AssetProxyOwner
+ const assetProxyOwner = await wrappers.AssetProxyOwnerContract.deployFrom0xArtifactAsync(
+ artifacts.AssetProxyOwner,
+ provider,
+ txDefaults,
+ owners,
+ [erc20Proxy.address, erc721Proxy.address],
+ confirmationsRequired,
+ secondsRequired,
+ );
+
+ await web3Wrapper.awaitTransactionSuccessAsync(
+ await erc20Proxy.addAuthorizedAddress.sendTransactionAsync(exchange.address, {
+ from: owner,
+ }),
+ );
+ await web3Wrapper.awaitTransactionSuccessAsync(
+ await erc20Proxy.transferOwnership.sendTransactionAsync(assetProxyOwner.address, {
+ from: owner,
+ }),
+ );
+ await web3Wrapper.awaitTransactionSuccessAsync(
+ await erc721Proxy.addAuthorizedAddress.sendTransactionAsync(exchange.address, {
+ from: owner,
+ }),
+ );
+ await web3Wrapper.awaitTransactionSuccessAsync(
+ await erc721Proxy.transferOwnership.sendTransactionAsync(assetProxyOwner.address, {
+ from: owner,
+ }),
+ );
+
+ // Register the Asset Proxies to the Exchange
+ await web3Wrapper.awaitTransactionSuccessAsync(
+ await exchange.registerAssetProxy.sendTransactionAsync(erc20Proxy.address),
+ );
+ await web3Wrapper.awaitTransactionSuccessAsync(
+ await exchange.registerAssetProxy.sendTransactionAsync(erc721Proxy.address),
+ );
+
+ // Dummy ERC20 tokens
+ for (const token of erc20TokenInfo) {
+ const totalSupply = new BigNumber(1000000000000000000000000000);
+ // tslint:disable-next-line:no-unused-variable
+ const dummyErc20Token = await wrappers.DummyERC20TokenContract.deployFrom0xArtifactAsync(
+ artifacts.DummyERC20Token,
+ provider,
+ txDefaults,
+ token.name,
+ token.symbol,
+ token.decimals,
+ totalSupply,
+ );
+ }
+
+ // ERC721
+ // tslint:disable-next-line:no-unused-variable
+ const cryptoKittieToken = await wrappers.DummyERC721TokenContract.deployFrom0xArtifactAsync(
+ artifacts.DummyERC721Token,
+ provider,
+ txDefaults,
+ erc721TokenInfo[0].name,
+ erc721TokenInfo[0].symbol,
+ );
+
+ // Forwarder
+ const forwarder = await wrappers.ForwarderContract.deployFrom0xArtifactAsync(
+ artifacts.Forwarder,
+ provider,
+ txDefaults,
+ exchange.address,
+ assetDataUtils.encodeERC20AssetData(zrxToken.address),
+ assetDataUtils.encodeERC20AssetData(etherToken.address),
+ );
+
+ // OrderValidator
+ const orderValidator = await wrappers.OrderValidatorContract.deployFrom0xArtifactAsync(
+ artifacts.OrderValidator,
+ provider,
+ txDefaults,
+ exchange.address,
+ zrxAssetData,
+ );
+
+ const migrationsResult = {
+ erc20Proxy,
+ erc721Proxy,
+ zrxToken,
+ etherToken,
+ exchange,
+ assetProxyOwner,
+ forwarder,
+ orderValidator,
+ };
+ _cachedMigrationsResult = migrationsResult;
+ return migrationsResult;
+}
+
+export function getContractAddresses(): ContractAddresses {
+ if (!_.isUndefined(_cachedContractAddresses)) {
+ return _cachedContractAddresses;
+ }
+ if (_.isUndefined(_cachedMigrationsResult)) {
+ throw new Error(
+ 'Migrations have not been run! You need to call runMigrationsAsync before getContractAddresses',
+ );
+ }
+ const contractAddresses = {
+ erc20Proxy: _cachedMigrationsResult.erc20Proxy.address,
+ erc721Proxy: _cachedMigrationsResult.erc721Proxy.address,
+ zrxToken: _cachedMigrationsResult.zrxToken.address,
+ etherToken: _cachedMigrationsResult.etherToken.address,
+ exchange: _cachedMigrationsResult.exchange.address,
+ assetProxyOwner: _cachedMigrationsResult.assetProxyOwner.address,
+ forwarder: _cachedMigrationsResult.forwarder.address,
+ orderValidator: _cachedMigrationsResult.orderValidator.address,
+ };
+ _cachedContractAddresses = contractAddresses;
+ return contractAddresses;
+}