From 2511d6ff9b12cc48834dccc1479f76c3321f57f1 Mon Sep 17 00:00:00 2001
From: fangxiang <fangxiang@cloudview.work>
Date: Sat, 26 Feb 2022 13:39:39 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E4=BF=AE=E6=94=B9=E8=BD=AE?=
 =?UTF-8?q?=E8=AF=A2=E5=8A=9F=E8=83=BD,=E4=BF=AE=E5=A4=8D=E6=B7=BB?=
 =?UTF-8?q?=E5=8A=A0=E8=BD=AE=E8=AF=A2=E4=B8=8D=E6=9B=B4=E6=96=B0=E8=BD=AE?=
 =?UTF-8?q?=E8=AF=A2=E5=88=97=E8=A1=A8=E7=9A=84BUG?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/common/ClientConnection.ts          |  28 ++++-
 src/common/RemoteDataExangeProcesser.ts |   2 +-
 src/components/PollingSettingDialog.vue |  30 +++++-
 src/components/PollingTree.vue          | 134 +++++++++++++++++++-----
 src/entities/WSProtocol.ts              |  57 ++++++++++
 src/i18n/zh-CN/index.ts                 |   5 +
 src/pages/WallPage.vue                  |  58 +++++-----
 src/store/index.ts                      |   1 -
 8 files changed, 255 insertions(+), 60 deletions(-)

diff --git a/src/common/ClientConnection.ts b/src/common/ClientConnection.ts
index 10d2443..f03e1f0 100644
--- a/src/common/ClientConnection.ts
+++ b/src/common/ClientConnection.ts
@@ -1,3 +1,4 @@
+import { PollingEntity } from "./../entities/PollingEntity";
 import ReconnectingWebSocket from "reconnecting-websocket";
 import NormalWindowRequestEntity from "src/entities/NormalWindowRequestEntity";
 import { PlanEntity } from "src/entities/PlanEntity";
@@ -375,6 +376,30 @@ export default class ClientConnection {
     }
   }
 
+  public async addPolling(
+    group_uuid: string,
+    name: string,
+    datas: StringKeyValueEntity[]
+  ) {
+    try {
+      return await this.doRpc<Protocol.AddPollingResponseEntity>(
+        new Protocol.AddPollingRequestEntity(0, group_uuid, name, datas, "")
+      );
+    } catch (e) {
+      console.error(e);
+    }
+  }
+
+  public async editPolling(entity: PollingEntity) {
+    try {
+      return await this.doRpc<Protocol.EditPollingResponseEntity>(
+        new Protocol.EditPollingRequestEntity(0, entity)
+      );
+    } catch (e) {
+      console.error(e);
+    }
+  }
+
   public async deletePolling(uuid: string) {
     try {
       return await this.doRpc<Protocol.DeletePollingResponseEntity>(
@@ -745,10 +770,11 @@ export default class ClientConnection {
 
   public async setPollingData(
     window_id: number,
+    name: string,
     datas?: StringKeyValueEntity[]
   ) {
     return await this.doRpc<Protocol.SetPollingDataResponseEntity>(
-      new Protocol.SetPollingDataRequestEntity(window_id, datas)
+      new Protocol.SetPollingDataRequestEntity(window_id, name, datas)
     );
   }
 
diff --git a/src/common/RemoteDataExangeProcesser.ts b/src/common/RemoteDataExangeProcesser.ts
index 0ecaefc..6a2d5e4 100644
--- a/src/common/RemoteDataExangeProcesser.ts
+++ b/src/common/RemoteDataExangeProcesser.ts
@@ -288,7 +288,7 @@ export default class RemoteDataExangeProcesser {
           }
           break;
 
-        case Protocol.Commands.kRpcAddPollingGroup:
+        case Protocol.Commands.kRpcAddPolling:
           {
             const temp = JSON.parse(
               notify.data
diff --git a/src/components/PollingSettingDialog.vue b/src/components/PollingSettingDialog.vue
index 71b0473..dff2906 100644
--- a/src/components/PollingSettingDialog.vue
+++ b/src/components/PollingSettingDialog.vue
@@ -40,6 +40,23 @@
 
         <q-card-section style="max-height: 70vh; width: 50vw" class="scroll">
           <q-list>
+            <q-item>
+              <q-item-section avatar>
+                <div class="text-h6 q-pb-md">
+                  {{ $t("polling name") }}:
+                </div></q-item-section
+              >
+              <q-item-section
+                ><q-input
+                  v-model="polling_name"
+                  lazy-rules
+                  :placeholder="$t('please input polling name')"
+                  :rules="[
+                    (val) =>
+                      (val && val.length > 0) || $t('Please type something'),
+                  ]"
+              /></q-item-section>
+            </q-item>
             <q-item>
               <q-item-section>
                 <q-table
@@ -194,6 +211,7 @@ export default defineComponent({
     let loading = ref(false);
     let datas: Ref<StringKeyValueEntity[]> = ref([]);
     let signals = ref(GlobalData.getInstance().signal_source);
+    let polling_name = ref("");
 
     const columns = [
       {
@@ -228,10 +246,11 @@ export default defineComponent({
       show_context_menu,
       target_dom,
       signals,
+      polling_name,
       loga(a: any) {
         console.log(a);
       },
-      showDialog(options: StringKeyValueEntity[] | null) {
+      showDialog(name: string, options: StringKeyValueEntity[] | null) {
         show_dialog.value = true;
         if (options) {
           datas.value = [];
@@ -239,8 +258,9 @@ export default defineComponent({
             datas.value.push(item);
           }
         }
+        polling_name.value = name ?? $t.t("new polling");
       },
-      showDialogAsync(options: StringKeyValueEntity[] | null) {
+      showDialogAsync(name: string, options: StringKeyValueEntity[] | null) {
         return new Promise((resolve) => {
           if (_resolve) {
             _resolve();
@@ -255,6 +275,7 @@ export default defineComponent({
             }
             signals.value = GlobalData.getInstance().signal_source;
           }
+          polling_name.value = name ?? $t.t("new polling");
         });
       },
       resetData() {
@@ -302,7 +323,10 @@ export default defineComponent({
         loading.value = true;
         try {
           if (_resolve) {
-            _resolve(datas.value);
+            _resolve({
+              name: polling_name.value,
+              datas: datas.value,
+            });
             _resolve = null;
           }
           show_dialog.value = false;
diff --git a/src/components/PollingTree.vue b/src/components/PollingTree.vue
index 0ec63c9..0ac2086 100644
--- a/src/components/PollingTree.vue
+++ b/src/components/PollingTree.vue
@@ -44,31 +44,21 @@
                   clickable
                   v-close-popup
                   v-ripple
-                  @click="
-                    $refs.polling_dialog.showDialog({
-                      type: 1,
-                      data: { uuid: prop.node.uuid },
-                    })
-                  "
+                  @click="addPolling($event, prop.node.item_data)"
                 >
                   <q-item-section avatar><q-icon name="add" /></q-item-section>
-                  <q-item-section>{{ $t("add polling item") }}</q-item-section>
+                  <q-item-section>{{ $t("add polling") }}</q-item-section>
                 </q-item>
                 <q-item
                   clickable
                   v-close-popup
                   v-if="
-                    prop.node.name == $t('root') ||
-                    (prop.node.is_group && prop.node.item_data)
+                    false &&
+                    (prop.node.name == $t('root') ||
+                      (prop.node.is_group && prop.node.item_data))
                   "
                   v-ripple
-                  @click="
-                    () =>
-                      $refs.group_dialog.showDialog({
-                        type: 1,
-                        data: prop.node,
-                      })
-                  "
+                  @click="addPollingGroup($event, prop.node.item_data)"
                 >
                   <q-item-section avatar
                     ><q-icon name="create_new_folder"
@@ -80,17 +70,8 @@
                   clickable
                   v-ripple
                   v-close-popup
-                  v-if="prop.node.item_data && false"
-                  @click="
-                    () =>
-                      (prop.node.is_group
-                        ? $refs.group_dialog
-                        : $refs.polling_dialog
-                      ).showDialog({
-                        type: 2,
-                        data: prop.node,
-                      })
-                  "
+                  v-if="!prop.node.is_group && prop.node.item_data"
+                  @click="editPolling($event, prop.node.item_data)"
                 >
                   <q-item-section avatar><q-icon name="edit" /></q-item-section>
                   <q-item-section>{{ $t("edit") }}</q-item-section>
@@ -117,10 +98,11 @@
       </q-tree>
     </q-scroll-area>
   </div>
+  <polling-setting-dialog ref="polling_setting_dialog" />
 </template>
 
 <script lang="ts">
-import { defineComponent, computed, onMounted, ref, nextTick } from "vue";
+import { defineComponent, computed, onMounted, ref, Ref, nextTick } from "vue";
 import { useStore } from "src/store";
 import { Common } from "src/common/Common";
 import GlobalData from "src/common/GlobalData";
@@ -129,10 +111,17 @@ import { useI18n } from "vue-i18n";
 import EventBus, { EventNamesDefine } from "src/common/EventBus";
 import { Protocol } from "src/entities/WSProtocol";
 import { NotifyMessage } from "src/common/ClientConnection";
-import { PollingTreeItemEntity } from "src/entities/PollingEntity";
+import {
+  PollingEntity,
+  PollingTreeItemEntity,
+} from "src/entities/PollingEntity";
+import PollingSettingDialog from "src/components/PollingSettingDialog.vue";
 
 export default defineComponent({
   name: "ComponentPollingTree",
+  components: {
+    PollingSettingDialog,
+  },
 
   setup() {
     const $store = useStore();
@@ -156,6 +145,7 @@ export default defineComponent({
     });
 
     const tree: any | null = ref(null);
+    const polling_setting_dialog: Ref<any> = ref(null);
 
     onMounted(async () => {
       while (!tree.value.nodes.length) {
@@ -166,6 +156,7 @@ export default defineComponent({
 
     return {
       tree,
+      polling_setting_dialog,
       tree_nodes,
       content_height,
       loga(a: any) {
@@ -209,6 +200,91 @@ export default defineComponent({
           e.dataTransfer.dropEffect = "move";
         }
       },
+      async editPolling(event: MouseEvent, data: PollingEntity) {
+        if (polling_setting_dialog.value) {
+          const result = await polling_setting_dialog.value.showDialogAsync(
+            data.name,
+            data.polling_signal_sources
+          );
+          if (result) {
+            const { name, datas } = result;
+            if (name && datas) {
+              try {
+                const new_data = new PollingEntity();
+                PollingEntity.copy(new_data, data);
+                new_data.name = name;
+                new_data.polling_signal_sources = datas;
+                const resposne = await GlobalData.getInstance()
+                  .getCurrentClient()
+                  ?.editPolling(new_data);
+                if (resposne) {
+                  if (resposne.success) {
+                    $q.notify({
+                      color: "positive",
+                      icon: "done",
+                      message: $t.t("edit polling") + $t.t("success") + "!",
+                      position: "top",
+                      timeout: 1500,
+                    });
+                  } else {
+                    $q.notify({
+                      color: "negative",
+                      icon: "warning",
+                      message: $t.t("edit polling") + $t.t("failed") + "!",
+                      position: "top",
+                      timeout: 1500,
+                    });
+                  }
+                }
+              } catch (e) {
+                console.log(e);
+              }
+            }
+          }
+        }
+      },
+      async addPolling(event: MouseEvent, data: any) {
+        if (polling_setting_dialog.value) {
+          const result = await polling_setting_dialog.value.showDialogAsync(
+            null,
+            []
+          );
+          if (result) {
+            const { name, datas } = result;
+            if (name && datas) {
+              try {
+                const resposne = await GlobalData.getInstance()
+                  .getCurrentClient()
+                  ?.addPolling(/* data.uuid */ "", name, datas);
+                if (resposne) {
+                  if (resposne.success) {
+                    $q.notify({
+                      color: "positive",
+                      icon: "done",
+                      message: $t.t("add polling") + $t.t("success") + "!",
+                      position: "top",
+                      timeout: 1500,
+                    });
+                  } else {
+                    $q.notify({
+                      color: "negative",
+                      icon: "warning",
+                      message: $t.t("add polling") + $t.t("failed") + "!",
+                      position: "top",
+                      timeout: 1500,
+                    });
+                  }
+                }
+              } catch (e) {
+                console.log(e);
+              }
+            }
+          }
+        }
+      },
+      addPollingGroup(event: MouseEvent, data: any) {
+        console.log(data);
+      },
     };
   },
 });
diff --git a/src/entities/WSProtocol.ts b/src/entities/WSProtocol.ts
index a94178b..f44462a 100644
--- a/src/entities/WSProtocol.ts
+++ b/src/entities/WSProtocol.ts
@@ -1166,6 +1166,60 @@ export namespace Protocol {
     }
   }
 
+  export class AddPollingRequestEntity extends Protocol.PacketEntity {
+    name: string;
+    group_uuid: string;
+    polling_signal_sources: StringKeyValueEntity[];
+    note: string;
+    constructor(
+      rcp_id?: number,
+      group_uuid?: string,
+      name?: string,
+      polling_signal_sources?: StringKeyValueEntity[],
+      note?: string
+    ) {
+      super();
+      this.rpc_id = rcp_id ?? 0;
+      this.command = Protocol.Commands.kRpcAddPolling;
+      this.name = name ?? "new pollong";
+      this.group_uuid = group_uuid ?? "";
+      this.polling_signal_sources = Array.isArray(polling_signal_sources)
+        ? polling_signal_sources
+        : [];
+      this.note = note ?? "";
+    }
+  }
+
+  export class AddPollingResponseEntity extends Protocol.PacketEntity {
+    success = false;
+
+    constructor() {
+      super();
+      this.command = Protocol.Commands.kRpcAddPolling;
+    }
+  }
+
+  export class EditPollingRequestEntity extends Protocol.PacketEntity {
+    entity: PollingEntity = new PollingEntity();
+    constructor(rcp_id?: number, entity?: PollingEntity) {
+      super();
+      this.rpc_id = rcp_id ?? 0;
+      this.command = Protocol.Commands.kRpcEditPolling;
+      if (entity) {
+        this.entity = entity;
+      }
+    }
+  }
+
+  export class EditPollingResponseEntity extends Protocol.PacketEntity {
+    success = false;
+
+    constructor() {
+      super();
+      this.command = Protocol.Commands.kRpcEditPolling;
+    }
+  }
+
   export class DeletePollingRequestEntity extends Protocol.PacketEntity {
     uuid: string = "";
     constructor(rcp_id?: number, uuid?: string) {
@@ -1699,9 +1753,11 @@ export namespace Protocol {
 
   export class SetPollingDataRequestEntity extends Protocol.PacketEntity {
     datas: StringKeyValueEntity[] = [];
+    name: string;
     window_id: number = 0;
     constructor(
       window_id: number,
+      name: string,
       datas?: StringKeyValueEntity[],
       rcp_id?: number
     ) {
@@ -1715,6 +1771,7 @@ export namespace Protocol {
         this.datas = datas;
       }
       this.window_id = window_id;
+      this.name = name;
     }
   }
 
diff --git a/src/i18n/zh-CN/index.ts b/src/i18n/zh-CN/index.ts
index b69cd23..93b1363 100644
--- a/src/i18n/zh-CN/index.ts
+++ b/src/i18n/zh-CN/index.ts
@@ -340,4 +340,9 @@ export default {
   usb7: "USB7",
   usb8: "USB8",
   polling: "轮询",
+  "polling name": "轮询名称",
+  "please input polling name": "请输入轮询名称",
+  "new polling": "新建轮询",
+  "add polling": "添加轮询",
+  "edit polling": "修改轮询",
 };
diff --git a/src/pages/WallPage.vue b/src/pages/WallPage.vue
index b498396..b6c0aa2 100644
--- a/src/pages/WallPage.vue
+++ b/src/pages/WallPage.vue
@@ -886,35 +886,43 @@ export default defineComponent({
           const polling = GlobalData.getInstance()._pollings.find(
             (element) => element && element.uuid == window.polling_uuid
           );
-          const datas = await polling_setting_dialog.value.showDialogAsync(
+          const signal_source = GlobalData.getInstance().signal_source.find(
+            (element) =>
+              element && element.uuid == window.signal_source_table_uuid
+          );
+          const result = await polling_setting_dialog.value.showDialogAsync(
+            polling ? polling.name : signal_source ? signal_source.name : null,
             polling ? polling.polling_signal_sources : []
           );
-          if (datas) {
-            try {
-              const resposne = await GlobalData.getInstance()
-                .getCurrentClient()
-                ?.setPollingData(window_id, datas);
-              if (resposne) {
-                if (resposne.success) {
-                  $q.notify({
-                    color: "positive",
-                    icon: "done",
-                    message: $t.t("set polling data") + $t.t("success") + "!",
-                    position: "top",
-                    timeout: 1500,
-                  });
-                } else {
-                  $q.notify({
-                    color: "negative",
-                    icon: "warning",
-                    message: $t.t("set polling data") + $t.t("failed") + "!",
-                    position: "top",
-                    timeout: 1500,
-                  });
+          if (result) {
+            const { name, datas } = result;
+            if (name && datas) {
+              try {
+                const resposne = await GlobalData.getInstance()
+                  .getCurrentClient()
+                  ?.setPollingData(window_id, name, datas);
+                if (resposne) {
+                  if (resposne.success) {
+                    $q.notify({
+                      color: "positive",
+                      icon: "done",
+                      message: $t.t("set polling data") + $t.t("success") + "!",
+                      position: "top",
+                      timeout: 1500,
+                    });
+                  } else {
+                    $q.notify({
+                      color: "negative",
+                      icon: "warning",
+                      message: $t.t("set polling data") + $t.t("failed") + "!",
+                      position: "top",
+                      timeout: 1500,
+                    });
+                  }
                 }
+              } catch (e) {
+                console.log(e);
               }
-            } catch (e) {
-              console.log(e);
             }
           }
         }
diff --git a/src/store/index.ts b/src/store/index.ts
index 9c972b0..011ddf1 100644
--- a/src/store/index.ts
+++ b/src/store/index.ts
@@ -791,7 +791,6 @@ export default store(function (/* { ssrContext } */) {
             ),
           playload
         );
-        console.log(state.polling_tree);
       },
       // polling tree end
       setDeviceIpAddress(state: StateInterface, playload?: any) {