1use utoipa::openapi::security::{ApiKey, ApiKeyValue, SecurityScheme};
10use utoipa::OpenApi;
11
12use crate::{
13 common::ApiError,
14 v1::{
15 BrainRegionDetail,
16 BrainRegionListResponse,
17 BrainRegionSummary,
19 BurstCountResponse,
20
21 ComponentReadiness,
22
23 ConnectivityStatsResponse,
24 ConnectomeAnalyticsResponse,
25 Coordinates3D,
26 CorticalAreaDetail,
27 CorticalAreaListResponse,
28 CorticalAreaStatsResponse,
29 CorticalAreaSummary,
31 CreateBrainRegionRequest,
32
33 CreateCorticalAreaRequest,
34 CreateNeuronRequest,
35 Dimensions3D,
36
37 GenomeInfoResponse,
39 HealthCheckResponseV1,
41 LoadGenomeRequest,
42 NeuronCountResponse,
43
44 NeuronDensityResponse,
45 NeuronInfoResponse,
47 NeuronListResponse,
48 PopulatedAreaInfo,
49 PopulatedAreasResponse,
50 ReadinessCheckResponseV1,
51 RuntimeStatusResponse,
53 SaveGenomeRequest,
54 SaveGenomeResponse,
55 SetFrequencyRequest,
56 SystemHealthResponse,
58 UpdateCorticalAreaRequest,
59 ValidateGenomeRequest,
60 ValidateGenomeResponse,
61 },
62};
63
64#[derive(OpenApi)]
66#[openapi(
67 info(
68 title = "FEAGI REST API",
69 version = "1.0.0",
70 description = "Comprehensive REST API for FEAGI (Foundational Engine for Artificial General Intelligence)",
71 license(
72 name = "Apache-2.0",
73 url = "https://www.apache.org/licenses/LICENSE-2.0"
74 ),
75 contact(
76 name = "FEAGI Team",
77 url = "https://feagi.org",
78 email = "contact@feagi.org"
79 )
80 ),
81 servers(
82 (url = "http://localhost:8000", description = "Default FEAGI server"),
83 (url = "http://localhost:8080", description = "Alternative port")
84 ),
85 paths(
86 crate::endpoints::agent::register_agent,
88 crate::endpoints::agent::heartbeat,
89 crate::endpoints::agent::list_agents,
90 crate::endpoints::agent::get_agent_properties,
91 crate::endpoints::agent::get_agent_properties_path,
92 crate::endpoints::agent::get_shared_memory,
93 crate::endpoints::agent::deregister_agent,
94 crate::endpoints::agent::manual_stimulation,
95 crate::endpoints::agent::get_fq_sampler_status,
96 crate::endpoints::agent::get_capabilities,
97 crate::endpoints::agent::get_agent_info,
98 crate::endpoints::agent::post_configure,
99 crate::endpoints::agent::export_device_registrations,
100 crate::endpoints::agent::import_device_registrations,
101
102 crate::endpoints::system::get_health_check,
104 crate::endpoints::system::get_cortical_area_visualization_skip_rate,
105 crate::endpoints::system::set_cortical_area_visualization_skip_rate,
106 crate::endpoints::system::get_cortical_area_visualization_suppression_threshold,
107 crate::endpoints::system::set_cortical_area_visualization_suppression_threshold,
108 crate::endpoints::system::get_version,
109 crate::endpoints::system::get_versions,
110 crate::endpoints::system::get_configuration,
111 crate::endpoints::system::get_user_preferences,
112 crate::endpoints::system::put_user_preferences,
113 crate::endpoints::system::get_cortical_area_types_list,
114 crate::endpoints::system::post_enable_visualization_fq_sampler,
115 crate::endpoints::system::post_disable_visualization_fq_sampler,
116 crate::endpoints::system::get_fcl_status_system,
117 crate::endpoints::system::post_fcl_reset_system,
118 crate::endpoints::system::get_processes,
119 crate::endpoints::system::get_unique_logs,
120 crate::endpoints::system::post_logs,
121 crate::endpoints::system::get_beacon_subscribers,
122 crate::endpoints::system::post_beacon_subscribe,
123 crate::endpoints::system::delete_beacon_unsubscribe,
124 crate::endpoints::system::get_global_activity_visualization,
125 crate::endpoints::system::put_global_activity_visualization,
126 crate::endpoints::system::post_circuit_library_path,
127 crate::endpoints::system::get_influxdb_test,
128 crate::endpoints::system::post_register_system,
129
130 crate::endpoints::cortical_area::get_ipu,
132 crate::endpoints::cortical_area::get_ipu_types,
133 crate::endpoints::cortical_area::get_opu,
134 crate::endpoints::cortical_area::get_opu_types,
135 crate::endpoints::cortical_area::get_cortical_area_id_list,
136 crate::endpoints::cortical_area::get_cortical_area_name_list,
137 crate::endpoints::cortical_area::get_cortical_id_name_mapping,
138 crate::endpoints::cortical_area::get_cortical_types,
139 crate::endpoints::cortical_area::get_cortical_map_detailed,
140 crate::endpoints::cortical_area::get_cortical_locations_2d,
141 crate::endpoints::cortical_area::get_cortical_area_geometry,
142 crate::endpoints::cortical_area::get_cortical_visibility,
143 crate::endpoints::cortical_area::post_cortical_name_location,
144 crate::endpoints::cortical_area::post_cortical_area_properties,
145 crate::endpoints::cortical_area::post_multi_cortical_area_properties,
146 crate::endpoints::cortical_area::post_cortical_area,
147 crate::endpoints::cortical_area::put_cortical_area,
148 crate::endpoints::cortical_area::delete_cortical_area,
149 crate::endpoints::cortical_area::post_custom_cortical_area,
150 crate::endpoints::cortical_area::post_clone,
151 crate::endpoints::cortical_area::put_multi_cortical_area,
152 crate::endpoints::cortical_area::delete_multi_cortical_area,
153 crate::endpoints::cortical_area::put_coord_2d,
154 crate::endpoints::cortical_area::put_suppress_cortical_visibility,
155 crate::endpoints::cortical_area::put_reset,
156 crate::endpoints::cortical_area::get_visualization,
157 crate::endpoints::cortical_area::post_batch_operations,
158 crate::endpoints::cortical_area::get_ipu_list,
159 crate::endpoints::cortical_area::get_opu_list,
160 crate::endpoints::cortical_area::put_coordinates_3d,
161 crate::endpoints::cortical_area::delete_bulk,
162 crate::endpoints::cortical_area::post_clone,
163 crate::endpoints::cortical_area::post_resize,
164 crate::endpoints::cortical_area::post_reposition,
165 crate::endpoints::cortical_area::post_voxel_neurons,
166 crate::endpoints::cortical_area::get_cortical_area_index_list,
167 crate::endpoints::cortical_area::get_cortical_idx_mapping,
168 crate::endpoints::cortical_area::get_mapping_restrictions_query,
169 crate::endpoints::cortical_area::get_memory_usage,
170 crate::endpoints::cortical_area::get_area_neuron_count,
171 crate::endpoints::cortical_area::post_cortical_type_options,
172 crate::endpoints::cortical_area::post_mapping_restrictions,
173 crate::endpoints::cortical_area::post_mapping_restrictions_between_areas,
174 crate::endpoints::cortical_area::put_coord_3d,
175
176 crate::endpoints::morphology::get_morphology_list,
178 crate::endpoints::morphology::get_morphology_types,
179 crate::endpoints::morphology::get_list_types,
180 crate::endpoints::morphology::get_morphologies,
181 crate::endpoints::morphology::post_morphology,
182 crate::endpoints::morphology::put_morphology,
183 crate::endpoints::morphology::delete_morphology_by_name,
184 crate::endpoints::morphology::delete_morphology, crate::endpoints::morphology::post_morphology_properties,
186 crate::endpoints::morphology::post_morphology_usage,
187 crate::endpoints::morphology::get_list,
188 crate::endpoints::morphology::get_info,
189 crate::endpoints::morphology::post_create,
190 crate::endpoints::morphology::put_update,
191
192 crate::endpoints::genome::get_file_name,
194 crate::endpoints::genome::get_circuits,
195 crate::endpoints::genome::post_amalgamation_destination,
196 crate::endpoints::genome::delete_amalgamation_cancellation,
197 crate::endpoints::genome::post_genome_append,
198 crate::endpoints::genome::post_upload_barebones_genome,
199 crate::endpoints::genome::post_upload_essential_genome,
200 crate::endpoints::genome::get_name,
201 crate::endpoints::genome::get_timestamp,
202 crate::endpoints::genome::post_save,
203 crate::endpoints::genome::post_load,
204 crate::endpoints::genome::post_upload,
205 crate::endpoints::genome::get_download,
206 crate::endpoints::genome::get_properties,
207 crate::endpoints::genome::post_validate,
208 crate::endpoints::genome::post_transform,
209 crate::endpoints::genome::post_clone,
210 crate::endpoints::genome::post_reset,
211 crate::endpoints::genome::get_metadata,
212 crate::endpoints::genome::post_merge,
213 crate::endpoints::genome::get_diff,
214 crate::endpoints::genome::post_export_format,
215 crate::endpoints::genome::get_amalgamation,
216 crate::endpoints::genome::get_amalgamation_history_exact,
217 crate::endpoints::genome::get_cortical_template,
218 crate::endpoints::genome::get_defaults_files,
219 crate::endpoints::genome::get_download_region,
220 crate::endpoints::genome::get_genome_number,
221 crate::endpoints::genome::post_amalgamation_by_filename,
222 crate::endpoints::genome::post_amalgamation_by_payload,
223 crate::endpoints::genome::post_amalgamation_by_upload,
224 crate::endpoints::genome::post_append_file,
225 crate::endpoints::genome::post_upload_file,
226 crate::endpoints::genome::post_upload_file_edit,
227 crate::endpoints::genome::post_upload_string,
228
229 crate::endpoints::cortical_mapping::post_afferents,
231 crate::endpoints::cortical_mapping::post_efferents,
232 crate::endpoints::cortical_mapping::post_mapping_properties,
233 crate::endpoints::cortical_mapping::put_mapping_properties,
234 crate::endpoints::cortical_mapping::get_mapping,
235 crate::endpoints::cortical_mapping::get_mapping_list,
236 crate::endpoints::cortical_mapping::delete_mapping,
237 crate::endpoints::cortical_mapping::post_batch_update,
238 crate::endpoints::cortical_mapping::post_mapping,
239 crate::endpoints::cortical_mapping::put_mapping,
240
241 crate::endpoints::region::get_regions_members,
243 crate::endpoints::region::post_region,
244 crate::endpoints::region::put_region,
245 crate::endpoints::region::delete_region,
246 crate::endpoints::region::post_clone,
247 crate::endpoints::region::put_relocate_members,
248 crate::endpoints::region::delete_region_and_members,
249 crate::endpoints::region::get_regions,
250 crate::endpoints::region::get_region_titles,
251 crate::endpoints::region::get_region_detail,
252 crate::endpoints::region::put_change_region_parent,
253 crate::endpoints::region::put_change_cortical_area_region,
254
255 crate::endpoints::connectome::get_cortical_areas_list_detailed,
257 crate::endpoints::connectome::get_properties_dimensions,
258 crate::endpoints::connectome::get_properties_mappings,
259 crate::endpoints::connectome::get_snapshot,
260 crate::endpoints::connectome::get_stats,
261 crate::endpoints::connectome::post_batch_neuron_operations,
262 crate::endpoints::connectome::post_batch_synapse_operations,
263 crate::endpoints::connectome::get_neuron_count,
264 crate::endpoints::connectome::get_synapse_count,
265 crate::endpoints::connectome::get_paths,
266 crate::endpoints::connectome::get_cumulative_stats,
267 crate::endpoints::connectome::get_area_details,
268 crate::endpoints::connectome::post_rebuild,
269 crate::endpoints::connectome::get_structure,
270 crate::endpoints::connectome::post_clear,
271 crate::endpoints::connectome::get_validation,
272 crate::endpoints::connectome::get_topology,
273 crate::endpoints::connectome::post_optimize,
274 crate::endpoints::connectome::get_connectivity_matrix,
275 crate::endpoints::connectome::post_neurons_batch,
276 crate::endpoints::connectome::post_synapses_batch,
277 crate::endpoints::connectome::get_cortical_areas_list_summary,
278 crate::endpoints::connectome::get_cortical_areas_list_transforming,
279 crate::endpoints::connectome::get_cortical_area_list_types,
280 crate::endpoints::connectome::get_cortical_area_neurons,
281 crate::endpoints::connectome::get_area_synapses,
282 crate::endpoints::connectome::get_cortical_info,
283 crate::endpoints::connectome::get_stats_cortical_cumulative,
284 crate::endpoints::connectome::get_neuron_properties_by_id,
285 crate::endpoints::connectome::get_neuron_properties_query,
286 crate::endpoints::connectome::get_neuron_properties_at_query,
287 crate::endpoints::connectome::get_area_neurons_query,
288 crate::endpoints::connectome::get_fire_queue_area,
289 crate::endpoints::connectome::get_plasticity_info,
290 crate::endpoints::connectome::get_path_query,
291 crate::endpoints::connectome::get_download_connectome,
292 crate::endpoints::connectome::get_download_cortical_area,
293 crate::endpoints::connectome::post_upload_connectome,
294 crate::endpoints::connectome::post_upload_cortical_area,
295
296 crate::endpoints::burst_engine::get_simulation_timestep,
298 crate::endpoints::burst_engine::post_simulation_timestep,
299 crate::endpoints::burst_engine::get_fcl,
300 crate::endpoints::burst_engine::get_fcl_neuron,
301 crate::endpoints::burst_engine::get_fire_queue,
302 crate::endpoints::burst_engine::get_fire_queue_neuron,
303 crate::endpoints::burst_engine::post_fcl_reset,
304 crate::endpoints::burst_engine::get_fcl_status,
305 crate::endpoints::burst_engine::get_fcl_sampler_config,
306 crate::endpoints::burst_engine::post_fcl_sampler_config,
307 crate::endpoints::burst_engine::get_area_fcl_sample_rate,
308 crate::endpoints::burst_engine::post_area_fcl_sample_rate,
309 crate::endpoints::burst_engine::get_fire_ledger_default_window_size,
310 crate::endpoints::burst_engine::put_fire_ledger_default_window_size,
311 crate::endpoints::burst_engine::get_fire_ledger_areas_window_config,
312 crate::endpoints::burst_engine::get_stats,
313 crate::endpoints::burst_engine::get_status,
314 crate::endpoints::burst_engine::post_control,
315 crate::endpoints::burst_engine::get_burst_counter,
316 crate::endpoints::burst_engine::post_start,
317 crate::endpoints::burst_engine::post_stop,
318 crate::endpoints::burst_engine::post_hold,
319 crate::endpoints::burst_engine::post_resume,
320 crate::endpoints::burst_engine::get_config,
321 crate::endpoints::burst_engine::put_config,
322 crate::endpoints::burst_engine::get_fire_ledger_area_window_size,
323 crate::endpoints::burst_engine::put_fire_ledger_area_window_size,
324 crate::endpoints::burst_engine::get_fire_ledger_history,
325 crate::endpoints::burst_engine::get_membrane_potentials,
326 crate::endpoints::burst_engine::put_membrane_potentials,
327 crate::endpoints::burst_engine::get_frequency_status,
328 crate::endpoints::burst_engine::post_measure_frequency,
329 crate::endpoints::burst_engine::get_frequency_history,
330 crate::endpoints::burst_engine::post_force_connectome_integration,
331
332 crate::endpoints::insight::post_neurons_membrane_potential_status,
334 crate::endpoints::insight::post_neuron_synaptic_potential_status,
335 crate::endpoints::insight::post_neurons_membrane_potential_set,
336 crate::endpoints::insight::post_neuron_synaptic_potential_set,
337
338 crate::endpoints::neuroplasticity::get_plasticity_queue_depth,
340 crate::endpoints::neuroplasticity::put_plasticity_queue_depth,
341 crate::endpoints::neuroplasticity::get_status,
342 crate::endpoints::neuroplasticity::get_transforming,
343 crate::endpoints::neuroplasticity::post_configure,
344 crate::endpoints::neuroplasticity::post_enable_area,
345 crate::endpoints::neuroplasticity::post_disable_area,
346
347 crate::endpoints::input::get_vision,
349 crate::endpoints::input::post_vision,
350
351 crate::endpoints::outputs::get_targets,
353 crate::endpoints::outputs::post_configure,
354
355 crate::endpoints::physiology::get_physiology,
357 crate::endpoints::physiology::put_physiology,
358
359 crate::endpoints::simulation::post_stimulation_upload,
361 crate::endpoints::simulation::post_reset,
362 crate::endpoints::simulation::get_status,
363 crate::endpoints::simulation::get_stats,
364 crate::endpoints::simulation::post_config,
365
366 crate::endpoints::training::post_shock,
368 crate::endpoints::training::get_shock_options,
369 crate::endpoints::training::get_shock_status,
370 crate::endpoints::training::post_reward_intensity,
371 crate::endpoints::training::post_punishment_intensity,
372 crate::endpoints::training::post_gameover,
373 crate::endpoints::training::get_brain_fitness,
374 crate::endpoints::training::get_fitness_criteria,
375 crate::endpoints::training::put_fitness_criteria,
376 crate::endpoints::training::get_fitness_stats,
377 crate::endpoints::training::get_training_report,
378 crate::endpoints::training::get_status,
379 crate::endpoints::training::get_stats,
380 crate::endpoints::training::post_config,
381
382 crate::endpoints::visualization::post_register_client,
384 crate::endpoints::visualization::post_unregister_client,
385 crate::endpoints::visualization::post_heartbeat,
386 crate::endpoints::visualization::get_status,
387
388 crate::endpoints::monitoring::get_status,
390 crate::endpoints::monitoring::get_metrics,
391 crate::endpoints::monitoring::get_data,
392
393 crate::endpoints::evolution::get_status,
395 crate::endpoints::evolution::post_config,
396
397 crate::endpoints::network::get_status,
409 crate::endpoints::network::post_config,
410 ),
411 components(
412 schemas(
413 ApiError,
415
416 HealthCheckResponseV1,
418 ReadinessCheckResponseV1,
419 ComponentReadiness,
420
421 crate::v1::AgentRegistrationRequest,
423
424 crate::v1::OutputTargetsResponse,
426 crate::v1::OutputConfigRequest,
427 crate::v1::OutputConfigResponse,
428
429 crate::v1::PhysiologyResponse,
431 crate::v1::PhysiologyParameters,
432 crate::v1::PhysiologyUpdateRequest,
433 crate::v1::PhysiologyUpdateResponse,
434
435 crate::v1::FCLResponse,
437 crate::v1::FireQueueResponse,
438 crate::v1::FCLStatusResponse,
439 crate::v1::FireLedgerConfigResponse,
440 crate::v1::BurstEngineStats,
441 crate::v1::BurstEngineStatus,
442 crate::v1::BurstEngineControlRequest,
443
444 crate::v1::MonitoringStatusResponse,
446 crate::v1::SystemMetricsResponse,
447 crate::v1::MonitoringData,
448 crate::v1::MonitoringDataResponse,
449
450 crate::v1::StimulationUploadRequest,
452 crate::v1::SimulationControlRequest,
453 crate::v1::SimulationStatusResponse,
454 crate::v1::SimulationStatsResponse,
455 crate::v1::SimulationSuccessResponse,
456
457 crate::v1::ShockConfigRequest,
459 crate::v1::ShockOptionsResponse,
460 crate::v1::ShockStatusResponse,
461 crate::v1::IntensityRequest,
462 crate::v1::BrainFitnessResponse,
463 crate::v1::FitnessCriteriaResponse,
464 crate::v1::FitnessCriteriaUpdateRequest,
465 crate::v1::FitnessStatsResponse,
466 crate::v1::TrainingReportResponse,
467 crate::v1::TrainingStatusResponse,
468 crate::v1::TrainingStatsResponse,
469 crate::v1::TrainingConfigRequest,
470 crate::v1::TrainingSuccessResponse,
471
472 crate::v1::VisualizationClientRequest,
474 crate::v1::VisualizationClientResponse,
475 crate::v1::VisualizationHeartbeatRequest,
476 crate::v1::VisualizationStatusResponse,
477 crate::v1::VisualizationSuccessResponse,
478
479 crate::v1::EvolutionStatusResponse,
481 crate::v1::EvolutionConfigRequest,
482 crate::v1::EvolutionSuccessResponse,
483
484 crate::v1::SnapshotCreateRequest,
486 crate::v1::SnapshotCreateResponse,
487 crate::v1::SnapshotRestoreRequest,
488 crate::v1::SnapshotListResponse,
489 crate::v1::SnapshotInfo,
490 crate::v1::SnapshotArtifactResponse,
491 crate::v1::SnapshotCompareRequest,
492 crate::v1::SnapshotCompareResponse,
493 crate::v1::SnapshotUploadRequest,
494 crate::v1::SnapshotUploadResponse,
495 crate::v1::SnapshotSuccessResponse,
496
497 crate::v1::NetworkStatusResponse,
499 crate::v1::NetworkConfigRequest,
500 crate::v1::NetworkSuccessResponse,
501 crate::v1::AgentRegistrationResponse,
502 crate::v1::HeartbeatRequest,
503 crate::v1::HeartbeatResponse,
504 crate::v1::AgentListResponse,
505 crate::v1::AgentPropertiesResponse,
506 crate::v1::AgentDeregistrationRequest,
507 crate::v1::SuccessResponse,
508 crate::v1::ManualStimulationRequest,
509 crate::v1::ManualStimulationResponse,
510 crate::v1::DeviceRegistrationExportResponse,
511 crate::v1::DeviceRegistrationImportRequest,
512 crate::v1::DeviceRegistrationImportResponse,
513
514 crate::endpoints::system::HealthCheckResponse,
516
517 crate::endpoints::cortical_area::CorticalAreaIdListResponse,
519 crate::endpoints::cortical_area::CorticalAreaNameListResponse,
520 crate::endpoints::cortical_area::CorticalTypeMetadata,
521
522 crate::endpoints::morphology::MorphologyListResponse,
524
525 CorticalAreaSummary,
527 CorticalAreaDetail,
528 CorticalAreaListResponse,
529 CreateCorticalAreaRequest,
530 UpdateCorticalAreaRequest,
531 Coordinates3D,
532 Dimensions3D,
533
534 BrainRegionSummary,
536 BrainRegionDetail,
537 BrainRegionListResponse,
538 CreateBrainRegionRequest,
539
540 GenomeInfoResponse,
542 LoadGenomeRequest,
543 SaveGenomeRequest,
544 SaveGenomeResponse,
545 ValidateGenomeRequest,
546 ValidateGenomeResponse,
547 crate::endpoints::genome::GenomeFileUploadForm,
548
549 NeuronInfoResponse,
551 CreateNeuronRequest,
552 NeuronListResponse,
553 NeuronCountResponse,
554
555 RuntimeStatusResponse,
557 SetFrequencyRequest,
558 BurstCountResponse,
559
560 SystemHealthResponse,
562 CorticalAreaStatsResponse,
563 ConnectivityStatsResponse,
564 ConnectomeAnalyticsResponse,
565 PopulatedAreasResponse,
566 PopulatedAreaInfo,
567 NeuronDensityResponse,
568 )
569 ),
570 tags(
571 (name = "agent", description = "Agent registration and heartbeat"),
572 (name = "system", description = "System health and configuration"),
573 (name = "cortical_area", description = "Cortical area management"),
574 (name = "morphology", description = "Morphology management"),
575 (name = "genome", description = "Genome operations"),
576 (name = "cortical_mapping", description = "Cortical mapping operations"),
577 (name = "region", description = "Brain region management"),
578 (name = "connectome", description = "Connectome operations"),
579 (name = "burst_engine", description = "Burst engine configuration and FCL/Fire Queue"),
580 (name = "insight", description = "Neuron insight operations"),
581 (name = "neuroplasticity", description = "Neuroplasticity configuration"),
582 (name = "input", description = "Input operations"),
583 (name = "outputs", description = "Output/motor target management"),
584 (name = "physiology", description = "Physiology parameter configuration"),
585 (name = "simulation", description = "Simulation control and stimulation"),
586 (name = "training", description = "Reinforcement learning and training"),
587 (name = "visualization", description = "Visualization client management"),
588 (name = "monitoring", description = "System monitoring and metrics"),
589 (name = "evolution", description = "Evolutionary algorithms"),
590 (name = "snapshot", description = "Brain snapshot management"),
591 (name = "network", description = "Network configuration"),
592 ),
593 modifiers(&SecurityAddon)
594)]
595pub struct ApiDoc;
596
597struct SecurityAddon;
599
600impl utoipa::Modify for SecurityAddon {
601 fn modify(&self, openapi: &mut utoipa::openapi::OpenApi) {
602 if let Some(components) = openapi.components.as_mut() {
603 components.add_security_scheme(
605 "api_key",
606 SecurityScheme::ApiKey(ApiKey::Header(ApiKeyValue::new("X-API-Key"))),
607 );
608
609 components.add_security_scheme(
611 "bearer_auth",
612 SecurityScheme::Http(
613 utoipa::openapi::security::HttpBuilder::new()
614 .scheme(utoipa::openapi::security::HttpAuthScheme::Bearer)
615 .bearer_format("JWT")
616 .build(),
617 ),
618 );
619 }
620 }
621}
622
623pub fn get_openapi_json() -> String {
625 ApiDoc::openapi()
626 .to_pretty_json()
627 .unwrap_or_else(|e| format!(r#"{{"error": "Failed to generate OpenAPI spec: {}"}}"#, e))
628}
629
630pub fn get_openapi_yaml() -> String {
632 "# OpenAPI YAML generation not yet implemented\n# Use /openapi.json instead".to_string()
635}
636
637#[cfg(test)]
638mod tests {
639 use super::*;
640
641 #[test]
642 fn test_openapi_generation() {
643 let json = get_openapi_json();
644 assert!(json.contains("FEAGI REST API"));
645 assert!(json.contains("health"));
646 assert!(json.contains("ready"));
647 }
648
649 #[test]
650 fn test_openapi_components() {
651 let openapi = ApiDoc::openapi();
652 assert!(openapi.components.is_some());
653
654 let components = openapi.components.unwrap();
655 assert!(components.schemas.contains_key("HealthCheckResponseV1"));
656 assert!(components.schemas.contains_key("ApiError"));
657 }
658
659 #[test]
660 fn test_openapi_amalgamation_by_upload_is_multipart_form() {
661 let openapi = ApiDoc::openapi();
662 let json = serde_json::to_value(&openapi).expect("serialize openapi");
663
664 let content =
665 &json["paths"]["/v1/genome/amalgamation_by_upload"]["post"]["requestBody"]["content"];
666
667 assert!(
668 content.get("multipart/form-data").is_some(),
669 "Expected multipart/form-data content for /v1/genome/amalgamation_by_upload requestBody. Got: {}",
670 content
671 );
672 }
673
674 #[test]
675 fn test_security_schemes() {
676 let openapi = ApiDoc::openapi();
677 let components = openapi.components.unwrap();
678
679 assert!(components.security_schemes.contains_key("api_key"));
680 assert!(components.security_schemes.contains_key("bearer_auth"));
681 }
682}