1use crate::response_selection::{ResponseSelectionMode, ResponseSelector};
7use crate::spec::OpenApiSpec;
8use mockforge_foundation::ai_response::AiResponseConfig;
9use mockforge_foundation::error::Result;
10use mockforge_foundation::intelligent_behavior::Persona;
11use openapiv3::{Operation, PathItem, ReferenceOr};
12use std::collections::BTreeMap;
13use std::sync::Arc;
14
15fn extract_path_parameters(path_template: &str) -> Vec<String> {
17 let mut params = Vec::new();
18 let mut in_param = false;
19 let mut current_param = String::new();
20
21 for ch in path_template.chars() {
22 match ch {
23 '{' => {
24 in_param = true;
25 current_param.clear();
26 }
27 '}' => {
28 if in_param {
29 params.push(current_param.clone());
30 in_param = false;
31 }
32 }
33 ch if in_param => {
34 current_param.push(ch);
35 }
36 _ => {}
37 }
38 }
39
40 params
41}
42
43#[derive(Debug, Clone)]
45pub struct OpenApiRoute {
46 pub method: String,
48 pub path: String,
50 pub operation: Operation,
52 pub metadata: BTreeMap<String, String>,
54 pub parameters: Vec<String>,
56 pub spec: Arc<OpenApiSpec>,
58 pub ai_config: Option<AiResponseConfig>,
60 pub response_selection_mode: ResponseSelectionMode,
62 pub response_selector: Arc<ResponseSelector>,
64 pub persona: Option<Arc<Persona>>,
66}
67
68impl OpenApiRoute {
69 pub fn new(method: String, path: String, operation: Operation, spec: Arc<OpenApiSpec>) -> Self {
71 Self::new_with_persona(method, path, operation, spec, None)
72 }
73
74 pub fn new_with_persona(
76 method: String,
77 path: String,
78 operation: Operation,
79 spec: Arc<OpenApiSpec>,
80 persona: Option<Arc<Persona>>,
81 ) -> Self {
82 let parameters = extract_path_parameters(&path);
83
84 let ai_config = Self::parse_ai_config(&operation);
86
87 let response_selection_mode = Self::parse_response_selection_mode(&operation);
89 let response_selector = Arc::new(ResponseSelector::new(response_selection_mode));
90
91 Self {
92 method,
93 path,
94 operation,
95 metadata: BTreeMap::new(),
96 parameters,
97 spec,
98 ai_config,
99 response_selection_mode,
100 response_selector,
101 persona,
102 }
103 }
104
105 fn parse_ai_config(operation: &Operation) -> Option<AiResponseConfig> {
107 if let Some(ai_config_value) = operation.extensions.get("x-mockforge-ai") {
109 match serde_json::from_value::<AiResponseConfig>(ai_config_value.clone()) {
111 Ok(config) => {
112 if config.is_active() {
113 tracing::debug!(
114 "Parsed AI config for operation {}: mode={:?}, prompt={:?}",
115 operation.operation_id.as_deref().unwrap_or("unknown"),
116 config.mode,
117 config.prompt
118 );
119 return Some(config);
120 }
121 }
122 Err(e) => {
123 tracing::warn!(
124 "Failed to parse x-mockforge-ai extension for operation {}: {}",
125 operation.operation_id.as_deref().unwrap_or("unknown"),
126 e
127 );
128 }
129 }
130 }
131 None
132 }
133
134 fn parse_response_selection_mode(operation: &Operation) -> ResponseSelectionMode {
136 let op_id = operation.operation_id.as_deref().unwrap_or("unknown");
138
139 if let Ok(op_env_var) = std::env::var(format!(
141 "MOCKFORGE_RESPONSE_SELECTION_{}",
142 op_id.to_uppercase().replace('-', "_")
143 )) {
144 if let Some(mode) = ResponseSelectionMode::from_str(&op_env_var) {
145 tracing::debug!(
146 "Using response selection mode from env var for operation {}: {:?}",
147 op_id,
148 mode
149 );
150 return mode;
151 }
152 }
153
154 if let Ok(global_mode_str) = std::env::var("MOCKFORGE_RESPONSE_SELECTION_MODE") {
156 if let Some(mode) = ResponseSelectionMode::from_str(&global_mode_str) {
157 tracing::debug!("Using global response selection mode from env var: {:?}", mode);
158 return mode;
159 }
160 }
161
162 if let Some(selection_value) = operation.extensions.get("x-mockforge-response-selection") {
164 if let Some(mode_str) = selection_value.as_str() {
166 if let Some(mode) = ResponseSelectionMode::from_str(mode_str) {
167 tracing::debug!(
168 "Parsed response selection mode for operation {}: {:?}",
169 op_id,
170 mode
171 );
172 return mode;
173 }
174 }
175 if let Some(obj) = selection_value.as_object() {
177 if let Some(mode_str) = obj.get("mode").and_then(|v| v.as_str()) {
178 if let Some(mode) = ResponseSelectionMode::from_str(mode_str) {
179 tracing::debug!(
180 "Parsed response selection mode for operation {}: {:?}",
181 op_id,
182 mode
183 );
184 return mode;
185 }
186 }
187 }
188 tracing::warn!(
189 "Failed to parse x-mockforge-response-selection extension for operation {}",
190 op_id
191 );
192 }
193 ResponseSelectionMode::First
195 }
196
197 pub fn from_operation(
199 method: &str,
200 path: String,
201 operation: &Operation,
202 spec: Arc<OpenApiSpec>,
203 ) -> Self {
204 Self::from_operation_with_persona(method, path, operation, spec, None)
205 }
206
207 pub fn from_operation_with_persona(
209 method: &str,
210 path: String,
211 operation: &Operation,
212 spec: Arc<OpenApiSpec>,
213 persona: Option<Arc<Persona>>,
214 ) -> Self {
215 Self::new_with_persona(method.to_string(), path, operation.clone(), spec, persona)
216 }
217
218 pub fn axum_path(&self) -> String {
220 let path = self.path.split('?').next().unwrap_or(&self.path);
223
224 if path.contains("()") {
226 let path = path.replace("()", "");
227 return path;
228 }
229
230 if path.contains('(') && path.contains('=') {
235 let mut result = String::with_capacity(path.len());
236 let mut chars = path.chars().peekable();
237
238 while let Some(ch) = chars.next() {
239 if ch == '(' {
240 let mut paren_content = String::new();
242 for c in chars.by_ref() {
243 if c == ')' {
244 break;
245 }
246 paren_content.push(c);
247 }
248 for part in paren_content.split(',') {
250 if let Some((_key, value)) = part.split_once('=') {
251 let param = value.trim_matches(|c| c == '\'' || c == '"');
252 result.push('/');
253 result.push_str(param);
254 }
255 }
256 } else {
257 result.push(ch);
258 }
259 }
260 return result;
261 }
262
263 path.to_string()
264 }
265
266 pub fn is_valid_axum_path(&self) -> bool {
271 let path = self.axum_path();
272 if path.contains('(') || path.contains(')') {
274 return false;
275 }
276 for segment in path.split('/') {
278 let brace_count = segment.matches('{').count();
279 if brace_count > 1 {
280 return false;
281 }
282 if brace_count == 1
285 && segment
286 != format!(
287 "{{{}}}",
288 segment
289 .trim_matches(|c: char| c != '{' && c != '}')
290 .trim_matches(|c| c == '{' || c == '}')
291 )
292 {
293 if !segment.starts_with('{') || !segment.ends_with('}') {
296 return false;
297 }
298 }
299 }
300 true
301 }
302
303 pub fn with_metadata(mut self, key: String, value: String) -> Self {
305 self.metadata.insert(key, value);
306 self
307 }
308
309 pub async fn mock_response_with_status_async(
318 &self,
319 context: &mockforge_foundation::ai_response::RequestContext,
320 ai_generator: Option<&dyn crate::response::AiGenerator>,
321 ) -> (u16, serde_json::Value) {
322 use crate::response::ResponseGenerator;
323
324 let status_code = self.find_first_available_status_code();
326
327 if let Some(ai_config) = &self.ai_config {
329 if ai_config.is_active() {
330 tracing::info!(
331 "Using AI-assisted response generation for {} {}",
332 self.method,
333 self.path
334 );
335
336 match ResponseGenerator::generate_ai_response(ai_config, context, ai_generator)
337 .await
338 {
339 Ok(response_body) => {
340 tracing::debug!(
341 "AI response generated successfully for {} {}: {:?}",
342 self.method,
343 self.path,
344 response_body
345 );
346 return (status_code, response_body);
347 }
348 Err(e) => {
349 tracing::warn!(
350 "AI response generation failed for {} {}: {}, falling back to standard generation",
351 self.method,
352 self.path,
353 e
354 );
355 }
357 }
358 }
359 }
360
361 let expand_tokens = std::env::var("MOCKFORGE_RESPONSE_TEMPLATE_EXPAND")
363 .map(|v| v == "1" || v.eq_ignore_ascii_case("true"))
364 .unwrap_or(false);
365
366 let mode = Some(self.response_selection_mode);
368 let selector = Some(self.response_selector.as_ref());
369
370 let persona_ref = self.persona.as_deref();
372
373 match ResponseGenerator::generate_response_with_expansion_and_mode_and_persona(
374 &self.spec,
375 &self.operation,
376 status_code,
377 Some("application/json"),
378 expand_tokens,
379 mode,
380 selector,
381 persona_ref,
382 ) {
383 Ok(response_body) => {
384 tracing::debug!(
385 "ResponseGenerator succeeded for {} {} with status {}: {:?}",
386 self.method,
387 self.path,
388 status_code,
389 response_body
390 );
391 (status_code, response_body)
392 }
393 Err(e) => {
394 tracing::debug!(
395 "ResponseGenerator failed for {} {}: {}, using fallback",
396 self.method,
397 self.path,
398 e
399 );
400 let response_body = serde_json::json!({
402 "message": format!("Mock response for {} {}", self.method, self.path),
403 "operation_id": self.operation.operation_id,
404 "status": status_code
405 });
406 (status_code, response_body)
407 }
408 }
409 }
410
411 pub fn mock_response_with_status(&self) -> (u16, serde_json::Value) {
416 self.mock_response_with_status_and_scenario(None)
417 }
418
419 pub fn mock_response_with_status_and_scenario(
431 &self,
432 scenario: Option<&str>,
433 ) -> (u16, serde_json::Value) {
434 self.mock_response_with_status_and_scenario_and_override(scenario, None)
435 }
436
437 pub fn mock_response_with_status_and_scenario_and_override(
443 &self,
444 scenario: Option<&str>,
445 status_override: Option<u16>,
446 ) -> (u16, serde_json::Value) {
447 let (status, body, _) =
448 self.mock_response_with_status_and_scenario_and_trace(scenario, status_override);
449 (status, body)
450 }
451
452 pub fn mock_response_with_status_and_scenario_and_trace(
456 &self,
457 scenario: Option<&str>,
458 status_override: Option<u16>,
459 ) -> (
460 u16,
461 serde_json::Value,
462 mockforge_foundation::response_generation_trace::ResponseGenerationTrace,
463 ) {
464 use crate::response_trace;
465 use mockforge_foundation::response_generation_trace::ResponseGenerationTrace;
466
467 if let Some(requested) = status_override {
480 if !self.has_response_for_status(requested) {
481 let available: Vec<String> =
482 self.operation.responses.responses.keys().map(|k| format!("{:?}", k)).collect();
483 mockforge_foundation::conformance_violations::record(
484 mockforge_foundation::conformance_violations::ServerConformanceViolation {
485 timestamp: chrono::Utc::now(),
486 method: self.method.clone(),
487 path: self.path.clone(),
488 client_ip: "unknown".to_string(),
489 status: requested,
490 reason: format!(
491 "spec defines no response for status {} on {} {}; available: {}",
492 requested,
493 self.method,
494 self.path,
495 available.join(", ")
496 ),
497 category: "response-shape".to_string(),
498 occurrences: 1,
499 client_mockforge_version: None,
506 client_sent_at: None,
507 },
508 );
509 }
510 }
511
512 let status_code = status_override
514 .filter(|code| self.has_response_for_status(*code))
515 .unwrap_or_else(|| self.find_first_available_status_code());
516
517 let expand_tokens = std::env::var("MOCKFORGE_RESPONSE_TEMPLATE_EXPAND")
519 .map(|v| v == "1" || v.eq_ignore_ascii_case("true"))
520 .unwrap_or(false);
521
522 let mode = Some(self.response_selection_mode);
524 let selector = Some(self.response_selector.as_ref());
525
526 match response_trace::generate_response_with_trace(
528 &self.spec,
529 &self.operation,
530 status_code,
531 Some("application/json"),
532 expand_tokens,
533 scenario,
534 mode,
535 selector,
536 None, ) {
538 Ok((response_body, trace)) => {
539 tracing::debug!(
540 "ResponseGenerator succeeded for {} {} with status {} and scenario {:?}: {:?}",
541 self.method,
542 self.path,
543 status_code,
544 scenario,
545 response_body
546 );
547 (status_code, response_body, trace)
548 }
549 Err(e) => {
550 tracing::debug!(
551 "ResponseGenerator failed for {} {}: {}, using fallback",
552 self.method,
553 self.path,
554 e
555 );
556 let response_body = serde_json::json!({
558 "message": format!("Mock response for {} {}", self.method, self.path),
559 "operation_id": self.operation.operation_id,
560 "status": status_code
561 });
562 let mut trace = ResponseGenerationTrace::new();
564 trace.set_final_payload(response_body.clone());
565 trace.add_metadata("fallback".to_string(), serde_json::json!(true));
566 trace.add_metadata("error".to_string(), serde_json::json!(e.to_string()));
567 (status_code, response_body, trace)
568 }
569 }
570 }
571
572 pub fn has_response_for_status(&self, code: u16) -> bool {
574 self.operation
575 .responses
576 .responses
577 .iter()
578 .any(|(status, _)| matches!(status, openapiv3::StatusCode::Code(c) if *c == code))
579 }
580
581 pub fn find_first_available_status_code(&self) -> u16 {
590 let mut lowest_2xx: Option<u16> = None;
591 let mut has_2xx_range = false;
592 let mut lowest_other: Option<u16> = None;
593
594 for (status, _) in &self.operation.responses.responses {
595 match status {
596 openapiv3::StatusCode::Code(code) => {
597 if (200..300).contains(code) {
598 lowest_2xx = Some(lowest_2xx.map_or(*code, |c| c.min(*code)));
599 } else {
600 lowest_other = Some(lowest_other.map_or(*code, |c| c.min(*code)));
601 }
602 }
603 openapiv3::StatusCode::Range(2) => has_2xx_range = true,
604 openapiv3::StatusCode::Range(_) => {}
605 }
606 }
607
608 if let Some(code) = lowest_2xx {
609 return code;
610 }
611 if has_2xx_range {
612 return 200;
613 }
614 if self.operation.responses.default.is_some() {
617 return 200;
618 }
619 lowest_other.unwrap_or(200)
620 }
621}
622
623#[derive(Debug, Clone)]
625pub struct OpenApiOperation {
626 pub method: String,
628 pub path: String,
630 pub operation: Operation,
632}
633
634impl OpenApiOperation {
635 pub fn new(method: String, path: String, operation: Operation) -> Self {
637 Self {
638 method,
639 path,
640 operation,
641 }
642 }
643}
644
645pub struct RouteGenerator;
647
648impl RouteGenerator {
649 pub fn generate_routes_from_path(
651 path: &str,
652 path_item: &ReferenceOr<PathItem>,
653 spec: &Arc<OpenApiSpec>,
654 ) -> Result<Vec<OpenApiRoute>> {
655 Self::generate_routes_from_path_with_persona(path, path_item, spec, None)
656 }
657
658 pub fn generate_routes_from_path_with_persona(
660 path: &str,
661 path_item: &ReferenceOr<PathItem>,
662 spec: &Arc<OpenApiSpec>,
663 persona: Option<Arc<Persona>>,
664 ) -> Result<Vec<OpenApiRoute>> {
665 let mut routes = Vec::new();
666
667 if let Some(item) = path_item.as_item() {
668 if let Some(op) = &item.get {
670 routes.push(OpenApiRoute::new_with_persona(
671 "GET".to_string(),
672 path.to_string(),
673 op.clone(),
674 spec.clone(),
675 persona.clone(),
676 ));
677 }
678 if let Some(op) = &item.post {
679 routes.push(OpenApiRoute::new_with_persona(
680 "POST".to_string(),
681 path.to_string(),
682 op.clone(),
683 spec.clone(),
684 persona.clone(),
685 ));
686 }
687 if let Some(op) = &item.put {
688 routes.push(OpenApiRoute::new_with_persona(
689 "PUT".to_string(),
690 path.to_string(),
691 op.clone(),
692 spec.clone(),
693 persona.clone(),
694 ));
695 }
696 if let Some(op) = &item.delete {
697 routes.push(OpenApiRoute::new_with_persona(
698 "DELETE".to_string(),
699 path.to_string(),
700 op.clone(),
701 spec.clone(),
702 persona.clone(),
703 ));
704 }
705 if let Some(op) = &item.patch {
706 routes.push(OpenApiRoute::new_with_persona(
707 "PATCH".to_string(),
708 path.to_string(),
709 op.clone(),
710 spec.clone(),
711 persona.clone(),
712 ));
713 }
714 if let Some(op) = &item.head {
715 routes.push(OpenApiRoute::new_with_persona(
716 "HEAD".to_string(),
717 path.to_string(),
718 op.clone(),
719 spec.clone(),
720 persona.clone(),
721 ));
722 }
723 if let Some(op) = &item.options {
724 routes.push(OpenApiRoute::new_with_persona(
725 "OPTIONS".to_string(),
726 path.to_string(),
727 op.clone(),
728 spec.clone(),
729 persona.clone(),
730 ));
731 }
732 if let Some(op) = &item.trace {
733 routes.push(OpenApiRoute::new_with_persona(
734 "TRACE".to_string(),
735 path.to_string(),
736 op.clone(),
737 spec.clone(),
738 persona.clone(),
739 ));
740 }
741 }
742
743 Ok(routes)
744 }
745}
746
747#[cfg(test)]
748mod status_code_selection_tests {
749 use super::OpenApiRoute;
750 use crate::spec::OpenApiSpec;
751 use serde_json::json;
752 use std::sync::Arc;
753
754 fn route_from_responses(responses: serde_json::Value) -> OpenApiRoute {
755 let operation: openapiv3::Operation =
756 serde_json::from_value(json!({ "responses": responses })).expect("valid operation");
757 let spec = OpenApiSpec::from_json(json!({
758 "openapi": "3.0.0",
759 "info": {"title": "t", "version": "1.0.0"},
760 "paths": {}
761 }))
762 .expect("valid spec");
763 OpenApiRoute::new("GET".to_string(), "/x".to_string(), operation, Arc::new(spec))
764 }
765
766 #[test]
767 fn prefers_2xx_over_earlier_listed_error_codes() {
768 let r = route_from_responses(json!({
771 "400": {"description": "bad"},
772 "404": {"description": "missing"},
773 "200": {"description": "ok"},
774 }));
775 assert_eq!(r.find_first_available_status_code(), 200);
776 }
777
778 #[test]
779 fn prefers_lowest_2xx() {
780 let r = route_from_responses(json!({
781 "204": {"description": "no content"},
782 "201": {"description": "created"},
783 }));
784 assert_eq!(r.find_first_available_status_code(), 201);
785 }
786
787 #[test]
788 fn uses_2xx_range_when_no_explicit_2xx() {
789 let r = route_from_responses(json!({ "2XX": {"description": "ok-ish"} }));
790 assert_eq!(r.find_first_available_status_code(), 200);
791 }
792
793 #[test]
794 fn default_only_returns_200() {
795 let r = route_from_responses(json!({ "default": {"description": "any"} }));
796 assert_eq!(r.find_first_available_status_code(), 200);
797 }
798
799 #[test]
800 fn error_only_returns_lowest_error() {
801 let r = route_from_responses(json!({
803 "500": {"description": "err"},
804 "404": {"description": "err"},
805 }));
806 assert_eq!(r.find_first_available_status_code(), 404);
807 }
808}