Fix a bug with needle highlighting

cc #186
rust-dev-tools · Apr 12, 2018 · 5414a0f · 5414a0f
1 parent aab3d9a
commit 5414a0f
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/static/search.js b/static/search.js
@@ -13,6 +13,13 @@ const utils = require('./utils');
 
 class ResultSet extends React.Component {
     componentDidMount() {
+        this.postRender();
+    }
+    componentDidUpdate() {
+        this.postRender();
+    }
+
+    postRender() {
         $(".div_search_results .src_link").removeClass("src_link");
         highlight_needle(this.props.input, this.props.kind);
     }
@@ -90,6 +97,13 @@ function FileResult(props) {
 
 class StructuredResultSet extends React.Component {
     componentDidMount() {
+        this.postRender();
+    }
+    componentDidUpdate() {
+        this.postRender();
+    }
+
+    postRender() {
         $(".div_search_group .src_link").removeClass("src_link");
         const defFile = { file_name: this.props.input.file, lines: [this.props.input.line] };
         highlight_needle([defFile], "def");