handle wildcard domains

pollystack · Nov 7, 2024 · 7ffc9e6 · 7ffc9e6
1 parent 829f979
commit 7ffc9e6
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/fast-server/server/server.go b/fast-server/server/server.go
@@ -113,7 +113,17 @@ func (s *Server) setupRoutes() {
 			var matchedDomain config.Domain
 			var matchedName string
 			for domainName, domain := range domainMap {
-				if strings.HasSuffix(host, domainName) {
+				// Handle wildcard domains
+				if strings.HasPrefix(domainName, "*.") {
+					suffix := domainName[1:] // Remove the *
+					if strings.HasSuffix(host, suffix) {
+						if len(suffix) > len(matchedName) {
+							matchedDomain = domain
+							matchedName = suffix
+						}
+					}
+				} else if strings.HasSuffix(host, domainName) {
+					// Handle exact domain matches
 					if len(domainName) > len(matchedName) {
 						matchedDomain = domain
 						matchedName = domainName