From d2e04a495d77b245a28e9cdd65efe5ac12338a5c Mon Sep 17 00:00:00 2001
From: nwin <nwin@sysnet.de>
Date: Thu, 21 Nov 2024 19:32:51 +0100
Subject: [PATCH] readd managedsave to current

---
 src/components/vm/overview/vmOverviewCard.jsx |  2 +-
 src/components/vm/vmActions.jsx               | 65 +++++++++++++++++++
 src/components/vms/hostvmslist.jsx            |  2 +
 src/helpers.js                                |  1 +
 src/libvirtApi/domain.js                      | 27 ++++++++
 5 files changed, 96 insertions(+), 1 deletion(-)

diff --git a/src/components/vm/overview/vmOverviewCard.jsx b/src/components/vm/overview/vmOverviewCard.jsx
index 2cd8aa14e..fa142e338 100644
--- a/src/components/vm/overview/vmOverviewCard.jsx
+++ b/src/components/vm/overview/vmOverviewCard.jsx
@@ -165,7 +165,7 @@ class VmOverviewCard extends React.Component {
                             <DescriptionListTerm>{_("State")}</DescriptionListTerm>
                             <DescriptionListDescription>
                                 <StateIcon error={vm.error}
-                                               state={vm.state}
+                                               state={vm.state + (vm.suspendImage ? " (suspended)" : "")}
                                                valueId={`${idPrefix}-${vm.connectionName}-state`}
                                                dismissError={() => store.dispatch(updateVm({
                                                    connectionName: vm.connectionName,
diff --git a/src/components/vm/vmActions.jsx b/src/components/vm/vmActions.jsx
index cf5868bcd..91aa86ef2 100644
--- a/src/components/vm/vmActions.jsx
+++ b/src/components/vm/vmActions.jsx
@@ -50,6 +50,8 @@ import {
     domainCanPause,
     domainCanShutdown,
     domainForceOff,
+    domainCanSuspendToDisk,
+    domainCanSuspenImageRemove,
     domainForceReboot,
     domainInstall,
     domainPause,
@@ -59,6 +61,9 @@ import {
     domainShutdown,
     domainStart,
     domainAddTPM,
+    domainSuspendToDisk,
+    domainSuspendImageRemove,
+    domainHasSuspendImage
 } from '../../libvirtApi/domain.js';
 import store from "../../store.js";
 
@@ -183,6 +188,43 @@ const onSendNMI = (vm) => domainSendNMI({ name: vm.name, id: vm.id, connectionNa
     );
 });
 
+const onSuspendToDisk = (vm) => domainSuspendToDisk({ name: vm.name, id: vm.id, connectionName: vm.connectionName, flags: vm.state == 'running' ? 2 : 4 }).catch(ex => {
+    store.dispatch(
+        updateVm({
+            connectionName: vm.connectionName,
+            name: vm.name,
+            error: {
+                text: cockpit.format(_("VM $0 failed to save "), vm.name),
+                detail: ex.message,
+            }
+        })
+    );
+});
+
+const onSuspendImageRemove = (vm) => domainSuspendImageRemove({ name: vm.name, id: vm.id, connectionName: vm.connectionName })
+        .then(() => domainHasSuspendImage({ name: vm.name, id: vm.id, connectionName: vm.connectionName }))
+        .then((SuspendImage) => {
+            store.dispatch(
+                updateVm({
+                    connectionName: vm.connectionName,
+                    name: vm.name,
+                    suspendImage: SuspendImage[0],
+                })
+            );
+        })
+        .catch(ex => {
+            store.dispatch(
+                updateVm({
+                    connectionName: vm.connectionName,
+                    name: vm.name,
+                    error: {
+                        text: cockpit.format(_("VM $0 failed to remove save image "), vm.name),
+                        detail: ex.message,
+                    }
+                })
+            );
+        });
+
 const onAddTPM = (vm, onAddErrorNotification) => domainAddTPM({ connectionName: vm.connectionName, vmName: vm.name })
         .catch(ex => onAddErrorNotification({
             text: cockpit.format(_("Failed to add TPM to VM $0"), vm.name),
@@ -208,6 +250,7 @@ const VmActions = ({ vm, vms, onAddErrorNotification, isDetailsPage }) => {
 
     const id = `${vmId(vm.name)}-${vm.connectionName}`;
     const state = vm.state;
+    const suspendImage = vm.suspendImage;
     const hasInstallPhase = vm.metadata && vm.metadata.hasInstallPhase;
     const dropdownItems = [];
 
@@ -235,6 +278,28 @@ const VmActions = ({ vm, vms, onAddErrorNotification, isDetailsPage }) => {
         dropdownItems.push(<Divider key="separator-resume" />);
     }
 
+    if (domainCanSuspendToDisk(state)) {
+        dropdownItems.push(
+            <DropdownItem key={`${id}-save`}
+                          id={`${id}-save`}
+                          onClick={() => onSuspendToDisk(vm)}>
+                {_("Suspend to disk")}
+            </DropdownItem>
+        );
+        dropdownItems.push(<Divider key="separator-suspend" />);
+    }
+
+    if (domainCanSuspenImageRemove(state, suspendImage)) {
+        dropdownItems.push(
+            <DropdownItem key={`${id}-saveRemove`}
+                          id={`${id}-savedRemove`}
+                          onClick={() => onSuspendImageRemove(vm)}>
+                {_("Remove Suspend Image")}
+            </DropdownItem>
+        );
+        dropdownItems.push(<Divider key="separator-suspend" />);
+    }
+
     if (domainCanShutdown(state)) {
         shutdown = (
             <Button key='action-shutdown'
diff --git a/src/components/vms/hostvmslist.jsx b/src/components/vms/hostvmslist.jsx
index b59bff15b..357c608e1 100644
--- a/src/components/vms/hostvmslist.jsx
+++ b/src/components/vms/hostvmslist.jsx
@@ -52,6 +52,8 @@ const VmState = ({ vm, vms, dismissError }) => {
         state = cockpit.format(_("Downloading: $0%"), vm.downloadProgress);
     } else if (vm.createInProgress) {
         state = _("Creating VM");
+    } else if (vm.state == "shut off" && vm.suspendImage === true) {
+            state = `${vm.state} (suspended)`;
     } else {
         state = vm.state;
     }
diff --git a/src/helpers.js b/src/helpers.js
index e15af415d..7e1f87a13 100644
--- a/src/helpers.js
+++ b/src/helpers.js
@@ -213,6 +213,7 @@ const transform = {
         paused: _("Paused"),
         shutdown: _("Shutting down"),
         'shut off': _("Shut off"),
+        'shut off (suspended)': _("Shut off (Suspended)"),        
         crashed: _("Crashed"),
         dying: _("Dying"),
         pmsuspended: _("Suspended (PM)"),
diff --git a/src/libvirtApi/domain.js b/src/libvirtApi/domain.js
index eab234179..f04ae185f 100644
--- a/src/libvirtApi/domain.js
+++ b/src/libvirtApi/domain.js
@@ -87,6 +87,8 @@ export const domainCanShutdown = (vmState) => domainCanReset(vmState);
 export const domainCanPause = (vmState) => vmState == 'running';
 export const domainCanRename = (vmState) => vmState == 'shut off';
 export const domainCanResume = (vmState) => vmState == 'paused';
+export const domainCanSuspendToDisk = (vmState) => vmState == 'running' || vmState == 'paused';
+export const domainCanSuspenImageRemove = (vmState, saveImage) => vmState == "shut off" && saveImage == true;
 export const domainIsRunning = (vmState) => domainCanReset(vmState);
 export const domainSerialConsoleCommand = ({ vm, alias }) => {
     if (vm.displays.find(display => display.type == 'pty'))
@@ -663,6 +665,9 @@ export async function domainGet({
             supportsTPM: getDomainCapSupportsTPM(domCaps),
         };
 
+        const [suspendImage] = await call(connectionName, objPath, 'org.libvirt.Domain', 'HasManagedSaveImage', [0], { timeout, type: 'u' });
+        props.suspendImage = suspendImage;
+
         const [state] = await call(connectionName, objPath, 'org.libvirt.Domain', 'GetState', [0], { timeout, type: 'u' });
         const stateStr = DOMAINSTATE[state[0]];
 
@@ -874,6 +879,28 @@ export function domainResume({
     return call(connectionName, objPath, 'org.libvirt.Domain', 'Resume', [], { timeout, type: '' });
 }
 
+export function domainSuspendToDisk({
+    connectionName,
+    id: objPath,
+    flags
+}) {
+    return call(connectionName, objPath, 'org.libvirt.Domain', 'ManagedSave', [flags], { timeout, type: 'u' });
+}
+
+export function domainSuspendImageRemove({
+    connectionName,
+    id: objPath,
+}) {
+    return call(connectionName, objPath, 'org.libvirt.Domain', 'ManagedSaveRemove', [0], { timeout, type: 'u' });
+}
+
+export function domainHasSuspendImage({
+    connectionName,
+    id: objPath,
+}) {
+    return call(connectionName, objPath, 'org.libvirt.Domain', 'HasManagedSaveImage', [0], { timeout, type: 'u' });
+}
+
 export function domainSendKey({ connectionName, id, keyCodes }) {
     const holdTime = 0;
     const flags = 0;