Add getInitialQuery to documentsearch interface

packages/csvviewer-extension/src/searchprovider.ts

@@ -17,6 +17,17 @@ export class CSVSearchProvider implements ISearchProvider {
     );
   }
 
+  /**
+   * Get an initial query value if applicable so that it can be entered
+   * into the search box as an initial query
+   *
+   * @returns Initial value used to populate the search box.
+   */
+  getInitialQuery(searchTarget: IDocumentWidget<CSVViewer>): any {
+    // CSV Viewer does not support selection
+    return null;
+  }
+
   /**
    * Initialize the search using the provided options.  Should update the UI
    * to highlight all matches and "select" whatever the first match should be.

packages/documentsearch/src/interfaces.ts

@@ -106,6 +106,15 @@ export interface ISearchProviderConstructor {
 }
 
 export interface ISearchProvider {
+  /**
+   * Get an initial query value if applicable so that it can be entered
+   * into the search box as an initial query
+   *
+   * @param searchTarget The widget to be searched
+   *
+   * @returns Initial value used to populate the search box.
+   */
+  getInitialQuery(searchTarget: Widget): any;
   /**
    * Initialize the search using the provided options.  Should update the UI
    * to highlight all matches and "select" whatever the first match should be.

packages/documentsearch/src/providers/codemirrorsearchprovider.ts

@@ -44,6 +44,21 @@ import { FileEditor } from '@jupyterlab/fileeditor';
 type MatchMap = { [key: number]: { [key: number]: ISearchMatch } };
 
 export class CodeMirrorSearchProvider implements ISearchProvider {
+  /**
+   * Get an initial query value if applicable so that it can be entered
+   * into the search box as an initial query
+   *
+   * @returns Initial value used to populate the search box.
+   */
+  getInitialQuery(searchTarget: Widget): any {
+    const target = searchTarget as MainAreaWidget;
+    const content = target.content as FileEditor;
+    const cm = content.editor as CodeMirrorEditor;
+    const selection = cm.doc.getSelection();
+    // if there are newlines, just return empty string
+    return selection.search(/\r?\n|\r/g) === -1 ? selection : '';
+  }
+
   /**
    * Initialize the search using the provided options.  Should update the UI
    * to highlight all matches and "select" whatever the first match should be.
@@ -63,8 +78,8 @@ export class CodeMirrorSearchProvider implements ISearchProvider {
 
     // Extract the codemirror object from the editor widget. Each of these casts
     // is justified by the canSearchOn call above.
-    let target = searchTarget as MainAreaWidget;
-    let content = target.content as FileEditor;
+    const target = searchTarget as MainAreaWidget;
+    const content = target.content as FileEditor;
     this._cm = content.editor as CodeMirrorEditor;
     return this._startQuery(query);
   }

packages/documentsearch/src/providers/notebooksearchprovider.ts

@@ -17,6 +17,19 @@ interface ICellSearchPair {
 }
 
 export class NotebookSearchProvider implements ISearchProvider {
+  /**
+   * Get an initial query value if applicable so that it can be entered
+   * into the search box as an initial query
+   *
+   * @returns Initial value used to populate the search box.
+   */
+  getInitialQuery(searchTarget: NotebookPanel): any {
+    const activeCell = searchTarget.content.activeCell;
+    const selection = (activeCell.editor as CodeMirrorEditor).doc.getSelection();
+    // if there are newlines, just return empty string
+    return selection.search(/\r?\n|\r/g) === -1 ? selection : '';
+  }
+
   /**
    * Initialize the search using the provided options.  Should update the UI
    * to highlight all matches and "select" whatever the first match should be.

packages/documentsearch/src/searchinstance.ts

@@ -17,6 +17,9 @@ export class SearchInstance implements IDisposable {
     this._widget = widget;
     this._activeProvider = searchProvider;
 
+    const initialQuery = this._activeProvider.getInitialQuery(this._widget);
+    this._displayState.searchText = initialQuery || '';
+
     this._searchWidget = createSearchOverlay({
       widgetChanged: this._displayUpdateSignal,
       overlayState: this._displayState,

packages/documentsearch/src/searchoverlay.tsx

@@ -243,6 +243,12 @@ class SearchOverlay extends React.Component<
     this.state = props.overlayState;
   }
 
+  componentDidMount() {
+    if (this.state.searchText) {
+      this._executeSearch(true, this.state.searchText);
+    }
+  }
+
   private _onSearchChange(event: React.ChangeEvent) {
     const searchText = (event.target as HTMLInputElement).value;
     this.setState({ searchText: searchText });