# 推送服务设置支持解绑

spug_api/apps/setting/urls.py

@@ -13,5 +13,6 @@ urlpatterns = [
     url(r'^email_test/$', email_test),
     url(r'^mfa/$', MFAView.as_view()),
     url(r'^about/$', get_about),
-    url(r'^balance/$', get_push_balance),
+    url(r'^push/bind/$', handle_push_bind),
+    url(r'^push/balance/$', handle_push_balance),
 ]

spug_api/apps/setting/utils.py

@@ -31,6 +31,10 @@ class AppSetting:
         else:
             raise KeyError('invalid key')
 
+    @classmethod
+    def delete(cls, key):
+        Setting.objects.filter(key=key).delete()
+
     @classmethod
     def get_ssh_key(cls):
         public_key = cls.get_default('public_key')

spug_api/apps/setting/views.py

@@ -21,6 +21,9 @@ class SettingView(AdminView):
     def get(self, request):
         response = deepcopy(KEYS_DEFAULT)
         for item in Setting.objects.all():
+            if item.key == 'spug_push_key':
+                response[item.key] = f'{item.real_val[:8]}********{item.real_val[-8:]}'
+            else:
                 response[item.key] = item.real_val
         return json_response(response)
 
@@ -126,7 +129,27 @@ def get_about(request):
 
 
 @auth('admin')
-def get_push_balance(request):
+def handle_push_bind(request):
+    form, error = JsonParser(
+        Argument('spug_push_key', required=False),
+    ).parse(request.body)
+    if error is None:
+        if not form.spug_push_key:
+            AppSetting.delete('spug_push_key')
+            return json_response()
+
+        try:
+            res = get_balance(form.spug_push_key)
+        except Exception as e:
+            return json_response(error=f'绑定失败：{e}')
+
+        AppSetting.set('spug_push_key', form.spug_push_key)
+        return json_response(res)
+    return json_response(error=error)
+
+
+@auth('admin')
+def handle_push_balance(request):
     token = AppSetting.get_default('spug_push_key')
     if not token:
         return json_response(error='请先配置推送服务绑定账户')

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

@@ -14,17 +14,18 @@ import store from './store';
 export default observer(function () {
   const [loading, setLoading] = useState(false);
   const [fetching, setFetching] = useState(false);
-  const [balance, setBalance] = useState({})
+  const [balance, setBalance] = useState({});
+  const [pushKey, setPushKey] = useState(store.settings.spug_push_key);
 
   useEffect(() => {
-    if (store.settings.spug_push_key) {
+    if (pushKey) {
       fetchBalance()
     }
   }, []);
 
   function fetchBalance() {
     setFetching(true)
-    http.get('/api/setting/balance/')
+    http.get('/api/setting/push/balance/')
       .then(res => setBalance(res))
       .finally(() => {
         setLoading(false)
@@ -33,19 +34,31 @@ export default observer(function () {
   }
 
   function handleBind() {
-    const spug_push_key = store.settings.spug_push_key
+    if (!pushKey) return message.error('请输入要绑定的推送助手用户ID')
-    if (!spug_push_key) return message.error('请输入要绑定的推送助手用户ID')
     setLoading(true);
-    http.post('/api/setting/', {data: [{key: 'spug_push_key', value: spug_push_key}]})
+    http.post('/api/setting/push/bind/', {spug_push_key: pushKey})
-      .then(() => {
+      .then(res => {
-        message.success('保存成功');
+        message.success('绑定成功');
         store.fetchSettings();
-        fetchBalance()
+        setBalance(res)
       })
-      .finally(() => store.loading = false)
+      .finally(() => setLoading(false))
   }
 
-  const isVip = true
+  function handleUnbind() {
+    setLoading(true);
+    http.post('/api/setting/push/bind/', {spug_push_key: ''})
+      .then(() => {
+        message.success('解绑成功');
+        store.fetchSettings();
+        setBalance({})
+        setPushKey('')
+      })
+      .finally(() => setLoading(false))
+  }
+
+  const isVip = balance.is_vip
+  const spugPushKey = store.settings.spug_push_key
   return (
     <React.Fragment>
       <div className={css.title}>推送服务设置</div>
@@ -54,10 +67,22 @@ export default observer(function () {
                    extra={<div>请登录 <Link href="https://push.spug.cc/login" title="推送助手"/>，至个人中心 /
                      个人设置查看用户ID，注意保密该ID请勿泄漏给第三方。</div>}>
 
+          {spugPushKey ? (
+            <Input.Group compact>
+              <div className={css.keyText}
+                   style={{width: 'calc(100% - 100px)', lineHeight: '32px', fontWeight: 'bold'}}>{spugPushKey}</div>
+              <Button
+                ghost
+                type="danger"
+                style={{width: 80, marginLeft: 20}}
+                onClick={handleUnbind}
+                loading={loading}>解绑</Button>
+            </Input.Group>
+          ) : (
             <Input.Group compact>
               <Input
-              value={store.settings.spug_push_key}
+                value={pushKey}
-              onChange={e => store.settings.spug_push_key = e.target.value}
+                onChange={e => setPushKey(e.target.value)}
                 style={{width: 'calc(100% - 100px)'}}
                 placeholder="请输入要绑定的推送助手用户ID"/>
               <Button
@@ -66,13 +91,12 @@ export default observer(function () {
                 onClick={handleBind}
                 loading={loading}>确定</Button>
             </Input.Group>
-          {/*<Input.Group compact>*/}
+
-          {/*  <Input bordered={false} style={{width: 'calc(100% - 100px)', paddingLeft: 0}} value="32uu73******64823d"/>*/}
+          )}
-          {/*  <Button style={{width: 80, marginLeft: 20}}>解绑</Button>*/}
-          {/*</Input.Group>*/}
         </Form.Item>
       </div>
 
+      {spugPushKey ? (
         <Form.Item style={{marginTop: 24}}
                    extra={<div> 如需充值请至 <Link href="https://push.spug.cc/buy/sms" title="推送助手"/>，具体计费规则及说明请查看推送助手官网。
                    </div>}>
@@ -109,6 +133,7 @@ export default observer(function () {
             </Spin>
           </div>
         </Form.Item>
+      ) : null}
     </React.Fragment>
   )
 })