A 新增上传自定义密钥功能

spug_api/apps/setting/views.py

@@ -14,8 +14,8 @@ import smtplib
 
 class SettingView(View):
     def get(self, request):
-        settings = Setting.objects.exclude(key__in=('public_key', 'private_key'))
-        return json_response(settings)
+        data = Setting.objects.all()
+        return json_response(data)
 
     def post(self, request):
         form, error = JsonParser(

spug_web/src/pages/system/setting/KeySetting.js

@@ -0,0 +1,75 @@
+/**
+ * Copyright (c) OpenSpug Organization. https://github.com/openspug/spug
+ * Copyright (c) <spug.dev@gmail.com>
+ * Released under the MIT License.
+ */
+import React from 'react';
+import { observer } from 'mobx-react';
+import { Alert, Button, Form, Input, Modal, message } from 'antd';
+import styles from './index.module.css';
+import http from 'libs/http';
+import store from './store';
+import lds from 'lodash';
+
+
+export default observer(function () {
+  function handleSubmit() {
+    Modal.confirm({
+      title: '密钥修改确认',
+      content: <span style={{color: '#f5222d'}}>请谨慎修改密钥对，修改密钥对会让现有的主机都无法进行验证，影响与主机相关的各项功能！</span>,
+      onOk: () => {
+        Modal.confirm({
+          title: '小提示',
+          content: <div>修改密钥对需要<span style={{color: '#f5222d'}}>重启服务后生效</span>，已添加的主机需要重新进行编辑验证后才可以正常连接。</div>,
+          onOk: doModify
+        })
+      }
+    })
+  }
+
+  function doModify() {
+    const public_key = lds.get(store.settings, 'public_key.value');
+    const private_key = lds.get(store.settings, 'private_key.value');
+    return http.post('/api/setting/', {data: [{key: 'public_key', value: public_key}, {key: 'private_key', value: private_key}]})
+      .then(() => {
+        message.success('保存成功');
+        store.fetchSettings()
+      })
+      .finally(() => store.loading = false)
+  }
+
+  return (
+    <React.Fragment>
+      <div className={styles.title}>密钥设置</div>
+      <Alert
+        closable
+        showIcon
+        type="info"
+        style={{width: 650}}
+        message="小提示"
+        description="在你没有上传密钥的情况下，Spug会在首次添加主机时自动生成密钥对。"
+      />
+      <Form style={{maxWidth: 650}}>
+        <Form.Item label="公钥" help="一般位于 ~/.ssh/id_rsa.pub">
+          <Input.TextArea
+            rows={7}
+            spellCheck={false}
+            value={lds.get(store.settings, 'public_key.value')}
+            onChange={e => lds.set(store.settings, 'public_key.value', e.target.value)}
+            placeholder="请输入公钥"/>
+        </Form.Item>
+        <Form.Item label="私钥" help="一般位于 ~/.ssh/id_rsa">
+          <Input.TextArea
+            rows={14}
+            spellCheck={false}
+            value={lds.get(store.settings, 'private_key.value')}
+            onChange={e => lds.set(store.settings, 'private_key.value', e.target.value)}
+            placeholder="请输入私钥"/>
+        </Form.Item>
+        <Form.Item>
+          <Button type="primary" loading={store.loading} onClick={handleSubmit}>保存设置</Button>
+        </Form.Item>
+      </Form>
+    </React.Fragment>
+  )
+})

spug_web/src/pages/system/setting/index.js

@@ -10,6 +10,7 @@ import BasicSetting from './BasicSetting';
 import AlarmSetting from './AlarmSetting';
 import LDAPSetting from './LDAPSetting';
 import OpenService from './OpenService';
+import KeySetting from './KeySetting';
 import About from './About';
 import styles from './index.module.css';
 import store from './store';
@@ -39,6 +40,7 @@ class Index extends React.Component {
             onSelect={({selectedKeys}) => this.setState({selectedKeys})}>
             <Menu.Item key="basic">基本设置</Menu.Item>
             <Menu.Item key="ldap">LDAP设置</Menu.Item>
+            <Menu.Item key="key">密钥设置</Menu.Item>
             <Menu.Item key="alarm">报警服务设置</Menu.Item>
             <Menu.Item key="service">开放服务设置</Menu.Item>
             <Menu.Item key="about">关于</Menu.Item>
@@ -49,6 +51,7 @@ class Index extends React.Component {
           {selectedKeys[0] === 'ldap' && <LDAPSetting />}
           {selectedKeys[0] === 'alarm' && <AlarmSetting />}
           {selectedKeys[0] === 'service' && <OpenService />}
+          {selectedKeys[0] === 'key' && <KeySetting />}
           {selectedKeys[0] === 'about' && <About />}
         </div>
       </AuthDiv>