local registry

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::{InstantServer, LocalRegistry},
+    CoLink, Participant, ProtocolEntry,
+};
 
 struct Initiator;
 #[colink::async_trait]
@@ -33,6 +36,7 @@ impl ProtocolEntry for Receiver {
 
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn std::error::Error + Send + Sync + 'static>> {
+    let lr = LocalRegistry::new().await;
     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,8 @@
 use crate::{utils::get_colink_home, CoLink};
 use rand::Rng;
 use std::{
+    fs::File,
+    io::Write,
     path::Path,
     process::{Child, Command, Stdio},
 };
@@ -103,3 +105,29 @@ impl InstantServer {
         CoLink::new(&format!("http://127.0.0.1:{}", self.port), &self.host_token)
     }
 }
+
+pub struct LocalRegistry {
+    _instant_server: InstantServer,
+}
+
+impl Drop for LocalRegistry {
+    fn drop(&mut self) {
+        let colink_home = get_colink_home().unwrap();
+        let registry_file = Path::new(&colink_home).join("registry.conf");
+        std::fs::remove_file(&registry_file).unwrap();
+    }
+}
+
+impl LocalRegistry {
+    pub async fn new() -> Self {
+        let is = InstantServer::new();
+        let colink_home = get_colink_home().unwrap();
+        let registry_file = Path::new(&colink_home).join("registry.conf");
+        let mut file = File::create(&registry_file).unwrap();
+        file.write_all(b"new_registry").unwrap();
+        is.get_colink().switch_to_generated_user().await.unwrap();
+        Self {
+            _instant_server: is,
+        }
+    }
+}