From 0198f6f0ec86361de4ed5f25ba49ded9371ce8a1 Mon Sep 17 00:00:00 2001
From: zu1k <i@lgf.im>
Date: Thu, 16 Dec 2021 13:20:24 +0800
Subject: [PATCH] fix: CName match #66

Signed-off-by: zu1k <i@lgf.im>
---
 pkg/cdn/cdn.go | 42 +++++++++++++++++++++++++-----------------
 1 file changed, 25 insertions(+), 17 deletions(-)

diff --git a/pkg/cdn/cdn.go b/pkg/cdn/cdn.go
index 30f221f7..6ff54190 100644
--- a/pkg/cdn/cdn.go
+++ b/pkg/cdn/cdn.go
@@ -58,27 +58,35 @@ func NewCDN(filePath string) *CDN {
 
 func (db CDN) Find(query string, params ...string) (result fmt.Stringer, err error) {
 	baseCname := parseBaseCname(query)
-	if baseCname == "" {
-		return nil, errors.New("base domain parse failed")
-	}
-	cdnResult, found := db.Data[baseCname]
-	if found {
-		return cdnResult, nil
-	}
+	for _, domain := range baseCname {
+		if domain != "" {
+			cdnResult, found := db.Data[domain]
+			if found {
+				return cdnResult, nil
+			}
+		}
 
-	if strings.Contains(baseCname, "kunlun") {
-		return CDNResult{
-			Name: "阿里云 CDN",
-		}, nil
+		if strings.Contains(domain, "kunlun") {
+			return CDNResult{
+				Name: "阿里云 CDN",
+			}, nil
+		}
 	}
+
 	return nil, errors.New("not found")
 }
 
-func parseBaseCname(domain string) string {
-	hostParts := strings.Split(domain, ".")
-	if len(hostParts) < 2 {
-		return domain
+func parseBaseCname(domain string) (result []string) {
+	parts := strings.Split(domain, ".")
+	size := len(parts)
+	if size == 0 {
+		return []string{}
+	}
+	domain = parts[size-1]
+	result = append(result, domain)
+	for i := len(parts) - 2; i >= 0; i-- {
+		domain = parts[i] + "." + domain
+		result = append(result, domain)
 	}
-	baseCname := hostParts[len(hostParts)-2] + "." + hostParts[len(hostParts)-1]
-	return baseCname
+	return result
 }