Genesis config for pallet-ddc-payouts (#207)

Cerebellum-Network · Dec 20, 2023 · ab4c93e · ab4c93e
1 parent 5febf20
commit ab4c93e
Show file tree

Hide file tree

Showing 3 changed files with 39 additions and 1 deletion.
diff --git a/node/service/chain-specs/example.json b/node/service/chain-specs/example.json
@@ -193,7 +193,7 @@
               "mode": "Storage"
             }
           }
-        ],
+        ]
       },
       "ddcClusters": {
         "clusters": [
@@ -233,6 +233,16 @@
             ]
           ]
         ]
+      },
+      "ddcPayouts": {
+        "authorisedCaller": "5GrwvaEF5zXb26Fz9rcQpDWS57CtERHpNehXCPcNoHGKutQY",
+        "debtorCustomers": [
+          [
+            "0x0000000000000000000000000000000000000001",
+            "5GrwvaEF5zXb26Fz9rcQpDWS57CtERHpNehXCPcNoHGKutQY",
+            10000000000
+          ]
+        ]
       }
     }
   }

diff --git a/node/service/src/chain_spec.rs b/node/service/src/chain_spec.rs
@@ -235,6 +235,7 @@ pub fn cere_dev_genesis(
 		nomination_pools: Default::default(),
 		ddc_clusters: Default::default(),
 		ddc_nodes: Default::default(),
+		ddc_payouts: Default::default(),
 	}
 }
 

diff --git a/pallets/ddc-payouts/src/lib.rs b/pallets/ddc-payouts/src/lib.rs
@@ -286,6 +286,33 @@ pub mod pallet {
 		Finalized = 7,
 	}
 
+	#[pallet::genesis_config]
+	pub struct GenesisConfig<T: Config> {
+		pub authorised_caller: Option<T::AccountId>,
+		pub debtor_customers: Vec<(ClusterId, T::AccountId, u128)>,
+	}
+
+	#[cfg(feature = "std")]
+	impl<T: Config> Default for GenesisConfig<T> {
+		fn default() -> Self {
+			GenesisConfig {
+				authorised_caller: Default::default(),
+				debtor_customers: Default::default(),
+			}
+		}
+	}
+
+	#[pallet::genesis_build]
+	impl<T: Config> GenesisBuild<T> for GenesisConfig<T> {
+		fn build(&self) {
+			AuthorisedCaller::<T>::set(self.authorised_caller.clone());
+
+			for (cluster_id, customer_id, debt) in &self.debtor_customers {
+				DebtorCustomers::<T>::insert(cluster_id, customer_id, debt);
+			}
+		}
+	}
+
 	#[pallet::call]
 	impl<T: Config> Pallet<T> {
 		#[pallet::weight(T::WeightInfo::set_authorised_caller())]