From eedf1b08f63edec42164c0df4e20a754e21db9a5 Mon Sep 17 00:00:00 2001 From: bashrc Date: Mon, 23 Feb 2026 14:55:54 +0000 Subject: [PATCH] Get or remove domain port for yggdrasil --- utils.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/utils.py b/utils.py index 39a3f98bc..244355bf3 100644 --- a/utils.py +++ b/utils.py @@ -2963,7 +2963,10 @@ def remove_domain_port(domain: str) -> str: if ':' in domain: if domain.startswith('did:'): return domain - domain = domain.split(':')[0] + if ']:' not in domain: + domain = domain.split(':')[0] + else: + domain = domain.split(']:')[0] + ']' return domain @@ -2974,7 +2977,10 @@ def get_port_from_domain(domain: str) -> int: if ':' in domain: if domain.startswith('did:'): return None - port_str = domain.split(':')[1] + if ']:' not in domain: + port_str = domain.split(':')[1] + else: + port_str = domain.split(']:')[1] if port_str.isdigit(): return int(port_str) return None