chore: CRP-2839 add key name initialization to notes dapp (#204)

Author: altkdf

examples/encrypted_notes_dapp_vetkd/motoko/backend/main.mo

@@ -16,7 +16,7 @@ import Hex "./utils/Hex";
 
 // Declare a shared actor class
 // Bind the caller and the initializer
-shared ({ caller = initializer }) actor class () {
+shared ({ caller = initializer }) actor class (keyName: Text) {
 
     // Currently, a single canister smart contract is limited to 4 GB of heap size.
     // For the current limits see https://internetcomputer.org/docs/current/developer-docs/production/resource-limits.
@@ -328,7 +328,7 @@ shared ({ caller = initializer }) actor class () {
         let { public_key } = await management_canister.vetkd_public_key({
             canister_id = null;
             context = Text.encodeUtf8("note_symmetric_key");
-            key_id = { curve = #bls12_381_g2; name = "test_key_1" };
+            key_id = { curve = #bls12_381_g2; name = keyName };
         });
         Hex.encode(Blob.toArray(public_key));
     };
@@ -348,7 +348,7 @@ shared ({ caller = initializer }) actor class () {
         let { encrypted_key } = await (with cycles = 26_153_846_153) management_canister.vetkd_derive_key({
             input;
             context = Text.encodeUtf8("note_symmetric_key");
-            key_id = { curve = #bls12_381_g2; name = "test_key_1" };
+            key_id = { curve = #bls12_381_g2; name = keyName };
             transport_public_key;
         });
         Hex.encode(Blob.toArray(encrypted_key));

examples/encrypted_notes_dapp_vetkd/motoko/dfx.json

@@ -2,7 +2,9 @@
     "canisters": {
       "encrypted_notes": {
         "main": "backend/main.mo",
-        "type": "motoko"
+        "type": "motoko",
+        "args": "--enhanced-orthogonal-persistence",
+        "init_arg": "(\"test_key_1\")"
       },
       "internet-identity": {
         "candid": "https://github.com/dfinity/internet-identity/releases/latest/download/internet_identity.did",

examples/encrypted_notes_dapp_vetkd/rust/backend/src/encrypted_notes_rust.did

@@ -1,20 +1,17 @@
-type anon_class_15_1 = 
- service {
-   create_note: () -> (nat);
-   delete_note: (nat) -> ();
-   get_notes: () -> (vec EncryptedNote);
-   update_note: (nat, text) -> ();
-   add_user: (nat, text) -> ();
-   remove_user: (nat, text) -> ();
-   whoami: () -> (text);
-   symmetric_key_verification_key_for_note: () -> (text);
-   encrypted_symmetric_key_for_note: (nat, blob) -> (text);
- };
-type EncryptedNote = 
- record {
-   encrypted_text: text;
-   id: nat;
-   owner: text;
-   users: vec text;
- };
-service : () -> anon_class_15_1
+type EncryptedNote = record {
+  id : nat;
+  encrypted_text : text;
+  owner : text;
+  users : vec text;
+};
+service : (text) -> {
+  add_user : (nat, text) -> ();
+  create_note : () -> (nat);
+  delete_note : (nat) -> ();
+  encrypted_symmetric_key_for_note : (nat, blob) -> (text);
+  get_notes : () -> (vec EncryptedNote);
+  remove_user : (nat, text) -> ();
+  symmetric_key_verification_key_for_note : () -> (text);
+  update_note : (nat, text) -> ();
+  whoami : () -> (text);
+}

examples/encrypted_notes_dapp_vetkd/rust/backend/src/lib.rs

@@ -1,5 +1,6 @@
 use candid::{CandidType, Decode, Deserialize, Encode, Principal};
 use ic_cdk::api::msg_caller;
+use ic_cdk::init;
 use ic_cdk::update;
 use ic_stable_structures::memory_manager::{MemoryId, MemoryManager, VirtualMemory};
 use ic_stable_structures::{
@@ -98,6 +99,21 @@ thread_local! {
             MEMORY_MANAGER.with_borrow(|m| m.get(MemoryId::new(3))),
         )
     );
+    static KEY_NAME: RefCell<StableCell<String, Memory>> =
+        RefCell::new(StableCell::init(
+            MEMORY_MANAGER.with(|m| m.borrow().get(MemoryId::new(1))),
+            String::new(),
+        )
+        .expect("failed to initialize key name"));
+}
+
+#[init]
+fn init(key_name_string: String) {
+    KEY_NAME.with_borrow_mut(|key_name| {
+        key_name
+            .set(key_name_string)
+            .expect("failed to set key name");
+    });
 }
 
 /// Unlike Motoko, the caller identity is not built into Rust.
@@ -365,7 +381,7 @@ async fn symmetric_key_verification_key_for_note() -> String {
     let request = VetKDPublicKeyArgs {
         canister_id: None,
         context: b"note_symmetric_key".to_vec(),
-        key_id: bls12_381_g2_test_key_1(),
+        key_id: key_id(),
     };
 
     let response: VetKDPublicKeyResult = ic_cdk::management_canister::vetkd_public_key(&request)
@@ -394,7 +410,7 @@ async fn encrypted_symmetric_key_for_note(
                     buf // prefix-free
                 },
                 context: b"note_symmetric_key".to_vec(),
-                key_id: bls12_381_g2_test_key_1(),
+                key_id: key_id(),
                 transport_public_key,
             }
         } else {
@@ -409,9 +425,9 @@ async fn encrypted_symmetric_key_for_note(
     hex::encode(response.encrypted_key)
 }
 
-fn bls12_381_g2_test_key_1() -> VetKDKeyId {
+fn key_id() -> VetKDKeyId {
     VetKDKeyId {
         curve: VetKDCurve::Bls12_381_G2,
-        name: "test_key_1".to_string(),
+        name: KEY_NAME.with_borrow(|key_name| key_name.get().clone()),
     }
 }

examples/encrypted_notes_dapp_vetkd/rust/dfx.json

@@ -3,7 +3,8 @@
     "encrypted_notes": {
       "type": "rust",
       "candid": "backend/src/encrypted_notes_rust.did",
-      "package": "encrypted_notes_backend"
+      "package": "encrypted_notes_backend",
+      "init_arg": "(\"test_key_1\")"
     },
     "internet-identity": {
       "candid": "https://github.com/dfinity/internet-identity/releases/latest/download/internet_identity.did",