Environment deletion and better actions menu

src-tauri/sqlx-data.json

@@ -472,6 +472,16 @@
     },
     "query": "\n            INSERT INTO environments (\n                id,\n                workspace_id,\n                name,\n                data\n            )\n            VALUES (?, ?, ?, ?)\n        "
   },
+  "aeb0712785a9964d516dc8939bc54aa8206ad852e608b362d014b67a0f21b0ed": {
+    "describe": {
+      "columns": [],
+      "nullable": [],
+      "parameters": {
+        "Right": 1
+      }
+    },
+    "query": "\n            DELETE FROM environments\n            WHERE id = ?\n        "
+  },
   "b19c275180909a39342b13c3cdcf993781636913ae590967f5508c46a56dc961": {
     "describe": {
       "columns": [],

src-tauri/src/main.rs

@@ -479,6 +479,19 @@ async fn delete_request(
     emit_and_return(&window, "deleted_model", req)
 }
 
+#[tauri::command]
+async fn delete_environment(
+    window: Window<Wry>,
+    db_instance: State<'_, Mutex<Pool<Sqlite>>>,
+    environment_id: &str,
+) -> Result<models::Environment, String> {
+    let pool = &*db_instance.lock().await;
+    let req = models::delete_environment(environment_id, pool)
+        .await
+        .expect("Failed to delete environment");
+    emit_and_return(&window, "deleted_model", req)
+}
+
 #[tauri::command]
 async fn list_requests(
     workspace_id: &str,
@@ -600,10 +613,10 @@ async fn new_window(window: Window<Wry>, url: &str) -> Result<(), String> {
 async fn delete_workspace(
     window: Window<Wry>,
     db_instance: State<'_, Mutex<Pool<Sqlite>>>,
-    id: &str,
+    workspace_id: &str,
 ) -> Result<models::Workspace, String> {
     let pool = &*db_instance.lock().await;
-    let workspace = models::delete_workspace(id, pool)
+    let workspace = models::delete_workspace(workspace_id, pool)
         .await
         .expect("Failed to delete workspace");
     emit_and_return(&window, "deleted_model", workspace)
@@ -643,6 +656,7 @@ fn main() {
             create_request,
             create_workspace,
             delete_all_responses,
+            delete_environment,
             delete_request,
             delete_response,
             delete_workspace,

src-tauri/src/models.rs

@@ -252,13 +252,27 @@ pub async fn create_environment(
     get_environment(&id, pool).await
 }
 
+pub async fn delete_environment(id: &str, pool: &Pool<Sqlite>) -> Result<Environment, sqlx::Error> {
+    let env = get_environment(id, pool).await?;
+    let _ = sqlx::query!(
+        r#"
+            DELETE FROM environments
+            WHERE id = ?
+        "#,
+        id,
+    )
+    .execute(pool)
+    .await;
+
+    Ok(env)
+}
+
 pub async fn update_environment(
     id: &str,
     name: &str,
     data: HashMap<String, JsonValue>,
     pool: &Pool<Sqlite>,
 ) -> Result<Environment, sqlx::Error> {
-    println!("DATA: {}", data.clone().len());
     let json_data = Json(data);
     sqlx::query!(
         r#"
@@ -457,6 +471,10 @@ pub async fn get_request(id: &str, pool: &Pool<Sqlite>) -> Result<HttpRequest, s
 
 pub async fn delete_request(id: &str, pool: &Pool<Sqlite>) -> Result<HttpRequest, sqlx::Error> {
     let req = get_request(id, pool).await?;
+
+    // DB deletes will cascade but this will delete the files
+    delete_all_responses(id, pool).await?;
+
     let _ = sqlx::query!(
         r#"
             DELETE FROM http_requests
@@ -467,8 +485,6 @@ pub async fn delete_request(id: &str, pool: &Pool<Sqlite>) -> Result<HttpRequest
     .execute(pool)
     .await;
 
-    delete_all_responses(id, pool).await?;
-
     Ok(req)
 }
 

src-web/components/EnvironmentActionsDropdown.tsx

@@ -13,6 +13,7 @@ import { usePrompt } from '../hooks/usePrompt';
 import { useDialog } from './DialogContext';
 import { EnvironmentEditDialog } from './EnvironmentEditDialog';
 import { useAppRoutes } from '../hooks/useAppRoutes';
+import { useDeleteEnvironment } from '../hooks/useDeleteEnvironment';
 
 type Props = {
   className?: string;
@@ -24,13 +25,14 @@ export const EnvironmentActionsDropdown = memo(function EnvironmentActionsDropdo
   const environments = useEnvironments();
   const activeEnvironment = useActiveEnvironment();
   const updateEnvironment = useUpdateEnvironment(activeEnvironment?.id ?? null);
+  const deleteEnvironment = useDeleteEnvironment(activeEnvironment);
   const createEnvironment = useCreateEnvironment();
   const prompt = usePrompt();
   const dialog = useDialog();
   const routes = useAppRoutes();
 
   const items: DropdownItem[] = useMemo(() => {
-    const environmentItems = environments.map(
+    const environmentItems: DropdownItem[] = environments.map(
       (e) => ({
         key: e.id,
         label: e.name,
@@ -41,57 +43,58 @@ export const EnvironmentActionsDropdown = memo(function EnvironmentActionsDropdo
       }),
       [activeEnvironment?.id],
     );
-    const activeEnvironmentItems: DropdownItem[] =
-      environments.length <= 1
-        ? []
-        : [
-            ...environmentItems,
-            {
-              type: 'separator',
-              label: activeEnvironment?.name,
-            },
-          ];
 
     return [
-      ...activeEnvironmentItems,
-      {
-        key: 'edit',
-        label: 'Edit',
-        leftSlot: <Icon icon="sun" />,
-        onSelect: async () => {
-          dialog.show({
-            title: 'Environments',
-            render: () => <EnvironmentEditDialog />,
-          });
-        },
-      },
-      {
-        key: 'rename',
-        label: 'Rename',
-        leftSlot: <Icon icon="pencil" />,
-        onSelect: async () => {
-          const name = await prompt({
-            title: 'Rename Environment',
-            description: (
-              <>
-                Enter a new name for <InlineCode>{activeEnvironment?.name}</InlineCode>
-              </>
-            ),
-            name: 'name',
-            label: 'Name',
-            defaultValue: activeEnvironment?.name,
-          });
-          updateEnvironment.mutate({ name });
-        },
-      },
-      // {
-      //   key: 'delete',
-      //   label: 'Delete',
-      //   leftSlot: <Icon icon="trash" />,
-      //   onSelect: deleteEnv.mutate,
-      //   variant: 'danger',
-      // },
-      { type: 'separator' },
+      ...environmentItems,
+      ...((environmentItems.length > 0
+        ? [{ type: 'separator', label: activeEnvironment?.name }]
+        : []) as DropdownItem[]),
+      ...((activeEnvironment != null
+        ? [
+            {
+              key: 'rename',
+              label: 'Rename',
+              leftSlot: <Icon icon="pencil" />,
+              onSelect: async () => {
+                const name = await prompt({
+                  title: 'Rename Environment',
+                  description: (
+                    <>
+                      Enter a new name for <InlineCode>{activeEnvironment?.name}</InlineCode>
+                    </>
+                  ),
+                  name: 'name',
+                  label: 'Name',
+                  defaultValue: activeEnvironment?.name,
+                });
+                updateEnvironment.mutate({ name });
+              },
+            },
+            {
+              key: 'delete',
+              label: 'Delete',
+              leftSlot: <Icon icon="trash" />,
+              onSelect: deleteEnvironment.mutate,
+              variant: 'danger',
+            },
+            { type: 'separator' },
+          ]
+        : []) as DropdownItem[]),
+      ...((environments.length > 0
+        ? [
+            {
+              key: 'edit',
+              label: 'Manage Environments',
+              leftSlot: <Icon icon="gear" />,
+              onSelect: async () => {
+                dialog.show({
+                  title: 'Environments',
+                  render: () => <EnvironmentEditDialog />,
+                });
+              },
+            },
+          ]
+        : []) as DropdownItem[]),
       {
         key: 'create-environment',
         label: 'Create Environment',
@@ -110,12 +113,13 @@ export const EnvironmentActionsDropdown = memo(function EnvironmentActionsDropdo
     ];
   }, [
     // deleteEnvironment.mutate,
-    activeEnvironment?.name,
+    activeEnvironment,
     createEnvironment,
     dialog,
     environments,
     prompt,
     updateEnvironment,
+    deleteEnvironment,
     routes,
   ]);
 

src-web/hooks/useCreateEnvironment.ts

@@ -3,11 +3,9 @@ import { invoke } from '@tauri-apps/api';
 import type { Environment } from '../lib/models';
 import { environmentsQueryKey } from './useEnvironments';
 import { useActiveWorkspaceId } from './useActiveWorkspaceId';
-import { useActiveEnvironmentId } from './useActiveEnvironmentId';
 import { useAppRoutes } from './useAppRoutes';
 
 export function useCreateEnvironment() {
-  const environmentId = useActiveEnvironmentId();
   const workspaceId = useActiveWorkspaceId();
   const queryClient = useQueryClient();
   const routes = useAppRoutes();
@@ -18,7 +16,7 @@ export function useCreateEnvironment() {
     },
     onSuccess: async (environment) => {
       if (workspaceId == null) return;
-      routes.navigate('workspace', { workspaceId, environmentId });
+      routes.setEnvironment(environment);
       queryClient.setQueryData<Environment[]>(
         environmentsQueryKey({ workspaceId }),
         (environments) => [...(environments ?? []), environment],

src-web/hooks/useDeleteEnvironment.tsx

@@ -0,0 +1,36 @@
+import { useMutation, useQueryClient } from '@tanstack/react-query';
+import { invoke } from '@tauri-apps/api';
+import { InlineCode } from '../components/core/InlineCode';
+import type { Environment, Workspace } from '../lib/models';
+import { useConfirm } from './useConfirm';
+import { environmentsQueryKey } from './useEnvironments';
+
+export function useDeleteEnvironment(environment: Environment | null) {
+  const queryClient = useQueryClient();
+  const confirm = useConfirm();
+
+  return useMutation<Environment | null, string>({
+    mutationFn: async () => {
+      const confirmed = await confirm({
+        title: 'Delete Environment',
+        variant: 'delete',
+        description: (
+          <>
+            Permanently delete <InlineCode>{environment?.name}</InlineCode>?
+          </>
+        ),
+      });
+      if (!confirmed) return null;
+      return invoke('delete_environment', { environmentId: environment?.id });
+    },
+    onSuccess: async (environment) => {
+      if (environment === null) return;
+
+      const { id: environmentId, workspaceId } = environment;
+      queryClient.setQueryData<Workspace[]>(
+        environmentsQueryKey({ workspaceId }),
+        (environments) => environments?.filter((e) => e.id !== environmentId),
+      );
+    },
+  });
+}

src-web/hooks/useDeleteWorkspace.tsx

@@ -26,7 +26,7 @@ export function useDeleteWorkspace(workspace: Workspace | null) {
         ),
       });
       if (!confirmed) return null;
-      return invoke('delete_workspace', { id: workspace?.id });
+      return invoke('delete_workspace', { workspaceId: workspace?.id });
     },
     onSuccess: async (workspace) => {
       if (workspace === null) return;