update template url (#4110)

projectdiscovery · Sep 1, 2023 · 680fc55 · 680fc55
1 parent 388c1e8
commit 680fc55
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/v2/pkg/output/output.go b/v2/pkg/output/output.go
@@ -215,7 +215,7 @@ func NewStandardWriter(options *types.Options) (*StandardWriter, error) {
 func (w *StandardWriter) Write(event *ResultEvent) error {
 	// Enrich the result event with extra metadata on the template-path and url.
 	if event.TemplatePath != "" {
-		event.Template, event.TemplateURL = utils.TemplatePathURL(types.ToString(event.TemplatePath))
+		event.Template, event.TemplateURL = utils.TemplatePathURL(types.ToString(event.TemplatePath), types.ToString(event.TemplateID))
 	}
 	event.Timestamp = time.Now()
 
@@ -327,7 +327,7 @@ func (w *StandardWriter) WriteFailure(wrappedEvent *InternalWrappedEvent) error
 	// if no results were found, manually create a failure event
 	event := wrappedEvent.InternalEvent
 
-	templatePath, templateURL := utils.TemplatePathURL(types.ToString(event["template-path"]))
+	templatePath, templateURL := utils.TemplatePathURL(types.ToString(event["template-path"]), types.ToString(event["template-id"]))
 	var templateInfo model.Info
 	if event["template-info"] != nil {
 		templateInfo = event["template-info"].(model.Info)

diff --git a/v2/pkg/testutils/testutils.go b/v2/pkg/testutils/testutils.go
@@ -141,7 +141,7 @@ func (m *MockOutputWriter) WriteFailure(wrappedEvent *output.InternalWrappedEven
 	if m.WriteCallback != nil {
 		// create event
 		event := wrappedEvent.InternalEvent
-		templatePath, templateURL := utils.TemplatePathURL(types.ToString(event["template-path"]))
+		templatePath, templateURL := utils.TemplatePathURL(types.ToString(event["template-path"]), types.ToString(event["template-id"]))
 		var templateInfo model.Info
 		if ti, ok := event["template-info"].(model.Info); ok {
 			templateInfo = ti

diff --git a/v2/pkg/utils/template_path.go b/v2/pkg/utils/template_path.go
@@ -8,11 +8,11 @@ import (
 
 const (
 	// TemplatesRepoURL is the URL for files in nuclei-templates repository
-	TemplatesRepoURL = "https://github.com/projectdiscovery/nuclei-templates/blob/main/"
+	TemplatesRepoURL = "https://templates.nuclei.sh/public/"
 )
 
 // TemplatePathURL returns the Path and URL for the provided template
-func TemplatePathURL(fullPath string) (string, string) {
+func TemplatePathURL(fullPath, templateId string) (string, string) {
 	var templateDirectory string
 	configData := config.DefaultConfig
 	if configData.TemplatesDirectory != "" && strings.HasPrefix(fullPath, configData.TemplatesDirectory) {
@@ -22,6 +22,6 @@ func TemplatePathURL(fullPath string) (string, string) {
 	}
 
 	finalPath := strings.TrimPrefix(strings.TrimPrefix(fullPath, templateDirectory), "/")
-	templateURL := TemplatesRepoURL + finalPath
+	templateURL := TemplatesRepoURL + templateId
 	return finalPath, templateURL
 }