Skip to main content

mockforge_tui/api/
client.rs

1//! HTTP client for the MockForge admin API.
2
3use anyhow::{Context, Result};
4use reqwest::Client;
5use serde::de::DeserializeOwned;
6
7use super::models::*;
8
9/// HTTP client wrapping `reqwest` with base URL and optional auth.
10#[derive(Clone)]
11pub struct MockForgeClient {
12    client: Client,
13    base_url: String,
14    token: Option<String>,
15}
16
17impl MockForgeClient {
18    /// Create a new client pointing at the given admin server.
19    pub fn new(base_url: String, token: Option<String>) -> Result<Self> {
20        let client = Client::builder()
21            .timeout(std::time::Duration::from_secs(10))
22            .build()
23            .context("failed to create HTTP client")?;
24
25        let base_url = base_url.trim_end_matches('/').to_string();
26
27        Ok(Self {
28            client,
29            base_url,
30            token,
31        })
32    }
33
34    /// Base URL for SSE stream connections.
35    pub fn base_url(&self) -> &str {
36        &self.base_url
37    }
38
39    /// Build a GET request with auth header if configured.
40    fn get(&self, path: &str) -> reqwest::RequestBuilder {
41        let url = format!("{}{path}", self.base_url);
42        let mut req = self.client.get(&url);
43        if let Some(ref token) = self.token {
44            req = req.bearer_auth(token);
45        }
46        req
47    }
48
49    /// Build a POST request with auth header and JSON body.
50    fn post<T: serde::Serialize>(&self, path: &str, body: &T) -> reqwest::RequestBuilder {
51        let url = format!("{}{path}", self.base_url);
52        let mut req = self.client.post(&url).json(body);
53        if let Some(ref token) = self.token {
54            req = req.bearer_auth(token);
55        }
56        req
57    }
58
59    /// Send a GET, check for JSON content type, and unwrap the `ApiResponse<T>` envelope.
60    async fn get_api<T: DeserializeOwned>(&self, path: &str) -> Result<T> {
61        let resp = self.get(path).send().await.with_context(|| format!("GET {path}"))?;
62
63        let status = resp.status();
64        if !status.is_success() {
65            anyhow::bail!("HTTP {status} from {path}");
66        }
67
68        // Guard against HTML responses from the SPA fallback (endpoint doesn't exist).
69        let ct = resp
70            .headers()
71            .get(reqwest::header::CONTENT_TYPE)
72            .and_then(|v| v.to_str().ok())
73            .unwrap_or("");
74        if ct.contains("text/html") {
75            anyhow::bail!("endpoint {path} not available (got HTML)");
76        }
77
78        let body = resp.text().await.with_context(|| format!("read body from {path}"))?;
79
80        let envelope: ApiResponse<T> = serde_json::from_str(&body)
81            .with_context(|| format!("deserialise response from {path}"))?;
82
83        if envelope.success {
84            envelope.data.context("API returned success but no data")
85        } else {
86            anyhow::bail!("API error: {}", envelope.error.unwrap_or_else(|| "unknown".into()))
87        }
88    }
89
90    /// Send a GET, check for JSON content type, and return raw JSON.
91    async fn get_raw<T: DeserializeOwned>(&self, path: &str) -> Result<T> {
92        let resp = self.get(path).send().await.with_context(|| format!("GET {path}"))?;
93
94        let status = resp.status();
95        if !status.is_success() {
96            anyhow::bail!("HTTP {status} from {path}");
97        }
98
99        let ct = resp
100            .headers()
101            .get(reqwest::header::CONTENT_TYPE)
102            .and_then(|v| v.to_str().ok())
103            .unwrap_or("");
104        if ct.contains("text/html") {
105            anyhow::bail!("endpoint {path} not available (got HTML)");
106        }
107
108        let body = resp.text().await.with_context(|| format!("read body from {path}"))?;
109
110        serde_json::from_str(&body).with_context(|| format!("deserialise response from {path}"))
111    }
112
113    /// POST helper that expects an `ApiResponse<String>` result.
114    async fn post_api(&self, path: &str, body: &serde_json::Value) -> Result<String> {
115        let resp = self.post(path, body).send().await.with_context(|| format!("POST {path}"))?;
116
117        let status = resp.status();
118        if !status.is_success() {
119            anyhow::bail!("HTTP {status} from {path}");
120        }
121
122        let ct = resp
123            .headers()
124            .get(reqwest::header::CONTENT_TYPE)
125            .and_then(|v| v.to_str().ok())
126            .unwrap_or("");
127        if ct.contains("text/html") {
128            anyhow::bail!("endpoint {path} not available");
129        }
130
131        let body_text = resp.text().await.context("read POST response body")?;
132        let envelope: ApiResponse<String> = serde_json::from_str(&body_text)
133            .with_context(|| format!("deserialise response from {path}"))?;
134
135        if envelope.success {
136            Ok(envelope.data.unwrap_or_default())
137        } else {
138            anyhow::bail!("API error: {}", envelope.error.unwrap_or_else(|| "unknown".into()))
139        }
140    }
141
142    // ── Tier 1 endpoints ─────────────────────────────────────────────
143
144    pub async fn get_dashboard(&self) -> Result<DashboardData> {
145        self.get_api("/__mockforge/dashboard").await
146    }
147
148    pub async fn get_routes(&self) -> Result<Vec<RouteInfo>> {
149        // Server may return ApiResponse<Vec<RouteInfo>> or {"routes": [...]}
150        let resp = self
151            .get("/__mockforge/routes")
152            .send()
153            .await
154            .context("GET /__mockforge/routes")?;
155
156        let status = resp.status();
157        if !status.is_success() {
158            anyhow::bail!("HTTP {status} from /__mockforge/routes");
159        }
160
161        let ct = resp
162            .headers()
163            .get(reqwest::header::CONTENT_TYPE)
164            .and_then(|v| v.to_str().ok())
165            .unwrap_or("");
166        if ct.contains("text/html") {
167            anyhow::bail!("endpoint /__mockforge/routes not available");
168        }
169
170        let body = resp.text().await.context("read routes response")?;
171
172        // Try ApiResponse envelope first
173        if let Ok(envelope) = serde_json::from_str::<ApiResponse<Vec<RouteInfo>>>(&body) {
174            if envelope.success {
175                return envelope.data.context("routes: no data");
176            }
177        }
178
179        // Try {"routes": [...]} wrapper
180        if let Ok(wrapper) = serde_json::from_str::<RoutesWrapper>(&body) {
181            return Ok(wrapper.routes);
182        }
183
184        // Try raw array
185        serde_json::from_str::<Vec<RouteInfo>>(&body).context("deserialise routes response")
186    }
187
188    pub async fn get_logs(&self, limit: Option<u32>) -> Result<Vec<RequestLog>> {
189        let path = match limit {
190            Some(n) => format!("/__mockforge/logs?limit={n}"),
191            None => "/__mockforge/logs".into(),
192        };
193        self.get_api(&path).await
194    }
195
196    pub async fn get_metrics(&self) -> Result<MetricsData> {
197        self.get_api("/__mockforge/metrics").await
198    }
199
200    pub async fn get_config(&self) -> Result<ConfigState> {
201        self.get_api("/__mockforge/config").await
202    }
203
204    pub async fn get_health(&self) -> Result<HealthCheck> {
205        self.get_raw("/__mockforge/health").await
206    }
207
208    pub async fn get_server_info(&self) -> Result<ServerInfo> {
209        // Server may return ApiResponse<ServerInfo> or raw ServerInfo
210        let resp = self
211            .get("/__mockforge/server-info")
212            .send()
213            .await
214            .context("GET /__mockforge/server-info")?;
215
216        let status = resp.status();
217        if !status.is_success() {
218            anyhow::bail!("HTTP {status} from /__mockforge/server-info");
219        }
220
221        let ct = resp
222            .headers()
223            .get(reqwest::header::CONTENT_TYPE)
224            .and_then(|v| v.to_str().ok())
225            .unwrap_or("");
226        if ct.contains("text/html") {
227            anyhow::bail!("endpoint /__mockforge/server-info not available");
228        }
229
230        let body = resp.text().await.context("read server-info response")?;
231
232        // Try ApiResponse envelope first
233        if let Ok(envelope) = serde_json::from_str::<ApiResponse<ServerInfo>>(&body) {
234            if envelope.success {
235                return envelope.data.context("server-info: no data");
236            }
237        }
238
239        // Try raw ServerInfo
240        serde_json::from_str::<ServerInfo>(&body).context("deserialise server-info response")
241    }
242
243    pub async fn get_plugins(&self) -> Result<Vec<PluginInfo>> {
244        // Server returns ApiResponse<{"plugins": [...], "total": N}>
245        let resp = self
246            .get("/__mockforge/plugins")
247            .send()
248            .await
249            .context("GET /__mockforge/plugins")?;
250
251        let status = resp.status();
252        if !status.is_success() {
253            anyhow::bail!("HTTP {status} from /__mockforge/plugins");
254        }
255
256        let ct = resp
257            .headers()
258            .get(reqwest::header::CONTENT_TYPE)
259            .and_then(|v| v.to_str().ok())
260            .unwrap_or("");
261        if ct.contains("text/html") {
262            anyhow::bail!("endpoint /__mockforge/plugins not available");
263        }
264
265        let body = resp.text().await.context("read plugins response")?;
266
267        // Try ApiResponse<Vec<PluginInfo>> first
268        if let Ok(envelope) = serde_json::from_str::<ApiResponse<Vec<PluginInfo>>>(&body) {
269            if envelope.success {
270                return envelope.data.context("plugins: no data");
271            }
272        }
273
274        // Try ApiResponse<PluginsWrapper>
275        if let Ok(envelope) = serde_json::from_str::<ApiResponse<PluginsWrapper>>(&body) {
276            if envelope.success {
277                return Ok(envelope.data.map(|w| w.plugins).unwrap_or_default());
278            }
279        }
280
281        Ok(Vec::new())
282    }
283
284    pub async fn get_fixtures(&self) -> Result<Vec<FixtureInfo>> {
285        self.get_api("/__mockforge/fixtures").await
286    }
287
288    pub async fn get_smoke_tests(&self) -> Result<Vec<SmokeTestResult>> {
289        self.get_api("/__mockforge/smoke").await
290    }
291
292    pub async fn run_smoke_tests(&self) -> Result<Vec<SmokeTestResult>> {
293        self.get_api("/__mockforge/smoke/run").await
294    }
295
296    pub async fn get_workspaces(&self) -> Result<Vec<WorkspaceInfo>> {
297        self.get_api("/__mockforge/workspaces").await
298    }
299
300    // ── Tier 2 endpoints ─────────────────────────────────────────────
301
302    pub async fn get_chaos_status(&self) -> Result<serde_json::Value> {
303        self.get_api("/__mockforge/chaos").await
304    }
305
306    /// Snapshot of chaos fault-injection counters (issue-#79 follow-up).
307    pub async fn get_chaos_stats(&self) -> Result<serde_json::Value> {
308        self.get_api("/__mockforge/chaos/stats").await
309    }
310
311    pub async fn toggle_chaos(&self, enabled: bool) -> Result<String> {
312        self.post_api("/__mockforge/chaos/toggle", &serde_json::json!({ "enabled": enabled }))
313            .await
314    }
315
316    pub async fn get_chaos_scenarios(&self) -> Result<serde_json::Value> {
317        self.get_api("/__mockforge/chaos/scenarios/predefined").await
318    }
319
320    pub async fn start_chaos_scenario(&self, name: &str) -> Result<String> {
321        self.post_api(&format!("/__mockforge/chaos/scenarios/{name}"), &serde_json::json!({}))
322            .await
323    }
324
325    pub async fn stop_chaos_scenario(&self, name: &str) -> Result<String> {
326        let url = format!("{}/__mockforge/chaos/scenarios/{name}", self.base_url);
327        let resp = self.client.delete(&url).send().await.context("DELETE chaos/scenarios stop")?;
328
329        let status = resp.status();
330        if !status.is_success() {
331            anyhow::bail!("HTTP {status} from chaos stop");
332        }
333
334        let body = resp.text().await.context("read chaos stop response")?;
335        let envelope: ApiResponse<String> =
336            serde_json::from_str(&body).context("deserialise chaos stop response")?;
337        if envelope.success {
338            Ok(envelope.data.unwrap_or_default())
339        } else {
340            anyhow::bail!(
341                "stop scenario failed: {}",
342                envelope.error.unwrap_or_else(|| "unknown".into())
343            )
344        }
345    }
346
347    pub async fn get_time_travel_status(&self) -> Result<TimeTravelStatus> {
348        // Server returns raw TimeTravelStatus, not ApiResponse-wrapped
349        self.get_raw("/__mockforge/time-travel/status").await
350    }
351
352    pub async fn get_chains(&self) -> Result<Vec<ChainInfo>> {
353        self.get_api("/__mockforge/chains").await
354    }
355
356    pub async fn get_audit_logs(&self) -> Result<Vec<AuditEntry>> {
357        self.get_api("/__mockforge/audit/logs").await
358    }
359
360    pub async fn get_analytics_summary(&self) -> Result<AnalyticsSummary> {
361        self.get_api("/__mockforge/analytics/summary").await
362    }
363
364    // ── Tier 3 endpoints ─────────────────────────────────────────────
365
366    pub async fn get_federation_peers(&self) -> Result<Vec<FederationPeer>> {
367        self.get_api("/__mockforge/federation/peers").await
368    }
369
370    pub async fn get_conformance_violations(&self) -> Result<ConformanceViolationsResponse> {
371        let resp = self
372            .get("/__mockforge/api/conformance/violations")
373            .send()
374            .await
375            .context("GET /__mockforge/api/conformance/violations")?;
376        let status = resp.status();
377        if !status.is_success() {
378            anyhow::bail!("HTTP {status} from conformance/violations");
379        }
380        let ct = resp
381            .headers()
382            .get(reqwest::header::CONTENT_TYPE)
383            .and_then(|v| v.to_str().ok())
384            .unwrap_or("");
385        if ct.contains("text/html") {
386            anyhow::bail!("endpoint conformance/violations not available");
387        }
388        let body = resp.text().await.context("read conformance response")?;
389        serde_json::from_str::<ConformanceViolationsResponse>(&body)
390            .context("deserialise conformance response")
391    }
392
393    /// Clear the server-side conformance violation buffer.
394    /// Returns the number of entries that were cleared.
395    pub async fn clear_conformance_violations(&self) -> Result<usize> {
396        let url = format!("{}/__mockforge/api/conformance/violations", self.base_url);
397        let mut req = self.client.delete(&url);
398        if let Some(ref token) = self.token {
399            req = req.bearer_auth(token);
400        }
401        let resp = req.send().await.context("DELETE /__mockforge/api/conformance/violations")?;
402        let status = resp.status();
403        if !status.is_success() {
404            anyhow::bail!("HTTP {status} from DELETE conformance/violations");
405        }
406        #[derive(serde::Deserialize)]
407        struct Cleared {
408            cleared: usize,
409        }
410        let body = resp.text().await.context("read clear-conformance response")?;
411        serde_json::from_str::<Cleared>(&body)
412            .map(|c| c.cleared)
413            .context("deserialise clear-conformance response")
414    }
415
416    pub async fn get_contract_diff_captures(&self) -> Result<Vec<ContractDiffCapture>> {
417        // Server returns {"captures": [...]} not ApiResponse-wrapped
418        let resp = self
419            .get("/__mockforge/contract-diff/captures")
420            .send()
421            .await
422            .context("GET /__mockforge/contract-diff/captures")?;
423
424        let status = resp.status();
425        if !status.is_success() {
426            anyhow::bail!("HTTP {status} from contract-diff/captures");
427        }
428
429        let ct = resp
430            .headers()
431            .get(reqwest::header::CONTENT_TYPE)
432            .and_then(|v| v.to_str().ok())
433            .unwrap_or("");
434        if ct.contains("text/html") {
435            anyhow::bail!("endpoint contract-diff/captures not available");
436        }
437
438        let body = resp.text().await.context("read contract-diff response")?;
439
440        // Try {"captures": [...]} wrapper first (actual server format)
441        if let Ok(wrapper) = serde_json::from_str::<ContractDiffWrapper>(&body) {
442            return Ok(wrapper.captures);
443        }
444
445        // Try ApiResponse envelope
446        if let Ok(envelope) = serde_json::from_str::<ApiResponse<Vec<ContractDiffCapture>>>(&body) {
447            if envelope.success {
448                return envelope.data.context("contract-diff: no data");
449            }
450        }
451
452        // Try raw array
453        serde_json::from_str::<Vec<ContractDiffCapture>>(&body)
454            .context("deserialise contract-diff response")
455    }
456
457    // ── Behavioral cloning / VBR ───────────────────────────────────
458
459    pub async fn get_vbr_status(&self) -> Result<serde_json::Value> {
460        self.get_api("/__mockforge/vbr/status").await
461    }
462
463    // ── Config mutations ─────────────────────────────────────────────
464
465    pub async fn update_latency(&self, config: &LatencyConfig) -> Result<String> {
466        self.post_api("/__mockforge/config/latency", &serde_json::to_value(config)?)
467            .await
468    }
469
470    pub async fn update_faults(&self, config: &FaultConfig) -> Result<String> {
471        self.post_api("/__mockforge/config/faults", &serde_json::to_value(config)?)
472            .await
473    }
474
475    pub async fn update_proxy(&self, config: &ProxyConfig) -> Result<String> {
476        self.post_api("/__mockforge/config/proxy", &serde_json::to_value(config)?).await
477    }
478
479    // ── Verification ─────────────────────────────────────────────────
480
481    pub async fn verify(&self, query: &serde_json::Value) -> Result<VerificationResult> {
482        let resp = self
483            .post("/__mockforge/verification/verify", query)
484            .send()
485            .await
486            .context("POST verification/verify")?;
487
488        let status = resp.status();
489        if !status.is_success() {
490            anyhow::bail!("HTTP {status} from verification/verify");
491        }
492
493        let body = resp.text().await.context("read verification response")?;
494        let envelope: ApiResponse<VerificationResult> =
495            serde_json::from_str(&body).context("deserialise verification response")?;
496
497        if envelope.success {
498            envelope.data.context("verification returned no data")
499        } else {
500            anyhow::bail!(
501                "verification failed: {}",
502                envelope.error.unwrap_or_else(|| "unknown".into())
503            )
504        }
505    }
506
507    // ── Time travel mutations ────────────────────────────────────────
508
509    pub async fn enable_time_travel(&self) -> Result<String> {
510        self.post_api("/__mockforge/time-travel/enable", &serde_json::json!({})).await
511    }
512
513    pub async fn disable_time_travel(&self) -> Result<String> {
514        self.post_api("/__mockforge/time-travel/disable", &serde_json::json!({})).await
515    }
516
517    // ── Chain execution ──────────────────────────────────────────────
518
519    pub async fn execute_chain(&self, id: &str) -> Result<serde_json::Value> {
520        let path = format!("/__mockforge/chains/{id}/execute");
521        let resp = self
522            .post(&path, &serde_json::json!({}))
523            .send()
524            .await
525            .with_context(|| format!("POST {path}"))?;
526
527        let status = resp.status();
528        if !status.is_success() {
529            anyhow::bail!("HTTP {status} from {path}");
530        }
531
532        let body = resp.text().await.context("read chain execution response")?;
533        let envelope: ApiResponse<serde_json::Value> =
534            serde_json::from_str(&body).context("deserialise chain execution response")?;
535
536        if envelope.success {
537            envelope.data.context("chain execution returned no data")
538        } else {
539            anyhow::bail!(
540                "chain execution failed: {}",
541                envelope.error.unwrap_or_else(|| "unknown".into())
542            )
543        }
544    }
545
546    // ── Import ───────────────────────────────────────────────────────
547
548    pub async fn get_import_history(&self) -> Result<serde_json::Value> {
549        self.get_api("/__mockforge/import/history").await
550    }
551
552    pub async fn clear_import_history(&self) -> Result<String> {
553        self.post_api("/__mockforge/import/history/clear", &serde_json::json!({})).await
554    }
555
556    // ── Recorder ─────────────────────────────────────────────────────
557
558    pub async fn get_recorder_status(&self) -> Result<serde_json::Value> {
559        self.get_api("/__mockforge/recorder/status").await
560    }
561
562    pub async fn toggle_recorder(&self, enable: bool) -> Result<String> {
563        let path = if enable {
564            "/__mockforge/recorder/start"
565        } else {
566            "/__mockforge/recorder/stop"
567        };
568        self.post_api(path, &serde_json::json!({})).await
569    }
570
571    // ── Workspace activation ──────────────────────────────────────────
572
573    pub async fn activate_workspace(&self, workspace_id: &str) -> Result<String> {
574        self.post_api(
575            &format!("/__mockforge/workspaces/{workspace_id}/activate"),
576            &serde_json::json!({}),
577        )
578        .await
579    }
580
581    // ── World State ──────────────────────────────────────────────────
582
583    pub async fn get_world_state(&self) -> Result<serde_json::Value> {
584        self.get_api("/__mockforge/world-state").await
585    }
586
587    // ── Connectivity check ───────────────────────────────────────────
588
589    /// Quick ping to verify the admin server is reachable.
590    pub async fn ping(&self) -> bool {
591        self.get("/__mockforge/health")
592            .timeout(std::time::Duration::from_secs(3))
593            .send()
594            .await
595            .is_ok()
596    }
597}
598
599#[cfg(test)]
600mod tests {
601    use super::*;
602
603    #[test]
604    fn client_strips_trailing_slash() {
605        let client = MockForgeClient::new("http://localhost:9080/".into(), None).unwrap();
606        assert_eq!(client.base_url(), "http://localhost:9080");
607    }
608
609    #[test]
610    fn client_preserves_clean_url() {
611        let client = MockForgeClient::new("http://localhost:9080".into(), None).unwrap();
612        assert_eq!(client.base_url(), "http://localhost:9080");
613    }
614}