From 36fe65328b4e72e9ac1b80c31ae70797e37c89ed Mon Sep 17 00:00:00 2001
From: Mykola Morhun <mmorhun@redhat.com>
Date: Mon, 25 Nov 2019 16:44:34 +0200
Subject: [PATCH] Fix webview after changes in upstream.

Co-authored-by: Oleksandr Andriienko <oandriie@redhat.com>
Signed-off-by: Mykola Morhun <mmorhun@redhat.com>
---
 dockerfiles/theia/Dockerfile               |  1 +
 dockerfiles/theia/src/entrypoint.sh        |  3 +-
 dockerfiles/theia/src/get-webview-route.js | 73 ++++++++++++++++++++++
 3 files changed, 76 insertions(+), 1 deletion(-)
 create mode 100644 dockerfiles/theia/src/get-webview-route.js

diff --git a/dockerfiles/theia/Dockerfile b/dockerfiles/theia/Dockerfile
index 0f72b3bce..5da75e0c9 100644
--- a/dockerfiles/theia/Dockerfile
+++ b/dockerfiles/theia/Dockerfile
@@ -163,5 +163,6 @@ COPY --from=builder /home/theia-dev/theia-source-code/production/plugins /defaul
 COPY --chown=theia:root --from=builder /home/theia-dev/theia-source-code/production /home/theia
 USER theia
 WORKDIR /projects
+COPY src/get-webview-route.js ${HOME}/get-webview-route.js
 COPY src/entrypoint.sh /entrypoint.sh
 ENTRYPOINT ["/entrypoint.sh"]
diff --git a/dockerfiles/theia/src/entrypoint.sh b/dockerfiles/theia/src/entrypoint.sh
index 99c92a1b2..58706b219 100755
--- a/dockerfiles/theia/src/entrypoint.sh
+++ b/dockerfiles/theia/src/entrypoint.sh
@@ -76,7 +76,8 @@ if [ "${NOCDN}" == "true" ]; then
 fi
 shopt -u nocasematch
 
-# run che
+# run Che Theia
+export THEIA_WEBVIEW_EXTERNAL_ENDPOINT=$(node get-webview-route.js)
 node src-gen/backend/main.js /projects --hostname=0.0.0.0 --port=${THEIA_PORT} &
 
 PID=$!
diff --git a/dockerfiles/theia/src/get-webview-route.js b/dockerfiles/theia/src/get-webview-route.js
new file mode 100644
index 000000000..aa3282bd9
--- /dev/null
+++ b/dockerfiles/theia/src/get-webview-route.js
@@ -0,0 +1,73 @@
+/*********************************************************************
+ * Copyright (c) 2019 Red Hat, Inc.
+ *
+ * This program and the accompanying materials are made
+ * available under the terms of the Eclipse Public License 2.0
+ * which is available at https://www.eclipse.org/legal/epl-2.0/
+ *
+ * SPDX-License-Identifier: EPL-2.0
+ **********************************************************************/
+
+const fs = require('fs');
+const workspaceClient = require('@eclipse-che/workspace-client');
+
+// Assume that self-signed certificate is located by the following path
+const SS_CRT_PATH = '/tmp/che/secret/ca.crt';
+
+// Configure workspace API client
+const restAPIConfig = {
+    baseUrl: process.env.CHE_API
+};
+
+const token = process.env.CHE_MACHINE_TOKEN;
+if (token) {
+    restAPIConfig.headers = {};
+    restAPIConfig.headers['Authorization'] = 'Bearer ' + token;
+}
+
+if (fs.existsSync(SS_CRT_PATH)) {
+    restAPIConfig.ssCrtPath = SS_CRT_PATH;
+}
+
+// Create rest API workspace client
+const restApiClient = workspaceClient.default.getRestApi(restAPIConfig);
+
+// Search for IDE route
+function getIdeRoute() {
+    return new Promise((resolve, reject) => {
+        restApiClient.getById(process.env.CHE_WORKSPACE_ID).then(workspace => {
+            const containers = workspace.runtime.machines;
+            Object.keys(containers).forEach(containerName => {
+                const container = containers[containerName];
+                const servers = container['servers'];
+                if (servers) {
+                    const ideServerName = Object.keys(servers).find(serverName => servers[serverName].attributes && servers[serverName].attributes['type'] === 'ide');
+                    if (ideServerName) {
+                        resolve(servers[ideServerName].url);
+                    }
+                }
+            });
+            reject('Server with type "ide" not found.');
+        }).catch(error => {
+            reject(error);
+        });
+    });
+}
+
+getIdeRoute().then(ideRoute => {
+    if (!ideRoute) {
+        throw new Error('Failed to get ide route.');
+    }
+
+    // Remove trailing slash if any
+    if (ideRoute.endsWith('/')) {
+        ideRoute = ideRoute.substring(0, ideRoute.length - 1);
+    }
+    // Remove protocol
+    const webviewDomain = ideRoute.replace(/^https?:\/\//, '');
+    // Return result to shell by writing into stdout
+    process.stdout.write(webviewDomain);
+}).catch(error => {
+    console.error('Unable to get IDE route. Webviews might not work. Cause:', error);
+    // Just exit this script without returning a value to the shell.
+});