More identifier checks in CoreDocument

bindings/wasm/src/did/wasm_core_document.rs

@@ -56,9 +56,12 @@ impl WasmCoreDocument {
   }
 
   /// Sets the DID of the document.
+  ///
+  ///
+  /// Warning: Improper use can lead to broken URI dereferencing.
   #[wasm_bindgen(js_name = setId)]
   pub fn set_id(&mut self, id: &WasmCoreDID) {
-    *self.0.id_mut() = id.0.clone();
+    *self.0.id_mut_unchecked() = id.0.clone();
   }
 
   /// Returns a copy of the document controllers.
@@ -79,7 +82,8 @@ impl WasmCoreDocument {
   /// Sets the controllers of the DID Document.
   ///
   /// Note: Duplicates will be ignored.
-  /// Use `null` to remove all controllers.
+  /// Use `null` to remove all controllers. Furthermore improper use can lead to
+  /// broken URI dereferencing.
   #[wasm_bindgen(js_name = setController)]
   pub fn set_controller(&mut self, controllers: &OptionOneOrManyCoreDID) -> Result<()> {
     let controllers: Option<OneOrMany<CoreDID>> = controllers.into_serde().wasm_result()?;
@@ -92,7 +96,7 @@ impl WasmCoreDocument {
     } else {
       None
     };
-    *self.0.controller_mut() = controller_set;
+    *self.0.controller_mut_unchecked() = controller_set;
     Ok(())
   }
 
@@ -110,6 +114,8 @@ impl WasmCoreDocument {
   }
 
   /// Sets the `alsoKnownAs` property in the DID document.
+  ///
+  /// Warning: Improper use can lead to broken URI dereferencing.
   #[wasm_bindgen(js_name = setAlsoKnownAs)]
   pub fn set_also_known_as(&mut self, urls: &OptionOneOrManyString) -> Result<()> {
     let urls: Option<OneOrMany<String>> = urls.into_serde().wasm_result()?;
@@ -119,7 +125,7 @@ impl WasmCoreDocument {
         urls_set.append(Url::parse(url).wasm_result()?);
       }
     }
-    *self.0.also_known_as_mut() = urls_set;
+    *self.0.also_known_as_mut_unchecked() = urls_set;
     Ok(())
   }
 
@@ -233,10 +239,10 @@ impl WasmCoreDocument {
     let value: Option<serde_json::Value> = value.into_serde().wasm_result()?;
     match value {
       Some(value) => {
-        self.0.properties_mut().insert(key, value);
+        self.0.properties_mut_unchecked().insert(key, value);
       }
       None => {
-        self.0.properties_mut().remove(&key);
+        self.0.properties_mut_unchecked().remove(&key);
       }
     }
     Ok(())
@@ -262,19 +268,19 @@ impl WasmCoreDocument {
 
   /// Add a new {@link CoreService} to the document.
   ///
-  /// Returns `true` if the service was added.
+  /// Errors if there already exists a service or (verification) method with the same id.
   #[wasm_bindgen(js_name = insertService)]
-  pub fn insert_service(&mut self, service: &WasmCoreService) -> bool {
-    self.0.service_mut().append(service.0.clone())
+  pub fn insert_service(&mut self, service: &WasmCoreService) -> Result<()> {
+    self.0.insert_service(service.0.clone()).wasm_result()
   }
 
   /// Remoce a {@link CoreService} identified by the given {@link CoreDIDUrl} from the document.
   ///
   /// Returns `true` if the service was removed.
   #[wasm_bindgen(js_name = removeService)]
   #[allow(non_snake_case)]
-  pub fn remove_service(&mut self, didUrl: &WasmCoreDIDUrl) -> bool {
-    self.0.service_mut().remove(&didUrl.0.clone())
+  pub fn remove_service(&mut self, didUrl: &WasmCoreDIDUrl) -> Option<WasmCoreService> {
+    self.0.remove_service(&didUrl.0.clone()).map(Into::into)
   }
 
   /// Returns the first {@link CoreService} with an `id` property matching the provided `query`,
@@ -330,14 +336,13 @@ impl WasmCoreDocument {
   /// Adds a new `method` to the document in the given `scope`.
   #[wasm_bindgen(js_name = insertMethod)]
   pub fn insert_method(&mut self, method: &WasmCoreVerificationMethod, scope: &WasmMethodScope) -> Result<()> {
-    self.0.insert_method(method.0.clone(), scope.0).wasm_result()?;
-    Ok(())
+    self.0.insert_method(method.0.clone(), scope.0).wasm_result()
   }
 
   /// Removes all references to the specified Verification Method.
   #[wasm_bindgen(js_name = removeMethod)]
-  pub fn remove_method(&mut self, did: &WasmCoreDIDUrl) -> Result<()> {
-    self.0.remove_method(&did.0).wasm_result()
+  pub fn remove_method(&mut self, did: &WasmCoreDIDUrl) -> Option<WasmCoreVerificationMethod> {
+    self.0.remove_method(&did.0).map(Into::into)
   }
 
   /// Returns a copy of the first verification method with an `id` property

bindings/wasm/src/iota/iota_document.rs

@@ -126,7 +126,7 @@ impl WasmIotaDocument {
         urls_set.append(Url::parse(url).wasm_result()?);
       }
     }
-    *self.0.also_known_as_mut() = urls_set;
+    *self.0.also_known_as_mut_unchecked() = urls_set;
     Ok(())
   }
 
@@ -146,10 +146,10 @@ impl WasmIotaDocument {
     let value: Option<serde_json::Value> = value.into_serde().wasm_result()?;
     match value {
       Some(value) => {
-        self.0.properties_mut().insert(key, value);
+        self.0.properties_mut_unchecked().insert(key, value);
       }
       None => {
-        self.0.properties_mut().remove(&key);
+        self.0.properties_mut_unchecked().remove(&key);
       }
     }
     Ok(())
@@ -177,16 +177,16 @@ impl WasmIotaDocument {
   ///
   /// Returns `true` if the service was added.
   #[wasm_bindgen(js_name = insertService)]
-  pub fn insert_service(&mut self, service: &WasmIotaService) -> bool {
-    self.0.insert_service(service.0.clone())
+  pub fn insert_service(&mut self, service: &WasmIotaService) -> Result<()> {
+    self.0.insert_service(service.0.clone()).wasm_result()
   }
 
   /// Remove a {@link IotaService} identified by the given {@link IotaDIDUrl} from the document.
   ///
   /// Returns `true` if a service was removed.
   #[wasm_bindgen(js_name = removeService)]
-  pub fn remove_service(&mut self, did: &WasmIotaDIDUrl) -> bool {
-    self.0.remove_service(&did.0)
+  pub fn remove_service(&mut self, did: &WasmIotaDIDUrl) -> Option<WasmIotaService> {
+    self.0.remove_service(&did.0).map(Into::into)
   }
 
   /// Returns the first {@link IotaService} with an `id` property matching the provided `query`,
@@ -233,8 +233,8 @@ impl WasmIotaDocument {
 
   /// Removes all references to the specified Verification Method.
   #[wasm_bindgen(js_name = removeMethod)]
-  pub fn remove_method(&mut self, did: &WasmIotaDIDUrl) -> Result<()> {
-    self.0.remove_method(&did.0).wasm_result()
+  pub fn remove_method(&mut self, did: &WasmIotaDIDUrl) -> Option<WasmIotaVerificationMethod> {
+    self.0.remove_method(&did.0).map(Into::into)
   }
 
   /// Returns a copy of the first verification method with an `id` property

bindings/wasm/tests/core.ts

@@ -273,8 +273,8 @@ describe("CoreDocument", function() {
             assert.deepStrictEqual(list.length, 1);
             assert.deepStrictEqual(list[0].toJSON(), resolved.toJSON());
             // Remove
-            const remove = doc.removeService(resolved.id());
-            assert.deepStrictEqual(remove, true);
+            const removed = doc.removeService(resolved.id());
+            assert.deepStrictEqual(removed.toJSON(), resolved.toJSON());
             assert.deepStrictEqual(doc.resolveService(fragment1), undefined);
             assert.deepStrictEqual(doc.service().length, 0);
         });

bindings/wasm/tests/iota.ts

@@ -159,8 +159,8 @@ describe("IotaDocument", function() {
             assert.deepStrictEqual(list.length, 1);
             assert.deepStrictEqual(list[0].toJSON(), resolved.toJSON());
             // Remove
-            const remove = doc.removeService(resolved.id());
-            assert.deepStrictEqual(remove, true);
+            const removed = doc.removeService(resolved.id());
+            assert.deepStrictEqual(removed.toJSON(), resolved.toJSON());
             assert.deepStrictEqual(doc.resolveService(fragment1), undefined);
             assert.deepStrictEqual(doc.service().length, 0);
         });

examples/0_basic/1_update_did.rs

@@ -67,7 +67,7 @@ async fn main() -> anyhow::Result<()> {
     "type": "LinkedDomains",
     "serviceEndpoint": "https://iota.org/"
   }))?;
-  assert!(document.insert_service(service));
+  assert!(document.insert_service(service).is_ok());
   document.metadata.updated = Some(Timestamp::now_utc());
 
   // Remove a verification method.

examples/1_advanced/5_alias_output_history.rs

@@ -70,7 +70,7 @@ async fn main() -> anyhow::Result<()> {
   ];
   for service in services {
     let service: IotaService = Service::from_json_value(service)?;
-    assert!(document.insert_service(service));
+    assert!(document.insert_service(service).is_ok());
     document.metadata.updated = Some(Timestamp::now_utc());
 
     // Increase the storage deposit and publish the update.

examples_legacy/account/unchecked.rs

@@ -41,7 +41,7 @@ async fn main() -> Result<()> {
 
   // Add a custom property to the document.
   document
-    .properties_mut()
+    .properties_mut_unchecked()
     .insert("myCustomPropertyKey".into(), "value".into());
 
   // Override the updated field timestamp to 24 hours (= 86400 seconds) in the future,

identity_account/src/tests/account.rs

@@ -478,8 +478,8 @@ async fn test_account_sync_integration_msg_update() {
 
       let client: Client = Client::builder().network(network).build().await.unwrap();
       let mut new_doc: IotaDocument = account.document().clone();
-      new_doc.properties_mut().insert("foo".into(), 123u32.into());
-      new_doc.properties_mut().insert("bar".into(), 456u32.into());
+      new_doc.properties_mut_unchecked().insert("foo".into(), 123u32.into());
+      new_doc.properties_mut_unchecked().insert("bar".into(), 456u32.into());
       new_doc.metadata.previous_message_id = *account.chain_state().last_integration_message_id();
       new_doc.metadata.updated = Some(Timestamp::now_utc());
       account
@@ -521,8 +521,8 @@ async fn test_account_sync_diff_msg_update() {
 
       let client: Client = Client::builder().network(network).build().await.unwrap();
       let mut new_doc: IotaDocument = account.document().clone();
-      new_doc.properties_mut().insert("foo".into(), 123u32.into());
-      new_doc.properties_mut().insert("bar".into(), 456u32.into());
+      new_doc.properties_mut_unchecked().insert("foo".into(), 123u32.into());
+      new_doc.properties_mut_unchecked().insert("bar".into(), 456u32.into());
       new_doc.metadata.updated = Some(Timestamp::now_utc());
       let mut diff_msg: DiffMessage = DiffMessage::new(
         account.document(),

identity_account/src/updates/update.rs

@@ -118,7 +118,7 @@ impl Update {
         // Check method identifier is not duplicated.
         let method_url: IotaDIDUrl = did.to_url().join(fragment.identifier())?;
         if document.resolve_method(method_url, None).is_some() {
-          return Err(crate::Error::DIDError(identity_did::Error::MethodAlreadyExists));
+          return Err(crate::Error::DIDError(identity_did::Error::MethodInsertionError));
         }
 
         // Generate or extract the private key and/or retrieve the public key.
@@ -235,11 +235,11 @@ impl Update {
         document.remove_service(&service_url);
       }
       Self::SetController { controllers } => {
-        *document.controller_mut() = controllers;
+        *document.controller_mut_unchecked() = controllers;
       }
 
       Self::SetAlsoKnownAs { urls } => {
-        *document.also_known_as_mut() = urls;
+        *document.also_known_as_mut_unchecked() = urls;
       }
     }