From ae12ef88ffe6db8de08057750e5ddbd2a2542057 Mon Sep 17 00:00:00 2001
From: wh1te909 <dcparsi@gmail.com>
Date: Sun, 12 Jan 2020 10:50:26 +0000
Subject: [PATCH] add live process manager

---
 package.json                      |   2 +-
 src/components/ProcessManager.vue | 213 ++++++++++++++++++++++++++++++
 src/views/RemoteBackground.vue    |  10 +-
 3 files changed, 222 insertions(+), 3 deletions(-)
 create mode 100644 src/components/ProcessManager.vue

diff --git a/package.json b/package.json
index e2fd33c..e08095a 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "web",
-  "version": "0.1.7",
+  "version": "0.1.8",
   "private": true,
   "scripts": {
     "serve": "vue-cli-service serve",
diff --git a/src/components/ProcessManager.vue b/src/components/ProcessManager.vue
new file mode 100644
index 0000000..082f09d
--- /dev/null
+++ b/src/components/ProcessManager.vue
@@ -0,0 +1,213 @@
+<template>
+  <div class="q-pa-md">
+    <q-table
+      dense
+      class="procs-sticky-header-table"
+      :data="procs"
+      :columns="columns"
+      :pagination.sync="pagination"
+      :filter="filter"
+      row-key="id"
+      binary-state-sort
+      hide-bottom
+    >
+      <template v-slot:top>
+        <q-btn v-if="poll" dense flat push @click="stopPoll" icon="stop" label="Stop Live Refresh" />
+        <q-btn
+          v-else
+          dense
+          flat
+          push
+          @click="startPoll"
+          icon="play_arrow"
+          label="Resume Live Refresh"
+        />
+        <q-space />
+        <q-input v-model="filter" outlined label="Search" dense clearable>
+          <template v-slot:prepend>
+            <q-icon name="search" />
+          </template>
+        </q-input>
+      </template>
+      <template slot="body" slot-scope="props" :props="props">
+        <q-tr :props="props">
+          <q-menu context-menu>
+            <q-list dense style="min-width: 200px">
+              <q-item clickable v-close-popup @click="killProc(props.row.pid, props.row.name)">
+                <q-item-section thumbnail>
+                  <q-icon name="fas fa-trash-alt" size="xs" />
+                </q-item-section>
+                <q-item-section>End task</q-item-section>
+              </q-item>
+            </q-list>
+          </q-menu>
+          <q-td>{{ props.row.name }}</q-td>
+          <q-td>{{ Math.ceil(props.row.cpu_percent) }}%</q-td>
+          <q-td>{{ convert(props.row.memory_percent) }} MB</q-td>
+          <q-td>{{ props.row.username }}</q-td>
+          <q-td>{{ props.row.pid }}</q-td>
+          <q-td>{{ props.row.status }}</q-td>
+        </q-tr>
+      </template>
+    </q-table>
+  </div>
+</template>
+
+<script>
+import axios from "axios";
+import mixins from "@/mixins/mixins";
+
+export default {
+  name: "ProcessManager",
+  props: ["pk"],
+  mixins: [mixins],
+  data() {
+    return {
+      polling: null,
+      mem: null,
+      poll: true,
+      procs: [],
+      filter: "",
+      pagination: {
+        rowsPerPage: 99999,
+        sortBy: "cpu_percent",
+        descending: true
+      },
+      columns: [
+        {
+          name: "name",
+          label: "Name",
+          field: "name",
+          align: "left",
+          sortable: true
+        },
+        {
+          name: "cpu_percent",
+          label: "CPU",
+          field: "cpu_percent",
+          align: "left",
+          sortable: true
+        },
+        {
+          name: "memory_percent",
+          label: "Memory",
+          field: "memory_percent",
+          align: "left",
+          sortable: true
+        },
+        {
+          name: "username",
+          label: "User",
+          field: "username",
+          align: "left",
+          sortable: true
+        },
+        {
+          name: "pid",
+          label: "PID",
+          field: "pid",
+          align: "left",
+          sortable: true
+        },
+        {
+          name: "status",
+          label: "Status",
+          field: "status",
+          align: "left",
+          sortable: true
+        }
+      ]
+    };
+  },
+  methods: {
+    getProcesses() {
+      this.procs = [];
+      this.$q.loading.show({ message: "Loading Processes..." });
+      axios
+        .get(`/agents/${this.pk}/getprocs/`)
+        .then(r => {
+          this.procs = r.data;
+          this.$q.loading.hide();
+        })
+        .catch(e => {
+          this.$q.loading.hide();
+          this.notifyError(e.response.data.error);
+        });
+    },
+    refreshProcs() {
+      this.polling = setInterval(() => {
+        axios
+          .get(`/agents/${this.pk}/getprocs/`)
+          .then(r => (this.procs = r.data))
+          .catch(() => this.notifyError("Unable to contact the agent"));
+      }, 10000);
+    },
+    killProc(pid, name) {
+      this.$q.loading.show({ message: `Attempting to kill process ${name}` });
+      axios
+        .get(`/agents/${this.pk}/${pid}/killproc/`)
+        .then(r => {
+          this.$q.loading.hide();
+          this.notifySuccess(`${name} was terminated!`);
+        })
+        .catch(e => {
+          this.$q.loading.hide();
+          this.notifyError(e.response.data.error);
+        });
+    },
+    stopPoll() {
+      this.poll = false;
+      clearInterval(this.polling);
+    },
+    startPoll() {
+      this.poll = true;
+      axios
+        .get(`/agents/${this.pk}/getprocs/`)
+        .then(r => (this.procs = r.data));
+      this.refreshProcs();
+    },
+    getAgent() {
+      axios
+        .get(`/agents/${this.pk}/agentdetail/`)
+        .then(r => (this.mem = r.data.total_ram));
+    },
+    convert(percent) {
+      const mb = this.mem * 1024;
+      return Math.ceil((percent * mb) / 100).toLocaleString();
+    }
+  },
+  beforeDestroy() {
+    clearInterval(this.polling);
+  },
+  created() {
+    this.getAgent();
+    // disable loading bar
+    this.$q.loadingBar.setDefaults({ size: "0px" });
+
+    this.getProcesses();
+  },
+  mounted() {
+    this.refreshProcs();
+  }
+};
+</script>
+
+<style lang="stylus">
+.procs-sticky-header-table {
+  /* max height is important */
+  .q-table__middle {
+    max-height: 650px;
+  }
+
+  .q-table__top, .q-table__bottom, thead tr:first-child th {
+    background-color: #f5f4f2;
+  }
+
+  thead tr:first-child th {
+    position: sticky;
+    top: 0;
+    opacity: 1;
+    z-index: 1;
+  }
+}
+</style>
\ No newline at end of file
diff --git a/src/views/RemoteBackground.vue b/src/views/RemoteBackground.vue
index fc99468..968b757 100644
--- a/src/views/RemoteBackground.vue
+++ b/src/views/RemoteBackground.vue
@@ -13,7 +13,8 @@
       <q-tab name="terminal" icon="fas fa-terminal" label="Terminal" />
       <q-tab name="filebrowser" icon="far fa-folder-open" label="File Browser" />
       <q-tab name="services" icon="fas fa-cogs" label="Services" />
-      <q-tab name="eventlog" icon="fas fa-cogs" label="Event Log" />
+      <q-tab name="processes" icon="fas fa-chart-area" label="Processes" />
+      <q-tab name="eventlog" icon="fas fa-clipboard-list" label="Event Log" />
     </q-tabs>
     <q-separator />
     <q-tab-panels v-model="tab">
@@ -24,6 +25,9 @@
         >
         </iframe>
       </q-tab-panel>
+      <q-tab-panel name="processes">
+        <ProcessManager :pk="pk" />
+      </q-tab-panel>
       <q-tab-panel name="services">
         <Services :pk="pk" />
       </q-tab-panel>
@@ -43,6 +47,7 @@
 
 <script>
 import axios from "axios";
+import ProcessManager from "@/components/ProcessManager";
 import Services from "@/components/Services";
 import EventLog from "@/components/EventLog";
 
@@ -50,7 +55,8 @@ export default {
   name: "RemoteBackground",
   components: {
     Services,
-    EventLog
+    EventLog,
+    ProcessManager
   },
   data() {
     return {