Merge pull request #15 from LFDT-Lockness/party-hard

Add methods to party to change its components

Author: maurges

round-based/CHANGELOG.md

@@ -1,3 +1,6 @@
+## v0.4.1
+* Add methods to MpcParty to change its components [#15]
+
 ## v0.4.0
 * BREAKING: Improve ergonomics of protocol simulation, which is used for writing tests [#14]
   * Remove `dev` feature, it's replaced with `sim` and `sim-async`

round-based/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "round-based"
-version = "0.4.0"
+version = "0.4.1"
 edition = "2021"
 license = "MIT OR Apache-2.0"
 description = "Driver for MPC protocols"

round-based/src/party.rs

@@ -123,6 +123,26 @@ impl<M, D, X> MpcParty<M, D, X>
 where
     D: Delivery<M>,
 {
+    /// Modify the delivery of this party while keeping everything else the same
+    pub fn map_delivery<D2>(self, f: impl FnOnce(D) -> D2) -> MpcParty<M, D2, X> {
+        let delivery = f(self.delivery);
+        MpcParty {
+            delivery,
+            runtime: self.runtime,
+            _msg: self._msg,
+        }
+    }
+
+    /// Modify the runtime of this party while keeping everything else the same
+    pub fn map_runtime<R>(self, f: impl FnOnce(X) -> R) -> MpcParty<M, D, R> {
+        let runtime = f(self.runtime);
+        MpcParty {
+            delivery: self.delivery,
+            runtime,
+            _msg: self._msg,
+        }
+    }
+
     /// Specifies a [async runtime](runtime)
     pub fn set_runtime<R>(self, runtime: R) -> MpcParty<M, D, R>
     where