chore: add alias when creating peer node connection

stakwork · Nov 28, 2024 · a0d5dd2 · a0d5dd2
1 parent ae94328
commit a0d5dd2
Show file tree

Hide file tree

Showing 5 changed files with 61 additions and 11 deletions.
diff --git a/app/src/api/cln.ts b/app/src/api/cln.ts
@@ -86,6 +86,11 @@ export async function create_channel(
   return await clnCmd("AddChannel", tag, { pubkey, amount, satsperbyte });
 }
 
-export async function add_peer(tag: string, pubkey: string, host: string) {
-  return await clnCmd("AddPeer", tag, { pubkey, host });
+export async function add_peer(
+  tag: string,
+  pubkey: string,
+  host: string,
+  alias?: string
+) {
+  return await clnCmd("AddPeer", tag, { pubkey, host, alias });
 }
diff --git a/app/src/api/lnd.ts b/app/src/api/lnd.ts
@@ -59,8 +59,13 @@ export async function list_channels(tag: string) {
   return await lndCmd("ListChannels", tag);
 }
 
-export async function add_peer(tag: string, pubkey: string, host: string) {
-  return await lndCmd("AddPeer", tag, { pubkey, host });
+export async function add_peer(
+  tag: string,
+  pubkey: string,
+  host: string,
+  alias?: string
+) {
+  return await lndCmd("AddPeer", tag, { pubkey, host, alias });
 }
 
 export async function list_peers(tag: string) {

diff --git a/app/src/lnd/Peers.svelte b/app/src/lnd/Peers.svelte
@@ -47,11 +47,12 @@
   let error_notification = false;
   let error_message = false;
   let isUpdate = false;
+  let peerAlias = "";
 
   async function addPeer() {
     message = "";
     if (type === "Cln") {
-      const peer = await CLN.add_peer(tag, pubkey, host);
+      const peer = await CLN.add_peer(tag, pubkey, host, peerAlias);
       show_notification = true;
 
       if (typeof peer === "string") {
@@ -69,6 +70,8 @@
       if (peer) {
         pubkey = "";
         host = "";
+        peerAlias = "";
+        await handleGetLightningPeers();
         const peersData = await CLN.list_peers(tag);
         const thepeers = await parseClnListPeerRes(peersData);
         peersStore.update((peer) => {
@@ -77,11 +80,12 @@
         createdPeerForOnboarding.update(() => true);
       }
     } else {
-      if (await add_peer(tag, pubkey, host)) {
+      if (await add_peer(tag, pubkey, host, peerAlias)) {
         show_notification = true;
         pubkey = "";
         host = "";
-
+        peerAlias = "";
+        await handleGetLightningPeers();
         setTimeout(async () => {
           const peersData = await list_peers(tag);
           peersStore.update((ps) => {
@@ -217,6 +221,12 @@
       bind:value={host}
     />
     <div class="spacer" />
+    <TextInput
+      labelText={"Alias"}
+      placeholder={"Peer Alias"}
+      bind:value={peerAlias}
+    />
+    <div class="spacer" />
     <center
       ><Button
         disabled={addDisabled}

diff --git a/src/cmd.rs b/src/cmd.rs
@@ -197,6 +197,7 @@ pub struct TestMine {
 pub struct AddPeer {
     pub pubkey: String,
     pub host: String,
+    pub alias: Option<String>,
 }
 
 #[derive(Serialize, Deserialize, Debug, Clone)]

diff --git a/src/handler.rs b/src/handler.rs
@@ -5,6 +5,7 @@ use crate::auth;
 use crate::builder;
 use crate::cmd::*;
 use crate::config;
+use crate::config::LightningPeer;
 use crate::config::Role;
 use crate::config::User;
 use crate::config::{Clients, Node, Stack, State, STATE};
@@ -523,8 +524,22 @@ pub async fn handle(
                     Some(serde_json::to_string(&channel_list.channels)?)
                 }
                 LndCmd::AddPeer(peer) => {
-                    let result = client.add_peer(peer).await?;
-                    Some(serde_json::to_string(&result)?)
+                    if let Some(alias) = peer.alias.clone() {
+                        add_new_lightning_peer(
+                            &mut state,
+                            LightningPeer {
+                                pubkey: peer.pubkey.clone(),
+                                alias,
+                            },
+                            &mut must_save_stack,
+                        );
+                        let client = state.clients.lnd.get_mut(tag).context("no lnd client")?;
+                        let result = client.add_peer(peer).await?;
+                        Some(serde_json::to_string(&result)?)
+                    } else {
+                        let result = client.add_peer(peer).await?;
+                        Some(serde_json::to_string(&result)?)
+                    }
                 }
                 LndCmd::ListPeers => {
                     let result = client.list_peers().await?;
@@ -604,8 +619,22 @@ pub async fn handle(
                     } else {
                         peer.host
                     };
-                    let result = client.connect_peer(&peer.pubkey, &host, port).await?;
-                    Some(serde_json::to_string(&result)?)
+                    if let Some(alias) = peer.alias.clone() {
+                        add_new_lightning_peer(
+                            &mut state,
+                            LightningPeer {
+                                alias,
+                                pubkey: peer.pubkey.clone(),
+                            },
+                            &mut must_save_stack,
+                        );
+                        let client = state.clients.cln.get_mut(tag).context("no cln client")?;
+                        let result = client.connect_peer(&peer.pubkey, &host, port).await?;
+                        Some(serde_json::to_string(&result)?)
+                    } else {
+                        let result = client.connect_peer(&peer.pubkey, &host, port).await?;
+                        Some(serde_json::to_string(&result)?)
+                    }
                 }
                 ClnCmd::AddChannel(channel) => {
                     let channel = client