forked from projectdiscovery/katana
-
Notifications
You must be signed in to change notification settings - Fork 0
/
format_screen.go
49 lines (42 loc) · 1.15 KB
/
format_screen.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
package output
import (
"bytes"
"fmt"
)
// formatScreen formats the output for showing on screen.
func (w *StandardWriter) formatScreen(output *Result) ([]byte, error) {
builder := &bytes.Buffer{}
if w.fields != "" {
result := formatField(output, w.fields)
for _, fop := range result {
if w.verbose {
builder.WriteRune('[')
builder.WriteString(w.aurora.Blue(fop.field).String())
builder.WriteRune(']')
builder.WriteRune(' ')
}
builder.WriteString(fmt.Sprintf("%s\n", fop.value))
}
return builder.Bytes(), nil
}
if w.verbose && output.Request.Tag != "" {
builder.WriteRune('[')
builder.WriteString(w.aurora.Blue(output.Request.Tag).String())
builder.WriteRune(']')
builder.WriteRune(' ')
}
if output.Request.Method != "" && w.verbose {
builder.WriteRune('[')
builder.WriteString(w.aurora.Green(output.Request.Method).String())
builder.WriteRune(']')
builder.WriteRune(' ')
}
builder.WriteString(output.Request.URL)
if output.Request.Body != "" && w.verbose {
builder.WriteRune(' ')
builder.WriteRune('[')
builder.WriteString(output.Request.Body)
builder.WriteRune(']')
}
return builder.Bytes(), nil
}