From c14041f82bd8169a159df3de07c0a0babc589769 Mon Sep 17 00:00:00 2001
From: fangxiang <fangxiang@cloudview.work>
Date: Wed, 16 Mar 2022 16:45:52 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=9C=A8=E7=BA=BF=E6=B3=A8?=
 =?UTF-8?q?=E5=86=8C=E9=80=BB=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/components/RegisterDialog.vue | 113 +++++++++++++++++++++++++++++-
 src/entities/WSProtocol.ts        |   1 -
 2 files changed, 111 insertions(+), 3 deletions(-)

diff --git a/src/components/RegisterDialog.vue b/src/components/RegisterDialog.vue
index 22771c9..daaf706 100644
--- a/src/components/RegisterDialog.vue
+++ b/src/components/RegisterDialog.vue
@@ -55,6 +55,7 @@
             <div>
               <q-btn
                 :loading="loading"
+                :disable="loading"
                 flat
                 round
                 icon="close"
@@ -86,6 +87,8 @@
                     val="offline"
                     :label="$t('offline register')"
                     class="offset-md-1 col"
+                    :loading="loading"
+                    :disable="loading"
                   />
                   <q-radio
                     v-model="register_type"
@@ -94,6 +97,8 @@
                     val="online"
                     :label="$t('online register')"
                     class="col"
+                    :loading="loading"
+                    :disable="loading"
                   />
                 </div>
               </q-item-section>
@@ -121,6 +126,8 @@
                   flat
                   icon="content_copy"
                   color="green"
+                  :loading="loading"
+                  :disable="loading"
                   @click="
                     copyToClipboard(register_code).then(() =>
                       $q.notify({
@@ -149,6 +156,8 @@
                   v-model="active_code"
                   lazy-rules
                   autofocus
+                  :loading="loading"
+                  :disable="loading"
                   :rules="[
                     (val) =>
                       (val && val.length > 0) || $t('Please type something'),
@@ -160,6 +169,8 @@
                   icon="attach_file"
                   round
                   flat
+                  :loading="loading"
+                  :disable="loading"
                   class="rotate-90"
                   color="green"
                   @click="$refs.select_file_dialog.pickFiles($event)"
@@ -185,8 +196,10 @@
               </q-item-section>
               <q-item-section>
                 <q-input
-                  v-model="active_code"
+                  v-model="server_address"
                   autofocus
+                  :loading="loading"
+                  :disable="loading"
                   lazy-rules
                   :rules="[
                     (val) =>
@@ -208,6 +221,8 @@
                     val="forever"
                     :label="$t('forever active')"
                     class="offset-md-1 col"
+                    :loading="loading"
+                    :disable="loading"
                   />
                   <q-radio
                     v-model="active_type"
@@ -216,6 +231,8 @@
                     val="limit"
                     :label="$t('time limit active')"
                     class="col"
+                    :loading="loading"
+                    :disable="loading"
                   />
                 </div>
               </q-item-section>
@@ -229,7 +246,9 @@
                   v-model="active_hour"
                   type="number"
                   min="1"
-                  max="65535"
+                  max="120"
+                  :loading="loading"
+                  :disable="loading"
                 >
                   <template v-slot:append>
                     {{ $t("day") }}
@@ -245,6 +264,7 @@
         <q-card-actions align="right">
           <q-btn
             :loading="loading"
+            :disable="loading"
             flat
             :label="$t('Cancel')"
             color="primary"
@@ -255,6 +275,7 @@
             flat
             :label="$t('register')"
             :loading="loading"
+            :disable="loading"
             type="submit"
             color="primary"
           />
@@ -308,6 +329,7 @@ export default defineComponent({
     const secret_key = ref("");
     const active_hour = ref(1);
     const ext_flag = ref(false);
+    const server_address = ref("");
 
     const trial_days = ref(0);
     const last_days = ref(0);
@@ -366,6 +388,7 @@ export default defineComponent({
       register_flag,
       licence_file,
       select_file_dialog,
+      server_address,
       copyToClipboard,
       isShow() {
         return show_dialog.value;
@@ -375,7 +398,93 @@ export default defineComponent({
 
       async onSubmit() {
         loading.value = true;
+        if (active_hour.value > 30 * 2) {
+          active_hour.value = 30 * 2;
+        }
+        if (active_hour.value < 1) {
+          active_hour.value = 1;
+        }
+
         try {
+          if (register_type.value == "online") {
+            let licence_str = "";
+            const requested = await new Promise((resolve) => {
+              const timeout_handler = setTimeout(() => {
+                resolve(false);
+              }, 1000 * 5);
+              try {
+                let ws = new WebSocket(
+                  "ws://" + server_address.value + ":37631"
+                );
+                ws.onmessage = (data) => {
+                  interface IResult {
+                    id: 1;
+                    action: "license";
+                    result: {
+                      err?: "";
+                      update_key: "";
+                      lic: "";
+                    };
+                  }
+                  try {
+                    const result = JSON.parse(data.data) as IResult;
+                    if (result && !result.result.err) {
+                      licence_str = result.result.lic;
+                    }
+                    console.log(result);
+                    clearTimeout(timeout_handler);
+                    resolve(true);
+                  } catch {
+                    clearTimeout(timeout_handler);
+                    resolve(false);
+                  }
+                };
+                ws.onerror = () => {
+                  clearTimeout(timeout_handler);
+                  resolve(false);
+                };
+                ws.onclose = ws.onerror;
+                ws.onopen = () => {
+                  ws.send(
+                    JSON.stringify({
+                      id: 1,
+                      action: "license",
+                      params: {
+                        name: "MSFusion",
+                        days:
+                          active_type.value == "forever"
+                            ? 0
+                            : parseInt(active_hour.value.toString()),
+                        minutes: 0,
+                        key: register_code.value,
+                      },
+                    })
+                  );
+                };
+              } catch (e) {
+                console.error(e);
+                clearTimeout(timeout_handler);
+                resolve(false);
+              }
+            });
+            console.log(licence_str);
+            if (requested && licence_str) {
+              active_code.value = licence_str;
+              console.log(licence_str);
+            } else {
+              $q.notify({
+                color: "negative",
+                icon: "warning",
+                message: $t.t("register device") + $t.t("fail") + "!",
+                position: "top",
+                timeout: 1500,
+              });
+
+              loading.value = false;
+              return;
+            }
+          }
+
           let success = false;
           const response = await GlobalData.getInstance()
             .getCurrentClient()
diff --git a/src/entities/WSProtocol.ts b/src/entities/WSProtocol.ts
index efd76e5..3f3040e 100644
--- a/src/entities/WSProtocol.ts
+++ b/src/entities/WSProtocol.ts
@@ -1496,7 +1496,6 @@ export namespace Protocol {
     ) {
       super();
       this.rpc_id = rcp_id ?? 0;
-      this.timeout = 1000 * 60 * 5; // 6911写固件比较耗时, 请求超时5分钟
       this.command = Protocol.Commands.kRpcRegisterDevice;
       this.timestamp = new Date().getMilliseconds();
       this.register_code = register_code;