Increase airdrop throughput

Author: Sukhveer Sanghera

Diff for: contracts/PolyDistribution.sol

@@ -41,6 +41,15 @@ contract PolyDistribution is Ownable {
     uint256 amountClaimed;  // Total tokens claimed
   }
   mapping (address => Allocation) public allocations;
+  
+  mapping (address => bool) public airdropAdmins;
+  
+  mapping (address => bool) public airdrops;
+
+  modifier onlyOwnerOrAdmin() {
+    require(msg.sender == owner || airdropAdmins[msg.sender]);
+    _;
+  }
 
   event LogNewAllocation(address indexed _recipient, AllocationType indexed _fromSupply, uint256 _totalAllocated, uint256 _grandTotalAllocated);
   event LogPolyClaimed(address indexed _recipient, uint8 indexed _fromSupply, uint256 _amountClaimed, uint256 _totalAllocated, uint256 _grandTotalClaimed);
@@ -72,9 +81,6 @@ contract PolyDistribution is Ownable {
     } else if (_supply == AllocationType.FOUNDER) {
       AVAILABLE_FOUNDER_SUPPLY = AVAILABLE_FOUNDER_SUPPLY.sub(_totalAllocated);
       allocations[_recipient] = Allocation(uint8(AllocationType.FOUNDER), startTime + 1 years, startTime + 4 years, _totalAllocated, 0);
-    } else if (_supply == AllocationType.AIRDROP) {
-      AVAILABLE_AIRDROP_SUPPLY = AVAILABLE_AIRDROP_SUPPLY.sub(_totalAllocated);
-      allocations[_recipient] = Allocation(uint8(AllocationType.AIRDROP), 0, 0, _totalAllocated, 0);
     } else if (_supply == AllocationType.ADVISOR) {
       AVAILABLE_ADVISOR_SUPPLY = AVAILABLE_ADVISOR_SUPPLY.sub(_totalAllocated);
       allocations[_recipient] = Allocation(uint8(AllocationType.ADVISOR), startTime + 212 days, 0, _totalAllocated, 0);
@@ -94,6 +100,35 @@ contract PolyDistribution is Ownable {
     AVAILABLE_TOTAL_SUPPLY = AVAILABLE_TOTAL_SUPPLY.sub(_totalAllocated);
     LogNewAllocation(_recipient, _supply, _totalAllocated, grandTotalAllocated());
   }
+  
+  /**
+    * @dev Add an airdrop admin
+    * @param _admin
+    * @param _isAdmin
+    */
+  function setAirdropAdmin(address _admin, bool _isAdmin) public onlyOwner {
+    airdropAdmins[_admin] = _isAdmin;
+  }
+  
+  /**
+    * @dev perform a transfer of allocations
+    * @param _reciepients
+    * @param _allocated
+    */
+  function airdropTokens(address[] _recipient) public onlyOwnerOrAdmin {
+    require(_startTime >= now);
+    uint airdropped;
+    for(uint8 i = 0; i< _recipient.length; i++)
+    {
+        if (!airdrops[_recipient[i]]) {
+          airdrops[_recipient[i]] = true;
+          require(POLY.transfer(_recipient[i], 250*(decimalsFactor));
+          airdropped = airdropped.add(250*decimalsFactor);
+        }
+    }
+    AVAILABLE_AIRDROP_SUPPLY = AVAILABLE_AIRDROP_SUPPLY.sub(airdropped);
+    AVAILABLE_TOTAL_SUPPLY = AVAILABLE_TOTAL_SUPPLY.sub(airdropped);
+  }
 
   /**
     * @dev Transfer a recipients available allocation to their address