From 913a09276df03645ac2b15e399130892302419f5 Mon Sep 17 00:00:00 2001
From: Abhishek Kumar <abhishek.mrt22@gmail.com>
Date: Mon, 4 May 2020 15:32:37 +0530
Subject: [PATCH] compute: ssh keypair create/import form (#256)

Fixes #243

Signed-off-by: Abhishek Kumar <abhishek.mrt22@gmail.com>
Signed-off-by: Rohit Yadav <rohit.yadav@shapeblue.com>
---
 ui/src/config/section/compute.js          |  10 +-
 ui/src/locales/en.json                    |   1 +
 ui/src/views/compute/CreateSSHKeyPair.vue | 224 ++++++++++++++++++++++
 3 files changed, 227 insertions(+), 8 deletions(-)
 create mode 100644 ui/src/views/compute/CreateSSHKeyPair.vue

diff --git a/ui/src/config/section/compute.js b/ui/src/config/section/compute.js
index 4d30841d2d5..43985497153 100644
--- a/ui/src/config/section/compute.js
+++ b/ui/src/config/section/compute.js
@@ -457,14 +457,8 @@ export default {
           icon: 'plus',
           label: 'Create SSH Key Pair',
           listView: true,
-          args: ['name', 'account', 'domainid']
-        },
-        {
-          api: 'registerSSHKeyPair',
-          icon: 'key',
-          label: 'Register SSH Public Key',
-          listView: true,
-          args: ['name', 'account', 'domainid', 'publickey']
+          popup: true,
+          component: () => import('@/views/compute/CreateSSHKeyPair.vue')
         },
         {
           api: 'deleteSSHKeyPair',
diff --git a/ui/src/locales/en.json b/ui/src/locales/en.json
index 264fa656d55..7174a08b1f0 100644
--- a/ui/src/locales/en.json
+++ b/ui/src/locales/en.json
@@ -744,6 +744,7 @@
 "message.network.removeNIC": "Please confirm that want to remove this NIC, which will also remove the associated network from the VM.",
 "message.network.secondaryIP" : "Please confirm that you would like to acquire a new secondary IP for this NIC. \n NOTE: You need to manually configure the newly-acquired secondary IP inside the virtual machine.",
 "message.network.updateIp": "Please confirm that you would like to change the IP address for this NIC on VM.",
+"message.desc.create.ssh.key.pair": "Please fill in the following data to create or register a ssh key pair.<br><br>(1) If public key is set, CloudStack will register the public key. You can use it through your private key.<br><br>(2) If public key is not set, CloudStack will create a new SSH Key pair. In this case, please copy and save the private key. CloudStack will not keep it.<br>",
 "mincpunumber": "Min CPU Cores",
 "minInstance": "Min Instances",
 "minIops": "Min IOPS",
diff --git a/ui/src/views/compute/CreateSSHKeyPair.vue b/ui/src/views/compute/CreateSSHKeyPair.vue
new file mode 100644
index 00000000000..4e0fa514924
--- /dev/null
+++ b/ui/src/views/compute/CreateSSHKeyPair.vue
@@ -0,0 +1,224 @@
+// Licensed to the Apache Software Foundation (ASF) under one
+// or more contributor license agreements.  See the NOTICE file
+// distributed with this work for additional information
+// regarding copyright ownership.  The ASF licenses this file
+// to you under the Apache License, Version 2.0 (the
+// "License"); you may not use this file except in compliance
+// with the License.  You may obtain a copy of the License at
+//
+//   http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing,
+// software distributed under the License is distributed on an
+// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+// KIND, either express or implied.  See the License for the
+// specific language governing permissions and limitations
+// under the License.
+
+<template>
+  <div class="form-layout">
+    <a-spin :spinning="loading">
+      <p v-html="$t('message.desc.create.ssh.key.pair')"></p>
+      <a-form
+        :form="form"
+        @submit="handleSubmit"
+        layout="vertical">
+        <a-form-item :label="$t('name')">
+          <a-input
+            v-decorator="['name', {
+              rules: [{ required: true, message: 'Please enter name' }]
+            }]"
+            :placeholder="apiParams.name.description"/>
+        </a-form-item>
+        <a-form-item :label="$t('publickey')">
+          <a-input
+            v-decorator="['publickey', {}]"
+            :placeholder="apiParams.publickey.description"/>
+        </a-form-item>
+        <a-form-item :label="$t('domainid')" v-if="this.isAdminOrDomainAdmin()">
+          <a-select
+            id="domain-selection"
+            v-decorator="['domainid', {}]"
+            showSearch
+            optionFilterProp="children"
+            :filterOption="(input, option) => {
+              return option.componentOptions.children[0].text.toLowerCase().indexOf(input.toLowerCase()) >= 0
+            }"
+            :loading="domainLoading"
+            :placeholder="apiParams.domainid.description"
+            @change="val => { this.handleDomainChanged(this.domains[val]) }">
+            <a-select-option v-for="(opt, optIndex) in this.domains" :key="optIndex">
+              {{ opt.name || opt.description }}
+            </a-select-option>
+          </a-select>
+        </a-form-item>
+        <a-form-item :label="$t('account')" v-if="this.isAdminOrDomainAdmin() && !this.isObjectEmpty(this.selectedDomain) && this.selectedDomain.id !== null">
+          <a-input
+            v-decorator="['name', {}]"
+            :placeholder="apiParams.account.description"/>
+        </a-form-item>
+
+        <div :span="24" class="action-button">
+          <a-button @click="closeAction">{{ this.$t('Cancel') }}</a-button>
+          <a-button :loading="loading" type="primary" @click="handleSubmit">{{ this.$t('OK') }}</a-button>
+        </div>
+      </a-form>
+    </a-spin>
+  </div>
+</template>
+
+<script>
+import { api } from '@/api'
+
+export default {
+  name: 'CreateSSHKeyPair',
+  props: {},
+  data () {
+    return {
+      domains: [],
+      domainLoading: false,
+      selectedDomain: {},
+      loading: false
+    }
+  },
+  beforeCreate () {
+    this.form = this.$form.createForm(this)
+    this.apiConfig = this.$store.getters.apis.createSSHKeyPair || {}
+    this.apiParams = {}
+    this.apiConfig.params.forEach(param => {
+      this.apiParams[param.name] = param
+    })
+    this.apiConfig = this.$store.getters.apis.registerSSHKeyPair || {}
+    this.apiConfig.params.forEach(param => {
+      if (!(param.name in this.apiParams)) {
+        this.apiParams[param.name] = param
+      }
+    })
+  },
+  created () {
+    this.domains = [
+      {
+        id: null,
+        name: ''
+      }
+    ]
+  },
+  mounted () {
+    this.fetchData()
+  },
+  methods: {
+    fetchData () {
+      if (this.isAdminOrDomainAdmin()) {
+        this.fetchDomainData()
+      }
+    },
+    isAdminOrDomainAdmin () {
+      return ['Admin', 'DomainAdmin'].includes(this.$store.getters.userInfo.roletype)
+    },
+    isValidValueForKey (obj, key) {
+      return key in obj && obj[key] != null
+    },
+    arrayHasItems (array) {
+      return array !== null && array !== undefined && Array.isArray(array) && array.length > 0
+    },
+    isObjectEmpty (obj) {
+      return !(obj !== null && obj !== undefined && Object.keys(obj).length > 0 && obj.constructor === Object)
+    },
+    fetchDomainData () {
+      const params = {}
+      this.domainLoading = true
+      api('listDomains', params).then(json => {
+        const listdomains = json.listdomainsresponse.domain
+        this.domains = this.domains.concat(listdomains)
+      }).finally(() => {
+        this.domainLoading = false
+        if (this.arrayHasItems(this.domains)) {
+          this.form.setFieldsValue({
+            domainid: 0
+          })
+          this.handleDomainChanged(this.domains[0])
+        }
+      })
+    },
+    handleDomainChanged (domain) {
+      this.selectedDomain = domain
+    },
+    handleSubmit (e) {
+      e.preventDefault()
+      this.form.validateFields((err, values) => {
+        if (err) {
+          return
+        }
+        this.loading = true
+        const params = {
+          name: values.name
+        }
+        if (this.isValidValueForKey(values, 'domainid') &&
+          this.arrayHasItems(this.domains) &&
+          this.domains[values.domainid].id !== null) {
+          params.domainid = this.domains[values.domainid].id
+        }
+        if (this.isValidValueForKey(values, 'account') && values.account.length > 0) {
+          params.account = values.account
+        }
+        if (this.isValidValueForKey(values, 'publickey') && values.publickey.length > 0) {
+          params.publickey = values.publickey
+          api('registerSSHKeyPair', params).then(json => {
+            this.$message.success('Successfully registered SSH key pair: ' + values.name)
+          }).catch(error => {
+            this.$notification.error({
+              message: 'Request Failed',
+              description: (error.response && error.response.headers && error.response.headers['x-description']) || error.message
+            })
+          }).finally(() => {
+            this.$emit('refresh-data')
+            this.loading = false
+            this.closeAction()
+          })
+        } else {
+          api('createSSHKeyPair', params).then(json => {
+            this.$message.success('Successfully created SSH key pair: ' + values.name)
+            if (json.createsshkeypairresponse && json.createsshkeypairresponse.keypair && json.createsshkeypairresponse.keypair.privatekey) {
+              this.$notification.info({
+                message: this.$t('label.create.ssh.key.pair'),
+                description: (<span domPropsInnerHTML={'<pre>' + json.createsshkeypairresponse.keypair.privatekey + '</pre>'}></span>),
+                duration: 0
+              })
+            }
+          }).catch(error => {
+            this.$notification.error({
+              message: 'Request Failed',
+              description: (error.response && error.response.headers && error.response.headers['x-description']) || error.message
+            })
+          }).finally(() => {
+            this.$emit('refresh-data')
+            this.loading = false
+            this.closeAction()
+          })
+        }
+      })
+    },
+    closeAction () {
+      this.$emit('close-action')
+    }
+  }
+}
+</script>
+
+<style scoped lang="less">
+  .form-layout {
+    width: 80vw;
+
+    @media (min-width: 600px) {
+      width: 450px;
+    }
+  }
+
+  .action-button {
+    text-align: right;
+
+    button {
+      margin-right: 5px;
+    }
+  }
+</style>