From f8060715ce091facadc0a5d39024faf3d54412f7 Mon Sep 17 00:00:00 2001 From: John Smith Date: Sat, 11 Mar 2023 13:56:40 -0500 Subject: [PATCH] fix debug --- veilid-core/src/veilid_api/debug.rs | 41 +++++++++++++++++++++++------ 1 file changed, 33 insertions(+), 8 deletions(-) diff --git a/veilid-core/src/veilid_api/debug.rs b/veilid-core/src/veilid_api/debug.rs index 5997eec2..494ac1b8 100644 --- a/veilid-core/src/veilid_api/debug.rs +++ b/veilid-core/src/veilid_api/debug.rs @@ -30,7 +30,7 @@ fn get_string(text: &str) -> Option { Some(text.to_owned()) } -fn get_route_id(rss: RouteSpecStore) -> impl Fn(&str) -> Option { +fn get_route_id(rss: RouteSpecStore, allow_remote: bool) -> impl Fn(&str) -> Option { return move |text: &str| { if text.is_empty() { return None; @@ -41,6 +41,12 @@ fn get_route_id(rss: RouteSpecStore) -> impl Fn(&str) -> Option { if routes.contains(&key) { return Some(key); } + if allow_remote { + let rroutes = rss.list_remote_routes(|k, _| Some(*k)); + if rroutes.contains(&key) { + return Some(key); + } + } } None => { let routes = rss.list_allocated_routes(|k, _| Some(*k)); @@ -50,6 +56,15 @@ fn get_route_id(rss: RouteSpecStore) -> impl Fn(&str) -> Option { return Some(r); } } + if allow_remote { + let routes = rss.list_remote_routes(|k, _| Some(*k)); + for r in routes { + let rkey = r.encode(); + if rkey.starts_with(text) { + return Some(r); + } + } + } } } None @@ -74,7 +89,7 @@ fn get_safety_selection(text: &str, routing_table: RoutingTable) -> Option "Released".to_owned(), + true => { + // release imported + let mut dc = DEBUG_CACHE.lock(); + for (n, ir) in dc.imported_routes.iter().enumerate() { + if *ir == route_id { + dc.imported_routes.remove(n); + break; + } + } + "Released".to_owned() + } false => "Route does not exist".to_owned(), }; @@ -671,7 +696,7 @@ impl VeilidAPI { 1, "debug_route", "route_id", - get_route_id(rss.clone()), + get_route_id(rss.clone(), false), )?; let full = { if args.len() > 2 { @@ -723,7 +748,7 @@ impl VeilidAPI { 1, "debug_route", "route_id", - get_route_id(rss.clone()), + get_route_id(rss.clone(), false), )?; // Unpublish route @@ -745,7 +770,7 @@ impl VeilidAPI { 1, "debug_route", "route_id", - get_route_id(rss.clone()), + get_route_id(rss.clone(), true), )?; match rss.debug_route(&route_id) { @@ -807,7 +832,7 @@ impl VeilidAPI { 1, "debug_route", "route_id", - get_route_id(rss.clone()), + get_route_id(rss.clone(), true), )?; let success = rss