[pitboss] phase 17: Track L.15 — Gin / Echo / Fiber / Chi adapters + Axum / Actix / Rocket / Warp adapters

2026-06-09 19:45:13 +02:00 · 2026-05-20 12:24:31 -05:00 · 2026-05-20 12:24:31 -05:00 · 2b96c6005b
commit 2b96c6005b
parent 5393fe22f2
33 changed files with 3247 additions and 27 deletions
--- a/src/dynamic/framework/adapters/rust_rocket.rs
+++ b/src/dynamic/framework/adapters/rust_rocket.rs
@ -0,0 +1,125 @@
+//! Rocket [`super::super::FrameworkAdapter`] (Phase 17 — Track L.15).
+//!
+//! Recognises rocket's `#[get("/path")]` / `#[post("/path")]`
+//! attribute macros plus the `routes![handler]` macro:
+//!
+//! ```rust
+//! #[get("/users/<id>")]
+//! fn show(id: String) -> String { id }
+//!
+//! #[launch]
+//! fn rocket() -> _ { rocket::build().mount("/", routes![show]) }
+//! ```
+//!
+//! Rocket's placeholder syntax `<id>` plus brace syntax `<id..>`
+//! resolve via [`super::rust_routes::extract_rust_path_placeholders`].
+//! The adapter shares the attribute-walk path with actix; the only
+//! difference is the source-import discriminator.
+
+use crate::dynamic::framework::{FrameworkAdapter, FrameworkBinding, RouteShape};
+use crate::evidence::EntryKind;
+use crate::summary::FuncSummary;
+use crate::symbol::Lang;
+use tree_sitter::Node;
+
+use super::rust_routes::{
+    bind_rust_path_params, find_method_attribute, find_rust_function, rust_formal_names,
+    source_imports_rocket,
+};
+
+pub struct RustRocketAdapter;
+
+const ADAPTER_NAME: &str = "rust-rocket";
+
+impl FrameworkAdapter for RustRocketAdapter {
+    fn name(&self) -> &'static str {
+        ADAPTER_NAME
+    }
+
+    fn lang(&self) -> Lang {
+        Lang::Rust
+    }
+
+    fn detect(
+        &self,
+        summary: &FuncSummary,
+        ast: Node<'_>,
+        file_bytes: &[u8],
+    ) -> Option<FrameworkBinding> {
+        if !source_imports_rocket(file_bytes) {
+            return None;
+        }
+        let func = find_rust_function(ast, file_bytes, &summary.name)?;
+        let (method, path) = find_method_attribute(func, file_bytes)?;
+        let formals = rust_formal_names(func, file_bytes);
+        let request_params = bind_rust_path_params(&formals, &path);
+        Some(FrameworkBinding {
+            adapter: ADAPTER_NAME.to_owned(),
+            kind: EntryKind::HttpRoute,
+            route: Some(RouteShape { method, path }),
+            request_params,
+            response_writer: None,
+            middleware: Vec::new(),
+        })
+    }
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+    use crate::dynamic::framework::{HttpMethod, ParamSource};
+
+    fn parse(src: &[u8]) -> tree_sitter::Tree {
+        let mut parser = tree_sitter::Parser::new();
+        let lang = tree_sitter::Language::from(tree_sitter_rust::LANGUAGE);
+        parser.set_language(&lang).unwrap();
+        parser.parse(src, None).unwrap()
+    }
+
+    fn summary(name: &str) -> FuncSummary {
+        FuncSummary {
+            name: name.into(),
+            lang: "rust".into(),
+            ..Default::default()
+        }
+    }
+
+    #[test]
+    fn fires_on_get_with_angle_placeholder() {
+        let src: &[u8] = b"use rocket::get;\n#[get(\"/u/<id>\")]\nfn show(id: String) -> String { id }\n";
+        let tree = parse(src);
+        let binding = RustRocketAdapter
+            .detect(&summary("show"), tree.root_node(), src)
+            .expect("binding");
+        assert_eq!(binding.adapter, "rust-rocket");
+        let route = binding.route.expect("route");
+        assert_eq!(route.method, HttpMethod::GET);
+        assert_eq!(route.path, "/u/<id>");
+        let id = binding
+            .request_params
+            .iter()
+            .find(|p| p.name == "id")
+            .unwrap();
+        assert!(matches!(id.source, ParamSource::PathSegment(_)));
+    }
+
+    #[test]
+    fn fires_on_post_with_data_param() {
+        let src: &[u8] =
+            b"use rocket::post;\n#[post(\"/save\", data = \"<body>\")]\nfn save(body: String) {}\n";
+        let tree = parse(src);
+        let binding = RustRocketAdapter
+            .detect(&summary("save"), tree.root_node(), src)
+            .expect("binding");
+        assert_eq!(binding.route.unwrap().method, HttpMethod::POST);
+    }
+
+    #[test]
+    fn skips_when_rocket_not_imported() {
+        let src: &[u8] = b"#[get(\"/u\")]\nfn show() {}\n";
+        let tree = parse(src);
+        assert!(RustRocketAdapter
+            .detect(&summary("show"), tree.root_node(), src)
+            .is_none());
+    }
+}