supabase_client_query/
csv_select.rs

1use reqwest::header::HeaderValue;
2
3use supabase_client_core::SupabaseError;
4
5use crate::backend::QueryBackend;
6use crate::filter::Filterable;
7use crate::modifier::Modifiable;
8use crate::sql::{FilterCondition, ParamStore, SqlParts};
9
10/// Builder for SELECT queries that return CSV text.
11///
12/// Created by calling `.csv()` on a `SelectBuilder`. Returns raw CSV string
13/// instead of deserialized rows. REST-only (returns error for direct-sql backend).
14pub struct CsvSelectBuilder {
15    pub(crate) backend: QueryBackend,
16    pub(crate) parts: SqlParts,
17    pub(crate) params: ParamStore,
18}
19
20impl Filterable for CsvSelectBuilder {
21    fn filters_mut(&mut self) -> &mut Vec<FilterCondition> {
22        &mut self.parts.filters
23    }
24    fn params_mut(&mut self) -> &mut ParamStore {
25        &mut self.params
26    }
27}
28
29impl Modifiable for CsvSelectBuilder {
30    fn parts_mut(&mut self) -> &mut SqlParts {
31        &mut self.parts
32    }
33}
34
35impl CsvSelectBuilder {
36    /// Override the schema for this query.
37    pub fn schema(mut self, schema: &str) -> Self {
38        self.parts.schema_override = Some(schema.to_string());
39        self
40    }
41
42    /// Execute the SELECT query and return the response as a CSV string.
43    pub async fn execute(self) -> Result<String, SupabaseError> {
44        match &self.backend {
45            QueryBackend::Rest { http, base_url, api_key, schema } => {
46                let (url, mut headers) = crate::postgrest::build_postgrest_select(
47                    base_url, &self.parts, &self.params,
48                )
49                .map_err(SupabaseError::QueryBuilder)?;
50
51                // Override Accept to text/csv
52                headers.insert("Accept", HeaderValue::from_static("text/csv"));
53
54                // Standard auth headers
55                headers.insert("apikey", HeaderValue::from_str(api_key).unwrap());
56                headers.insert(
57                    "Authorization",
58                    HeaderValue::from_str(&format!("Bearer {}", api_key)).unwrap(),
59                );
60
61                // Schema profile
62                if let Some(ref so) = self.parts.schema_override {
63                    headers.insert(
64                        "Accept-Profile",
65                        HeaderValue::from_str(so).unwrap(),
66                    );
67                } else if schema != "public" {
68                    headers.entry("Accept-Profile")
69                        .or_insert_with(|| HeaderValue::from_str(schema).unwrap());
70                }
71
72                let response = http
73                    .get(&url)
74                    .headers(headers)
75                    .send()
76                    .await
77                    .map_err(|e| SupabaseError::Http(e.to_string()))?;
78
79                let status = response.status().as_u16();
80                let body = response
81                    .text()
82                    .await
83                    .map_err(|e| SupabaseError::Http(e.to_string()))?;
84
85                if status >= 400 {
86                    return Err(SupabaseError::postgrest(status, body, None));
87                }
88
89                Ok(body)
90            }
91            #[cfg(feature = "direct-sql")]
92            QueryBackend::DirectSql { .. } => {
93                Err(SupabaseError::query_builder(
94                    "CSV output is only supported with the REST (PostgREST) backend",
95                ))
96            }
97        }
98    }
99}
100
101#[cfg(test)]
102mod tests {
103    use super::*;
104    use crate::sql::{SqlOperation, SqlParts, ParamStore};
105
106    #[test]
107    fn test_csv_builder_modifiable() {
108        let mut builder = CsvSelectBuilder {
109            backend: QueryBackend::Rest {
110                http: reqwest::Client::new(),
111                base_url: "http://localhost".into(),
112                api_key: "key".into(),
113                schema: "public".to_string(),
114            },
115            parts: SqlParts::new(SqlOperation::Select, "public", "cities"),
116            params: ParamStore::new(),
117        };
118        builder = builder.limit(10);
119        assert_eq!(builder.parts.limit, Some(10));
120    }
121
122    #[test]
123    fn test_csv_builder_filterable() {
124        let builder = CsvSelectBuilder {
125            backend: QueryBackend::Rest {
126                http: reqwest::Client::new(),
127                base_url: "http://localhost".into(),
128                api_key: "key".into(),
129                schema: "public".to_string(),
130            },
131            parts: SqlParts::new(SqlOperation::Select, "public", "cities"),
132            params: ParamStore::new(),
133        };
134        let builder = builder.eq("name", "Auckland");
135        assert_eq!(builder.parts.filters.len(), 1);
136    }
137
138    #[test]
139    fn test_csv_accept_header() {
140        let parts = SqlParts::new(SqlOperation::Select, "public", "cities");
141        let params = ParamStore::new();
142        let (_, mut headers) = crate::postgrest::build_postgrest_select(
143            "http://localhost:64321", &parts, &params,
144        ).unwrap();
145        // Simulate what CsvSelectBuilder does
146        headers.insert("Accept", HeaderValue::from_static("text/csv"));
147        assert_eq!(headers.get("Accept").unwrap(), "text/csv");
148    }
149}
supabase_client_query/csv_select.rs

supabase_client_query/
csv_select.rs
