From c68dcf41ae6d52459e91fd08d0dab4d970a7a327 Mon Sep 17 00:00:00 2001
From: fuomag9 <1580624+fuomag9@users.noreply.github.com>
Date: Mon, 23 Mar 2026 14:16:21 +0100
Subject: [PATCH] feat: add feature badges for WAF, Geo, LB, mTLS, Authentik,
 Redirects, and Rewrite in proxy hosts table

Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
---
 .../proxy-hosts/ProxyHostsClient.tsx          | 65 +++++++++++++++----
 1 file changed, 51 insertions(+), 14 deletions(-)

diff --git a/app/(dashboard)/proxy-hosts/ProxyHostsClient.tsx b/app/(dashboard)/proxy-hosts/ProxyHostsClient.tsx
index 6ac0069a..b3b741a2 100644
--- a/app/(dashboard)/proxy-hosts/ProxyHostsClient.tsx
+++ b/app/(dashboard)/proxy-hosts/ProxyHostsClient.tsx
@@ -2,7 +2,7 @@
 
 import { useEffect, useRef, useState } from "react";
 import { useRouter, usePathname, useSearchParams } from "next/navigation";
-import { Globe, MoreHorizontal, ArrowRight, Shield } from "lucide-react";
+import { Globe, MoreHorizontal, ArrowRight, Shield, Bug, MapPin, Scale, KeyRound, UserCheck, CornerRightDown, Replace } from "lucide-react";
 import type { AccessList } from "@/lib/models/access-lists";
 import type { Certificate } from "@/lib/models/certificates";
 import type { ProxyHost } from "@/lib/models/proxy-hosts";
@@ -118,21 +118,58 @@ export default function ProxyHostsClient({ hosts, certificates, accessLists, caC
     {
       id: "features",
       label: "Features",
-      render: (host: ProxyHost) => (
-        <div className="flex flex-wrap gap-1">
-          {host.certificate_id && (
-            <Badge variant="info" className="text-[10px] px-1.5 py-0">TLS</Badge>
-          )}
-          {host.access_list_id && (
-            <Badge variant="warning" className="text-[10px] px-1.5 py-0">
+      render: (host: ProxyHost) => {
+        const badges = [
+          host.certificate_id && (
+            <Badge key="tls" variant="info" className="text-[10px] px-1.5 py-0">TLS</Badge>
+          ),
+          host.access_list_id && (
+            <Badge key="auth" variant="warning" className="text-[10px] px-1.5 py-0">
               <Shield className="h-2.5 w-2.5 mr-0.5" />Auth
             </Badge>
-          )}
-          {!host.certificate_id && !host.access_list_id && (
-            <span className="text-xs text-muted-foreground">—</span>
-          )}
-        </div>
-      ),
+          ),
+          host.authentik?.enabled && (
+            <Badge key="authentik" variant="secondary" className="text-[10px] px-1.5 py-0">
+              <UserCheck className="h-2.5 w-2.5 mr-0.5" />Authentik
+            </Badge>
+          ),
+          host.waf?.enabled && (
+            <Badge key="waf" variant="secondary" className="text-[10px] px-1.5 py-0">
+              <Bug className="h-2.5 w-2.5 mr-0.5" />WAF
+            </Badge>
+          ),
+          host.geoblock?.enabled && (
+            <Badge key="geo" variant="secondary" className="text-[10px] px-1.5 py-0">
+              <MapPin className="h-2.5 w-2.5 mr-0.5" />Geo
+            </Badge>
+          ),
+          host.load_balancer?.enabled && (
+            <Badge key="lb" variant="secondary" className="text-[10px] px-1.5 py-0">
+              <Scale className="h-2.5 w-2.5 mr-0.5" />LB
+            </Badge>
+          ),
+          host.mtls?.enabled && (
+            <Badge key="mtls" variant="secondary" className="text-[10px] px-1.5 py-0">
+              <KeyRound className="h-2.5 w-2.5 mr-0.5" />mTLS
+            </Badge>
+          ),
+          host.redirects?.length > 0 && (
+            <Badge key="redirects" variant="secondary" className="text-[10px] px-1.5 py-0">
+              <CornerRightDown className="h-2.5 w-2.5 mr-0.5" />Redirects
+            </Badge>
+          ),
+          host.rewrite && (
+            <Badge key="rewrite" variant="secondary" className="text-[10px] px-1.5 py-0">
+              <Replace className="h-2.5 w-2.5 mr-0.5" />Rewrite
+            </Badge>
+          ),
+        ].filter(Boolean);
+        return (
+          <div className="flex flex-wrap gap-1">
+            {badges.length > 0 ? badges : <span className="text-xs text-muted-foreground">—</span>}
+          </div>
+        );
+      },
     },
     {
       id: "status",