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::heartbeat,
88 crate::endpoints::agent::list_agents,
89 crate::endpoints::agent::get_agent_properties,
90 crate::endpoints::agent::get_agent_properties_path,
91 crate::endpoints::agent::get_shared_memory,
92 crate::endpoints::agent::deregister_agent,
93 crate::endpoints::agent::manual_stimulation,
94 crate::endpoints::agent::get_fq_sampler_status,
95 crate::endpoints::agent::get_capabilities,
96 crate::endpoints::agent::get_all_agent_capabilities,
97 crate::endpoints::agent::get_agent_info,
98 crate::endpoints::agent::post_configure,
99
100 crate::endpoints::system::get_health_check,
102 crate::endpoints::system::get_cortical_area_visualization_skip_rate,
103 crate::endpoints::system::set_cortical_area_visualization_skip_rate,
104 crate::endpoints::system::get_cortical_area_visualization_suppression_threshold,
105 crate::endpoints::system::set_cortical_area_visualization_suppression_threshold,
106 crate::endpoints::system::get_version,
107 crate::endpoints::system::get_versions,
108 crate::endpoints::system::get_configuration,
109 crate::endpoints::system::get_user_preferences,
110 crate::endpoints::system::put_user_preferences,
111 crate::endpoints::system::get_cortical_area_types_list,
112 crate::endpoints::system::post_enable_visualization_fq_sampler,
113 crate::endpoints::system::post_disable_visualization_fq_sampler,
114 crate::endpoints::system::get_fcl_status_system,
115 crate::endpoints::system::post_fcl_reset_system,
116 crate::endpoints::system::get_processes,
117 crate::endpoints::system::get_unique_logs,
118 crate::endpoints::system::post_logs,
119 crate::endpoints::system::get_beacon_subscribers,
120 crate::endpoints::system::post_beacon_subscribe,
121 crate::endpoints::system::delete_beacon_unsubscribe,
122 crate::endpoints::system::get_global_activity_visualization,
123 crate::endpoints::system::put_global_activity_visualization,
124 crate::endpoints::system::post_circuit_library_path,
125 crate::endpoints::system::get_influxdb_test,
126 crate::endpoints::system::post_register_system,
127
128 crate::endpoints::cortical_area::get_ipu,
130 crate::endpoints::cortical_area::get_ipu_types,
131 crate::endpoints::cortical_area::get_opu,
132 crate::endpoints::cortical_area::get_opu_types,
133 crate::endpoints::cortical_area::get_cortical_area_id_list,
134 crate::endpoints::cortical_area::get_cortical_area_name_list,
135 crate::endpoints::cortical_area::get_cortical_id_name_mapping,
136 crate::endpoints::cortical_area::get_cortical_types,
137 crate::endpoints::cortical_area::get_cortical_map_detailed,
138 crate::endpoints::cortical_area::get_cortical_locations_2d,
139 crate::endpoints::cortical_area::get_cortical_area_geometry,
140 crate::endpoints::cortical_area::get_cortical_visibility,
141 crate::endpoints::cortical_area::post_cortical_name_location,
142 crate::endpoints::cortical_area::post_cortical_area_properties,
143 crate::endpoints::cortical_area::post_multi_cortical_area_properties,
144 crate::endpoints::cortical_area::post_cortical_area,
145 crate::endpoints::cortical_area::put_cortical_area,
146 crate::endpoints::cortical_area::delete_cortical_area,
147 crate::endpoints::cortical_area::post_custom_cortical_area,
148 crate::endpoints::cortical_area::post_clone,
149 crate::endpoints::cortical_area::put_multi_cortical_area,
150 crate::endpoints::cortical_area::delete_multi_cortical_area,
151 crate::endpoints::cortical_area::put_coord_2d,
152 crate::endpoints::cortical_area::put_suppress_cortical_visibility,
153 crate::endpoints::cortical_area::put_reset,
154 crate::endpoints::cortical_area::get_visualization,
155 crate::endpoints::cortical_area::post_batch_operations,
156 crate::endpoints::cortical_area::get_ipu_list,
157 crate::endpoints::cortical_area::get_opu_list,
158 crate::endpoints::cortical_area::put_coordinates_3d,
159 crate::endpoints::cortical_area::delete_bulk,
160 crate::endpoints::cortical_area::post_clone,
161 crate::endpoints::cortical_area::post_resize,
162 crate::endpoints::cortical_area::post_reposition,
163 crate::endpoints::cortical_area::post_voxel_neurons,
164 crate::endpoints::cortical_area::get_cortical_area_index_list,
165 crate::endpoints::cortical_area::get_cortical_idx_mapping,
166 crate::endpoints::cortical_area::get_mapping_restrictions_query,
167 crate::endpoints::cortical_area::get_memory_usage,
168 crate::endpoints::cortical_area::get_area_neuron_count,
169 crate::endpoints::cortical_area::post_cortical_type_options,
170 crate::endpoints::cortical_area::post_mapping_restrictions,
171 crate::endpoints::cortical_area::post_mapping_restrictions_between_areas,
172 crate::endpoints::cortical_area::put_coord_3d,
173
174 crate::endpoints::morphology::get_morphology_list,
176 crate::endpoints::morphology::get_morphology_types,
177 crate::endpoints::morphology::get_list_types,
178 crate::endpoints::morphology::get_morphologies,
179 crate::endpoints::morphology::post_morphology,
180 crate::endpoints::morphology::put_morphology,
181 crate::endpoints::morphology::put_rename_morphology,
182 crate::endpoints::morphology::delete_morphology_by_name,
183 crate::endpoints::morphology::delete_morphology, crate::endpoints::morphology::post_morphology_properties,
185 crate::endpoints::morphology::post_morphology_usage,
186 crate::endpoints::morphology::get_list,
187 crate::endpoints::morphology::get_info,
188 crate::endpoints::morphology::post_create,
189 crate::endpoints::morphology::put_update,
190
191 crate::endpoints::genome::get_file_name,
193 crate::endpoints::genome::get_circuits,
194 crate::endpoints::genome::post_amalgamation_destination,
195 crate::endpoints::genome::delete_amalgamation_cancellation,
196 crate::endpoints::genome::post_genome_append,
197 crate::endpoints::genome::post_upload_barebones_genome,
198 crate::endpoints::genome::post_upload_essential_genome,
199 crate::endpoints::genome::get_name,
200 crate::endpoints::genome::get_timestamp,
201 crate::endpoints::genome::post_save,
202 crate::endpoints::genome::post_load,
203 crate::endpoints::genome::post_upload,
204 crate::endpoints::genome::get_download,
205 crate::endpoints::genome::get_properties,
206 crate::endpoints::genome::post_validate,
207 crate::endpoints::genome::post_transform,
208 crate::endpoints::genome::post_clone,
209 crate::endpoints::genome::post_reset,
210 crate::endpoints::genome::get_metadata,
211 crate::endpoints::genome::post_merge,
212 crate::endpoints::genome::get_diff,
213 crate::endpoints::genome::post_export_format,
214 crate::endpoints::genome::get_amalgamation,
215 crate::endpoints::genome::get_amalgamation_history_exact,
216 crate::endpoints::genome::get_cortical_template,
217 crate::endpoints::genome::get_defaults_files,
218 crate::endpoints::genome::get_download_region,
219 crate::endpoints::genome::get_genome_number,
220 crate::endpoints::genome::post_amalgamation_by_filename,
221 crate::endpoints::genome::post_amalgamation_by_payload,
222 crate::endpoints::genome::post_amalgamation_by_upload,
223 crate::endpoints::genome::post_append_file,
224 crate::endpoints::genome::post_upload_file,
225 crate::endpoints::genome::post_upload_file_edit,
226 crate::endpoints::genome::post_upload_string,
227
228 crate::endpoints::cortical_mapping::post_afferents,
230 crate::endpoints::cortical_mapping::post_efferents,
231 crate::endpoints::cortical_mapping::post_mapping_properties,
232 crate::endpoints::cortical_mapping::put_mapping_properties,
233 crate::endpoints::cortical_mapping::get_mapping,
234 crate::endpoints::cortical_mapping::get_mapping_list,
235 crate::endpoints::cortical_mapping::delete_mapping,
236 crate::endpoints::cortical_mapping::post_batch_update,
237 crate::endpoints::cortical_mapping::post_mapping,
238 crate::endpoints::cortical_mapping::put_mapping,
239
240 crate::endpoints::region::get_regions_members,
242 crate::endpoints::region::post_region,
243 crate::endpoints::region::put_region,
244 crate::endpoints::region::delete_region,
245 crate::endpoints::region::post_clone,
246 crate::endpoints::region::put_relocate_members,
247 crate::endpoints::region::delete_region_and_members,
248 crate::endpoints::region::get_regions,
249 crate::endpoints::region::get_region_titles,
250 crate::endpoints::region::get_region_detail,
251 crate::endpoints::region::put_change_region_parent,
252 crate::endpoints::region::put_change_cortical_area_region,
253
254 crate::endpoints::connectome::get_cortical_areas_list_detailed,
256 crate::endpoints::connectome::get_properties_dimensions,
257 crate::endpoints::connectome::get_properties_mappings,
258 crate::endpoints::connectome::get_snapshot,
259 crate::endpoints::connectome::get_stats,
260 crate::endpoints::connectome::post_batch_neuron_operations,
261 crate::endpoints::connectome::post_batch_synapse_operations,
262 crate::endpoints::connectome::get_neuron_count,
263 crate::endpoints::connectome::get_synapse_count,
264 crate::endpoints::connectome::get_paths,
265 crate::endpoints::connectome::get_cumulative_stats,
266 crate::endpoints::connectome::get_area_details,
267 crate::endpoints::connectome::post_rebuild,
268 crate::endpoints::connectome::get_structure,
269 crate::endpoints::connectome::post_clear,
270 crate::endpoints::connectome::get_validation,
271 crate::endpoints::connectome::get_topology,
272 crate::endpoints::connectome::post_optimize,
273 crate::endpoints::connectome::get_connectivity_matrix,
274 crate::endpoints::connectome::post_neurons_batch,
275 crate::endpoints::connectome::post_synapses_batch,
276 crate::endpoints::connectome::get_cortical_areas_list_summary,
277 crate::endpoints::connectome::get_cortical_areas_list_transforming,
278 crate::endpoints::connectome::get_cortical_area_list_types,
279 crate::endpoints::connectome::get_cortical_area_neurons,
280 crate::endpoints::connectome::get_area_synapses,
281 crate::endpoints::connectome::get_cortical_info,
282 crate::endpoints::connectome::get_stats_cortical_cumulative,
283 crate::endpoints::connectome::get_neuron_properties_by_id,
284 crate::endpoints::connectome::get_neuron_properties_query,
285 crate::endpoints::connectome::get_neuron_properties_at_query,
286 crate::endpoints::connectome::get_area_neurons_query,
287 crate::endpoints::connectome::get_fire_queue_area,
288 crate::endpoints::connectome::get_plasticity_info,
289 crate::endpoints::connectome::get_path_query,
290 crate::endpoints::connectome::get_download_connectome,
291 crate::endpoints::connectome::get_download_cortical_area,
292 crate::endpoints::connectome::post_upload_connectome,
293 crate::endpoints::connectome::post_upload_cortical_area,
294
295 crate::endpoints::burst_engine::get_simulation_timestep,
297 crate::endpoints::burst_engine::post_simulation_timestep,
298 crate::endpoints::burst_engine::get_fcl,
299 crate::endpoints::burst_engine::get_fcl_neuron,
300 crate::endpoints::burst_engine::get_fire_queue,
301 crate::endpoints::burst_engine::get_fire_queue_neuron,
302 crate::endpoints::burst_engine::post_fcl_reset,
303 crate::endpoints::burst_engine::get_fcl_status,
304 crate::endpoints::burst_engine::get_fcl_sampler_config,
305 crate::endpoints::burst_engine::post_fcl_sampler_config,
306 crate::endpoints::burst_engine::get_area_fcl_sample_rate,
307 crate::endpoints::burst_engine::post_area_fcl_sample_rate,
308 crate::endpoints::burst_engine::get_fire_ledger_default_window_size,
309 crate::endpoints::burst_engine::put_fire_ledger_default_window_size,
310 crate::endpoints::burst_engine::get_fire_ledger_areas_window_config,
311 crate::endpoints::burst_engine::get_stats,
312 crate::endpoints::burst_engine::get_status,
313 crate::endpoints::burst_engine::post_control,
314 crate::endpoints::burst_engine::get_burst_counter,
315 crate::endpoints::burst_engine::post_start,
316 crate::endpoints::burst_engine::post_stop,
317 crate::endpoints::burst_engine::post_hold,
318 crate::endpoints::burst_engine::post_resume,
319 crate::endpoints::burst_engine::get_config,
320 crate::endpoints::burst_engine::put_config,
321 crate::endpoints::burst_engine::get_fire_ledger_area_window_size,
322 crate::endpoints::burst_engine::put_fire_ledger_area_window_size,
323 crate::endpoints::burst_engine::get_fire_ledger_history,
324 crate::endpoints::burst_engine::get_membrane_potentials,
325 crate::endpoints::burst_engine::put_membrane_potentials,
326 crate::endpoints::burst_engine::get_frequency_status,
327 crate::endpoints::burst_engine::post_measure_frequency,
328 crate::endpoints::burst_engine::get_frequency_history,
329 crate::endpoints::burst_engine::post_force_connectome_integration,
330
331 crate::endpoints::insight::post_neurons_membrane_potential_status,
333 crate::endpoints::insight::post_neuron_synaptic_potential_status,
334 crate::endpoints::insight::post_neurons_membrane_potential_set,
335 crate::endpoints::insight::post_neuron_synaptic_potential_set,
336
337 crate::endpoints::neuroplasticity::get_plasticity_queue_depth,
339 crate::endpoints::neuroplasticity::put_plasticity_queue_depth,
340 crate::endpoints::neuroplasticity::get_status,
341 crate::endpoints::neuroplasticity::get_transforming,
342 crate::endpoints::neuroplasticity::post_configure,
343 crate::endpoints::neuroplasticity::post_enable_area,
344 crate::endpoints::neuroplasticity::post_disable_area,
345
346 crate::endpoints::input::get_vision,
348 crate::endpoints::input::post_vision,
349
350 crate::endpoints::outputs::get_targets,
352 crate::endpoints::outputs::post_configure,
353
354 crate::endpoints::physiology::get_physiology,
356 crate::endpoints::physiology::put_physiology,
357
358 crate::endpoints::simulation::post_stimulation_upload,
360 crate::endpoints::simulation::post_reset,
361 crate::endpoints::simulation::get_status,
362 crate::endpoints::simulation::get_stats,
363 crate::endpoints::simulation::post_config,
364
365 crate::endpoints::training::post_shock,
367 crate::endpoints::training::get_shock_options,
368 crate::endpoints::training::get_shock_status,
369 crate::endpoints::training::post_reward_intensity,
370 crate::endpoints::training::post_punishment_intensity,
371 crate::endpoints::training::post_gameover,
372 crate::endpoints::training::get_brain_fitness,
373 crate::endpoints::training::get_fitness_criteria,
374 crate::endpoints::training::put_fitness_criteria,
375 crate::endpoints::training::get_fitness_stats,
376 crate::endpoints::training::get_training_report,
377 crate::endpoints::training::get_status,
378 crate::endpoints::training::get_stats,
379 crate::endpoints::training::post_config,
380
381 crate::endpoints::visualization::post_register_client,
383 crate::endpoints::visualization::post_unregister_client,
384 crate::endpoints::visualization::post_heartbeat,
385 crate::endpoints::visualization::get_status,
386
387 crate::endpoints::monitoring::get_status,
389 crate::endpoints::monitoring::get_metrics,
390 crate::endpoints::monitoring::get_data,
391
392 crate::endpoints::evolution::get_status,
394 crate::endpoints::evolution::post_config,
395
396 crate::endpoints::network::get_status,
408 crate::endpoints::network::post_config,
409 crate::endpoints::network::get_connection_info,
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::NetworkConnectionInfo,
500 crate::v1::ConnectionInfoApi,
501 crate::v1::ConnectionInfoZmq,
502 crate::v1::ConnectionInfoZmqPorts,
503 crate::v1::ConnectionInfoZmqEndpoints,
504 crate::v1::ConnectionInfoWebSocket,
505 crate::v1::ConnectionInfoWebSocketPorts,
506 crate::v1::ConnectionInfoWebSocketEndpoints,
507 crate::v1::ConnectionInfoShm,
508 crate::v1::ConnectionInfoUdp,
509 crate::v1::ConnectionInfoBluetooth,
510 crate::v1::ConnectionInfoStreamStatus,
511 crate::v1::NetworkConfigRequest,
512 crate::v1::NetworkSuccessResponse,
513 crate::v1::AgentRegistrationResponse,
514 crate::v1::HeartbeatRequest,
515 crate::v1::HeartbeatResponse,
516 crate::v1::AgentListResponse,
517 crate::v1::AgentPropertiesResponse,
518 crate::v1::AgentCapabilitiesSummary,
519 crate::v1::AgentDeregistrationRequest,
520 crate::v1::SuccessResponse,
521 crate::v1::ManualStimulationRequest,
522 crate::v1::ManualStimulationResponse,
523 crate::v1::DeviceRegistrationExportResponse,
524 crate::v1::DeviceRegistrationImportRequest,
525 crate::v1::DeviceRegistrationImportResponse,
526
527 crate::endpoints::system::HealthCheckResponse,
529
530 crate::endpoints::cortical_area::CorticalAreaIdListResponse,
532 crate::endpoints::cortical_area::CorticalAreaNameListResponse,
533 crate::endpoints::cortical_area::CorticalTypeMetadata,
534
535 crate::endpoints::morphology::MorphologyListResponse,
537 crate::endpoints::morphology::RenameMorphologyRequest,
538
539 CorticalAreaSummary,
541 CorticalAreaDetail,
542 CorticalAreaListResponse,
543 CreateCorticalAreaRequest,
544 UpdateCorticalAreaRequest,
545 Coordinates3D,
546 Dimensions3D,
547
548 BrainRegionSummary,
550 BrainRegionDetail,
551 BrainRegionListResponse,
552 CreateBrainRegionRequest,
553
554 GenomeInfoResponse,
556 LoadGenomeRequest,
557 SaveGenomeRequest,
558 SaveGenomeResponse,
559 ValidateGenomeRequest,
560 ValidateGenomeResponse,
561 crate::endpoints::genome::GenomeFileUploadForm,
562
563 NeuronInfoResponse,
565 CreateNeuronRequest,
566 NeuronListResponse,
567 NeuronCountResponse,
568
569 RuntimeStatusResponse,
571 SetFrequencyRequest,
572 BurstCountResponse,
573
574 SystemHealthResponse,
576 CorticalAreaStatsResponse,
577 ConnectivityStatsResponse,
578 ConnectomeAnalyticsResponse,
579 PopulatedAreasResponse,
580 PopulatedAreaInfo,
581 NeuronDensityResponse,
582 )
583 ),
584 tags(
585 (name = "agent", description = "Agent registration and heartbeat"),
586 (name = "system", description = "System health and configuration"),
587 (name = "cortical_area", description = "Cortical area management"),
588 (name = "morphology", description = "Morphology management"),
589 (name = "genome", description = "Genome operations"),
590 (name = "cortical_mapping", description = "Cortical mapping operations"),
591 (name = "region", description = "Brain region management"),
592 (name = "connectome", description = "Connectome operations"),
593 (name = "burst_engine", description = "Burst engine configuration and FCL/Fire Queue"),
594 (name = "insight", description = "Neuron insight operations"),
595 (name = "neuroplasticity", description = "Neuroplasticity configuration"),
596 (name = "input", description = "Input operations"),
597 (name = "outputs", description = "Output/motor target management"),
598 (name = "physiology", description = "Physiology parameter configuration"),
599 (name = "simulation", description = "Simulation control and stimulation"),
600 (name = "training", description = "Reinforcement learning and training"),
601 (name = "visualization", description = "Visualization client management"),
602 (name = "monitoring", description = "System monitoring and metrics"),
603 (name = "evolution", description = "Evolutionary algorithms"),
604 (name = "snapshot", description = "Brain snapshot management"),
605 (name = "network", description = "Network configuration"),
606 ),
607 modifiers(&SecurityAddon)
608)]
609pub struct ApiDoc;
610
611struct SecurityAddon;
613
614impl utoipa::Modify for SecurityAddon {
615 fn modify(&self, openapi: &mut utoipa::openapi::OpenApi) {
616 if let Some(components) = openapi.components.as_mut() {
617 components.add_security_scheme(
619 "api_key",
620 SecurityScheme::ApiKey(ApiKey::Header(ApiKeyValue::new("X-API-Key"))),
621 );
622
623 components.add_security_scheme(
625 "bearer_auth",
626 SecurityScheme::Http(
627 utoipa::openapi::security::HttpBuilder::new()
628 .scheme(utoipa::openapi::security::HttpAuthScheme::Bearer)
629 .bearer_format("JWT")
630 .build(),
631 ),
632 );
633 }
634 }
635}
636
637pub fn get_openapi_json() -> String {
639 ApiDoc::openapi()
640 .to_pretty_json()
641 .unwrap_or_else(|e| format!(r#"{{"error": "Failed to generate OpenAPI spec: {}"}}"#, e))
642}
643
644pub fn get_openapi_yaml() -> String {
646 "# OpenAPI YAML generation not yet implemented\n# Use /openapi.json instead".to_string()
649}
650
651#[cfg(test)]
652mod tests {
653 use super::*;
654
655 #[test]
656 fn test_openapi_generation() {
657 let json = get_openapi_json();
658 assert!(json.contains("FEAGI REST API"));
659 assert!(json.contains("health"));
660 assert!(json.contains("ready"));
661 }
662
663 #[test]
664 fn test_openapi_components() {
665 let openapi = ApiDoc::openapi();
666 assert!(openapi.components.is_some());
667
668 let components = openapi.components.unwrap();
669 assert!(components.schemas.contains_key("HealthCheckResponseV1"));
670 assert!(components.schemas.contains_key("ApiError"));
671 }
672
673 #[test]
674 fn test_openapi_amalgamation_by_upload_is_multipart_form() {
675 let openapi = ApiDoc::openapi();
676 let json = serde_json::to_value(&openapi).expect("serialize openapi");
677
678 let content =
679 &json["paths"]["/v1/genome/amalgamation_by_upload"]["post"]["requestBody"]["content"];
680
681 assert!(
682 content.get("multipart/form-data").is_some(),
683 "Expected multipart/form-data content for /v1/genome/amalgamation_by_upload requestBody. Got: {}",
684 content
685 );
686 }
687
688 #[test]
689 fn test_security_schemes() {
690 let openapi = ApiDoc::openapi();
691 let components = openapi.components.unwrap();
692
693 assert!(components.security_schemes.contains_key("api_key"));
694 assert!(components.security_schemes.contains_key("bearer_auth"));
695 }
696}