Claim scripts for marketplaces

Universal-Page · Nov 27, 2023 · 3f8c7c1 · 3f8c7c1
1 parent bda83de
commit 3f8c7c1
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 10 deletions.
diff --git a/scripts/marketplace/lsp7/LSP7Marketplace.s.sol b/scripts/marketplace/lsp7/LSP7Marketplace.s.sol
@@ -47,6 +47,18 @@ contract Deploy is Script {
     }
 }
 
+contract Claim is Script {
+    function run() external {
+        address owner = vm.envAddress("OWNER_ADDRESS");
+        LSP7Marketplace marketplace = LSP7Marketplace(payable(vm.envAddress("CONTRACT_LSP7_MARKETPLACE_ADDRESS")));
+
+        if (address(marketplace).balance > 0) {
+            vm.broadcast(owner);
+            marketplace.withdraw(address(marketplace).balance);
+        }
+    }
+}
+
 contract Configure is Script {
     function run() external {
         address owner = vm.envAddress("OWNER_ADDRESS");
@@ -85,10 +97,5 @@ contract Configure is Script {
             vm.broadcast(owner);
             marketplace.setParticipant(participant);
         }
-
-        if (address(marketplace).balance > 0) {
-            vm.broadcast(owner);
-            marketplace.withdraw(address(marketplace).balance);
-        }
     }
 }
diff --git a/scripts/marketplace/lsp8/LSP8Marketplace.s.sol b/scripts/marketplace/lsp8/LSP8Marketplace.s.sol
@@ -48,6 +48,18 @@ contract Deploy is Script {
     }
 }
 
+contract Claim is Script {
+    function run() external {
+        address owner = vm.envAddress("OWNER_ADDRESS");
+        LSP8Marketplace marketplace = LSP8Marketplace(payable(vm.envAddress("CONTRACT_LSP8_MARKETPLACE_ADDRESS")));
+
+        if (address(marketplace).balance > 0) {
+            vm.broadcast(owner);
+            marketplace.withdraw(address(marketplace).balance);
+        }
+    }
+}
+
 contract Configure is Script {
     function run() external {
         address owner = vm.envAddress("OWNER_ADDRESS");
@@ -98,10 +110,5 @@ contract Configure is Script {
             vm.broadcast(owner);
             marketplace.setParticipant(participant);
         }
-
-        if (address(marketplace).balance > 0) {
-            vm.broadcast(owner);
-            marketplace.withdraw(address(marketplace).balance);
-        }
     }
 }