From 5625cbce20422f04cc0ed965ed45cbff5147d8ff Mon Sep 17 00:00:00 2001
From: Matt Bierner <matb@microsoft.com>
Date: Tue, 29 May 2018 14:47:55 -0700
Subject: [PATCH] Make sure we update the webview's state on setState

---
 .../workbench/parts/webview/electron-browser/webview-pre.js | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/vs/workbench/parts/webview/electron-browser/webview-pre.js b/src/vs/workbench/parts/webview/electron-browser/webview-pre.js
index 9d1ce8f49fc..51a9d42aa32 100644
--- a/src/vs/workbench/parts/webview/electron-browser/webview-pre.js
+++ b/src/vs/workbench/parts/webview/electron-browser/webview-pre.js
@@ -190,8 +190,10 @@
 								postMessage: function(msg) {
 									return originalPostMessage({ command: 'onmessage', data: msg }, '*');
 								},
-								setState: function(state) {
-									return originalPostMessage({ command: 'do-update-state', data: JSON.stringify(state) }, '*');
+								setState: function(newState) {
+									state = newState;
+									originalPostMessage({ command: 'do-update-state', data: JSON.stringify(newState) }, '*');
+									return newState;
 								},
 								getState: function() {
 									return state;