feat: add functionality to download DB Gate files

biersoeckli · Jan 28, 2025 · 88ce5f5 · 88ce5f5
1 parent 460ac6f
commit 88ce5f5
Show file tree

Hide file tree

Showing 4 changed files with 69 additions and 10 deletions.
diff --git a/src/app/project/app/[appId]/credentials/actions.ts b/src/app/project/app/[appId]/credentials/actions.ts
@@ -41,4 +41,11 @@ export const deleteDbGatDeploymentForAppIfExists = async (appId: string) =>
         await getAuthUserSession();
         await dbGateService.deleteDbGatDeploymentForAppIfExists(appId);
         return new SuccessActionResult();
-    }) as Promise<ServerActionResult<unknown, void>>;
+    }) as Promise<ServerActionResult<unknown, void>>;
+
+export const downloadDbGateFilesForApp = async (appId: string) =>
+    simpleAction(async () => {
+        await getAuthUserSession();
+        const url = await dbGateService.downloadDbGateFilesForApp(appId);
+        return new SuccessActionResult(url);
+    }) as Promise<ServerActionResult<unknown, string>>;
diff --git a/src/app/project/app/[appId]/credentials/db-crendentials.tsx b/src/app/project/app/[appId]/credentials/db-crendentials.tsx
@@ -1,15 +1,9 @@
 import { Card, CardContent, CardDescription, CardHeader, CardTitle } from "@/components/ui/card";
 import { AppExtendedModel } from "@/shared/model/app-extended.model";
 import { useEffect, useState } from "react";
-import { Button } from "@/components/ui/button";
-import { useConfirmDialog } from "@/frontend/states/zustand.states";
-import { Toast } from "@/frontend/utils/toast.utils";
-import { ClipboardCopy } from "lucide-react";
-import { toast } from "sonner";
 import { DatabaseTemplateInfoModel } from "@/shared/model/database-template-info.model";
 import { Actions } from "@/frontend/utils/nextjs-actions.utils";
 import { getDatabaseCredentials } from "./actions";
-import { Label } from "@/components/ui/label";
 import CopyInputField from "@/components/custom/copy-input-field";
 import FullLoadingSpinner from "@/components/ui/full-loading-spinnter";
 

diff --git a/src/app/project/app/[appId]/credentials/db-gate.tsx b/src/app/project/app/[appId]/credentials/db-gate.tsx
@@ -5,12 +5,14 @@ import { Button } from "@/components/ui/button";
 import { useConfirmDialog } from "@/frontend/states/zustand.states";
 import { Toast } from "@/frontend/utils/toast.utils";
 import { Actions } from "@/frontend/utils/nextjs-actions.utils";
-import { deleteDbGatDeploymentForAppIfExists, deployDbGate, getIsDbGateActive, getLoginCredentialsForRunningDbGate } from "./actions";
+import { deleteDbGatDeploymentForAppIfExists, deployDbGate, downloadDbGateFilesForApp, getIsDbGateActive, getLoginCredentialsForRunningDbGate } from "./actions";
 import { Label } from "@/components/ui/label";
 import FullLoadingSpinner from "@/components/ui/full-loading-spinnter";
 import { Switch } from "@/components/ui/switch";
 import { Code } from "@/components/custom/code";
 import LoadingSpinner from "@/components/ui/loading-spinner";
+import { Download } from "lucide-react";
+import { Tooltip, TooltipContent, TooltipProvider, TooltipTrigger } from "@/components/ui/tooltip";
 
 export default function DbGateCard({
     app
@@ -27,6 +29,19 @@ export default function DbGateCard({
         setIsDbGateActive(response);
     }
 
+    const downloadDbGateFilesForAppAsync = async () => {
+        try {
+            setLoading(true);
+            await Toast.fromAction(() => downloadDbGateFilesForApp(app.id)).then(x => {
+                if (x.status === 'success' && x.data) {
+                    window.open('/api/volume-data-download?fileName=' + x.data);
+                }
+            });
+        } finally {
+            setLoading(false);
+        }
+    }
+
     const openDbGateAsync = async () => {
         try {
             setLoading(true);
@@ -88,9 +103,23 @@ export default function DbGateCard({
                         }} />
                         <Label htmlFor="airplane-mode">DB Gate</Label>
                     </div>
+                    {isDbGateActive && <>
+                        <Button variant='outline' onClick={() => openDbGateAsync()}
+                            disabled={!isDbGateActive || loading}>Open DB Gate</Button>
+
+                        <TooltipProvider>
+                            <Tooltip delayDuration={300}>
+                                <TooltipTrigger>
+                                    <Button onClick={() => downloadDbGateFilesForAppAsync()} disabled={!isDbGateActive || loading}
+                                        variant="ghost"><Download /></Button>
+                                </TooltipTrigger>
+                                <TooltipContent>
+                                    <p>Download the "Files" folder from DB Gate.</p>
+                                </TooltipContent>
+                            </Tooltip>
+                        </TooltipProvider>
+                    </>}
                     {loading && <LoadingSpinner></LoadingSpinner>}
-                    {isDbGateActive && <Button variant='outline' onClick={() => openDbGateAsync()}
-                        disabled={!isDbGateActive || loading}>Open DB Gate</Button>}
                 </div>}
             </CardContent>
         </Card >

diff --git a/src/server/services/dbgate.service.ts b/src/server/services/dbgate.service.ts
@@ -13,6 +13,9 @@ import podService from "./pod.service";
 import { AppTemplateUtils } from "../utils/app-template.utils";
 import appService from "./app.service";
 import { AppExtendedModel } from "@/shared/model/app-extended.model";
+import { PathUtils } from "../utils/path.utils";
+import { FsUtils } from "../utils/fs.utils";
+import path from "path";
 
 class DbGateService {
 
@@ -39,6 +42,32 @@ class DbGateService {
         return true;
     }
 
+    async downloadDbGateFilesForApp(appId: string) {
+
+        const app = await appService.getExtendedById(appId);
+        const dbGateAppName = KubeObjectNameUtils.toDbGateId(app.id);
+        const pod = await podService.getPodsForApp(app.projectId, dbGateAppName);
+        if (pod.length === 0) {
+            throw new ServiceException(`There are no running pods for DBGate. Make sure the DB Gate is running.`);
+        }
+        const firstPod = pod[0];
+
+        const continerSourcePath = '/root/.dbgate/files';
+        const continerRootPath = '/root';
+
+        await podService.runCommandInPod(app.projectId, firstPod.podName, firstPod.containerName, ['cp', '-r', continerSourcePath, continerRootPath]);
+
+        const downloadPath = path.join(PathUtils.tempVolumeDownloadPath, dbGateAppName + '.tar.gz');
+        await FsUtils.createDirIfNotExistsAsync(PathUtils.tempVolumeDownloadPath, true);
+        await FsUtils.deleteDirIfExistsAsync(downloadPath, true);
+
+        console.log(`Downloading data from pod ${firstPod.podName} ${continerRootPath} to ${downloadPath}`);
+        await podService.cpFromPod(app.projectId, firstPod.podName, firstPod.containerName, continerRootPath, downloadPath, continerRootPath);
+
+        const fileName = path.basename(downloadPath);
+        return fileName;
+    }
+
     async getLoginCredentialsForRunningDbGate(appId: string) {
         const app = await appService.getExtendedById(appId);
         const dbGateAppName = KubeObjectNameUtils.toDbGateId(app.id);