nvd api

Signed-off-by: Prabhu Subramanian <[email protected]>
AppThreat · Jan 22, 2024 · b669b52 · b669b52
1 parent 3984d5e
commit b669b52
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/utils/utils.go b/utils/utils.go
@@ -19,6 +19,8 @@ import (
 	pb "gopkg.in/cheggaaa/pb.v1"
 )
 
+var vulnListDir = filepath.Join(CacheDir(), "vuln-list")
+
 func CacheDir() string {
 	cacheDir, err := os.UserCacheDir()
 	if err != nil {
@@ -28,8 +30,12 @@ func CacheDir() string {
 	return dir
 }
 
+func SetVulnListDir(dir string) {
+	vulnListDir = dir
+}
+
 func VulnListDir() string {
-	return filepath.Join(CacheDir(), "vuln-list")
+	return vulnListDir
 }
 
 func SaveCVEPerYear(dirPath string, cveID string, data interface{}) error {