From da18f2f6e9f3cb202fbf3aa790b8b9ea56d73a18 Mon Sep 17 00:00:00 2001
From: FredWuCZ <fredwucz@outlook.com>
Date: Fri, 27 Sep 2024 13:22:57 +0800
Subject: [PATCH] Add stop container

---
 desktop_env/providers/docker/provider.py | 30 +++++++++---------------
 1 file changed, 11 insertions(+), 19 deletions(-)

diff --git a/desktop_env/providers/docker/provider.py b/desktop_env/providers/docker/provider.py
index b60eaca..fb3f77b 100644
--- a/desktop_env/providers/docker/provider.py
+++ b/desktop_env/providers/docker/provider.py
@@ -12,22 +12,6 @@ logger.setLevel(logging.INFO)
 
 WAIT_TIME = 3
 
-
-def get_vmrun_type(return_list=False):
-    if platform.system() == 'Windows' or platform.system() == 'Linux':
-        if return_list:
-            return ['-T', 'ws']
-        else:
-            return '-T ws'
-    elif platform.system() == 'Darwin':  # Darwin is the system name for macOS
-        if return_list:
-            return ['-T', 'fusion']
-        else:
-            return '-T fusion'
-    else:
-        raise Exception("Unsupported operating system")
-
-
 class DockerProvider(Provider):
     def __init__(self, region: str):
         self.client = docker.from_env()
@@ -49,16 +33,24 @@ class DockerProvider(Provider):
             return None
 
     def start_emulator(self, path_to_vm: str, headless: bool, os_type: str):
-        pass
+        self.container = self.client.containers.run('qemux/qemu-docker', environment={"DISK_SIZE": "64G", "RAM_SIZE": "6G", "CPU_CORES": "8"}, volumes={"/Users/happysix/Programs/HKUNLP/Qemu/Ubuntu.qcow2": {"bind": "/Ubuntu.qcow2", "mode": "ro"}, "/Users/happysix/Programs/HKUNLP/Qemu/snapshot.qcow2": {"bind": "/boot.qcow2", "mode": "rw"}}, cap_add=["NET_ADMIN"], ports={8006: 8006, 5000: 5001}, detach=True)
 
     def get_ip_address(self, path_to_vm: str) -> str:
         pass
 
     def save_state(self, path_to_vm: str, snapshot_name: str):
-        pass
+        logger.info("Saving VM state...")
+        DockerProvider._execute_command(
+            ["qemu-img", "convert", "-O", "qcow2", snapshot_name, "temp.qcow2"])
+        DockerProvider._execute_command(
+            ["mv", "temp.qcow2", path_to_vm]
+        )
+        time.sleep(WAIT_TIME)  # Wait for the VM to save
 
     def revert_to_snapshot(self, path_to_vm: str, snapshot_name: str):
         pass
 
     def stop_emulator(self, path_to_vm: str):
-        pass
\ No newline at end of file
+        logger.info("Stopping VMware VM...")
+        self.container.stop(WAIT_TIME)
+        self.container.remove()
\ No newline at end of file