fix(core): add data source variant for file connectors (#1114)

Author: goldmedal

ibis-server/tests/routers/v2/connector/test_gcs_file.py

@@ -77,6 +77,7 @@
             "condition": "Customer.custkey = Orders.custkey",
         }
     ],
+    "dataSource": "gcs_file",
 }
 
 

ibis-server/tests/routers/v2/connector/test_local_file.py

@@ -72,6 +72,7 @@
             "condition": "Customer.custkey = Orders.custkey",
         }
     ],
+    "dataSource": "local_file",
 }
 
 

ibis-server/tests/routers/v2/connector/test_minio_file.py

@@ -75,6 +75,7 @@
             "condition": "Customer.custkey = Orders.custkey",
         }
     ],
+    "dataSource": "minio_file",
 }
 
 

ibis-server/tests/routers/v2/connector/test_s3_file.py

@@ -77,6 +77,7 @@
             "condition": "Customer.custkey = Orders.custkey",
         }
     ],
+    "dataSource": "s3_file",
 }
 
 

wren-core-base/manifest-macro/src/lib.rs

@@ -94,6 +94,14 @@ pub fn data_source(python_binding: proc_macro::TokenStream) -> proc_macro::Token
             Datafusion,
             #[serde(alias = "duckdb")]
             DuckDB,
+            #[serde(alias = "local_file")]
+            LocalFile,
+            #[serde(alias = "s3_file")]
+            S3File,
+            #[serde(alias = "gcs_file")]
+            GcsFile,
+            #[serde(alias = "minio_file")]
+            MinioFile,
         }
     };
     proc_macro::TokenStream::from(expanded)

wren-core-base/src/mdl/manifest.rs

@@ -102,6 +102,10 @@ impl Display for DataSource {
             DataSource::Snowflake => write!(f, "SNOWFLAKE"),
             DataSource::Datafusion => write!(f, "DATAFUSION"),
             DataSource::DuckDB => write!(f, "DUCKDB"),
+            DataSource::LocalFile => write!(f, "LOCAL_FILE"),
+            DataSource::S3File => write!(f, "S3_FILE"),
+            DataSource::GcsFile => write!(f, "GCS_FILE"),
+            DataSource::MinioFile => write!(f, "MINIO_FILE"),
         }
     }
 }