feat(image-processor): add upload info to response

image-processor/proto/scuffle/image_processor/service.proto

@@ -36,13 +36,24 @@ message ProcessImageResponse {
   // A unique identifier for the task
   string id = 1;
 
-  // If the task had an input upload, this will be the path to the uploaded image.
-  optional DrivePath upload_path = 2;
+  // If the task had an input upload, this will be the info of the uploaded image.
+  optional ProcessImageResponseUploadInfo upload_info = 2;
 
   // Errors that occurred when creating the task.
   optional Error error = 3;
 }
 
+message ProcessImageResponseUploadInfo {
+  // The path of the uploaded image
+  DrivePath path = 1;
+
+  // The content type of the uploaded image
+  string content_type = 2;
+
+  // The size of the uploaded image in bytes
+  uint64 size = 3;
+}
+
 // The Payload for a ImageProcessor.CancelTask request
 message CancelTaskRequest {
   // The unique identifier of the task to cancel

image-processor/src/management/grpc.rs

@@ -22,7 +22,7 @@ impl scuffle_image_processor_proto::image_processor_server::ImageProcessor for M
 			Ok(resp) => resp,
 			Err(err) => ProcessImageResponse {
 				id: "".to_owned(),
-				upload_path: None,
+				upload_info: None,
 				error: Some(err),
 			},
 		};

image-processor/src/management/http.rs

@@ -36,7 +36,7 @@ async fn process_image(
 		Ok(resp) => resp,
 		Err(err) => ProcessImageResponse {
 			id: "".to_owned(),
-			upload_path: None,
+			upload_info: None,
 			error: Some(err),
 		},
 	};

image-processor/src/management/mod.rs

@@ -5,7 +5,7 @@ use bson::oid::ObjectId;
 use bytes::Bytes;
 use scuffle_image_processor_proto::{
 	input, CancelTaskRequest, CancelTaskResponse, DrivePath, Error, ErrorCode, Input, ProcessImageRequest,
-	ProcessImageResponse,
+	ProcessImageResponse, ProcessImageResponseUploadInfo,
 };
 
 use crate::database::Job;
@@ -42,7 +42,7 @@ impl ManagementServer {
 
 		let id = ObjectId::new();
 
-		let upload_path = if let Some(input_upload) = request.input_upload {
+		let upload_info = if let Some(input_upload) = request.input_upload {
 			let drive_path = input_upload.drive_path.unwrap();
 			let drive = self.global.drive(&drive_path.drive).unwrap();
 
@@ -79,6 +79,8 @@ impl ManagementServer {
 				});
 			}
 
+			let upload_size = input_upload.binary.len() as u64;
+
 			drive
 				.write(
 					&path,
@@ -99,7 +101,11 @@ impl ManagementServer {
 					}
 				})?;
 
-			Some(drive_path)
+			Some(ProcessImageResponseUploadInfo {
+				path: Some(drive_path),
+				content_type: file_format.media_type().to_owned(),
+				size: upload_size,
+			})
 		} else {
 			None
 		};
@@ -116,7 +122,7 @@ impl ManagementServer {
 
 		Ok(ProcessImageResponse {
 			id: job.id.to_string(),
-			upload_path,
+			upload_info,
 			error: None,
 		})
 	}