Add StorageController trait to contracts

d0rich · Nov 22, 2023 · 2c48202 · 2c48202
1 parent 81a2f27
commit 2c48202
Show file tree

Hide file tree

Showing 4 changed files with 52 additions and 3 deletions.
diff --git a/contracts/contracts/d_social_network_blog.tact b/contracts/contracts/d_social_network_blog.tact
@@ -2,10 +2,11 @@ import "@stdlib/deploy";
 import "./lib/trait_nft_collection.tact";
 import "./lib/trait_nft_collection_royalty_extention.tact";
 import "./lib/trait_personal_fees.tact";
+import "./lib/trait_storage_controller.tact";
 import "./messages_d.tact";
 import "./d_social_network_post.tact";
 
-contract DSocialNetworkBlog with NftCollection, NftCollectionRoyaltyExtention, PersonalFees {
+contract DSocialNetworkBlog with NftCollection, NftCollectionRoyaltyExtention, PersonalFees, StorageController {
     blog_id: Int;
     next_item_index: Int as uint32 = 0;
     owner: Address;

diff --git a/contracts/contracts/d_social_network_master.tact b/contracts/contracts/d_social_network_master.tact
@@ -2,10 +2,11 @@ import "@stdlib/deploy";
 import "@stdlib/ownable";
 import "./lib/trait_accounting.tact";
 import "./lib/trait_personal_fees.tact";
+import "./lib/trait_storage_controller.tact";
 import "./messages_d.tact";
 import "./d_social_network_blog.tact";
 
-contract DSocialNetworkMaster with Deployable, OwnableTransferable, Accounting, PersonalFees  {
+contract DSocialNetworkMaster with Deployable, OwnableTransferable, Accounting, PersonalFees, StorageController  {
     version: String = "v0.0.1-beta";
     owner: Address;
     personal_fees_destination: Address;

diff --git a/contracts/contracts/d_social_network_post.tact b/contracts/contracts/d_social_network_post.tact
@@ -1,9 +1,10 @@
 import "@stdlib/deploy";
 import "@stdlib/ownable";
 import "./lib/trait_nft_item.tact";
+import "./lib/trait_storage_controller.tact";
 import "./messages_d.tact";
 
-contract DSocialNetworkPost with NftItem {
+contract DSocialNetworkPost with NftItem, StorageController {
     collection_address: Address;
     item_index: Int;
     owner: Address;

diff --git a/contracts/contracts/lib/trait_storage_controller.tact b/contracts/contracts/lib/trait_storage_controller.tact
@@ -0,0 +1,46 @@
+import "@stdlib/ownable";
+import "./trait_accounting";
+
+message Withdraw {
+    query_id: Int as uint64;
+    amount: Int;
+}
+
+message WithdrawAll {
+    query_id: Int as uint64;
+    destroy: Bool?;
+}
+
+message TopUp {
+    query_id: Int as uint64;
+}
+
+trait StorageController with Ownable, Accounting {
+    owner: Address;
+
+    receive(msg: TopUp) {}
+
+    receive(msg: Withdraw) {
+        self.requireOwner();
+        send(SendParameters{
+            to: self.owner,
+            value: min(msg.amount, self.calculateSendAmountToHoldEnoughTonForStorage()),
+            mode: SendIgnoreErrors,
+            body: "Withdraw".asComment()
+        });
+    }
+
+    receive(msg: WithdrawAll) {
+        self.requireOwner();
+        let additionalMode: Int = 0;
+        if (msg.destroy == true) {
+            additionalMode = additionalMode + SendDestroyIfZero;
+        }
+        send(SendParameters{
+            to: self.owner,
+            value: 0,
+            mode: SendRemainingBalance + SendIgnoreErrors + additionalMode,
+            body: "Withdraw all".asComment()
+        });
+    }
+}