From 4a09d25586b281b11f3b7d2cae397ee79e46ec2c Mon Sep 17 00:00:00 2001
From: Zephyruso <176294927+Zephyruso@users.noreply.github.com>
Date: Sun, 1 Dec 2024 14:31:03 +0800
Subject: [PATCH] feat: source ip filter

---
 src/components/sidebar/ConnectionCtrl.vue | 38 +++++++++++++++++++++--
 src/store/connections.ts                  | 12 ++++++-
 2 files changed, 47 insertions(+), 3 deletions(-)

diff --git a/src/components/sidebar/ConnectionCtrl.vue b/src/components/sidebar/ConnectionCtrl.vue
index 4060610d..468c21f5 100644
--- a/src/components/sidebar/ConnectionCtrl.vue
+++ b/src/components/sidebar/ConnectionCtrl.vue
@@ -45,10 +45,42 @@
         v-model="quickFilterEnabled"
       />
     </div>
-    <div class="join">
+    <select
+      v-if="!horizontal"
+      class="select select-bordered select-sm"
+      v-model="sourceIPFilter"
+    >
+      <option
+        value=""
+        label="all"
+      ></option>
+      <option
+        v-for="opt in sourceIPs"
+        :key="opt"
+        :value="opt"
+        :label="opt"
+      />
+    </select>
+    <div class="join flex-1">
+      <select
+        v-if="horizontal"
+        class="join-item select select-bordered select-sm min-w-24"
+        v-model="sourceIPFilter"
+      >
+        <option
+          value=""
+          label="all"
+        ></option>
+        <option
+          v-for="opt in sourceIPs"
+          :key="opt"
+          :value="opt"
+          :label="opt"
+        />
+      </select>
       <input
         type="text"
-        class="input input-sm join-item input-bordered flex-1"
+        class="input input-sm join-item input-bordered w-24 flex-1"
         v-model="connectionFilter"
       />
       <button
@@ -82,6 +114,8 @@ import {
   quickFilterEnabled,
   quickFilterRegex,
   renderConnections,
+  sourceIPFilter,
+  sourceIPs,
 } from '@/store/connections'
 import { PauseIcon, PlayIcon, XMarkIcon } from '@heroicons/vue/24/outline'
 import { twMerge } from 'tailwind-merge'
diff --git a/src/store/connections.ts b/src/store/connections.ts
index 436ed8a4..385128af 100644
--- a/src/store/connections.ts
+++ b/src/store/connections.ts
@@ -3,7 +3,7 @@ import { CONNECTION_TAB_TYPE, SORT_TYPE } from '@/config'
 import type { Connection, ConnectionRawMessage } from '@/types'
 import { useStorage } from '@vueuse/core'
 import dayjs from 'dayjs'
-import { differenceWith } from 'lodash'
+import _, { differenceWith } from 'lodash'
 import { computed, ref, watch } from 'vue'
 import { useConnectionCard } from './config'
 
@@ -137,6 +137,10 @@ export const renderConnections = computed(() => {
         }
       }
 
+      if (sourceIPFilter.value && conn.metadata.sourceIP === sourceIPFilter.value) {
+        return false
+      }
+
       if (connectionFilter.value) {
         return [
           conn.metadata.host,
@@ -166,3 +170,9 @@ export const renderConnections = computed(() => {
       return sortResult
     })
 })
+
+export const sourceIPFilter = ref('')
+
+export const sourceIPs = computed(() => {
+  return _.uniq(activeConnections.value.map((conn) => conn.metadata.sourceIP))
+})