34 lines
1.0 KiB
Rust
34 lines
1.0 KiB
Rust
use std::sync::Arc;
|
|
|
|
use crate::commands::connection::AppState;
|
|
use dbx_core::csv_export::{export_table_data_csv_core, format_csv, TableCsvExportOptions};
|
|
use serde::Deserialize;
|
|
use serde_json::Value;
|
|
use tauri::State;
|
|
|
|
#[derive(Debug, Clone, Deserialize)]
|
|
#[serde(rename_all = "camelCase")]
|
|
pub struct QueryResultCsvExportRequest {
|
|
pub file_path: String,
|
|
pub columns: Vec<String>,
|
|
pub rows: Vec<Vec<Value>>,
|
|
}
|
|
|
|
#[tauri::command]
|
|
pub async fn export_query_result_csv(request: QueryResultCsvExportRequest) -> Result<(), String> {
|
|
tauri::async_runtime::spawn_blocking(move || {
|
|
let csv = format_csv(&request.columns, &request.rows);
|
|
std::fs::write(&request.file_path, format!("\u{FEFF}{csv}")).map_err(|err| err.to_string())
|
|
})
|
|
.await
|
|
.map_err(|err| err.to_string())?
|
|
}
|
|
|
|
#[tauri::command]
|
|
pub async fn export_table_data_csv(
|
|
state: State<'_, Arc<AppState>>,
|
|
request: TableCsvExportOptions,
|
|
) -> Result<u64, String> {
|
|
export_table_data_csv_core(&state, request).await
|
|
}
|