jh163888/nali
1
0
Fork 0
mirror of https://github.com/zu1k/nali.git synced 2025-03-11 19:02:03 +08:00
Code Issues Projects Releases Wiki Activity

add trim for result

Browse Source
This commit is contained in:
zu1k 2020-07-21 17:54:57 +08:00
parent e69316b95d
commit ace8309f64
2 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cmd/root.go
View File

@ -66,7 +66,7 @@ Find document on: https://github.com/zu1k/nali
if line == "quit" || line == "exit" { if line == "quit" || line == "exit" {
return return
} }
fmt.Println(app.ReplaceIPInString(app.ReplaceCDNInString(line))) fmt.Printf("%s\n", app.ReplaceIPInString(app.ReplaceCDNInString(line)))
} }
} else { } else {
app.ParseIPs(args) app.ParseIPs(args)

11
internal/tools/replace.go
View File

@ -1,12 +1,15 @@
package tools package tools
import "strings" import (
"strings"
)
func ReplaceAdd(origin string, old string, new string) string { func ReplaceAdd(origin string, old string, new string) (result string) {
subLen := len(new) - len(old) subLen := len(new) - len(old)
wanted := old + strings.Repeat(" ", subLen) wanted := old + strings.Repeat(" ", subLen)
if strings.Contains(origin, wanted) { if strings.Contains(origin, wanted) {
return strings.ReplaceAll(origin, wanted, new) result = strings.ReplaceAll(origin, wanted, new)
} }
return strings.ReplaceAll(origin, old, new) result = strings.ReplaceAll(origin, old, new)
return strings.TrimRight(result, " \t")
} }