iface: add add_rights convenience method to builder

RGB-WG · Oct 17, 2024 · 9240917 · 9240917
1 parent 0bc16ef
commit 9240917
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 1 deletion.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/src/interface/builder.rs b/src/interface/builder.rs
@@ -252,6 +252,16 @@ impl ContractBuilder {
         Ok(self)
     }
 
+    pub fn add_rights(
+        mut self,
+        name: impl Into<FieldName>,
+        seal: impl Into<BuilderSeal<GenesisSeal>>,
+        attach: Option<AttachId>,
+    ) -> Result<Self, BuilderError> {
+        self.builder = self.builder.add_rights(name, seal, attach)?;
+        Ok(self)
+    }
+
     pub fn add_owned_state(
         mut self,
         name: impl Into<FieldName>,
@@ -513,6 +523,16 @@ impl TransitionBuilder {
         Ok(self)
     }
 
+    pub fn add_rights(
+        mut self,
+        name: impl Into<FieldName>,
+        seal: impl Into<BuilderSeal<GraphSeal>>,
+        attach: Option<AttachId>,
+    ) -> Result<Self, BuilderError> {
+        self.builder = self.builder.add_rights(name, seal, attach)?;
+        Ok(self)
+    }
+
     pub fn fulfill_owned_state(
         mut self,
         type_id: AssignmentType,
@@ -758,6 +778,18 @@ impl<Seal: ExposedSeal> OperationBuilder<Seal> {
         Ok(self)
     }
 
+    fn add_rights(
+        self,
+        name: impl Into<FieldName>,
+        seal: impl Into<BuilderSeal<Seal>>,
+        attach: Option<AttachId>,
+    ) -> Result<Self, BuilderError> {
+        let type_id = self.assignments_type(name)?;
+        let mut state = State::default();
+        state.attach = attach;
+        self.add_owned_state_raw(type_id, seal, state)
+    }
+
     fn add_owned_state(
         self,
         name: impl Into<FieldName>,