From 6b30ace7f6919d5e5122959d0abf463906fa17d7 Mon Sep 17 00:00:00 2001
From: Emiliano Ciavatta
Date: Mon, 19 Oct 2020 16:20:53 +0200
Subject: Highlight rules in stream pane

---
 frontend/src/components/panels/StreamsPane.js   | 40 ++++++++++++++++++++-----
 frontend/src/components/panels/StreamsPane.scss |  6 ++++
 2 files changed, 38 insertions(+), 8 deletions(-)

(limited to 'frontend/src/components/panels')

diff --git a/frontend/src/components/panels/StreamsPane.js b/frontend/src/components/panels/StreamsPane.js
index 9470d7d..cc3d53a 100644
--- a/frontend/src/components/panels/StreamsPane.js
+++ b/frontend/src/components/panels/StreamsPane.js
@@ -21,12 +21,14 @@ import {Row} from "react-bootstrap";
 import ReactJson from "react-json-view";
 import backend from "../../backend";
 import log from "../../log";
+import rules from "../../model/rules";
 import {downloadBlob, getHeaderValue} from "../../utils";
 import ButtonField from "../fields/ButtonField";
 import ChoiceField from "../fields/ChoiceField";
 import MessageAction from "../objects/MessageAction";
 import "./StreamsPane.scss";
 
+const reactStringReplace = require('react-string-replace')
 const classNames = require("classnames");
 
 class StreamsPane extends Component {
@@ -77,11 +79,9 @@ class StreamsPane extends Component {
     };
 
     tryParseConnectionMessage = (connectionMessage) => {
+        const isClient = connectionMessage["from_client"];
         if (connectionMessage.metadata == null) {
-            return connectionMessage.content;
-        }
-        if (connectionMessage["is_metadata_continuation"]) {
-            return <span style={{"fontSize": "12px"}}>**already parsed in previous messages**</span>;
+            return this.highlightRules(connectionMessage.content, isClient);
         }
 
         let unrollMap = (obj) => obj == null ? null : Object.entries(obj).map(([key, value]) =>
@@ -96,7 +96,7 @@ class StreamsPane extends Component {
                 return <span className="type-http-request">
                     <p style={{"marginBottom": "7px"}}><strong>{m.method}</strong> {url} {m.protocol}</p>
                     {unrollMap(m.headers)}
-                    <div style={{"margin": "20px 0"}}>{m.body}</div>
+                    <div style={{"margin": "20px 0"}}>{this.highlightRules(m.body, isClient)}</div>
                     {unrollMap(m.trailers)}
                 </span>;
             case "http-response":
@@ -108,20 +108,44 @@ class StreamsPane extends Component {
                         body = <ReactJson src={json} theme="grayscale" collapsed={false} displayDataTypes={false}/>;
                     } catch (e) {
                         log.error(e);
+                        body = m.body;
                     }
                 }
 
                 return <span className="type-http-response">
                     <p style={{"marginBottom": "7px"}}>{m.protocol} <strong>{m.status}</strong></p>
                     {unrollMap(m.headers)}
-                    <div style={{"margin": "20px 0"}}>{body}</div>
+                    <div style={{"margin": "20px 0"}}>{this.highlightRules(body, isClient)}</div>
                     {unrollMap(m.trailers)}
                 </span>;
             default:
-                return connectionMessage.content;
+                return this.highlightRules(connectionMessage.content, isClient);
         }
     };
 
+    highlightRules = (content, isClient) => {
+        let streamContent = content;
+        this.props.connection["matched_rules"].forEach(ruleId => {
+            const rule = rules.ruleById(ruleId);
+            rule.patterns.forEach(pattern => {
+                if ((!isClient && pattern.direction === 1) || (isClient && pattern.direction === 2)) {
+                    return;
+                }
+                let flags = "";
+                pattern["caseless"] && (flags += "i");
+                pattern["dot_all"] && (flags += "s");
+                pattern["multi_line"] && (flags += "m");
+                pattern["unicode_property"] && (flags += "u");
+                const regex = new RegExp("(" + pattern.regex + ")", flags);
+                streamContent = reactStringReplace(streamContent, regex, (match, i) => (
+                    <span key={i} className="matched-occurrence" style={{"backgroundColor": rule.color}}>{match}</span>
+                ));
+            });
+        });
+
+        return streamContent;
+    };
+
     connectionsActions = (connectionMessage) => {
         if (!connectionMessage.metadata) {
             return null;
@@ -187,7 +211,7 @@ class StreamsPane extends Component {
         };
         const content = this.state.messages || [];
 
-        let payload = content.map((c, i) =>
+        let payload = content.filter((c) => !c["is_metadata_continuation"]).map((c, i) =>
             <div key={`content-${i}`}
                  className={classNames("connection-message", c["from_client"] ? "from-client" : "from-server")}>
                 <div className="connection-message-header container-fluid">
diff --git a/frontend/src/components/panels/StreamsPane.scss b/frontend/src/components/panels/StreamsPane.scss
index 1d8a250..c44c7eb 100644
--- a/frontend/src/components/panels/StreamsPane.scss
+++ b/frontend/src/components/panels/StreamsPane.scss
@@ -18,6 +18,12 @@
     &:hover::-webkit-scrollbar-thumb {
       background: $color-secondary-2;
     }
+
+    .matched-occurrence {
+      color: $color-primary-4;
+      font-weight: 500;
+      border-radius: 2px;
+    }
   }
 
   .connection-message {
-- 
cgit v1.2.3-70-g09d2