mirror of
https://github.com/SAP/jenkins-library.git
synced 2024-12-14 11:03:09 +02:00
81 lines
3.0 KiB
Go
81 lines
3.0 KiB
Go
package whitesource
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/SAP/jenkins-library/pkg/log"
|
|
"github.com/SAP/jenkins-library/pkg/piperutils"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// ReportOptions defines options for downloading reports after scanning.
|
|
type ReportOptions struct {
|
|
// ReportDirectory defines the target directory for downloading reports.
|
|
ReportDirectory string
|
|
// VulnerabilityReportFormat defines the requested file format of the vulnerability report (i.e. pdf).
|
|
VulnerabilityReportFormat string
|
|
}
|
|
|
|
type scanUtils interface {
|
|
MkdirAll(path string, perm os.FileMode) error
|
|
FileWrite(path string, content []byte, perm os.FileMode) error
|
|
}
|
|
|
|
// DownloadReports downloads a Project's risk and vulnerability reports
|
|
func (s *Scan) DownloadReports(options ReportOptions, utils scanUtils, sys whitesource) ([]piperutils.Path, error) {
|
|
if err := utils.MkdirAll(options.ReportDirectory, os.ModePerm); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
var paths []piperutils.Path
|
|
for _, project := range s.scannedProjects {
|
|
vulnPath, err := downloadVulnerabilityReport(options, project, utils, sys)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
riskPath, err := downloadRiskReport(options, project, utils, sys)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
paths = append(paths, *vulnPath, *riskPath)
|
|
}
|
|
return paths, nil
|
|
}
|
|
|
|
func downloadVulnerabilityReport(options ReportOptions, project Project, utils scanUtils, sys whitesource) (*piperutils.Path, error) {
|
|
reportBytes, err := sys.GetProjectVulnerabilityReport(project.Token, options.VulnerabilityReportFormat)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "unable to download vulnerability report from url")
|
|
}
|
|
|
|
rptFileName := fmt.Sprintf("%s-vulnerability-report.%s", strings.ReplaceAll(project.Name, "/", "_"), options.VulnerabilityReportFormat)
|
|
rptFileName = filepath.Join(options.ReportDirectory, rptFileName)
|
|
if err := utils.FileWrite(rptFileName, reportBytes, 0644); err != nil {
|
|
return nil, errors.Wrapf(err, "unable to copy content from url to file %v", rptFileName)
|
|
}
|
|
|
|
log.Entry().Infof("Successfully downloaded vulnerability report to %s", rptFileName)
|
|
pathName := fmt.Sprintf("%s Vulnerability Report", project.Name)
|
|
return &piperutils.Path{Name: pathName, Target: rptFileName}, nil
|
|
}
|
|
|
|
func downloadRiskReport(options ReportOptions, project Project, utils scanUtils, sys whitesource) (*piperutils.Path, error) {
|
|
reportBytes, err := sys.GetProjectRiskReport(project.Token)
|
|
if err != nil {
|
|
return nil, errors.Wrapf(err, "unable to download risk report from url")
|
|
}
|
|
|
|
rptFileName := fmt.Sprintf("%s-risk-report.pdf", strings.ReplaceAll(project.Name, "/", "_"))
|
|
rptFileName = filepath.Join(options.ReportDirectory, rptFileName)
|
|
if err := utils.FileWrite(rptFileName, reportBytes, 0644); err != nil {
|
|
return nil, errors.Wrapf(err, "unable to copy content from url to file %v", rptFileName)
|
|
}
|
|
|
|
log.Entry().Infof("Successfully downloaded risk report to %s", rptFileName)
|
|
pathName := fmt.Sprintf("%s PDF Risk Report", project.Name)
|
|
return &piperutils.Path{Name: pathName, Target: rptFileName}, nil
|
|
}
|