(core) Fixing bug with collapsed custom widget.

Summary:
Fix for a bug. Custom widget when collapsed and expanded was disconnecting from
Grist, as WidgetFrame was disposed to early.

Test Plan: Added new

Reviewers: georgegevoian

Reviewed By: georgegevoian

Differential Revision: https://phab.getgrist.com/D4109

app/client/components/CustomView.ts

@@ -242,7 +242,7 @@ export class CustomView extends Disposable {
     const {baseUrl, access, showAfterReady, widgetId, pluginId} = options;
     const documentSettings = this.gristDoc.docData.docSettings();
     const readonly = this.gristDoc.isReadonly.get();
-    return grains.create(WidgetFrame, {
+    const frame = WidgetFrame.create(null,  {
       url: baseUrl || this.getEmptyWidgetPage(),
       widgetId,
       pluginId,
@@ -304,6 +304,12 @@ export class CustomView extends Disposable {
       gristDoc: this.gristDoc,
     });
 
+    // Can't use dom.create() because it seems buggy in this context. This dom will be detached
+    // and attached several times, and dom.create() doesn't seem to handle that well as it returns an
+    // array of nodes (comment, node, comment) and it somehow breaks the dispose order. Collapsed widgets
+    // relay on a correct order of dispose, and are detaching nodes just before they are disposed, so if
+    // the order is wrong, the node is disposed without being detached first.
+    return grains.update(frame.buildDom(), dom.autoDispose(frame));
   }
 }
 

app/client/components/WidgetFrame.ts

@@ -185,8 +185,7 @@ export class WidgetFrame extends DisposableWithEvents {
 
   public buildDom() {
     const onElem = this._options.onElem ?? ((el: HTMLIFrameElement) => el);
-    return onElem(
+    this._iframe = dom(
-      (this._iframe = dom(
       'iframe',
       dom.style('visibility', use => use(this._visible) ? 'visible' : 'hidden'),
       dom.cls('clipboard_focus'),
@@ -194,8 +193,9 @@ export class WidgetFrame extends DisposableWithEvents {
       dom.attr('src', use => this._getUrl(use(this._widget))),
       hooks.iframeAttributes,
       testId('ready', this._readyCalled),
-      ))
+      self => void onElem(self),
     );
+    return this._iframe;
   }
 
   private _getUrl(widget: ICustomWidget|null): string {

test/nbrowser/ViewLayoutCollapse.ts

@@ -4,11 +4,12 @@ import {getCollapsedSection, openCollapsedSectionMenu} from 'test/nbrowser/ViewL
 import {assert, driver, Key, WebElement, WebElementPromise} from 'mocha-webdriver';
 import {arrayRepeat} from 'app/plugin/gutil';
 import {addStatic, serveSomething} from 'test/server/customUtil';
+import {AccessLevel} from 'app/common/CustomWidget';
 
 const GAP = 16;     // Distance between buttons representing collapsed widgets.
 
 describe("ViewLayoutCollapse", function() {
-  this.timeout(40000);
+  this.timeout('50s');
   const cleanup = setupTestSuite();
   gu.bigScreen();
   let session: gu.Session;
@@ -19,6 +20,45 @@ describe("ViewLayoutCollapse", function() {
     await gu.openPage("Overview");
   });
 
+  it('fix: custom widget should restart when added back after collapsing', async function() {
+    const revert = await gu.begin();
+
+    // Add custom section.
+    await gu.addNewPage('Table', 'Companies');
+    await gu.addNewSection('Custom', 'Companies', { selectBy: 'COMPANIES'});
+
+    // Serve custom widget.
+    const widgetServer = await serveSomething(app => {
+      addStatic(app);
+    });
+    cleanup.addAfterAll(widgetServer.shutdown);
+    await gu.openWidgetPanel();
+    await gu.setWidgetUrl(widgetServer.url + '/probe/index.html');
+    await gu.widgetAccess(AccessLevel.full);
+
+    // Collapse it.
+    await collapseByMenu('COMPANIES Custom');
+
+    // Now restore its position.
+    await addToMainByMenu('COMPANIES Custom');
+
+    // Collapsed widget used to lost connection with Grist as it was disposed to early.
+    // Make sure that this widget can call the API.
+    await gu.doInIframe(async () => {
+      await gu.waitToPass(async () => {
+        assert.equal(await driver.find('#output').getText(),
+          `["Companies","Investments","Companies_summary_category_code","Investments_summary_funded_year",` +
+          `"Investments_summary_Company_category_code_funded_year","Investments_summary_Company_category_code"]`
+        );
+      });
+    });
+
+
+    // Make sure we don't have an error.
+    await gu.checkForErrors();
+    await revert();
+  });
+
   it('fix: custom widget should not throw errors when collapsed', async function() {
     const revert = await gu.begin();
 
@@ -33,9 +73,7 @@ describe("ViewLayoutCollapse", function() {
     cleanup.addAfterAll(widgetServer.shutdown);
     await gu.openWidgetPanel();
     await gu.setWidgetUrl(widgetServer.url + '/probe/index.html');
-    await driver.find('.test-config-widget-access .test-select-open').click();
+    await gu.widgetAccess(AccessLevel.full);
-    await driver.findContent('.test-select-menu li', 'Full document access').click();
-    await gu.waitForServer();
 
     // Collapse it.
     await collapseByMenu('COMPANIES Custom');

test/nbrowser/gristUtils.ts

@@ -894,8 +894,17 @@ export async function importUrlDialog(url: string): Promise<void> {
  * Executed passed function in the context of given iframe, and then switching back to original context
  *
  */
-export async function doInIframe<T>(iframe: WebElement, func: () => Promise<T>) {
+export async function doInIframe<T>(func: () => Promise<T>): Promise<T>
+export async function doInIframe<T>(iframe: WebElement, func: () => Promise<T>): Promise<T>
+export async function doInIframe<T>(frameOrFunc: WebElement|(() => Promise<T>), func?: () => Promise<T>): Promise<T> {
   try {
+    let iframe: WebElement;
+    if (!func) {
+      func = frameOrFunc as () => Promise<T>;
+      iframe = await driver.findWait('iframe', 5000);
+    } else {
+      iframe = frameOrFunc as WebElement;
+    }
     await driver.switchTo().frame(iframe);
     return await func();
   } finally {