Merge pull request RGB-WG#219 from cymqqqq/more-operations

add rights, attachment operations
crisdut · Jun 18, 2024 · d2f99cf · d2f99cf
2 parents 41c881c + e27ece8
commit d2f99cf
Showing 1 changed file with 34 additions and 0 deletions.
diff --git a/src/interface/contract.rs b/src/interface/contract.rs
@@ -443,4 +443,38 @@ impl ContractIface {
             witness_filter,
         ))
     }
+
+    pub fn rights_ops<C: StateChange<State = VoidState>>(
+        &self,
+        name: impl Into<FieldName>,
+        witness_filter: impl WitnessFilter + Copy,
+        outpoint_filter: impl OutpointFilter + Copy,
+    ) -> Result<HashMap<XWitnessId, IfaceOp<C>>, ContractError> {
+        Ok(self.operations(
+            self.state
+                .rights()
+                .iter()
+                .cloned()
+                .map(OutputAssignment::transmute),
+            self.rights(name, outpoint_filter)?,
+            witness_filter,
+        ))
+    }
+
+    pub fn attachment_ops<C: StateChange<State = AttachedState>>(
+        &self,
+        name: impl Into<FieldName>,
+        witness_filter: impl WitnessFilter + Copy,
+        outpoint_filter: impl OutpointFilter + Copy,
+    ) -> Result<HashMap<XWitnessId, IfaceOp<C>>, ContractError> {
+        Ok(self.operations(
+            self.state
+                .attach()
+                .iter()
+                .cloned()
+                .map(OutputAssignment::transmute),
+            self.attachments(name, outpoint_filter)?,
+            witness_filter,
+        ))
+    }
 }