From e58cfd89c58649ed3ae32e2304495f31b057d865 Mon Sep 17 00:00:00 2001
From: Alexis Svinartchouk <alexis@resin.io>
Date: Tue, 10 Nov 2020 13:21:00 +0100
Subject: [PATCH] Add successBannerURL setting

Change-type: patch
---
 lib/gui/app/components/finish/finish.tsx | 38 ++++++++++++++++--------
 1 file changed, 26 insertions(+), 12 deletions(-)

diff --git a/lib/gui/app/components/finish/finish.tsx b/lib/gui/app/components/finish/finish.tsx
index d8bf0593..508bb3fa 100644
--- a/lib/gui/app/components/finish/finish.tsx
+++ b/lib/gui/app/components/finish/finish.tsx
@@ -20,6 +20,7 @@ import { v4 as uuidV4 } from 'uuid';
 
 import * as flashState from '../../models/flash-state';
 import * as selectionState from '../../models/selection-state';
+import * as settings from '../../models/settings';
 import { Actions, store } from '../../models/store';
 import * as analytics from '../../modules/analytics';
 import { FlashAnother } from '../flash-another/flash-another';
@@ -39,8 +40,19 @@ function restart(goToMain: () => void) {
 	goToMain();
 }
 
+async function getSuccessBannerURL() {
+	return (
+		(await settings.get('successBannerURL')) ??
+		'https://www.balena.io/etcher/success-banner?borderTop=false&darkBackground=true'
+	);
+}
+
 function FinishPage({ goToMain }: { goToMain: () => void }) {
 	const [webviewShowing, setWebviewShowing] = React.useState(false);
+	const [successBannerURL, setSuccessBannerURL] = React.useState('');
+	(async () => {
+		setSuccessBannerURL(await getSuccessBannerURL());
+	})();
 	const flashResults = flashState.getFlashResults();
 	const errors: FlashError[] = (
 		store.getState().toJS().failedDeviceErrors || []
@@ -96,18 +108,20 @@ function FinishPage({ goToMain }: { goToMain: () => void }) {
 					}}
 				/>
 			</Flex>
-			<SafeWebview
-				src="https://www.balena.io/etcher/success-banner?borderTop=false&darkBackground=true"
-				onWebviewShow={setWebviewShowing}
-				style={{
-					display: webviewShowing ? 'flex' : 'none',
-					position: 'absolute',
-					right: 0,
-					bottom: 0,
-					width: '63.8vw',
-					height: '100vh',
-				}}
-			/>
+			{successBannerURL.length && (
+				<SafeWebview
+					src={successBannerURL}
+					onWebviewShow={setWebviewShowing}
+					style={{
+						display: webviewShowing ? 'flex' : 'none',
+						position: 'absolute',
+						right: 0,
+						bottom: 0,
+						width: '63.8vw',
+						height: '100vh',
+					}}
+				/>
+			)}
 		</Flex>
 	);
 }