Merge pull request #20 from conlan0/develop

Add Shutdown option to agent action menu
amidaware · Feb 22, 2024 · 07cc6ac · 07cc6ac
2 parents 23bf81e + 38dc709
commit 07cc6ac
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 0 deletions.
diff --git a/src/api/agents.js b/src/api/agents.js
@@ -191,6 +191,11 @@ export async function agentRebootNow(agent_id) {
   return data;
 }
 
+export async function agentShutdown(agent_id) {
+  const { data } = await axios.post(`${baseUrl}/${agent_id}/shutdown/`);
+  return data;
+}
+
 export async function sendAgentRecoverMesh(agent_id, params = {}) {
   const { data } = await axios.post(
     `${baseUrl}/${agent_id}/meshcentral/recover/`,

diff --git a/src/components/agents/AgentActionMenu.vue b/src/components/agents/AgentActionMenu.vue
@@ -176,6 +176,13 @@
       </q-menu>
     </q-item>
 
+    <q-item clickable v-close-popup @click="shutdown(agent)">
+      <q-item-section side>
+        <q-icon size="xs" name="power" />
+      </q-item-section>
+      <q-item-section>Shutdown</q-item-section>
+    </q-item>
+
     <q-item clickable v-close-popup @click="showPolicyAdd(agent)">
       <q-item-section side>
         <q-icon size="xs" name="policy" />
@@ -231,6 +238,7 @@ import { fetchURLActions, runURLAction } from "@/api/core";
 import {
   editAgent,
   agentRebootNow,
+  agentShutdown,
   sendAgentPing,
   removeAgent,
   runRemoteBackground,
@@ -437,6 +445,32 @@ export default {
       });
     }
 
+    function shutdown(agent) {
+      $q.dialog({
+        title:
+          'Please type <code style="color:red">yes</code> in the box below to confirm shutdown.',
+        prompt: {
+          model: "",
+          type: "text",
+          isValid: (val) => val === "yes",
+        },
+        cancel: true,
+        ok: { label: "Shutdown", color: "negative" },
+        persistent: true,
+        html: true,
+      }).onOk(async () => {
+        $q.loading.show();
+        try {
+          await agentShutdown(agent.agent_id);
+          notifySuccess(`${agent.hostname} will now be shutdown`);
+          $q.loading.hide();
+        } catch (e) {
+          $q.loading.hide();
+          console.error(e);
+        }
+      });
+    }
+    
     function showPolicyAdd(agent) {
       $q.dialog({
         component: PolicyAdd,
@@ -534,6 +568,7 @@ export default {
       runChecks,
       showRebootLaterModal,
       rebootNow,
+      shutdown,
       showPolicyAdd,
       showAgentRecovery,
       pingAgent,