Merge pull request #30 from CoLearn-Dev/local_registry

- local registry
- fix update_registries

Author: stneng

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "colink"
-version = "0.2.2"
+version = "0.2.3"
 edition = "2021"
 description = "CoLink Rust SDK"
 license = "MIT"

README.md

@@ -9,7 +9,7 @@ CoLink SDK helps both application adnd protocol developers access the functional
 Add this to your Cargo.toml:
 ```toml
 [dependencies]
-colink = "0.2.1"
+colink = "0.2.3"
 ```
 
 ## Getting Started

examples/protocol_greetings_with_instant_server.rs

@@ -1,5 +1,8 @@
 #![allow(unused_variables)]
-use colink::{extensions::instant_server::InstantServer, CoLink, Participant, ProtocolEntry};
+use colink::{
+    extensions::instant_server::{InstantRegistry, InstantServer},
+    CoLink, Participant, ProtocolEntry,
+};
 
 struct Initiator;
 #[colink::async_trait]
@@ -33,6 +36,8 @@ impl ProtocolEntry for Receiver {
 
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync + 'static>> {
+    let ir = InstantRegistry::new().await;
+    // The instant server will automatically use the instant registry when there is one.
     let is0 = InstantServer::new();
     let is1 = InstantServer::new();
     let cl0 = is0.get_colink().switch_to_generated_user().await?;

src/extensions/instant_server.rs

@@ -1,6 +1,7 @@
 use crate::{utils::get_colink_home, CoLink};
 use rand::Rng;
 use std::{
+    fs::File,
     path::Path,
     process::{Child, Command, Stdio},
 };
@@ -103,3 +104,32 @@ impl InstantServer {
         CoLink::new(&format!("http://127.0.0.1:{}", self.port), &self.host_token)
     }
 }
+
+pub struct InstantRegistry {
+    _instant_server: InstantServer,
+}
+
+impl Drop for InstantRegistry {
+    fn drop(&mut self) {
+        let colink_home = get_colink_home().unwrap();
+        let registry_file = Path::new(&colink_home).join("reg_config");
+        std::fs::remove_file(&registry_file).unwrap();
+    }
+}
+
+impl InstantRegistry {
+    pub async fn new() -> Self {
+        let is = InstantServer::new();
+        let colink_home = get_colink_home().unwrap();
+        let registry_file = Path::new(&colink_home).join("reg_config");
+        let _file = File::options()
+            .write(true)
+            .create_new(true)
+            .open(&registry_file)
+            .unwrap();
+        is.get_colink().switch_to_generated_user().await.unwrap();
+        Self {
+            _instant_server: is,
+        }
+    }
+}

src/extensions/registry.rs

@@ -15,8 +15,10 @@ impl crate::application::CoLink {
         }];
         let mut payload = vec![];
         registries.encode(&mut payload).unwrap();
-        self.run_task("registry", &payload, &participants, false)
+        let task_id = self
+            .run_task("registry", &payload, &participants, false)
             .await?;
+        self.wait_task(&task_id).await?;
         Ok(())
     }
 }