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