Merge pull request #1488 from projectdiscovery/issue-1459-new-templates-loader

Excluding non yaml file from new additions loading

v2/internal/runner/runner.go

@@ -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()

v2/pkg/templates/templates.go

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