Introduce ScopedCapability{Keeper,Reader}

informalsystems · Apr 8, 2022 · ec20b5a · ec20b5a
1 parent a7b3b48
commit ec20b5a
Show file tree

Hide file tree

Showing 2 changed files with 58 additions and 0 deletions.
diff --git a/modules/src/core/ics05_port/capabilities.rs b/modules/src/core/ics05_port/capabilities.rs
@@ -92,6 +92,11 @@ impl CapabilityName {
             Err(InvalidCapabilityName)
         }
     }
+
+    pub(crate) fn prefixed_with(&self, mut prefix: String) -> Self {
+        prefix.push_str(&self.0);
+        Self(prefix)
+    }
 }
 
 impl fmt::Display for CapabilityName {

diff --git a/modules/src/core/ics05_port/context.rs b/modules/src/core/ics05_port/context.rs
@@ -101,3 +101,56 @@ fn port_capability_name(port_id: PortId) -> CapabilityName {
         .parse()
         .expect("PortsPath cannot be empty string")
 }
+
+pub struct ScopedCapabilityReader<CR, M> {
+    capability_reader: CR,
+    module_id: M,
+}
+
+impl<CR: CapabilityReader, M: ToString> CapabilityReader for ScopedCapabilityReader<CR, M> {
+    fn lookup_module(&self, name: &CapabilityName) -> Result<(ModuleId, Capability), Error> {
+        self.capability_reader
+            .lookup_module(&name.prefixed_with(self.module_id.to_string()))
+    }
+
+    fn get_capability(&self, name: &CapabilityName) -> Result<Capability, Error> {
+        self.capability_reader
+            .get_capability(&name.prefixed_with(self.module_id.to_string()))
+    }
+
+    fn authenticate_capability(
+        &self,
+        name: &CapabilityName,
+        capability: &Capability,
+    ) -> Result<(), Error> {
+        self.capability_reader
+            .authenticate_capability(&name.prefixed_with(self.module_id.to_string()), capability)
+    }
+
+    fn create_capability(&self, name: CapabilityName) -> Result<Capability, Error> {
+        self.capability_reader
+            .create_capability(name.prefixed_with(self.module_id.to_string()))
+    }
+}
+
+pub struct ScopedCapabilityKeeper<CK, M> {
+    capability_keeper: CK,
+    module_id: M,
+}
+
+impl<CK: CapabilityKeeper, M: ToString> CapabilityKeeper for ScopedCapabilityKeeper<CK, M> {
+    fn new_capability(&mut self, name: CapabilityName) -> Result<Capability, Error> {
+        self.capability_keeper
+            .new_capability(name.prefixed_with(self.module_id.to_string()))
+    }
+
+    fn claim_capability(&mut self, name: CapabilityName, capability: Capability) {
+        self.capability_keeper
+            .claim_capability(name.prefixed_with(self.module_id.to_string()), capability)
+    }
+
+    fn release_capability(&mut self, name: CapabilityName, capability: Capability) {
+        self.capability_keeper
+            .release_capability(name.prefixed_with(self.module_id.to_string()), capability)
+    }
+}