From 2df239d14e030131a1ea82230e2de59e79963dd7 Mon Sep 17 00:00:00 2001
From: Quobix <dave@quobix.com>
Date: Wed, 16 Aug 2023 11:18:34 -0400
Subject: [PATCH] A few tune-ups from bugs discovered during testing.

Signed-off-by: Quobix <dave@quobix.com>
---
 ui/index.html                          | 5 ++---
 ui/src/components/controls/controls.ts | 4 ++--
 ui/src/model/http_transaction.ts       | 2 +-
 ui/src/wiretap.ts                      | 6 +++++-
 4 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/ui/index.html b/ui/index.html
index 90da189..0724b18 100644
--- a/ui/index.html
+++ b/ui/index.html
@@ -6,15 +6,14 @@
     <meta name="description"
           content="wiretap is a tool that sniffs your API traffic and validates it against an OpenAPI specification.">
     <script>
-        sessionStorage.setItem('wiretapPort', '%WIRETAP_PORT%')
-        //sessionStorage.setItem('wiretapPort', '9092')
+        localStorage.setItem('wiretapPort', '%WIRETAP_PORT%')
+        //localStorage.setItem('wiretapPort', '9092');
         const version = '%WIRETAP_VERSION%';
         const existingVersion = localStorage.getItem('wiretapVersion');
         if (existingVersion !== version) {
             localStorage.setItem('wiretapVersion', version);
             localStorage.setItem('wiretapWipeCache', 'true');
         }
-
     </script>
     <script type="module" src="/src/index.ts" defer></script>
     <script type="module" src="/src/workers/link_cache_worker.ts"></script>
diff --git a/ui/src/components/controls/controls.ts b/ui/src/components/controls/controls.ts
index 4b1bba6..a570f72 100644
--- a/ui/src/components/controls/controls.ts
+++ b/ui/src/components/controls/controls.ts
@@ -131,7 +131,7 @@ export class WiretapControlsComponent extends LitElement {
                 body: JSON.stringify(
                     {
                         id: RanchUtils.genUUID(),
-                        requestCommand: ChangeDelayCommand,
+                        request: ChangeDelayCommand,
                         payload: {
                             delay: delay
                         }
@@ -155,7 +155,7 @@ export class WiretapControlsComponent extends LitElement {
             body: JSON.stringify(
                 {
                     id: RanchUtils.genUUID(),
-                    requestCommand: RequestReportCommand,
+                    request: RequestReportCommand,
                     payload: {}
                 }
             ),
diff --git a/ui/src/model/http_transaction.ts b/ui/src/model/http_transaction.ts
index 9c3cb3c..f294042 100644
--- a/ui/src/model/http_transaction.ts
+++ b/ui/src/model/http_transaction.ts
@@ -135,7 +135,7 @@ export class HttpTransaction extends HttpTransactionBase {
     }
 
     matchesMethodFilter(filter: WiretapFilters): Filter | boolean {
-        if (filter?.filterMethod?.keyword.toLowerCase() === this.httpRequest.method.toLowerCase()) {
+        if (filter?.filterMethod?.keyword?.toLowerCase() === this.httpRequest?.method?.toLowerCase()) {
             return filter.filterMethod;
         }
         return false;
diff --git a/ui/src/wiretap.ts b/ui/src/wiretap.ts
index b6910c4..fbe8b52 100644
--- a/ui/src/wiretap.ts
+++ b/ui/src/wiretap.ts
@@ -79,7 +79,11 @@ export class WiretapComponent extends LitElement {
         });
 
         // extract port from session storage.
-        this._wiretapPort = sessionStorage.getItem("wiretapPort");
+        this._wiretapPort = localStorage.getItem("wiretapPort");
+
+        if (!this._wiretapPort) {
+            this._wiretapPort = "9092"; // default port
+        }
 
         // extract version from session storage.
         this._wiretapVersion = localStorage.getItem("wiretapVersion");