From e91ca71812135b23fab7515645cf0785d2a106fa Mon Sep 17 00:00:00 2001
From: ibuler <ibuler@qq.com>
Date: Sat, 7 May 2022 15:00:54 +0800
Subject: [PATCH] =?UTF-8?q?perf:=20=E4=BF=AE=E6=94=B9=E4=BD=BF=E7=94=A8=20?=
 =?UTF-8?q?rsa=20=E5=8A=A0=E5=AF=86=E6=95=B0=E6=8D=AE?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 package.json                                  |  1 +
 .../FormFields/EncryptedPassword.vue          | 40 +++++++++++++++++++
 src/components/FormFields/PasswordInput.vue   |  5 ++-
 src/components/FormFields/UserPassword.vue    |  1 +
 src/components/FormFields/index.js            |  7 +++-
 .../GenericCreateUpdateForm/index.vue         | 19 +++++++++
 src/utils/jms.js                              | 15 +++++++
 src/views/assets/Asset/TreeMenu.vue           |  3 +-
 .../CommonUser/telnet.vue                     |  1 +
 .../profile/ProfileUpdate/PasswordUpdate.vue  | 11 +++--
 yarn.lock                                     |  5 +++
 11 files changed, 97 insertions(+), 11 deletions(-)
 create mode 100644 src/components/FormFields/EncryptedPassword.vue

diff --git a/package.json b/package.json
index bb5f32208..4b24583d1 100644
--- a/package.json
+++ b/package.json
@@ -32,6 +32,7 @@
     "install": "^0.13.0",
     "jquery": "^3.5.0",
     "js-cookie": "2.2.0",
+    "jsencrypt": "^3.2.1",
     "krry-transfer": "^1.7.3",
     "less": "^3.10.3",
     "less-loader": "^5.0.0",
diff --git a/src/components/FormFields/EncryptedPassword.vue b/src/components/FormFields/EncryptedPassword.vue
new file mode 100644
index 000000000..390610a09
--- /dev/null
+++ b/src/components/FormFields/EncryptedPassword.vue
@@ -0,0 +1,40 @@
+<template>
+  <PasswordInput
+    :value="value"
+    :attrs="attrs"
+    @input="handleInput"
+  />
+</template>
+
+<script>
+import PasswordInput from './PasswordInput'
+import { encryptPassword } from '@/utils/jms'
+
+export default {
+  name: 'EncryptedPassword',
+  components: { PasswordInput },
+  props: {
+    value: {
+      type: String,
+      default: ''
+    }
+  },
+  data() {
+    return {
+      attrs: {
+      }
+    }
+  },
+  computed: {
+  },
+  methods: {
+    handleInput(value) {
+      const cipher = encryptPassword(value)
+      this.$emit('input', cipher)
+    }
+  }
+}
+</script>
+
+<style lang="scss" scoped>
+</style>
diff --git a/src/components/FormFields/PasswordInput.vue b/src/components/FormFields/PasswordInput.vue
index fcf234053..7911f663e 100644
--- a/src/components/FormFields/PasswordInput.vue
+++ b/src/components/FormFields/PasswordInput.vue
@@ -1,5 +1,5 @@
 <template>
-  <password
+  <Password
     :value="value"
     v-bind="iAttrs"
     class="el-input password-input"
@@ -32,7 +32,8 @@ export default {
       const defaultAttrs = {
         secureLength: 7,
         defaultClass: 'el-input__inner',
-        toggle: true
+        toggle: true,
+        showStrengthMeter: false
       }
       return Object.assign(defaultAttrs, this.attrs)
     }
diff --git a/src/components/FormFields/UserPassword.vue b/src/components/FormFields/UserPassword.vue
index b20923be1..15623da11 100644
--- a/src/components/FormFields/UserPassword.vue
+++ b/src/components/FormFields/UserPassword.vue
@@ -66,6 +66,7 @@ export default {
   data() {
     return {
       attrs: {
+        showStrengthMeter: true
       }
     }
   },
diff --git a/src/components/FormFields/index.js b/src/components/FormFields/index.js
index 383adb2f4..89d204fc8 100644
--- a/src/components/FormFields/index.js
+++ b/src/components/FormFields/index.js
@@ -1,6 +1,7 @@
 import DatetimeRangePicker from './DatetimeRangePicker'
 import Link from './Link'
 import PasswordInput from './PasswordInput'
+import EncryptedPassword from './EncryptedPassword'
 import Select2 from './Select2'
 import Swicher from './Swicher'
 import UploadField from './UploadField'
@@ -23,7 +24,8 @@ export default {
   WeekCronSelect,
   UpdateToken,
   JsonEditor,
-  Text
+  Text,
+  EncryptedPassword
 }
 
 export {
@@ -38,5 +40,6 @@ export {
   WeekCronSelect,
   UpdateToken,
   JsonEditor,
-  Text
+  Text,
+  EncryptedPassword
 }
diff --git a/src/layout/components/GenericCreateUpdateForm/index.vue b/src/layout/components/GenericCreateUpdateForm/index.vue
index e2a60fd85..648e4b28a 100644
--- a/src/layout/components/GenericCreateUpdateForm/index.vue
+++ b/src/layout/components/GenericCreateUpdateForm/index.vue
@@ -16,6 +16,8 @@
 <script>
 import AutoDataForm from '@/components/AutoDataForm'
 import { getUpdateObjURL } from '@/utils/common'
+import { encryptPassword } from '@/utils/jms'
+
 export default {
   name: 'GenericCreateUpdateForm',
   components: {
@@ -218,6 +220,10 @@ export default {
     hasDetailInMsg: {
       type: Boolean,
       default: true
+    },
+    encryptedFields: {
+      type: Array,
+      default: () => ['password', 'token']
     }
   },
   data() {
@@ -265,10 +271,23 @@ export default {
     isUpdateMethod() {
       return ['put', 'patch'].indexOf(this.method.toLowerCase()) > -1
     },
+    encryptFields(values) {
+      values = { ...values }
+      for (const field of this.encryptedFields) {
+        let value = values[field]
+        if (!value || typeof value !== 'string') {
+          continue
+        }
+        value = encryptPassword(value)
+        values[field] = value
+      }
+      return values
+    },
     handleSubmit(values, formName, addContinue) {
       let handler = this.onSubmit || this.defaultOnSubmit
       handler = handler.bind(this)
       values = this.cleanFormValue(values)
+      values = this.encryptFields(values)
       return handler(values, formName, addContinue)
     },
     defaultOnSubmit(validValues, formName, addContinue) {
diff --git a/src/utils/jms.js b/src/utils/jms.js
index c57b8d35e..1f9cce32b 100644
--- a/src/utils/jms.js
+++ b/src/utils/jms.js
@@ -1,5 +1,7 @@
 import store from '@/store'
 import { constantRoutes } from '@/router'
+import { JSEncrypt } from 'jsencrypt'
+import VueCookie from 'vue-cookie'
 
 export function openTaskPage(taskId) {
   window.open(`/#/ops/celery/task/${taskId}/log/`, '', 'width=900,height=600')
@@ -132,3 +134,16 @@ export function getConstRouteName() {
   addRoutes(names, constRoutes)
   return names
 }
+
+export function encryptPassword(password) {
+  if (!password) {
+    return ''
+  }
+  var rsaPublicKeyText = VueCookie.get('jms_public_key')
+    .replaceAll('"', '')
+  var rsaPublicKey = atob(rsaPublicKeyText)
+  var jsencrypt = new JSEncrypt()
+  jsencrypt.setPublicKey(rsaPublicKey)
+  var value = jsencrypt.encrypt(password)
+  return value
+}
diff --git a/src/views/assets/Asset/TreeMenu.vue b/src/views/assets/Asset/TreeMenu.vue
index fdc847b42..08ab70ed5 100644
--- a/src/views/assets/Asset/TreeMenu.vue
+++ b/src/views/assets/Asset/TreeMenu.vue
@@ -69,7 +69,8 @@ export default {
   props: {
     tree: {
       type: Object,
-      require: true
+      require: true,
+      default: () => ({})
     }
   },
   data() {
diff --git a/src/views/assets/SystemUser/SystemUserCreateUpdate/CommonUser/telnet.vue b/src/views/assets/SystemUser/SystemUserCreateUpdate/CommonUser/telnet.vue
index 851e6c890..7894bd97e 100644
--- a/src/views/assets/SystemUser/SystemUserCreateUpdate/CommonUser/telnet.vue
+++ b/src/views/assets/SystemUser/SystemUserCreateUpdate/CommonUser/telnet.vue
@@ -29,6 +29,7 @@ export default {
         [this.$t('common.Command filter'), ['cmd_filters']],
         [this.$t('common.Other'), ['priority', 'comment']]
       ],
+      encryptedFields: ['password'],
       fieldsMeta: {
         login_mode: fields.login_mode,
         username: fields.username,
diff --git a/src/views/profile/ProfileUpdate/PasswordUpdate.vue b/src/views/profile/ProfileUpdate/PasswordUpdate.vue
index b0cdb22b2..b6fc6c1b5 100644
--- a/src/views/profile/ProfileUpdate/PasswordUpdate.vue
+++ b/src/views/profile/ProfileUpdate/PasswordUpdate.vue
@@ -2,6 +2,7 @@
   <IBox>
     <GenericCreateUpdateForm
       :fields="fields"
+      :encrypted-fields="encryptedFields"
       :fields-meta="fieldsMeta"
       :initial="object"
       :url="url"
@@ -17,6 +18,7 @@ import GenericCreateUpdateForm from '@/layout/components/GenericCreateUpdateForm
 import UserPassword from '@/components/FormFields/UserPassword'
 import { IBox } from '@/components'
 import rules from '@/components/DataForm/rules'
+import { PasswordInput } from '@/components/FormFields'
 
 export default {
   name: 'PasswordUpdate',
@@ -34,12 +36,11 @@ export default {
     return {
       url: '/api/v1/users/profile/password/',
       fields: ['old_password', 'new_password', 'new_password_again'],
+      encryptedFields: ['old_password', 'new_password', 'new_password_again'],
       fieldsMeta: {
         old_password: {
           label: this.$t('users.OldPassword'),
-          el: {
-            type: 'password'
-          }
+          component: PasswordInput
         },
         new_password: {
           label: this.$t('users.NewPassword'),
@@ -48,9 +49,7 @@ export default {
         },
         new_password_again: {
           label: this.$t('users.ConfirmPassword'),
-          el: {
-            type: 'password'
-          }
+          component: PasswordInput
         }
       },
       updateSuccessNextRoute: {
diff --git a/yarn.lock b/yarn.lock
index 0ffadf87c..cf74f6fb5 100644
--- a/yarn.lock
+++ b/yarn.lock
@@ -6338,6 +6338,11 @@ jsdom@^11.5.1:
     ws "^5.2.0"
     xml-name-validator "^3.0.0"
 
+jsencrypt@^3.2.1:
+  version "3.2.1"
+  resolved "https://registry.npmmirror.com/jsencrypt/-/jsencrypt-3.2.1.tgz#09766983cc760088ff26b12fe7e574252af97727"
+  integrity sha512-k1sD5QV0KPn+D8uG9AdGzTQuamt82QZ3A3l6f7TRwMU6Oi2Vg0BsL+wZIQBONcraO1pc78ExMdvmBBJ8WhNYUA==
+
 jsesc@^1.3.0:
   version "1.3.0"
   resolved "https://registry.yarnpkg.com/jsesc/-/jsesc-1.3.0.tgz#46c3fec8c1892b12b0833db9bc7622176dbab34b"