Adding snapshots feature with some other features.

Author: Bl4ckk98

src/renderer/lib/config.ts

@@ -21,6 +21,9 @@ export type WinboatConfigObj = {
     advancedFeatures: boolean
     multiMonitor: number
     rdpArgs: RdpArg[]
+    snapshotMaxCount: number;
+    snapshotCompression: boolean;
+    snapshotPath: string;
 };
 
 const defaultConfig: WinboatConfigObj = {
@@ -33,10 +36,13 @@ const defaultConfig: WinboatConfigObj = {
     experimentalFeatures: false,
     advancedFeatures: false,
     multiMonitor: 0,
-    rdpArgs:[]
+    rdpArgs:[],
+    snapshotMaxCount: 3,
+    snapshotCompression: true,
+    snapshotPath: path.join(WINBOAT_DIR, "snapshots"),
 };
 
-export class WinboatConfig { 
+export class WinboatConfig {
     private static instance: WinboatConfig;
     #configPath: string = path.join(WINBOAT_DIR, "winboat.config.json");
     #configData: WinboatConfigObj = { ...defaultConfig };
@@ -113,4 +119,4 @@ export class WinboatConfig {
             return { ...defaultConfig };
         }
     }
-}
+}

src/renderer/lib/snapshot.ts

@@ -13,6 +13,7 @@ import { QMPManager } from "./qmp";
 import { assert } from "@vueuse/core";
 import { setIntervalImmediately } from "../utils/interval";
 import { ComposePortEntry, PortManager } from "../utils/port";
+import { SnapshotManager } from "./snapshot";
 
 const nodeFetch: typeof import('node-fetch').default = require('node-fetch');
 const fs: typeof import('fs') = require('fs');
@@ -37,7 +38,7 @@ const presetApps: WinApp[] = [
         Icon: AppIcons[InternalApps.WINDOWS_DESKTOP],
         Source: "internal",
         Path: InternalApps.WINDOWS_DESKTOP,
-        Usage: 0 
+        Usage: 0
     },
     {
         Name: "⚙️ Windows Explorer",
@@ -95,7 +96,7 @@ class AppManager {
     appCache: WinApp[] = []
     appUsageCache: { [key: string]: number } = {};
     #wbConfig: WinboatConfig | null = null;
-    
+
     constructor() {
         if(!fs.existsSync(USAGE_PATH)) {
             fs.writeFileSync(USAGE_PATH, "{}");
@@ -126,8 +127,8 @@ class AppManager {
         }
 
         // Get the usage object that's on the disk
-        const fsUsage = Object.entries(JSON.parse(fs.readFileSync(USAGE_PATH, 'utf-8'))) as any[]; 
-        this.appCache = []; 
+        const fsUsage = Object.entries(JSON.parse(fs.readFileSync(USAGE_PATH, 'utf-8'))) as any[];
+        this.appCache = [];
 
         // Populate appCache with dummy WinApp object containing data from the disk
         for (let i = 0; i < fsUsage.length; i++) {
@@ -227,13 +228,14 @@ export class Winboat {
     appMgr: AppManager | null = null;
     qmpMgr: QMPManager | null = null;
     portMgr: Ref<PortManager | null> = ref(null);
+    snapshotMgr: SnapshotManager | null = null;
 
 
     constructor() {
         if (Winboat.instance) {
             return Winboat.instance;
         }
-        
+
         // This is a special interval which will never be destroyed
         this.#containerInterval = setInterval(async () => {
             const _containerStatus = await this.getContainerStatus();
@@ -252,13 +254,43 @@ export class Winboat {
 
         this.#wbConfig = new WinboatConfig();
 
+        this.snapshotMgr = new SnapshotManager();
+
         this.appMgr = new AppManager();
 
         Winboat.instance = this;
 
         return Winboat.instance;
     }
 
+    // Helper method to get storage folder
+    getStorageInfo(): { type: "volume" | "bind"; path: string } {
+        const compose = this.parseCompose();
+        const storageVol = compose.services.windows.volumes.find((v) => v.includes("/storage"));
+
+        if (!storageVol) {
+            throw new Error("Storage volume not found in compose file");
+        }
+
+        // Check if it's a named volume (eg. "data:/storage")
+        if (storageVol.startsWith("data:")) {
+            return { type: "volume", path: "winboat_data" };
+        }
+
+        // Bind mount otherwise (es. "/path/to/folder:/storage")
+        return { type: "bind", path: storageVol.split(":")[0] };
+    }
+
+    /**
+     * Recreates the SnapshotManager instance with updated configuration.
+     * Call this when the snapshot path changes.
+     */
+    recreateSnapshotManager(): void {
+        logger.info("Recreating SnapshotManager with updated configuration...");
+        this.snapshotMgr = new SnapshotManager();
+        logger.info("SnapshotManager recreated successfully");
+    }
+
     /**
      * Creates the intervals which rely on the Winboat Guest API.
      */
@@ -438,7 +470,7 @@ export class Winboat {
 
     /**
      * Returns the host port that maps to the given guest port
-     * 
+     *
      * @param guestPort The port that gets looked up
      * @returns The host port that maps to the given guest port, or null if not found
      */
@@ -487,7 +519,7 @@ export class Winboat {
             // QMP either doesn't exist or is disconnected
             await this.#connectQMPManager();
             logger.info("[QMPInterval] Created new QMP Manager");
-            
+
         }, QMP_WAIT_MS);
     }
 
@@ -567,15 +599,15 @@ export class Winboat {
         this.containerActionLoading.value = true;
 
         const composeFilePath = path.join(WINBOAT_DIR, 'docker-compose.yml');
-        
+
         if (restart) {
             // 1. Compose down the current container
             await execAsync(`docker compose -f ${composeFilePath} down`);
         }
 
         // 2. Create a backup directory if it doesn't exist
         const backupDir = path.join(WINBOAT_DIR, 'backup');
-        
+
         if (!fs.existsSync(backupDir)) {
             fs.mkdirSync(backupDir);
             logger.info(`Created compose backup dir: ${backupDir}`)
@@ -590,13 +622,13 @@ export class Winboat {
         const newComposeYAML = PrettyYAML.stringify(composeConfig).replaceAll("null", "");
         fs.writeFileSync(composeFilePath, newComposeYAML, { encoding: 'utf8' });
         logger.info(`Wrote new compose file to: ${composeFilePath}`);
-        
+
         if (restart) {
             // 5. Deploy the container with the new compose file
             await execAsync(`docker compose -f ${composeFilePath} up -d`);
             remote.getCurrentWindow().reload();
         }
-      
+
         logger.info("Replace compose config completed, successfully deployed new container");
 
         this.containerActionLoading.value = false;
@@ -608,7 +640,7 @@ export class Winboat {
         // 1. Stop container
         await this.stopContainer();
         console.info("Stopped container");
-        
+
         // 2. Remove the container
         await execAsync("docker rm WinBoat")
         console.info("Removed container")
@@ -652,7 +684,7 @@ export class Winboat {
         const rdpHostPort = this.getHostPort(GUEST_RDP_PORT);
 
         logger.info(`Launching app: ${app.Name} at path ${app.Path}`);
-        
+
         const freeRDPBin = await getFreeRDP();
 
         logger.info(`Using FreeRDP Command: '${freeRDPBin}'`);
@@ -728,7 +760,7 @@ export class Winboat {
             : path.join(remote.app.getAppPath(), '..', '..', 'guest_server', 'winboat_guest_server.zip');
 
         logger.info("ZIP Path", zipPath)
-        
+
         // 4. Send the payload to the guest server, as a multipart/form-data with updateFile
         const formData = new FormData();
         formData.append('updateFile', fs.createReadStream(zipPath));
@@ -747,7 +779,7 @@ export class Winboat {
             const resJson = await res.json() as GuestServerUpdateResponse;
             logger.info(`Update params: ${JSON.stringify(resJson, null, 4)}`);
             logger.info("Successfully sent update payload to guest server");
-            
+
         } catch(e) {
             logger.error("Failed to send update payload to guest server");
             logger.error(e);
@@ -774,4 +806,4 @@ export class Winboat {
     get hasQMPInterval() {
         return this.#qmpInterval !== null;
     }
-}
+}

src/renderer/lib/winboat.ts

@@ -7,18 +7,20 @@ import Apps from './views/Apps.vue'
 import About from './views/About.vue'
 import Blank from './views/Blank.vue'
 import Config from './views/Config.vue'
+import Snapshots from './views/Snapshots.vue'
 
 export const routes: RouteRecordRaw[] = [
     { path: '/', name: "Loading", component: Blank, meta: { icon: 'line-md:loading-loop' } },
     { path: '/home', name: "Home", component: Home, meta: { icon: 'fluent:home-32-filled' } },
     { path: '/setup', name: "SetupUI", component: SetupUI, meta: { icon: 'fluent-mdl2:install-to-drive' } },
     { path: '/apps', name: "Apps", component: Apps, meta: { icon: 'fluent:apps-32-filled' } },
     { path: '/configuration', name: "Configuration", component: Config, meta: { icon: 'icon-park-outline:config' } },
+    { path: '/snapshots', name: "Snapshots", component: Snapshots, meta: { icon: 'mdi:camera' } },
     { path: '/about', name: "About", component: About, meta: { icon: 'fluent:info-32-filled' } },
 
 ]
 
 export const router = createRouter({
     history: createMemoryHistory(),
     routes,
-})
+})