remove IsCustomTemplate func

projectdiscovery · Nov 6, 2023 · 3ebfb7a · 3ebfb7a
1 parent cdb4689
commit 3ebfb7a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 8 deletions.
diff --git a/pkg/utils/template_path.go b/pkg/utils/template_path.go
@@ -13,11 +13,15 @@ const (
 
 // TemplatePathURL returns the Path and URL for the provided template
 func TemplatePathURL(fullPath, templateId string) (string, string) {
-	if IsCustomTemplate(fullPath) {
+	var templateDirectory string
+	configData := config.DefaultConfig
+	if configData.TemplatesDirectory != "" && strings.HasPrefix(fullPath, configData.TemplatesDirectory) {
+		templateDirectory = configData.TemplatesDirectory
+	} else {
 		return "", ""
 	}
 
-	relativePath := strings.TrimPrefix(strings.TrimPrefix(fullPath, config.DefaultConfig.TemplatesDirectory), "/")
+	finalPath := strings.TrimPrefix(strings.TrimPrefix(fullPath, templateDirectory), "/")
 	templateURL := TemplatesRepoURL + templateId
-	return relativePath, templateURL
+	return finalPath, templateURL
 }
diff --git a/pkg/utils/utils.go b/pkg/utils/utils.go
@@ -78,8 +78,3 @@ func StringSliceContains(slice []string, item string) bool {
 	}
 	return false
 }
-
-// IsCustomTemplate checks if the template is a custom template
-func IsCustomTemplate(templatePath string) bool {
-	return !strings.HasPrefix(templatePath, config.DefaultConfig.TemplatesDirectory)
-}