From e812e3ff89ee7802a8ef2ae03a3f368773706196 Mon Sep 17 00:00:00 2001
From: ibuler <ibuler@qq.com>
Date: Tue, 5 Dec 2023 14:29:08 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BC=98=E5=8C=96=20endpoint=20?=
 =?UTF-8?q?=E7=9A=84=20ipv6=20=E6=94=AF=E6=8C=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 apps/terminal/models/component/endpoint.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/apps/terminal/models/component/endpoint.py b/apps/terminal/models/component/endpoint.py
index 9b73dbea0..8e73bb17d 100644
--- a/apps/terminal/models/component/endpoint.py
+++ b/apps/terminal/models/component/endpoint.py
@@ -131,5 +131,11 @@ class EndpointRule(JMSBaseModel):
             endpoint = Endpoint.get_or_create_default(request)
         if not endpoint.host and request:
             # 动态添加 current request host
-            endpoint.host = request.get_host().split(':')[0]
+            host_port = request.get_host()
+            # IPv6
+            if host_port.startswith('['):
+                host = host_port.split(']:')[0].rstrip(']') + ']'
+            else:
+                host = host_port.split(':')[0]
+            endpoint.host = host
         return endpoint