From bcc82442d55e731849846900c5aff03bae0247b5 Mon Sep 17 00:00:00 2001
From: Ramana Reddy <ramanaredy.manda@gmail.com>
Date: Tue, 5 Sep 2023 09:54:18 +0530
Subject: [PATCH] fix filtering active subdomains when using as lib

---
 v2/pkg/runner/enumerate.go | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/v2/pkg/runner/enumerate.go b/v2/pkg/runner/enumerate.go
index 5ec1e0f0d..8c38d0acf 100644
--- a/v2/pkg/runner/enumerate.go
+++ b/v2/pkg/runner/enumerate.go
@@ -150,8 +150,14 @@ func (r *Runner) EnumerateSingleDomainWithCtx(ctx context.Context, domain string
 	}
 
 	if r.options.ResultCallback != nil {
-		for _, v := range uniqueMap {
-			r.options.ResultCallback(&v)
+		if r.options.RemoveWildcard {
+			for host, result := range foundResults {
+				r.options.ResultCallback(&resolve.HostEntry{Domain: host, Host: result.Host, Source: result.Source})
+			}
+		} else {
+			for _, v := range uniqueMap {
+				r.options.ResultCallback(&v)
+			}
 		}
 	}
 	gologger.Info().Msgf("Found %d subdomains for %s in %s\n", numberOfSubDomains, domain, duration)