Refactor grouping

crates/pile-dataset/src/serve/extract.rs

@@ -1,14 +1,19 @@
 use axum::{
 	Json,
+	body::Body,
 	extract::{Query, RawQuery, State},
 	http::{StatusCode, header},
 	response::{IntoResponse, Response},
 };
 use percent_encoding::percent_decode_str;
 use pile_config::{Label, objectpath::ObjectPath};
-use pile_value::{extract::traits::ExtractState, value::PileValue};
+use pile_value::{
+	extract::traits::ExtractState,
+	value::{BinaryPileValue, PileValue},
+};
 use serde::Deserialize;
 use std::{sync::Arc, time::Instant};
+use tokio_util::io::ReaderStream;
 use tracing::debug;
 use utoipa::ToSchema;
 
@@ -141,15 +146,30 @@ pub async fn get_extract(
 			s.to_string(),
 		)
 			.into_response(),
-		PileValue::Blob { mime, bytes } => (
-			StatusCode::OK,
-			[
-				(header::CONTENT_TYPE, mime.to_string()),
-				(header::CONTENT_DISPOSITION, disposition),
-			],
-			bytes.as_ref().clone(),
-		)
-			.into_response(),
+
+		PileValue::Binary(binary) => {
+			let mime = binary.mime().to_string();
+			let body = match binary {
+				BinaryPileValue::Blob { bytes, .. } => Body::from(bytes.0.to_vec()),
+				BinaryPileValue::File { path, .. } => match tokio::fs::File::open(&path).await {
+					Ok(file) => Body::from_stream(ReaderStream::new(file)),
+					Err(e) => {
+						return (StatusCode::INTERNAL_SERVER_ERROR, format!("{e:?}"))
+							.into_response();
+					}
+				},
+			};
+			(
+				StatusCode::OK,
+				[
+					(header::CONTENT_TYPE, mime),
+					(header::CONTENT_DISPOSITION, disposition),
+				],
+				body,
+			)
+				.into_response()
+		}
+
 		_ => match value.to_json(&extract_state).await {
 			Ok(json) => (
 				StatusCode::OK,