External/nuclei
1
0
Fork 0
mirror of https://github.com/projectdiscovery/nuclei.git synced 2025-12-17 23:25:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

add callback support to StandardWriter (#4839)

Browse Source 
* add callback support to StandardWriter

* minor

* use multiwriter

* minor
This commit is contained in:
Dogan Can Bakir 2024-03-11 14:37:07 +03:00 committed by GitHub
parent fe4d5f05a1
commit e9625d621f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 23 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
lib/sdk_private.go
View File

@ -34,7 +34,6 @@ import (
// applyRequiredDefaults to options // applyRequiredDefaults to options
func (e *NucleiEngine) applyRequiredDefaults() { func (e *NucleiEngine) applyRequiredDefaults() {
if e.customWriter == nil {
mockoutput := testutils.NewMockOutputWriter(e.opts.OmitTemplate) mockoutput := testutils.NewMockOutputWriter(e.opts.OmitTemplate)
mockoutput.WriteCallback = func(event *output.ResultEvent) { mockoutput.WriteCallback = func(event *output.ResultEvent) {
if len(e.resultCallbacks) > 0 { if len(e.resultCallbacks) > 0 {
@ -57,8 +56,13 @@ func (e *NucleiEngine) applyRequiredDefaults() {
if e.onFailureCallback != nil { if e.onFailureCallback != nil {
mockoutput.FailureCallback = e.onFailureCallback mockoutput.FailureCallback = e.onFailureCallback
} }
if e.customWriter != nil {
e.customWriter = output.NewMultiWriter(e.customWriter, mockoutput)
} else {
e.customWriter = mockoutput e.customWriter = mockoutput
} }
if e.customProgress == nil { if e.customProgress == nil {
e.customProgress = &testutils.MockProgressClient{} e.customProgress = &testutils.MockProgressClient{}
} }