Merge pull request #27 from mbits-imaging/mwwl-proposal

feat(mwl): Implement MWL-RS proposal

Author: nickamzol

src/api/mod.rs

@@ -3,6 +3,7 @@ use axum::Router;
 
 mod aets;
 mod home;
+pub mod mwl;
 pub mod qido;
 pub mod stow;
 pub mod wado;
@@ -16,7 +17,8 @@ pub fn routes(base_path: &str) -> Router<AppState> {
 			Router::new()
 				.merge(qido::routes())
 				.merge(wado::routes())
-				.merge(stow::routes()),
+				.merge(stow::routes())
+				.merge(mwl::routes()),
 		);
 
 	// axum no longer supports nesting at the root

src/api/mwl/mod.rs

@@ -0,0 +1,39 @@
+mod routes;
+mod service;
+
+pub use routes::routes;
+pub use service::*;
+
+use dicom::core::Tag;
+use dicom::dictionary_std::tags;
+
+/// <https://dicom.nema.org/medical/dicom/current/output/chtml/part04/sect_K.6.html#table_K.6-1>
+pub const WORKITEM_SEARCH_TAGS: &[Tag] = &[
+	// Scheduled Procedure Step
+	tags::SCHEDULED_PROCEDURE_STEP_SEQUENCE,
+	tags::SCHEDULED_STATION_AE_TITLE,
+	tags::SCHEDULED_PROCEDURE_STEP_START_DATE,
+	tags::SCHEDULED_PROCEDURE_STEP_START_TIME,
+	tags::MODALITY,
+	tags::SCHEDULED_PERFORMING_PHYSICIAN_NAME,
+	tags::SCHEDULED_PROCEDURE_STEP_DESCRIPTION,
+	tags::SCHEDULED_STATION_NAME,
+	tags::SCHEDULED_PROCEDURE_STEP_LOCATION,
+	tags::REFERENCED_DEFINED_PROTOCOL_SEQUENCE,
+	tags::REFERENCED_SOP_CLASS_UID,
+	tags::REFERENCED_SOP_INSTANCE_UID,
+	// Requested Procedure
+	tags::REQUESTED_PROCEDURE_ID,
+	tags::REQUESTED_PROCEDURE_DESCRIPTION,
+	tags::REQUESTED_PROCEDURE_CODE_SEQUENCE,
+	tags::STUDY_INSTANCE_UID,
+	tags::STUDY_DATE,
+	tags::STUDY_TIME,
+	// Patient Identification
+	tags::PATIENT_NAME,
+	tags::PATIENT_ID,
+	tags::ISSUER_OF_PATIENT_ID,
+	// Patient Demographics
+	tags::PATIENT_BIRTH_DATE,
+	tags::PATIENT_SEX,
+];

src/api/mwl/routes.rs

@@ -0,0 +1,70 @@
+use crate::backend::ServiceProvider;
+use crate::AppState;
+use axum::http::header;
+use axum::http::StatusCode;
+use axum::response::IntoResponse;
+use axum::routing::get;
+use axum::Router;
+use axum_extra::extract::Query;
+use axum_streams::StreamBodyAs;
+use dicom::object::InMemDicomObject;
+use dicom_json::DicomJson;
+use futures::TryStreamExt;
+use tracing::instrument;
+
+use super::{MwlQueryParameters, MwlRequestHeaderFields, MwlSearchError, MwlSearchRequest};
+
+/// HTTP Router for the Modality Worklist.
+///
+/// <https://www.dicomstandard.org/news-dir/current/docs/sups/sup246.pdf>
+#[rustfmt::skip]
+pub fn routes() -> Router<AppState> {
+    Router::new()
+        .route("/modality-scheduled-procedure-steps", get(all_workitems))
+}
+
+// MWL-RS implementation
+async fn mwl_handler(provider: ServiceProvider, request: MwlSearchRequest) -> impl IntoResponse {
+	if let Some(mwl) = provider.mwl {
+		let response = mwl.search(request).await;
+		let matches: Result<Vec<InMemDicomObject>, MwlSearchError> =
+			response.stream.try_collect().await;
+
+		match matches {
+			Ok(matches) => {
+				if matches.is_empty() {
+					StatusCode::NO_CONTENT.into_response()
+				} else {
+					let json: Vec<DicomJson<InMemDicomObject>> =
+						matches.into_iter().map(DicomJson::from).collect();
+
+					axum::response::Response::builder()
+						.status(StatusCode::OK)
+						.header(header::CONTENT_TYPE, mime::APPLICATION_JSON.as_ref())
+						.body(StreamBodyAs::json_array(futures::stream::iter(json)))
+						.unwrap()
+						.into_response()
+				}
+			}
+			Err(err) => (StatusCode::INTERNAL_SERVER_ERROR, err.to_string()).into_response(),
+		}
+	} else {
+		(
+			StatusCode::SERVICE_UNAVAILABLE,
+			"MWL-RS endpoint is disabled",
+		)
+			.into_response()
+	}
+}
+
+#[instrument(skip_all)]
+async fn all_workitems(
+	provider: ServiceProvider,
+	Query(parameters): Query<MwlQueryParameters>,
+) -> impl IntoResponse {
+	let request = MwlSearchRequest {
+		parameters,
+		headers: MwlRequestHeaderFields::default(),
+	};
+	mwl_handler(provider, request).await
+}