mol-plugin-ui: close Screenshot/snapshot panel after saving/opening state

src/mol-plugin-ui/state/snapshots.tsx

@@ -42,12 +42,14 @@ export class StateSnapshots extends PluginUIComponent<{}> {
     }
 }
 
-export class StateExportImportControls extends PluginUIComponent {
+export class StateExportImportControls extends PluginUIComponent<{ onAction?: () => void }> {
     downloadToFileJson = () => {
+        this.props.onAction?.();
         PluginCommands.State.Snapshots.DownloadToFile(this.plugin, { type: 'json' });
     }
 
     downloadToFileZip = () => {
+        this.props.onAction?.();
         PluginCommands.State.Snapshots.DownloadToFile(this.plugin, { type: 'zip' });
     }
 
@@ -57,6 +59,7 @@ export class StateExportImportControls extends PluginUIComponent {
             return;
         }
 
+        this.props.onAction?.();
         PluginCommands.State.Snapshots.OpenFile(this.plugin, { file: e.target.files[0] });
     }
 

src/mol-plugin-ui/viewport/screenshot.tsx

@@ -146,7 +146,7 @@ export class DownloadScreenshotControls extends PluginUIComponent<{ close: () =>
             </div>
             <ParameterControls params={this.plugin.helpers.viewportScreenshot!.params} values={this.plugin.helpers.viewportScreenshot!.values} onChange={this.setProps} isDisabled={this.state.isDisabled} />
             <ExpandGroup header='State Snapshot'>
-                <StateExportImportControls />
+                <StateExportImportControls onAction={this.props.close} />
                 <ExpandGroup header='Save Options' initiallyExpanded={false} noOffset>
                     <LocalStateSnapshotParams />
                 </ExpandGroup>