mirror of
https://github.com/projectdiscovery/nuclei.git
synced 2025-12-18 15:05:26 +00:00
be73fde0db
* use go-homedir instead of standard os.userhomedir * set r.templatesConfig before write attempt to avoid panic
44 lines
1.1 KiB
Go
44 lines
1.1 KiB
Go
package utils
|
|
|
|
import (
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/mitchellh/go-homedir"
|
|
"github.com/projectdiscovery/nuclei/v2/pkg/catalog/config"
|
|
)
|
|
|
|
const (
|
|
// TemplatesRepoURL is the URL for files in nuclei-templates repository
|
|
TemplatesRepoURL = "https://github.com/projectdiscovery/nuclei-templates/blob/master/"
|
|
)
|
|
|
|
var configData *config.Config
|
|
|
|
func init() {
|
|
configData, _ = config.ReadConfiguration()
|
|
}
|
|
|
|
// TemplatePathURL returns the Path and URL for the provided template
|
|
func TemplatePathURL(fullPath string) (string, string) {
|
|
var templateDirectory string
|
|
if configData != nil && configData.TemplatesDirectory != "" && strings.HasPrefix(fullPath, configData.TemplatesDirectory) {
|
|
templateDirectory = configData.TemplatesDirectory
|
|
} else {
|
|
return "", ""
|
|
}
|
|
|
|
finalPath := strings.TrimPrefix(strings.TrimPrefix(fullPath, templateDirectory), "/")
|
|
templateURL := TemplatesRepoURL + finalPath
|
|
return finalPath, templateURL
|
|
}
|
|
|
|
// GetDefaultTemplatePath on default settings
|
|
func GetDefaultTemplatePath() (string, error) {
|
|
home, err := homedir.Dir()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return filepath.Join(home, "nuclei-templates"), nil
|
|
}
|