github/kubescape
1
0
Fork 0
mirror of https://github.com/kubescape/kubescape.git synced 2026-04-15 06:58:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

each artifact handle print seprately

Browse Source
This commit is contained in:
yiscah
2022-01-10 09:07:06 +02:00
parent 59498361e7
commit 46e5aff5f9
1 changed files with 8 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
clihandler/clidownload.go
+8 -9
View File
@@ -2,11 +2,9 @@ package clihandler
import (
"fmt"
"path/filepath"
"strings"
"github.com/armosec/armoapi-go/armotypes"
"github.com/armosec/kubescape/cautils"
"github.com/armosec/kubescape/cautils/getter"
@@ -41,7 +39,6 @@ func downloadArtifact(downloadInfo *cautils.DownloadInfo, downloadArtifactFunc m
if err := f(downloadInfo); err != nil {
return err
}
fmt.Printf("'%s' downloaded successfully and saved at: '%s'\n", downloadInfo.Target, filepath.Join(downloadInfo.Path, downloadInfo.FileName))
return nil
}
return fmt.Errorf("unknown command to download")
@@ -91,6 +88,7 @@ func downloadConfigInputs(downloadInfo *cautils.DownloadInfo) error {
if err != nil {
return err
}
fmt.Printf("'%s' downloaded successfully and saved at: '%s'\n", downloadInfo.Target, filepath.Join(downloadInfo.Path, downloadInfo.FileName))
return nil
}
@@ -113,6 +111,7 @@ func downloadExceptions(downloadInfo *cautils.DownloadInfo) error {
if err != nil {
return err
}
fmt.Printf("'%s' downloaded successfully and saved at: '%s'\n", downloadInfo.Target, filepath.Join(downloadInfo.Path, downloadInfo.FileName))
return nil
}
@@ -123,8 +122,6 @@ func downloadFramework(downloadInfo *cautils.DownloadInfo) error {
if downloadInfo.Name == "" {
// if framework name not specified - download all frameworks
downloadInfo.Target = "frameworks"
downloadInfo.FileName = fmt.Sprintf("%s.json", "<framework.Name>")
frameworks, err := g.GetFrameworks()
if err != nil {
return err
@@ -134,6 +131,7 @@ func downloadFramework(downloadInfo *cautils.DownloadInfo) error {
if err != nil {
return err
}
fmt.Printf("'%s': '%s' downloaded successfully and saved at: '%s'\n", downloadInfo.Target, fw.Name, filepath.Join(downloadInfo.Path, (strings.ToLower(fw.Name)+".json")))
}
// return fmt.Errorf("missing framework name")
} else {
@@ -148,6 +146,7 @@ func downloadFramework(downloadInfo *cautils.DownloadInfo) error {
if err != nil {
return err
}
fmt.Printf("'%s' downloaded successfully and saved at: '%s'\n", downloadInfo.Target, filepath.Join(downloadInfo.Path, downloadInfo.FileName))
}
return nil
}
@@ -161,17 +160,17 @@ func downloadControl(downloadInfo *cautils.DownloadInfo) error {
// TODO - support
return fmt.Errorf("missing control name")
}
filename := downloadInfo.FileName
if filename == "" {
filename = fmt.Sprintf("%s.json", "controls-inputs")
if downloadInfo.FileName == "" {
downloadInfo.FileName = fmt.Sprintf("%s.json", downloadInfo.Name)
}
controls, err := g.GetControl(downloadInfo.Name)
if err != nil {
return err
}
err = getter.SaveInFile(controls, filepath.Join(downloadInfo.Path, filename))
err = getter.SaveInFile(controls, filepath.Join(downloadInfo.Path, downloadInfo.FileName))
if err != nil {
return err
}
fmt.Printf("'%s' downloaded successfully and saved at: '%s'\n", downloadInfo.Target, filepath.Join(downloadInfo.Path, downloadInfo.FileName))
return nil
}