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 },
500 );
501 }
502 }
503
504 let status_code = status_override
506 .filter(|code| self.has_response_for_status(*code))
507 .unwrap_or_else(|| self.find_first_available_status_code());
508
509 let expand_tokens = std::env::var("MOCKFORGE_RESPONSE_TEMPLATE_EXPAND")
511 .map(|v| v == "1" || v.eq_ignore_ascii_case("true"))
512 .unwrap_or(false);
513
514 let mode = Some(self.response_selection_mode);
516 let selector = Some(self.response_selector.as_ref());
517
518 match response_trace::generate_response_with_trace(
520 &self.spec,
521 &self.operation,
522 status_code,
523 Some("application/json"),
524 expand_tokens,
525 scenario,
526 mode,
527 selector,
528 None, ) {
530 Ok((response_body, trace)) => {
531 tracing::debug!(
532 "ResponseGenerator succeeded for {} {} with status {} and scenario {:?}: {:?}",
533 self.method,
534 self.path,
535 status_code,
536 scenario,
537 response_body
538 );
539 (status_code, response_body, trace)
540 }
541 Err(e) => {
542 tracing::debug!(
543 "ResponseGenerator failed for {} {}: {}, using fallback",
544 self.method,
545 self.path,
546 e
547 );
548 let response_body = serde_json::json!({
550 "message": format!("Mock response for {} {}", self.method, self.path),
551 "operation_id": self.operation.operation_id,
552 "status": status_code
553 });
554 let mut trace = ResponseGenerationTrace::new();
556 trace.set_final_payload(response_body.clone());
557 trace.add_metadata("fallback".to_string(), serde_json::json!(true));
558 trace.add_metadata("error".to_string(), serde_json::json!(e.to_string()));
559 (status_code, response_body, trace)
560 }
561 }
562 }
563
564 pub fn has_response_for_status(&self, code: u16) -> bool {
566 self.operation
567 .responses
568 .responses
569 .iter()
570 .any(|(status, _)| matches!(status, openapiv3::StatusCode::Code(c) if *c == code))
571 }
572
573 pub fn find_first_available_status_code(&self) -> u16 {
582 let mut lowest_2xx: Option<u16> = None;
583 let mut has_2xx_range = false;
584 let mut lowest_other: Option<u16> = None;
585
586 for (status, _) in &self.operation.responses.responses {
587 match status {
588 openapiv3::StatusCode::Code(code) => {
589 if (200..300).contains(code) {
590 lowest_2xx = Some(lowest_2xx.map_or(*code, |c| c.min(*code)));
591 } else {
592 lowest_other = Some(lowest_other.map_or(*code, |c| c.min(*code)));
593 }
594 }
595 openapiv3::StatusCode::Range(2) => has_2xx_range = true,
596 openapiv3::StatusCode::Range(_) => {}
597 }
598 }
599
600 if let Some(code) = lowest_2xx {
601 return code;
602 }
603 if has_2xx_range {
604 return 200;
605 }
606 if self.operation.responses.default.is_some() {
609 return 200;
610 }
611 lowest_other.unwrap_or(200)
612 }
613}
614
615#[derive(Debug, Clone)]
617pub struct OpenApiOperation {
618 pub method: String,
620 pub path: String,
622 pub operation: Operation,
624}
625
626impl OpenApiOperation {
627 pub fn new(method: String, path: String, operation: Operation) -> Self {
629 Self {
630 method,
631 path,
632 operation,
633 }
634 }
635}
636
637pub struct RouteGenerator;
639
640impl RouteGenerator {
641 pub fn generate_routes_from_path(
643 path: &str,
644 path_item: &ReferenceOr<PathItem>,
645 spec: &Arc<OpenApiSpec>,
646 ) -> Result<Vec<OpenApiRoute>> {
647 Self::generate_routes_from_path_with_persona(path, path_item, spec, None)
648 }
649
650 pub fn generate_routes_from_path_with_persona(
652 path: &str,
653 path_item: &ReferenceOr<PathItem>,
654 spec: &Arc<OpenApiSpec>,
655 persona: Option<Arc<Persona>>,
656 ) -> Result<Vec<OpenApiRoute>> {
657 let mut routes = Vec::new();
658
659 if let Some(item) = path_item.as_item() {
660 if let Some(op) = &item.get {
662 routes.push(OpenApiRoute::new_with_persona(
663 "GET".to_string(),
664 path.to_string(),
665 op.clone(),
666 spec.clone(),
667 persona.clone(),
668 ));
669 }
670 if let Some(op) = &item.post {
671 routes.push(OpenApiRoute::new_with_persona(
672 "POST".to_string(),
673 path.to_string(),
674 op.clone(),
675 spec.clone(),
676 persona.clone(),
677 ));
678 }
679 if let Some(op) = &item.put {
680 routes.push(OpenApiRoute::new_with_persona(
681 "PUT".to_string(),
682 path.to_string(),
683 op.clone(),
684 spec.clone(),
685 persona.clone(),
686 ));
687 }
688 if let Some(op) = &item.delete {
689 routes.push(OpenApiRoute::new_with_persona(
690 "DELETE".to_string(),
691 path.to_string(),
692 op.clone(),
693 spec.clone(),
694 persona.clone(),
695 ));
696 }
697 if let Some(op) = &item.patch {
698 routes.push(OpenApiRoute::new_with_persona(
699 "PATCH".to_string(),
700 path.to_string(),
701 op.clone(),
702 spec.clone(),
703 persona.clone(),
704 ));
705 }
706 if let Some(op) = &item.head {
707 routes.push(OpenApiRoute::new_with_persona(
708 "HEAD".to_string(),
709 path.to_string(),
710 op.clone(),
711 spec.clone(),
712 persona.clone(),
713 ));
714 }
715 if let Some(op) = &item.options {
716 routes.push(OpenApiRoute::new_with_persona(
717 "OPTIONS".to_string(),
718 path.to_string(),
719 op.clone(),
720 spec.clone(),
721 persona.clone(),
722 ));
723 }
724 if let Some(op) = &item.trace {
725 routes.push(OpenApiRoute::new_with_persona(
726 "TRACE".to_string(),
727 path.to_string(),
728 op.clone(),
729 spec.clone(),
730 persona.clone(),
731 ));
732 }
733 }
734
735 Ok(routes)
736 }
737}
738
739#[cfg(test)]
740mod status_code_selection_tests {
741 use super::OpenApiRoute;
742 use crate::spec::OpenApiSpec;
743 use serde_json::json;
744 use std::sync::Arc;
745
746 fn route_from_responses(responses: serde_json::Value) -> OpenApiRoute {
747 let operation: openapiv3::Operation =
748 serde_json::from_value(json!({ "responses": responses })).expect("valid operation");
749 let spec = OpenApiSpec::from_json(json!({
750 "openapi": "3.0.0",
751 "info": {"title": "t", "version": "1.0.0"},
752 "paths": {}
753 }))
754 .expect("valid spec");
755 OpenApiRoute::new("GET".to_string(), "/x".to_string(), operation, Arc::new(spec))
756 }
757
758 #[test]
759 fn prefers_2xx_over_earlier_listed_error_codes() {
760 let r = route_from_responses(json!({
763 "400": {"description": "bad"},
764 "404": {"description": "missing"},
765 "200": {"description": "ok"},
766 }));
767 assert_eq!(r.find_first_available_status_code(), 200);
768 }
769
770 #[test]
771 fn prefers_lowest_2xx() {
772 let r = route_from_responses(json!({
773 "204": {"description": "no content"},
774 "201": {"description": "created"},
775 }));
776 assert_eq!(r.find_first_available_status_code(), 201);
777 }
778
779 #[test]
780 fn uses_2xx_range_when_no_explicit_2xx() {
781 let r = route_from_responses(json!({ "2XX": {"description": "ok-ish"} }));
782 assert_eq!(r.find_first_available_status_code(), 200);
783 }
784
785 #[test]
786 fn default_only_returns_200() {
787 let r = route_from_responses(json!({ "default": {"description": "any"} }));
788 assert_eq!(r.find_first_available_status_code(), 200);
789 }
790
791 #[test]
792 fn error_only_returns_lowest_error() {
793 let r = route_from_responses(json!({
795 "500": {"description": "err"},
796 "404": {"description": "err"},
797 }));
798 assert_eq!(r.find_first_available_status_code(), 404);
799 }
800}