From def8e5e1afd0cd853c4ab48257b0603b54512c69 Mon Sep 17 00:00:00 2001
From: Deadshot Ryker <ryker@zerolend.xyz>
Date: Tue, 1 Oct 2024 09:50:27 +0400
Subject: [PATCH] wrote a script to update NFT vests

---
 contracts/vesting/UpdateZeroNFTScript.sol | 43 +++++++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 contracts/vesting/UpdateZeroNFTScript.sol

diff --git a/contracts/vesting/UpdateZeroNFTScript.sol b/contracts/vesting/UpdateZeroNFTScript.sol
new file mode 100644
index 0000000..c528d53
--- /dev/null
+++ b/contracts/vesting/UpdateZeroNFTScript.sol
@@ -0,0 +1,43 @@
+// SPDX-License-Identifier: UNLICENSED
+pragma solidity ^0.8.20;
+
+// ███████╗███████╗██████╗  ██████╗
+// ╚══███╔╝██╔════╝██╔══██╗██╔═══██╗
+//   ███╔╝ █████╗  ██████╔╝██║   ██║
+//  ███╔╝  ██╔══╝  ██╔══██╗██║   ██║
+// ███████╗███████╗██║  ██║╚██████╔╝
+// ╚══════╝╚══════╝╚═╝  ╚═╝ ╚═════╝
+
+// Website: https://zerolend.xyz
+// Discord: https://discord.gg/zerolend
+// Twitter: https://twitter.com/zerolendxyz
+
+import {IVestedZeroNFT} from "../interfaces/IVestedZeroNFT.sol";
+import {AccessControlEnumerable} from "@openzeppelin/contracts/access/extensions/AccessControlEnumerable.sol";
+
+interface IVestedZeroNFTWithLock is IVestedZeroNFT {
+    function tokenIdToLockDetails(uint256) external view returns (LockDetails memory);
+}
+
+contract UpdateZeroNFTScript is AccessControlEnumerable {
+    IVestedZeroNFTWithLock public vest;
+
+    constructor(address _vest) {
+        vest = IVestedZeroNFTWithLock(_vest);
+        _grantRole(DEFAULT_ADMIN_ROLE, msg.sender);
+    }
+
+    function update(uint256[] memory ids) external onlyRole(DEFAULT_ADMIN_ROLE) {
+        uint256[] memory linearDurations = new uint256[](ids.length);
+        uint256[] memory cliffDurations = new uint256[](ids.length);
+
+        for (uint256 i = 0; i < ids.length; i++) {
+            IVestedZeroNFT.LockDetails memory lock = vest.tokenIdToLockDetails(ids[i]);
+            require(lock.category == IVestedZeroNFT.VestCategory.AIRDROP, "!category");
+            linearDurations[i] = 86400 * 30 * 3; // 3 mo linear
+            cliffDurations[i] = 86400 * 30 * 3; // 3 mo cliff
+        }
+
+        vest.updateCliffDuration(ids, linearDurations, cliffDurations);
+    }
+}