Mirror/brk
1
0
Fork 0
mirror of https://github.com/bitcoinresearchkit/brk.git synced 2026-07-11 02:58:14 -07:00
Code Issues Packages Projects Releases Wiki Activity

global: snapshot

Browse Source
This commit is contained in:
nym21
2025-12-25 22:21:12 +01:00
parent eadf93b804
commit bbb74b76c8
68 changed files with 6497 additions and 2659 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/brk_server/src/lib.rs
+24 -15
View File
@@ -14,7 +14,7 @@ use axum::{
};
use brk_error::Result;
use brk_logger::OwoColorize;
use brk_mcp::route::MCPRoutes;
use brk_mcp::route::mcp_router;
use brk_query::AsyncQuery;
use log::{error, info};
use quick_cache::sync::Cache;
@@ -92,7 +92,6 @@ impl Server {
let vecs = state.query.inner().vecs();
let router = ApiRouter::new()
.add_api_routes()
.add_mcp_routes(&state.query, mcp)
.add_files_routes(state.path.as_ref())
.route(
"/discord",
@@ -136,24 +135,34 @@ impl Server {
let mut openapi = create_openapi();
let router = router.finish_api(&mut openapi);
let clients_path = PathBuf::from(env!("CARGO_MANIFEST_DIR"))
let workspace_root: PathBuf = PathBuf::from(env!("CARGO_MANIFEST_DIR"))
.parent()
.and_then(|p| p.parent())
.unwrap()
.join("brk_binder")
.join("clients");
if clients_path.exists() {
let openapi_json = serde_json::to_string(&openapi).unwrap();
let result = panic::catch_unwind(panic::AssertUnwindSafe(|| {
brk_binder::generate_clients(vecs, &openapi_json, &clients_path)
}));
.into();
let output_paths = brk_binder::ClientOutputPaths::new()
.rust(workspace_root.join("crates/brk_client/src/lib.rs"))
.javascript(workspace_root.join("modules/brk-client/index.js"))
.python(workspace_root.join("packages/brk_client/__init__.py"));
match result {
Ok(Ok(())) => info!("Generated clients at {}", clients_path.display()),
Ok(Err(e)) => error!("Failed to generate clients: {e}"),
Err(_) => error!("Client generation panicked"),
}
let openapi_json = Arc::new(serde_json::to_string(&openapi).unwrap());
let result = panic::catch_unwind(panic::AssertUnwindSafe(|| {
brk_binder::generate_clients(vecs, &openapi_json, &output_paths)
}));
match result {
Ok(Ok(())) => info!("Generated clients"),
Ok(Err(e)) => error!("Failed to generate clients: {e}"),
Err(_) => error!("Client generation panicked"),
}
let router = if mcp {
let base_url = format!("http://127.0.0.1:{port}");
router.merge(mcp_router(base_url, openapi_json))
} else {
router
};
serve(
listener,
router