From 59fcd4336794034f7d3408d555312701c8baf11e Mon Sep 17 00:00:00 2001
From: LJ <linghua@cocoindex.io>
Date: Tue, 18 Mar 2025 23:33:24 -0700
Subject: [PATCH] Add a field `mime_type` for `GoogleDrive` source.

---
 src/base/value.rs               |  9 +++++++++
 src/ops/sources/google_drive.rs | 12 ++++++++----
 2 files changed, 17 insertions(+), 4 deletions(-)

diff --git a/src/base/value.rs b/src/base/value.rs
index c05973b1..06d8a467 100644
--- a/src/base/value.rs
+++ b/src/base/value.rs
@@ -395,6 +395,15 @@ impl From<FieldValues> for Value {
     }
 }
 
+impl<T: Into<Value>> From<Option<T>> for Value {
+    fn from(value: Option<T>) -> Self {
+        match value {
+            Some(v) => v.into(),
+            None => Value::Null,
+        }
+    }
+}
+
 impl<VS> Value<VS> {
     pub fn from_alternative<AltVS>(value: Value<AltVS>) -> Self
     where
diff --git a/src/ops/sources/google_drive.rs b/src/ops/sources/google_drive.rs
index 2cdf7372..f42ec318 100644
--- a/src/ops/sources/google_drive.rs
+++ b/src/ops/sources/google_drive.rs
@@ -196,7 +196,7 @@ impl SourceExecutor for Executor {
             .doit()
             .await?;
 
-        let resp_body = if let Some(export_mime_type) = file
+        let (mime_type, resp_body) = if let Some(export_mime_type) = file
             .mime_type
             .as_ref()
             .and_then(|mime_type| EXPORT_MIME_TYPES.get(mime_type.as_str()))
@@ -206,13 +206,15 @@ impl SourceExecutor for Executor {
             } else {
                 export_mime_type.text
             };
-            self.drive_hub
+            let content = self
+                .drive_hub
                 .files()
                 .export(&file_id, target_mime_type)
                 .add_scope(Scope::Readonly)
                 .doit()
                 .await?
-                .into_body()
+                .into_body();
+            (Some(target_mime_type.to_string()), content)
         } else {
             let (resp, _) = self
                 .drive_hub
@@ -222,11 +224,12 @@ impl SourceExecutor for Executor {
                 .param("alt", "media")
                 .doit()
                 .await?;
-            resp.into_body()
+            (file.mime_type, resp.into_body())
         };
         let content = resp_body.collect().await?;
         let mut fields = Vec::with_capacity(2);
         fields.push(file.name.unwrap_or_default().into());
+        fields.push(mime_type.into());
         if self.binary {
             fields.push(content.to_bytes().to_vec().into());
         } else {
@@ -261,6 +264,7 @@ impl SourceFactoryBase for Factory {
             vec![
                 FieldSchema::new("file_id", make_output_type(BasicValueType::Str)),
                 FieldSchema::new("filename", make_output_type(BasicValueType::Str)),
+                FieldSchema::new("mime_type", make_output_type(BasicValueType::Str)),
                 FieldSchema::new(
                     "content",
                     make_output_type(if spec.binary {