admin/spray
1
0
Fork 0
mirror of https://github.com/chainreactors/spray.git synced 2025-06-21 18:30:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

-F新增从stdin中读, 并添加颜色参数

Browse Source
This commit is contained in:
M09Ic 2023-02-08 19:17:42 +08:00
parent 5ace37824a
commit 25188b24e8
2 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cmd/cmd.go
View File

@ -50,7 +50,7 @@ func Spray() {
} }
if option.Format != "" { if option.Format != "" {
internal.Format(option.Format) internal.Format(option.Format, !option.NoColor)
os.Exit(0) os.Exit(0)
} }

18
internal/format.go
View File

@ -6,10 +6,18 @@ import (
"github.com/chainreactors/logs" "github.com/chainreactors/logs"
"github.com/chainreactors/spray/pkg" "github.com/chainreactors/spray/pkg"
"io/ioutil" "io/ioutil"
"os"
) )
func Format(filename string) { func Format(filename string, color bool) {
content, err := ioutil.ReadFile(filename) var content []byte
var err error
if filename == "stdin" {
content, err = ioutil.ReadAll(os.Stdin)
} else {
content, err = ioutil.ReadFile(filename)
}
if err != nil { if err != nil {
return return
} }
@ -24,6 +32,10 @@ func Format(filename string) {
results = append(results, &result) results = append(results, &result)
} }
for _, result := range results { for _, result := range results {
logs.Log.Info(result.String()) if color {
logs.Log.Info(result.ColorString())
} else {
logs.Log.Info(result.String())
}
} }
} }