remove scheme (#1166)

* remove scheme * use replacer --------- Co-authored-by: Sandeep Singh <sandeep@projectdiscovery.io>
projectdiscovery · Mar 26, 2024 · e5ee088 · e5ee088
1 parent e8cb3d9
commit e5ee088
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/v2/pkg/runner/enumerate.go b/v2/pkg/runner/enumerate.go
@@ -18,6 +18,12 @@ import (
 
 const maxNumCount = 2
 
+var replacer = strings.NewReplacer(
+	"*.", "",
+	"http://", "",
+	"https://", "",
+)
+
 // EnumerateSingleDomain wraps EnumerateSingleDomainWithCtx with an empty context
 func (r *Runner) EnumerateSingleDomain(domain string, writers []io.Writer) error {
 	return r.EnumerateSingleDomainWithCtx(context.Background(), domain, writers)
@@ -62,7 +68,7 @@ func (r *Runner) EnumerateSingleDomainWithCtx(ctx context.Context, domain string
 					skippedCounts[result.Source]++
 					continue
 				}
-				subdomain := strings.ReplaceAll(strings.ToLower(result.Value), "*.", "")
+				subdomain := replacer.Replace(result.Value)
 
 				if matchSubdomain := r.filterAndMatchSubdomain(subdomain); matchSubdomain {
 					if _, ok := uniqueMap[subdomain]; !ok {