mirror of https://github.com/daffainfo/nuclei.git
Excluding non yaml file from new additions loading
parent
7ff2335935
commit
11df6da209
|
@ -38,6 +38,7 @@ import (
|
|||
"github.com/projectdiscovery/nuclei/v2/pkg/utils"
|
||||
"github.com/projectdiscovery/nuclei/v2/pkg/utils/stats"
|
||||
yamlwrapper "github.com/projectdiscovery/nuclei/v2/pkg/utils/yaml"
|
||||
"github.com/projectdiscovery/stringsutil"
|
||||
)
|
||||
|
||||
// Runner is a client for running the enumeration process.
|
||||
|
@ -452,8 +453,10 @@ func (r *Runner) readNewTemplatesFile() ([]string, error) {
|
|||
if text == "" {
|
||||
continue
|
||||
}
|
||||
if isNewTemplate(text) {
|
||||
templatesList = append(templatesList, text)
|
||||
}
|
||||
}
|
||||
return templatesList, nil
|
||||
}
|
||||
|
||||
|
@ -476,11 +479,19 @@ func (r *Runner) countNewTemplates() int {
|
|||
if text == "" {
|
||||
continue
|
||||
}
|
||||
|
||||
if isNewTemplate(text) {
|
||||
count++
|
||||
}
|
||||
|
||||
}
|
||||
return count
|
||||
}
|
||||
|
||||
func isNewTemplate(filename string) bool {
|
||||
return stringsutil.EqualFoldAny(filepath.Ext(filename), templates.TemplateExtension)
|
||||
}
|
||||
|
||||
// SaveResumeConfig to file
|
||||
func (r *Runner) SaveResumeConfig() error {
|
||||
resumeCfg := types.NewResumeCfg()
|
||||
|
|
|
@ -21,6 +21,11 @@ import (
|
|||
"gopkg.in/yaml.v2"
|
||||
)
|
||||
|
||||
const (
|
||||
// TemplateExtension defines the template default file extension
|
||||
TemplateExtension = ".yaml"
|
||||
)
|
||||
|
||||
// Template is a YAML input file which defines all the requests and
|
||||
// other metadata for a template.
|
||||
type Template struct {
|
||||
|
|
Loading…
Reference in New Issue