Added unique slice sorting

This commit is contained in:
perp 2024-07-09 16:53:06 +01:00
parent 85d8146c58
commit 5e214456c2
2 changed files with 19 additions and 2 deletions

View File

@ -6,6 +6,7 @@ import (
"strings"
"git.supernets.org/perp/blink/v1/internal/file"
"git.supernets.org/perp/blink/v1/internal/util"
"git.supernets.org/perp/blink/v1/pkg/dns"
"git.supernets.org/perp/blink/v1/pkg/runner"
"github.com/fatih/color"
@ -64,14 +65,14 @@ var rootCmd = &cobra.Command{
if err != nil {
panic(err)
}
opts.Wordlist = wordlistLines
opts.Wordlist = util.Sort(wordlistLines)
// Read resolvers file
resolversLines, err := file.Read(resolvers)
if err != nil {
panic(err)
}
opts.Resolvers = resolversLines
opts.Resolvers = util.Sort(resolversLines)
// Warnings
if opts.UDP {
@ -199,6 +200,7 @@ func main() {
}
// Todo:
// Clear bar on finish
// Clean up main.go
// Fix warning logs
// Fix wildcard log

15
v1/internal/util/sort.go Normal file
View File

@ -0,0 +1,15 @@
package util
// Sort a slice by unique
// https://stackoverflow.com/a/66751055
func Sort(strSlice []string) []string {
allKeys := make(map[string]bool)
list := []string{}
for _, item := range strSlice {
if _, value := allKeys[item]; !value {
allKeys[item] = true
list = append(list, item)
}
}
return list
}