From ca3e03e3e5d9c3b9989434ff1283de4918034585 Mon Sep 17 00:00:00 2001
From: lqqyt2423 <974923609@qq.com>
Date: Wed, 10 Feb 2021 15:52:09 +0800
Subject: [PATCH] addon web client add buildMessage

---
 addon/web/client/src/App.css |  5 +++++
 addon/web/client/src/App.js  | 26 ++++++++++++++++++++++++--
 2 files changed, 29 insertions(+), 2 deletions(-)

diff --git a/addon/web/client/src/App.css b/addon/web/client/src/App.css
index 13df450..127b9b2 100644
--- a/addon/web/client/src/App.css
+++ b/addon/web/client/src/App.css
@@ -3,6 +3,11 @@
     font-size: 0.8rem;
 }
 
+.main-table-wrap tbody tr.tr-selected {
+    background-color: rgb(35, 118, 229);
+    color: white;
+}
+
 .flow-detail {
     position: fixed;
     top: 0;
diff --git a/addon/web/client/src/App.js b/addon/web/client/src/App.js
index 03faf56..d53b9ea 100644
--- a/addon/web/client/src/App.js
+++ b/addon/web/client/src/App.js
@@ -53,6 +53,22 @@ const parseMessage = data => {
   return resp
 }
 
+/**
+ * 
+ * @param {number} messageType 
+ * @param {string} id 
+ * @param {string} content 
+ */
+const buildMessage = (messageType, id, content) => {
+  content = new TextEncoder().encode(content)
+  const data = new Uint8Array(38 + content.byteLength)
+  data[0] = 1
+  data[1] = messageType
+  data.set(new TextEncoder().encode(id), 2)
+  data.set(content, 38)
+  return data
+}
+
 class App extends React.Component {
 
   constructor(props) {
@@ -81,7 +97,13 @@ class App extends React.Component {
   initWs() {
     if (this.ws) return
 
-    this.ws = new WebSocket("ws://localhost:9081/echo")
+    let host;
+    if (process.env.NODE_ENV === 'development') {
+      host = 'localhost:9081'
+    } else {
+      host = new URL(document.URL).host
+    }
+    this.ws = new WebSocket(`ws://${host}/echo`)
     this.ws.binaryType = 'arraybuffer'
     this.ws.onopen = () => { console.log('OPEN') }
     this.ws.onclose = () => { console.log('CLOSE') }
@@ -217,7 +239,7 @@ class App extends React.Component {
                 const request = f.request
                 const response = f.response || {}
                 return (
-                  <tr key={f.id} onClick={() => {
+                  <tr className={(this.state.flow && this.state.flow.id === f.id) ? "tr-selected" : null} key={f.id} onClick={() => {
                     this.setState({ flow: f })
                   }}>
                     <td>{u.host}</td>