fix(panic): fix output panic

lc · Nov 2, 2023 · 6dae657 · 6dae657
1 parent c146c22
commit 6dae657
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/pkg/output/output.go b/pkg/output/output.go
@@ -22,12 +22,14 @@ func WriteURLs(writer io.Writer, results <-chan string, blacklistMap mapset.Set[
 		if err != nil {
 			continue
 		}
-		if blacklistMap.Contains(strings.ToLower(path.Ext(u.Path))) {
+		if path.Ext(u.Path) != "" && blacklistMap.Contains(strings.ToLower(path.Ext(u.Path))) {
 			continue
 		}
-		if RemoveParameters && !lastURL.Add(u.Host+u.Path) {
+
+		if RemoveParameters && !lastURL.Contains(u.Host+u.Path) {
 			continue
 		}
+		lastURL.Add(u.Host + u.Path)
 
 		buf.B = append(buf.B, []byte(result)...)
 		buf.B = append(buf.B, "\n"...)