From 7c49d7adbee37f33ce4d2b2043ef34e7af92bf4a Mon Sep 17 00:00:00 2001
From: HFO4 <912394456@qq.com>
Date: Sun, 31 Oct 2021 09:33:20 +0800
Subject: [PATCH] Add node

---
 src/component/Admin/Node/Guide/Aria2RPC.js    | 37 ++++++++++-----
 .../Admin/Node/Guide/Communication.js         |  2 +-
 src/component/Admin/Node/Guide/NodeGuide.js   | 45 ++++++++++++++++---
 3 files changed, 65 insertions(+), 19 deletions(-)

diff --git a/src/component/Admin/Node/Guide/Aria2RPC.js b/src/component/Admin/Node/Guide/Aria2RPC.js
index e4964e7..fcde8fe 100644
--- a/src/component/Admin/Node/Guide/Aria2RPC.js
+++ b/src/component/Admin/Node/Guide/Aria2RPC.js
@@ -100,8 +100,8 @@ export default function Aria2RPC(props) {
             type: props.node.Type,
             server: props.node.Server,
             secret: props.node.SlaveKey,
-            rpc: props.node.Aria2Options.Server,
-            token: props.node.Aria2Options.Token,
+            rpc: props.node.Aria2OptionsSerialized.Server,
+            token: props.node.Aria2OptionsSerialized.Token,
         })
             .then((response) => {
                 ToggleSnackbar(
@@ -219,7 +219,7 @@ export default function Aria2RPC(props) {
                             <br />
                             # RPC 授权令牌，可自行设定
                             <br />
-                            rpc-secure={props.node.Aria2Options.Token}
+                            rpc-secure={props.node.Aria2OptionsSerialized.Token}
                             <br />
                         </pre>
                         <Alert severity="info" style={{ marginBottom: 10 }}>
@@ -253,7 +253,9 @@ export default function Aria2RPC(props) {
                                 <Input
                                     required
                                     type={"url"}
-                                    value={props.node.Aria2Options.Server}
+                                    value={
+                                        props.node.Aria2OptionsSerialized.Server
+                                    }
                                     onChange={props.handleOptionChange(
                                         "Server"
                                     )}
@@ -279,7 +281,7 @@ export default function Aria2RPC(props) {
                         </Typography>
                         <div className={classes.form}>
                             <Input
-                                value={props.node.Aria2Options.Token}
+                                value={props.node.Aria2OptionsSerialized.Token}
                                 onChange={props.handleOptionChange("Token")}
                             />
                         </div>
@@ -298,7 +300,9 @@ export default function Aria2RPC(props) {
                         </Typography>
                         <div className={classes.form}>
                             <Input
-                                value={props.node.Aria2Options.TempPath}
+                                value={
+                                    props.node.Aria2OptionsSerialized.TempPath
+                                }
                                 onChange={props.handleOptionChange("TempPath")}
                             />
                         </div>
@@ -325,7 +329,10 @@ export default function Aria2RPC(props) {
                                         min: 1,
                                     }}
                                     required
-                                    value={props.node.Aria2Options.Interval}
+                                    value={
+                                        props.node.Aria2OptionsSerialized
+                                            .Interval
+                                    }
                                     onChange={props.handleOptionChange(
                                         "Interval"
                                     )}
@@ -347,7 +354,10 @@ export default function Aria2RPC(props) {
                                         min: 1,
                                     }}
                                     required
-                                    value={props.node.Aria2Options.Timeout}
+                                    value={
+                                        props.node.Aria2OptionsSerialized
+                                            .Timeout
+                                    }
                                     onChange={props.handleOptionChange(
                                         "Timeout"
                                     )}
@@ -365,7 +375,10 @@ export default function Aria2RPC(props) {
                                 <Input
                                     multiline
                                     required
-                                    value={props.node.Aria2Options.Options}
+                                    value={
+                                        props.node.Aria2OptionsSerialized
+                                            .Options
+                                    }
                                     onChange={props.handleOptionChange(
                                         "Options"
                                     )}
@@ -386,7 +399,8 @@ export default function Aria2RPC(props) {
                     </div>
                     <div className={classes.subStepContent}>
                         <Typography variant={"body2"}>
-                            完成以上步骤后，你可以点击下方的测试按钮测试{mode}
+                            完成以上步骤后，你可以点击下方的测试按钮测试
+                            {mode}
                             Cloudreve 向 Aria2 通信是否正常。
                             {props.node.Type === 0 &&
                                 "在进行测试前请先确保您已进行并通过上一页面中的“从机通信测试”。"}
@@ -416,10 +430,11 @@ export default function Aria2RPC(props) {
                     </Button>
                 )}
                 <Button
-                    disabled={loading}
+                    disabled={loading || props.loading}
                     type={"submit"}
                     variant={"contained"}
                     color={"primary"}
+                    onClick={props.onSubmit}
                 >
                     下一步
                 </Button>
diff --git a/src/component/Admin/Node/Guide/Communication.js b/src/component/Admin/Node/Guide/Communication.js
index d813378..3ed307f 100644
--- a/src/component/Admin/Node/Guide/Communication.js
+++ b/src/component/Admin/Node/Guide/Communication.js
@@ -263,7 +263,7 @@ export default function Communication(props) {
 
             <div className={classes.stepFooter}>
                 <Button
-                    disabled={loading}
+                    disabled={loading || props.loading}
                     type={"submit"}
                     variant={"contained"}
                     color={"primary"}
diff --git a/src/component/Admin/Node/Guide/NodeGuide.js b/src/component/Admin/Node/Guide/NodeGuide.js
index 913c9a4..f0eff06 100644
--- a/src/component/Admin/Node/Guide/NodeGuide.js
+++ b/src/component/Admin/Node/Guide/NodeGuide.js
@@ -8,6 +8,7 @@ import { toggleSnackbar } from "../../../../actions";
 import { randomStr } from "../../../../utils";
 import Communication from "./Communication";
 import Aria2RPC from "./Aria2RPC";
+import API from "../../../../middleware/Api";
 
 const steps = [
     {
@@ -26,11 +27,6 @@ const steps = [
             return <Aria2RPC {...p} />;
         },
     },
-    {
-        slaveOnly: false,
-        title: "杂项信息",
-        optional: false,
-    },
     {
         slaveOnly: false,
         title: "完成",
@@ -41,6 +37,7 @@ const steps = [
 export default function NodeGuide(props) {
     const [activeStep, setActiveStep] = useState(0);
     const [skipped, setSkipped] = React.useState(new Set());
+    const [loading, setLoading] = useState(false);
     const [node, setNode] = useState(
         props.node
             ? props.node
@@ -51,7 +48,7 @@ export default function NodeGuide(props) {
                   Server: "https://example.com:5212",
                   SlaveKey: randomStr(64),
                   MasterKey: randomStr(64),
-                  Aria2Options: {
+                  Aria2OptionsSerialized: {
                       Token: randomStr(32),
                       Options: "{}",
                       Interval: 10,
@@ -91,6 +88,39 @@ export default function NodeGuide(props) {
         [dispatch]
     );
 
+    const nextStep = () => {
+        if (props.node || activeStep + 1 === steps.length - 1) {
+            setLoading(true);
+
+            const nodeCopy = { ...node };
+            nodeCopy.Aria2OptionsSerialized = {
+                ...node.Aria2OptionsSerialized,
+            };
+            nodeCopy.Aria2Enabled = nodeCopy.Aria2Enabled === "true";
+            API.post("/admin/node", {
+                node: nodeCopy,
+            })
+                .then(() => {
+                    ToggleSnackbar(
+                        "top",
+                        "right",
+                        "存储策略已" + (props.policy ? "保存" : "添加"),
+                        "success"
+                    );
+                    setActiveStep(activeStep + 1);
+                    setLoading(false);
+                })
+                .catch((error) => {
+                    ToggleSnackbar("top", "right", error.message, "error");
+                })
+                .then(() => {
+                    setLoading(false);
+                });
+        } else {
+            setActiveStep(activeStep + 1);
+        }
+    };
+
     return (
         <div>
             <Typography variant={"h6"}>
@@ -121,8 +151,9 @@ export default function NodeGuide(props) {
             </Stepper>
 
             {usedSteps[activeStep].component({
-                onSubmit: (e) => setActiveStep(activeStep + 1),
+                onSubmit: (e) => nextStep(),
                 node: node,
+                loading: loading,
                 onBack: (e) => setActiveStep(activeStep - 1),
                 handleTextChange: handleTextChange,
                 activeStep: activeStep,