Excluding non yaml file from new additions loading

dev
mzack 2022-01-14 11:20:19 +01:00
parent 7ff2335935
commit 11df6da209
2 changed files with 18 additions and 2 deletions

View File

@ -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,7 +453,9 @@ func (r *Runner) readNewTemplatesFile() ([]string, error) {
if text == "" {
continue
}
templatesList = append(templatesList, text)
if isNewTemplate(text) {
templatesList = append(templatesList, text)
}
}
return templatesList, nil
}
@ -476,11 +479,19 @@ func (r *Runner) countNewTemplates() int {
if text == "" {
continue
}
count++
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()

View File

@ -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 {