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