[FIX] Improve cloud section

app/cloud/client/admin/cloud.html

@@ -16,25 +16,41 @@
 						</div>
 					</div>
 					<div class="section">
-						{{#if info.registeredWithWizard}}
-							{{#if info.workspaceConnected}}
+						{{#if info.connectToCloud}}
+							{{#if info.workspaceRegistered}}
 								<div class="section-content border-component-color">
-									{{#if info.userAssociated}}
-										<p>{{_ "Cloud_workspace_connected_plus_account"}}</p>
-									{{else}}
-										<p>{{_ "Cloud_workspace_connected_without_account"}}</p>
+									<p>{{_ "Cloud_workspace_connected"}}</p>
+									<div class="input-line double-col">
+										<label class="setting-label" title=""></label>
+										<div class="setting-field">
+											<a href="https://cloud.rocket.chat" class="rc-button rc-button--primary" target="_blank">{{_ "Cloud_login_to_cloud"}}</a>
+										</div>
+									</div>
+								</div>
+
+								<div class="section-content border-component-color">
+										<p>{{_ "Cloud_workspace_support"}}</p>
 										<div class="input-line double-col">
 											<label class="setting-label" title=""></label>
 											<div class="setting-field">
-												<button type="button" class="rc-button rc-button--primary action login-btn">{{_ "Cloud_login_to_cloud"}}</button>
+												<button type="button" class="rc-button rc-button--danger action sync-btn">{{_ "Sync"}}</button>
 											</div>
 										</div>
-									{{/if}}
+									</div>
+
+								<div class="section-content border-component-color">
+									<p>{{_ "Cloud_workspace_disconnect"}}</p>
+									<div class="input-line double-col">
+										<label class="setting-label" title=""></label>
+										<div class="setting-field">
+											<button type="button" class="rc-button rc-button--danger action disconnect-btn">{{_ "Disconnect"}}</button>
+										</div>
+									</div>
 								</div>
 							{{else}}
 								<div class="section-content border-component-color">
 									<div class="input-line double-col">
-										<label class="setting-label" title="cloudEmail">Email</label>
+										<label class="setting-label" title="cloudEmail">{{_ "Email"}}</label>
 										<div class="setting-field">
 											<input class="input-monitor rc-input__element" type="text" name="cloudEmail" value="{{ info.email }}">
 											<div class="settings-description secondary-font-color">{{_ "Cloud_address_to_send_registration_to"}}</div>
@@ -47,8 +63,6 @@
 										</div>
 									</div>
 
-									<pre>{{ registeredWithWizard }}</pre>
-
 									<div class="input-line double-col">
 										<label class="setting-label" title="cloudToken">{{_ "Token"}}</label>
 										<div class="setting-field">
@@ -62,6 +76,8 @@
 											<button type="button" class="rc-button rc-button--primary action connect-btn">{{_ "Connect"}}</button>
 										</div>
 									</div>
+
+									<p>If you still haven't received a registration email please make sure your email is updated above.  If you still have issues you can contact support at: <a href="mailto:support@rocket.chat?subject=[Self Hosted Registration]&body=WorkspaceId: {{ info.workspaceId }}%0D%0ADeployment Id: {{ info.uniqueId }}%0D%0AIssue: <please describe your issue here>">support@rocket.chat</a></p>
 								</div>
 							{{/if}}
 						{{else}}
@@ -72,7 +88,7 @@
 							</div>
 							<div class="section-content border-component-color">
 								<p>{{_ "Cloud_registration_required_description"}}</p>
-								<p><a href="./admin/Setup_Wizard">{{_ "Cloud_registration_requried_link_text"}}</a></p>
+								<button type="button" class="rc-button rc-button--primary action register-btn">{{_ "Cloud_registration_requried_link_text"}}</button>
 							</div>
 						{{/if}}
 					</div>

app/cloud/client/admin/cloud.js

@@ -34,7 +34,67 @@ Template.cloud.onCreated(function() {
 			}
 
 			if (!success) {
-				toastr.error('Invalid token');
+				toastr.error('An error occured connecting');
+				instance.loadRegStatus();
+				return;
+			}
+
+			toastr.success(t('Connected'));
+
+			instance.loadRegStatus();
+		});
+	};
+
+	instance.disconnectWorkspace = function _disconnectWorkspace() {
+		Meteor.call('cloud:disconnectWorkspace', (error, success) => {
+			if (error) {
+				toastr.error(error);
+				instance.loadRegStatus();
+				return;
+			}
+
+			if (!success) {
+				toastr.error('An error occured disconnecting');
+				instance.loadRegStatus();
+				return;
+			}
+
+			toastr.success(t('Disconnected'));
+
+			instance.loadRegStatus();
+		});
+	};
+
+	instance.syncWorkspace = function _syncWorkspace() {
+		Meteor.call('cloud:syncWorkspace', (error, success) => {
+			if (error) {
+				toastr.error(error);
+				instance.loadRegStatus();
+				return;
+			}
+
+			if (!success) {
+				toastr.error('An error occured syncing');
+				instance.loadRegStatus();
+				return;
+			}
+
+			toastr.success(t('Sync Complete'));
+
+			instance.loadRegStatus();
+		});
+	};
+
+	instance.registerWorkspace = function _registerWorkspace() {
+		Meteor.call('cloud:registerWorkspace', (error, success) => {
+			if (error) {
+				toastr.error(error);
+				instance.loadRegStatus();
+				return;
+			}
+
+			if (!success) {
+				toastr.error('An error occured');
 				instance.loadRegStatus();
 				return;
 			}
@@ -74,20 +134,21 @@ Template.cloud.events({
 		});
 	},
 
-	'click .login-btn'() {
-		Meteor.call('cloud:getOAuthAuthorizationUrl', (error, url) => {
-			if (error) {
-				console.warn(error);
-				return;
-			}
-
-			window.location.href = url;
-		});
-	},
-
 	'click .connect-btn'(e, i) {
 		const token = $('input[name=cloudToken]').val();
 
 		i.connectWorkspace(token);
 	},
+
+	'click .register-btn'(e, i) {
+		i.registerWorkspace();
+	},
+
+	'click .disconnect-btn'(e, i) {
+		i.disconnectWorkspace();
+	},
+
+	'click .sync-btn'(e, i) {
+		i.syncWorkspace();
+	},
 });

app/cloud/server/functions/connectWorkspace.js

@@ -8,9 +8,9 @@ import { getRedirectUri } from './getRedirectUri';
 import { retrieveRegistrationStatus } from './retrieveRegistrationStatus';
 
 export function connectWorkspace(token) {
-	const { registeredWithWizard } = retrieveRegistrationStatus();
-	if (!registeredWithWizard) {
-		return false;
+	const { connectToCloud } = retrieveRegistrationStatus();
+	if (!connectToCloud) {
+		Settings.updateValueById('Register_Server', true);
 	}
 
 	const redirectUri = getRedirectUri();

app/cloud/server/functions/disconnectWorkspace.js

@@ -0,0 +1,13 @@
+import { retrieveRegistrationStatus } from './retrieveRegistrationStatus';
+import { Settings } from '../../../models';
+
+export function disconnectWorkspace() {
+	const { connectToCloud } = retrieveRegistrationStatus();
+	if (!connectToCloud) {
+		return true;
+	}
+
+	Settings.updateValueById('Register_Server', false);
+
+	return true;
+}

app/cloud/server/functions/getWorkspaceAccessToken.js

@@ -5,9 +5,13 @@ import { settings } from '../../../settings';
 import { Settings } from '../../../models';
 
 import { getRedirectUri } from './getRedirectUri';
+import { retrieveRegistrationStatus } from './retrieveRegistrationStatus';
+import { unregisterWorkspace } from './unregisterWorkspace';
 
-export function getWorkspaceAccessToken() {
-	if (!settings.get('Register_Server')) {
+export function getWorkspaceAccessToken(forceNew = false) {
+	const { connectToCloud, workspaceRegistered } = retrieveRegistrationStatus();
+
+	if (!connectToCloud || !workspaceRegistered) {
 		return '';
 	}
 
@@ -19,7 +23,7 @@ export function getWorkspaceAccessToken() {
 	const expires = Settings.findOneById('Cloud_Workspace_Access_Token_Expires_At');
 	const now = new Date();
 
-	if (now < expires.value) {
+	if (now < expires.value && !forceNew) {
 		return settings.get('Cloud_Workspace_Access_Token');
 	}
 
@@ -39,6 +43,11 @@ export function getWorkspaceAccessToken() {
 			}),
 		});
 	} catch (e) {
+		if (e.response && e.response.data && e.response.data.errorCode === 'oauth_invalid_client_credentials') {
+			console.error('Server has been unregistered from cloud');
+			unregisterWorkspace();
+		}
+
 		return '';
 	}
 
@@ -48,6 +57,5 @@ export function getWorkspaceAccessToken() {
 	Settings.updateValueById('Cloud_Workspace_Access_Token', authTokenResult.data.access_token);
 	Settings.updateValueById('Cloud_Workspace_Access_Token_Expires_At', expiresAt);
 
-
 	return authTokenResult.data.access_token;
 }

app/cloud/server/functions/retrieveRegistrationStatus.js

@@ -3,9 +3,11 @@ import { Users } from '../../../models';
 
 export function retrieveRegistrationStatus() {
 	const info = {
-		registeredWithWizard: settings.get('Register_Server'),
-		workspaceConnected: (settings.get('Cloud_Workspace_Client_Id')) ? true : false,
+		connectToCloud: settings.get('Register_Server'),
+		workspaceRegistered: (settings.get('Cloud_Workspace_Client_Id')) ? true : false,
 		userAssociated: (settings.get('Cloud_Workspace_Account_Associated')) ? true : false,
+		workspaceId: settings.get('Cloud_Workspace_Id'),
+		uniqueId: settings.get('uniqueID'),
 		token: '',
 		email: '',
 	};

app/cloud/server/functions/startRegisterWorkspace.js

@@ -0,0 +1,66 @@
+import { HTTP } from 'meteor/http';
+import { settings } from '../../../settings';
+import { Settings } from '../../../models';
+
+import { retrieveRegistrationStatus } from './retrieveRegistrationStatus';
+
+import { statistics } from '../../../statistics';
+
+export function startRegisterWorkspace() {
+	const { workspaceRegistered, connectToCloud } = retrieveRegistrationStatus();
+	if ((workspaceRegistered && connectToCloud) || process.env.TEST_MODE) {
+		return true;
+	}
+
+	settings.updateById('Register_Server', true);
+
+	if (workspaceRegistered) {
+		return true;
+	}
+
+	const stats = statistics.get();
+
+	const address = settings.get('Site_Url');
+
+	// If we have it lets send it because likely an update
+	const workspaceId = settings.get('Cloud_Workspace_Id');
+
+	const regInfo = {
+		uniqueId: stats.uniqueId,
+		workspaceId,
+		address,
+		contactName: stats.wizard.contactName,
+		contactEmail: stats.wizard.contactEmail,
+		accountName: stats.wizard.organizationName,
+		siteName: stats.wizard.siteName,
+		deploymentMethod: stats.deploy.method,
+		deploymentPlatform: stats.deploy.platform,
+		version: stats.version,
+	};
+
+	const cloudUrl = settings.get('Cloud_Url');
+
+	let result;
+	try {
+		result = HTTP.post(`${ cloudUrl }/api/v2/register/workspace`, {
+			data: regInfo,
+		});
+	} catch (e) {
+		if (e.response && e.response.data && e.response.data.errorCode) {
+			console.error(`Failed to register with Rocket.Chat Cloud.  ErrorCode: ${ e.response.data.errorCode }`);
+		}
+		return false;
+	}
+
+	const { data } = result;
+
+	if (!data) {
+		return false;
+	}
+
+	Settings.updateValueById('Cloud_Workspace_Id', data.id);
+
+	console.log(data);
+
+	return true;
+}

app/cloud/server/functions/syncWorkspace.js

@@ -0,0 +1,58 @@
+import { HTTP } from 'meteor/http';
+import { settings } from '../../../settings';
+
+import { retrieveRegistrationStatus } from './retrieveRegistrationStatus';
+import { getWorkspaceAccessToken } from './getWorkspaceAccessToken';
+
+import { statistics } from '../../../statistics';
+import { getWorkspaceLicense } from './getWorkspaceLicense';
+
+export function syncWorkspace() {
+	const { workspaceRegistered, connectToCloud } = retrieveRegistrationStatus();
+	if (!workspaceRegistered || !connectToCloud) {
+		return false;
+	}
+
+	const stats = statistics.get();
+
+	const address = settings.get('Site_Url');
+
+	const info = {
+		uniqueId: stats.uniqueId,
+		address,
+		contactName: stats.wizard.contactName,
+		contactEmail: stats.wizard.contactEmail,
+		accountName: stats.wizard.organizationName,
+		siteName: stats.wizard.siteName,
+		deploymentMethod: stats.deploy.method,
+		deploymentPlatform: stats.deploy.platform,
+		version: stats.version,
+	};
+
+	const workspaceUrl = settings.get('Cloud_Workspace_Registration_Client_Uri');
+
+	try {
+		const headers = {};
+		const token = getWorkspaceAccessToken(true);
+
+		if (token) {
+			headers.Authorization = `Bearer ${ token }`;
+		} else {
+			return false;
+		}
+
+		HTTP.post(`${ workspaceUrl }/registration`, {
+			data: info,
+			headers,
+		});
+
+	} catch (e) {
+		if (e.response && e.response.data && e.response.data.errorCode) {
+			console.error(`Failed to sync with Rocket.Chat Cloud.  ErrorCode: ${ e.response.data.errorCode }`);
+		}
+
+		return false;
+	}
+
+	return getWorkspaceLicense();
+}