Deployer.sol

// SPDX-License-Identifier: GPL-3.0-only
pragma solidity ^0.6.0;

import { Ownable } from "@openzeppelin/contracts/access/Ownable.sol";
import { IERC20 } from "@openzeppelin/contracts/token/ERC20/IERC20.sol";

import { Buyback } from "./Buyback.sol";
import { CustomMasterChef } from "./CustomMasterChef.sol";
import { FeeCollector } from "./FeeCollector.sol";
import { MasterChefAdmin } from "./MasterChefAdmin.sol";
import { RewardCompoundingStrategyToken } from "./RewardCompoundingStrategyToken.sol";
import { WHEAT, stkWHEAT } from "./Tokens.sol";

import { Factory, Pair } from "./interop/UniswapV2.sol";

import { Transfers } from "./modules/Transfers.sol";
import { Wrapping } from "./modules/Wrapping.sol";

import { $ } from "./network/$.sol";

contract Deployer is Ownable
{
	address constant DEFAULT_ADMIN = 0xAD4E38B274720c1a6c7fB8B735C5FAD112DF9A13;
	address constant DEFAULT_TREASURY = 0x0d1d68C73b57a53B1DdCD287aCf4e66Ed745B759;
	address constant DEFAULT_DEV = 0x7674D2a14076e8af53AC4ba9bBCf0c19FeBe8899;
	address constant DEFAULT_YIELD = 0x0d1d68C73b57a53B1DdCD287aCf4e66Ed745B759;

	uint256 public constant INITIAL_WHEAT_PER_BLOCK = 0e18;

	uint256 public constant WHEAT_LIQUIDITY_ALLOCATION = 10000e18; // 10k WHEAT
	uint256 public constant WBNB_LIQUIDITY_ALLOCATION = 200e18; // ~100k USD at deploy

	address public admin;
	address public treasury;
	address public dev;
	address public yield;

	address public exchange;

	address public wheat;
	address public stkWheat;
	address public masterChef;
	address public masterChefAdmin;
	address public buyback;

	enum Stage {
		Deploy, Batch1, Batch2, Batch3, Done
	}

	Stage public stage = Stage.Deploy;

	constructor () public
	{
		require($.NETWORK == $.network(), "wrong network");
	}

	function deploy() external payable onlyOwner
	{
		require(stage == Stage.Deploy, "unavailable");

		uint256 _amount = msg.value;
		require(_amount == WBNB_LIQUIDITY_ALLOCATION, "BNB amount mismatch");

		// wraps LP liquidity BNB into WBNB
		Wrapping._wrap($.WBNB, WBNB_LIQUIDITY_ALLOCATION);

		admin = DEFAULT_ADMIN;
		treasury = DEFAULT_TREASURY;
		dev = DEFAULT_DEV;
		yield = DEFAULT_YIELD;

		exchange = $.WHEAT_EXCHANGE_IMPL;

		// configure MasterChef
		wheat = LibDeployer1.new_WHEAT();
		stkWheat = LibDeployer2.new_stkWHEAT(wheat);
		masterChef = LibDeployer1.new_CustomMasterChef(wheat, stkWheat, INITIAL_WHEAT_PER_BLOCK, block.number);
		masterChefAdmin = LibDeployer3.new_MasterChefAdmin(masterChef);

		CustomMasterChef(masterChef).set(0, 15000, false);

		address _factory = $.UniswapV2_Compatible_FACTORY;
		address _BNB_WHEAT = Factory(_factory).createPair($.WBNB, wheat);
		address _BNB_GRO = Factory(_factory).getPair($.WBNB, $.GRO);
		address _GRO_gROOT = Factory(_factory).getPair($.GRO, $.gROOT);
		address _BNB_gROOT = Factory(_factory).getPair($.WBNB, $.gROOT);
		CustomMasterChef(masterChef).add(30000, IERC20(_BNB_WHEAT), false);
		CustomMasterChef(masterChef).add(3000, IERC20(_BNB_GRO), false);
		CustomMasterChef(masterChef).add(1000, IERC20(_GRO_gROOT), false);
		CustomMasterChef(masterChef).add(1000, IERC20(_BNB_gROOT), false);

		// adds the liquidity to the WHEAT/BNB LP
		WHEAT(wheat).mint(_BNB_WHEAT, WHEAT_LIQUIDITY_ALLOCATION);
		Transfers._pushFunds($.WBNB, _BNB_WHEAT, WBNB_LIQUIDITY_ALLOCATION);
		Pair(_BNB_WHEAT).mint(DEFAULT_TREASURY);

		buyback = LibDeployer2.new_Buyback($.CAKE, $.WBNB, wheat, $.GRO, treasury, yield);
		Buyback(buyback).setExchange(exchange);

		require(Transfers._getBalance($.WBNB) == 0, "WBNB left over");
		require(Transfers._getBalance(wheat) == 0, "WHEAT left over");
		require(Transfers._getBalance(_BNB_WHEAT) == 0, "BNB/WHEAT LP shares left over");

		// transfer ownerships
		Ownable(wheat).transferOwnership(masterChef);
		Ownable(stkWheat).transferOwnership(masterChef);
		Ownable(masterChef).transferOwnership(masterChefAdmin);
		Ownable(buyback).transferOwnership(admin);

		stage = Stage.Batch1;
	}

	function batch1() external onlyOwner
	{
		require(stage == Stage.Batch1, "unavailable");
		if ($.NETWORK == $.Network.Bscmain) {
			_addStrategy("staked CAKE", "stkCAKE", 0, $.CAKE, 15000);
			_addStrategy("staked BNB/CAKE", "stkBNB/CAKE", 1, $.CAKE, 15000);
			_addStrategy("staked BNB/BUSD", "stkBNB/BUSD", 2, $.WBNB, 7000);
			_addStrategy("staked BNB/BTCB", "stkBNB/BTCB", 15, $.WBNB, 3000);
		}
		if ($.NETWORK == $.Network.Bsctest) {
			_addStrategy("staked CAKE", "stkCAKE", 0, $.CAKE, 15000);
			_addStrategy("staked BNB/CAKE", "stkBNB/CAKE", 2, $.CAKE, 15000);
			_addStrategy("staked BNB/ETH", "stkBNB/ETH", 4, $.WBNB, 3000);
		}
		stage = Stage.Batch2;
	}

	function batch2() external onlyOwner
	{
		require(stage == Stage.Batch2, "unavailable");
		if ($.NETWORK == $.Network.Bscmain) {
			_addStrategy("staked BNB/ETH", "stkBNB/ETH", 14, $.WBNB, 3000);
			_addStrategy("staked BNB/LINK", "stkBNB/LINK", 7, $.WBNB, 1000);
			_addStrategy("staked BNB/UNI", "stkBNB/UNI", 25, $.WBNB, 1000);
			_addStrategy("staked BNB/DOT", "stkBNB/DOT", 5, $.WBNB, 1000);
		}
		stage = Stage.Batch3;
	}

	function batch3() external onlyOwner
	{
		require(stage == Stage.Batch3, "unavailable");
		if ($.NETWORK == $.Network.Bscmain) {
			_addStrategy("staked BNB/ADA", "stkBNB/ADA", 3, $.WBNB, 1000);
			_addStrategy("staked BUSD/UST", "stkBUSD/UST", 63, $.BUSD, 1000);
			_addStrategy("staked BUSD/DAI", "stkBUSD/DAI", 52, $.BUSD, 1000);
			_addStrategy("staked BUSD/USDC", "stkBUSD/USDC", 53, $.BUSD, 1000);
		}

		// transfer ownerships
		Ownable(masterChefAdmin).transferOwnership(admin);
		renounceOwnership();

		stage = Stage.Done;
		emit DeployPerformed();
	}

	function _addStrategy(string memory _name, string memory _symbol, uint256 _pid, address _routingToken, uint256 _allocPoint) internal
	{
		uint256 _length = CustomMasterChef(masterChef).poolLength();
		MasterChefAdmin(masterChefAdmin).addRewardCompoundingStrategy(_name, _symbol, 18, $.PancakeSwap_MASTERCHEF, _pid, _routingToken, _allocPoint, buyback, exchange, dev, treasury);
		(IERC20 _lptoken,,,) = CustomMasterChef(masterChef).poolInfo(_length);
		address _strategy = address(_lptoken);
		address _collector = RewardCompoundingStrategyToken(_strategy).collector();
		Ownable(_collector).transferOwnership(admin);
		Ownable(_strategy).transferOwnership(admin);
	}

	event DeployPerformed();
}

library LibDeployer1
{
	function new_WHEAT() public returns (address _address)
	{
		return address(new WHEAT());
	}

	function new_CustomMasterChef(address _wheat, address _stkWheat, uint256 _cakePerBlock, uint256 _startBlock) public returns (address _address)
	{
		return address(new CustomMasterChef(_wheat, _stkWheat, _cakePerBlock, _startBlock));
	}
}

library LibDeployer2
{
	function new_stkWHEAT(address _wheat) public returns (address _address)
	{
		return address(new stkWHEAT(_wheat));
	}

	function new_Buyback(address _rewardToken, address _routingToken, address _buybackToken1, address _buybackToken2, address _treasury, address _yield) public returns (address _address)
	{
		return address(new Buyback(_rewardToken, _routingToken, _buybackToken1, _buybackToken2, _treasury, _yield));
	}
}

library LibDeployer3
{
	function new_MasterChefAdmin(address _masterChef) public returns (address _address)
	{
		return address(new MasterChefAdmin(_masterChef));
	}
}