starred/yaak
1
0
Fork 0
mirror of https://github.com/mountain-loop/yaak.git synced 2026-03-18 07:24:07 +01:00
Code Issues 8 Packages Projects Releases 10 Wiki Activity

Add ability to exclude environments from data export

Browse Source
This commit is contained in:
Gregory Schier
2025-01-11 11:36:00 -08:00
parent 88b410bf99
commit 8dff75ad4f
10 changed files with 59 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
src-tauri/yaak-models/src/queries.rs
View File

@@ -2175,7 +2175,8 @@ pub async fn batch_upsert<R: Runtime>(
pub async fn get_workspace_export_resources<R: Runtime>(
mgr: &impl Manager<R>,
workspace_ids: Vec<&str>,
) -> WorkspaceExport {
include_environments: bool,
) -> Result<WorkspaceExport> {
let mut data = WorkspaceExport {
yaak_version: mgr.package_info().version.clone().to_string(),
yaak_schema: 2,
@@ -2190,24 +2191,19 @@ pub async fn get_workspace_export_resources<R: Runtime>(
};
for workspace_id in workspace_ids {
data.resources
.workspaces
.push(get_workspace(mgr, workspace_id).await.expect("Failed to get workspace"));
data.resources.environments.append(
&mut list_environments(mgr, workspace_id).await.expect("Failed to get environments"),
);
data.resources
.folders
.append(&mut list_folders(mgr, workspace_id).await.expect("Failed to get folders"));
data.resources.http_requests.append(
&mut list_http_requests(mgr, workspace_id).await.expect("Failed to get http requests"),
);
data.resources.grpc_requests.append(
&mut list_grpc_requests(mgr, workspace_id).await.expect("Failed to get grpc requests"),
);
data.resources.workspaces.push(get_workspace(mgr, workspace_id).await?);
data.resources.environments.append(&mut list_environments(mgr, workspace_id).await?);
data.resources.folders.append(&mut list_folders(mgr, workspace_id).await?);
data.resources.http_requests.append(&mut list_http_requests(mgr, workspace_id).await?);
data.resources.grpc_requests.append(&mut list_grpc_requests(mgr, workspace_id).await?);
}
data
// Nuke environments if we don't want them
if !include_environments {
data.resources.environments.clear();
}
Ok(data)
}
// Generate the created_at or updated_at timestamps for an upsert operation, depending on the ID