From d5f81773a8a26b7ce88c14cc21d5a2e587c644f3 Mon Sep 17 00:00:00 2001
From: Yukino Song <nutosservice@gmail.com>
Date: Wed, 22 Jan 2025 22:28:57 +0800
Subject: [PATCH] Per client DO/UNDO commands frontend part - done

---
 src/confighttp.cpp                            |  3 +
 .../assets/web/configs/tabs/General.vue       | 16 ++--
 src_assets/common/assets/web/pin.html         | 93 ++++++++++++++++++-
 .../assets/web/public/assets/locale/en.json   |  4 +
 4 files changed, 105 insertions(+), 11 deletions(-)

diff --git a/src/confighttp.cpp b/src/confighttp.cpp
index e1781ff2..a321b9bd 100644
--- a/src/confighttp.cpp
+++ b/src/confighttp.cpp
@@ -1223,6 +1223,9 @@ namespace confighttp {
     pt::ptree outputTree;
     outputTree.put("status", false);
     outputTree.add_child("named_certs", named_certs);
+  #ifdef _WIN32
+    outputTree.put("platform", "windows");
+  #endif
     outputTree.put("status", true);
     send_response(response, outputTree);
   }
diff --git a/src_assets/common/assets/web/configs/tabs/General.vue b/src_assets/common/assets/web/configs/tabs/General.vue
index 807a9cae..f89c1202 100644
--- a/src_assets/common/assets/web/configs/tabs/General.vue
+++ b/src_assets/common/assets/web/configs/tabs/General.vue
@@ -23,13 +23,17 @@ const serverCmdTemplate = {
   cmd: ""
 }
 
-function addCmd(cmdArr, template) {
+function addCmd(cmdArr, template, idx) {
   const _tpl = Object.assign({}, template);
 
   if (props.platform === 'windows') {
     _tpl.elevated = false;
   }
-  cmdArr.push(_tpl);
+  if (idx < 0) {
+    cmdArr.push(_tpl);
+  } else {
+    cmdArr.splice(idx + 1, 0, _tpl);
+  }
 }
 
 function removeCmd(cmdArr, index) {
@@ -129,14 +133,14 @@ onMounted(() => {
             <button class="btn btn-danger me-2" @click="removeCmd(globalPrepCmd, i)">
               <i class="fas fa-trash"></i>
             </button>
-            <button class="btn btn-success" @click="addCmd(globalPrepCmd, prepCmdTemplate)">
+            <button class="btn btn-success" @click="addCmd(globalPrepCmd, prepCmdTemplate, i)">
               <i class="fas fa-plus"></i>
             </button>
           </td>
         </tr>
         </tbody>
       </table>
-      <button class="ms-0 mt-2 btn btn-success" style="margin: 0 auto" @click="addCmd(globalPrepCmd, prepCmdTemplate)">
+      <button class="ms-0 mt-2 btn btn-success" style="margin: 0 auto" @click="addCmd(globalPrepCmd, prepCmdTemplate, -1)">
         &plus; {{ $t('config.add') }}
       </button>
     </div>
@@ -178,14 +182,14 @@ onMounted(() => {
             <button class="btn btn-danger me-2" @click="removeCmd(serverCmd, i)">
               <i class="fas fa-trash"></i>
             </button>
-            <button class="btn btn-success" @click="addCmd(serverCmd, serverCmdTemplate)">
+            <button class="btn btn-success" @click="addCmd(serverCmd, serverCmdTemplate, i)">
               <i class="fas fa-plus"></i>
             </button>
           </td>
         </tr>
         </tbody>
       </table>
-      <button class="ms-0 mt-2 btn btn-success" style="margin: 0 auto" @click="addCmd(serverCmd, serverCmdTemplate)">
+      <button class="ms-0 mt-2 btn btn-success" style="margin: 0 auto" @click="addCmd(serverCmd, serverCmdTemplate, -1)">
         &plus; {{ $t('config.add') }}
       </button>
     </div>
diff --git a/src_assets/common/assets/web/pin.html b/src_assets/common/assets/web/pin.html
index ce7e002c..7086992c 100644
--- a/src_assets/common/assets/web/pin.html
+++ b/src_assets/common/assets/web/pin.html
@@ -104,6 +104,48 @@
                 </button>
               </div>
             </div>
+
+            <!-- connect/disconnect commands -->
+            <div class="mb-3 mt-2 d-flex flex-column" v-for="cmdType in ['do', 'undo']">
+              <label class="mb-0 orm-label">{{ $t(`config.client_${cmdType}_cmd`) }}</label>
+              <div class="form-text">{{ $t(`config.client_${cmdType}_cmd_desc`) }} <a href="https://github.com/ClassicOldSong/Apollo/wiki/Client-Commands" target="_blank">{{ $t('_common.learn_more') }}</a></div>
+              <table class="mt-2 table" v-if="client[`edit_${cmdType}`].length > 0">
+                <thead>
+                  <tr>
+                    <th style="width: 80%"><i class="fas fa-terminal"></i> {{ $t('_common.cmd_val') }}</th>
+                    <th style="min-width: 10em; max-width: 12em;" v-if="platform === 'windows'">
+                      <i class="fas fa-shield-alt"></i> {{ $t('_common.run_as') }}
+                    </th>
+                    <th style="min-width: 110px;"></th>
+                  </tr>
+                </thead>
+                <tbody>
+                <tr v-for="(c, i) in client[`edit_${cmdType}`]">
+                  <td>
+                    <input type="text" class="form-control monospace" v-model="c.cmd" />
+                  </td>
+                  <td v-if="platform === 'windows'">
+                    <div class="form-check">
+                      <input type="checkbox" class="form-check-input" :id="`client-${cmdType}-cmd-admin-${i}`" v-model="c.elevated"
+                             true-value="true" false-value="false" />
+                      <label :for="`client-${cmdType}-cmd-admin-${i}`" class="form-check-label">{{ $t('_common.elevated') }}</label>
+                    </div>
+                  </td>
+                  <td>
+                    <button class="btn btn-danger me-2" @click="removeCmd(client[`edit_${cmdType}`], i)">
+                      <i class="fas fa-trash"></i>
+                    </button>
+                    <button class="btn btn-success" @click="addCmd(client[`edit_${cmdType}`], i)">
+                      <i class="fas fa-plus"></i>
+                    </button>
+                  </td>
+                </tr>
+                </tbody>
+              </table>
+              <button class="ms-0 mt-2 btn btn-success" style="margin: 0 auto" @click="addCmd(client[`edit_${cmdType}`], -1)">
+                &plus; {{ $t('config.add') }}
+              </button>
+            </div>
           </div>
           <div v-else class="list-group-item d-flex align-items-center">
             <div class="p-2 flex-grow-1 d-flex align-items-center">
@@ -272,6 +314,7 @@
 
   const data = () => {
     return {
+      platform: '',
       editingHost: false,
       currentTab: location.hash || '#OTP',
       otp: '',
@@ -290,6 +333,11 @@
     }
   }
 
+  const cmdTpl = {
+    cmd: '',
+    elevated: 'false'
+  }
+
   let app = createApp({
     components: {
       Navbar
@@ -413,6 +461,17 @@
       clickedApplyBanner() {
         this.showApplyMessage = false;
       },
+      addCmd(arr, idx) {
+        const newCmd = Object.assign({}, cmdTpl);
+        if (idx < 0) {
+          arr.push(newCmd);
+        } else {
+          arr.splice(idx + 1, 0, newCmd);
+        }
+      },
+      removeCmd(arr, idx) {
+        arr.splice(idx, 1);
+      },
       editClient(client) {
         if (currentEditingClient) {
           this.cancelEdit(currentEditingClient);
@@ -421,6 +480,10 @@
         client.editing = true;
         client.editPerm = client.perm;
         client.editName = client.name;
+        client.edit_do = JSON.parse(JSON.stringify(client.do || []));
+        client.edit_undo = JSON.parse(JSON.stringify(client.undo || []));
+
+        console.log(client.do, client.undo)
       },
       cancelEdit(client) {
         client.editing = false;
@@ -434,7 +497,27 @@
         const editedClient = {
           uuid: client.uuid,
           name: client.editName,
-          perm: client.editPerm & permissionMapping._all
+          perm: client.editPerm & permissionMapping._all,
+          do: client.edit_do.reduce((filtered, {cmd: _cmd, elevated}) => {
+            const cmd = _cmd.trim()
+            if (cmd) {
+              filtered.push({
+                cmd,
+                elevated
+              })
+            }
+            return filtered
+          }, []),
+          undo: client.edit_undo.reduce((filtered, {cmd: _cmd, elevated}) => {
+            const cmd = _cmd.trim()
+            if (cmd) {
+              filtered.push({
+                cmd,
+                elevated
+              })
+            }
+            return filtered
+          }, [])
         }
         fetch("./api/clients/update", {
           credentials: 'include',
@@ -515,9 +598,9 @@
         fetch("./api/clients/list", { credentials: 'include' })
           .then((response) => response.json())
           .then((response) => {
-            const clientList = document.querySelector("#client-list");
             if (response.status === 'true' && response.named_certs && response.named_certs.length) {
-              this.clients = response.named_certs.map(({name, uuid, perm, connected}) => {
+              this.platform = response.platform
+              this.clients = response.named_certs.map(({name, uuid, perm, connected, do: _do, undo}) => {
                 const permInt = parseInt(perm, 10);
                 return {
                   name,
@@ -525,8 +608,8 @@
                   perm: permInt,
                   connected: connected === 'true',
                   editing: false,
-                  editPerm: permInt,
-                  editName: name
+                  do: _do,
+                  undo
                 }
               })
               currentEditingClient = null;
diff --git a/src_assets/common/assets/web/public/assets/locale/en.json b/src_assets/common/assets/web/public/assets/locale/en.json
index 292141a4..b0bb8525 100644
--- a/src_assets/common/assets/web/public/assets/locale/en.json
+++ b/src_assets/common/assets/web/public/assets/locale/en.json
@@ -169,6 +169,10 @@
     "channels": "Maximum Connected Clients",
     "channels_desc_1": "Apollo can allow a single streaming session to be shared with multiple clients simultaneously.",
     "channels_desc_2": "Some hardware encoders may have limitations that reduce performance with multiple streams.",
+    "client_do_cmd": "Client connect commands",
+    "client_do_cmd_desc": "Commands to be executed when client connects. All of the commands are executed detached.",
+    "client_undo_cmd": "Client disconnect commands",
+    "client_undo_cmd_desc": "Commands to be executed when client disconnects. All of the commands are executed detached.",
     "coder_cabac": "cabac -- context adaptive binary arithmetic coding - higher quality",
     "coder_cavlc": "cavlc -- context adaptive variable-length coding - faster decode",
     "configuration": "Configuration",