From e6582c140f370128d27f1ad738310f0683aa0128 Mon Sep 17 00:00:00 2001
From: Jason Robinson <mail@jasonrobinson.me>
Date: Fri, 13 Nov 2020 15:23:54 +0200
Subject: [PATCH] Dont' allow HostingSignupDialog to close

Except via confirmed cancel action or host setup success postmessage.
---
 .../structures/HostingSignupAction.tsx        | 27 +++++++--
 .../structures/HostingSignupDialog.tsx        | 58 +++++++++++++++----
 .../ConfirmCloseHostingSignupDialog.js        | 35 +++++++++++
 3 files changed, 104 insertions(+), 16 deletions(-)
 create mode 100644 src/components/views/dialogs/ConfirmCloseHostingSignupDialog.js

diff --git a/src/components/structures/HostingSignupAction.tsx b/src/components/structures/HostingSignupAction.tsx
index 7434abfff8..d0ed9d651e 100644
--- a/src/components/structures/HostingSignupAction.tsx
+++ b/src/components/structures/HostingSignupAction.tsx
@@ -23,15 +23,34 @@ interface IProps {}
 interface IState {}
 
 export default class HostingSignupAction extends React.PureComponent<IProps, IState> {
-    private static openDialog() {
-        Modal.createDialog(
-            HostingSignupDialog, {}, "mx_HostingSignupDialog",
+    closingAllowed = false;
+    modalRef: any;
+
+    private openDialog = () => {
+        this.modalRef = Modal.createTrackedDialog(
+            'Hosting Signup Open',
+            '',
+            HostingSignupDialog,
+            {
+                requestClose: this.requestClose,
+            },
+            "mx_HostingSignupDialog",
+            false,
+            true,
+            {
+                onBeforeClose: async () => this.closingAllowed,
+            },
         );
     }
 
+    private requestClose = () => {
+        this.closingAllowed = true;
+        this.modalRef.close();
+    }
+
     public render(): React.ReactNode {
         return (
-            <div onClick={HostingSignupAction.openDialog} className="mx_HostingSignupAction">
+            <div onClick={this.openDialog} className="mx_HostingSignupAction">
                 Get your own personal Element!
             </div>
         );
diff --git a/src/components/structures/HostingSignupDialog.tsx b/src/components/structures/HostingSignupDialog.tsx
index 7cd4e85046..0315504a8d 100644
--- a/src/components/structures/HostingSignupDialog.tsx
+++ b/src/components/structures/HostingSignupDialog.tsx
@@ -15,10 +15,14 @@ limitations under the License.
 */
 
 import * as React from "react";
+import * as sdk from '../../index';
+import Modal from "../../Modal";
 import SdkConfig from "../../SdkConfig";
 import {MatrixClientPeg} from "../../MatrixClientPeg";
 
-interface IProps {}
+interface IProps {
+    requestClose(): void,
+}
 
 interface IState {
     error: string,
@@ -49,6 +53,29 @@ export default class HostingSignupDialog extends React.PureComponent<IProps, ISt
                 // noinspection JSIgnoredPromiseFromCall
                 this.fetchOpenIDToken();
                 break;
+            case 'setup_complete':
+                this.onFinished(true);
+                break;
+        }
+    }
+
+    private onFinished = (result: boolean) => {
+        if (result) {
+            // We're done, close
+            this.props.requestClose();
+        } else {
+            const ConfirmDialog = sdk.getComponent('views.dialogs.ConfirmCloseHostingSignupDialog');
+            Modal.createDialog(
+                ConfirmDialog,
+                {
+                    onFinished: result => {
+                        if (result) {
+                            // TODO call an API endpoint to clean up?
+                            this.props.requestClose();
+                        }
+                    },
+                },
+            )
         }
     }
 
@@ -82,18 +109,25 @@ export default class HostingSignupDialog extends React.PureComponent<IProps, ISt
     }
 
     public render(): React.ReactNode {
+        const BaseDialog = sdk.getComponent('views.dialogs.BaseDialog');
         return (
-            <div className="mx_HostingSignupDialog_container">
-                <iframe
-                    src={this.hostingSignupUrl}
-                    ref={ref => this.iframeRef = ref}
-                />
-                {this.state.error &&
-                    <div>
-                        {this.state.error}
-                    </div>
-                }
-            </div>
+            <BaseDialog
+                onFinished={this.onFinished}
+                title="Set up your own personal Element host"
+                hasCancel={true}
+            >
+                <div className="mx_HostingSignupDialog_container">
+                    <iframe
+                        src={this.hostingSignupUrl}
+                        ref={ref => this.iframeRef = ref}
+                    />
+                    {this.state.error &&
+                        <div>
+                            {this.state.error}
+                        </div>
+                    }
+                </div>
+            </BaseDialog>
         );
     }
 }
diff --git a/src/components/views/dialogs/ConfirmCloseHostingSignupDialog.js b/src/components/views/dialogs/ConfirmCloseHostingSignupDialog.js
new file mode 100644
index 0000000000..c990663645
--- /dev/null
+++ b/src/components/views/dialogs/ConfirmCloseHostingSignupDialog.js
@@ -0,0 +1,35 @@
+/*
+Copyright 2020 Element
+
+Licensed 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.
+*/
+
+import React from 'react';
+import * as sdk from '../../../index';
+
+/*
+ * A dialog for confirming closing the Hosting signup setup dialog.
+ */
+export default class ConfirmCloseHostingSignupDialog extends React.Component {
+    render() {
+        const QuestionDialog = sdk.getComponent('views.dialogs.QuestionDialog');
+        return (
+            <QuestionDialog
+                onFinished={this.props.onFinished}
+                title="Confirm Abort Of Host Creation"
+                description="Are you sure you wish to abort creation of the host? The process cannot be continued."
+                button="Abort"
+            />
+        );
+    }
+}