Skip to content

Commit

Permalink
Merge pull request blackducksoftware#68 from christianprando/ud-7238-…
Browse files Browse the repository at this point in the history
…rapid-scan-initiate-request

UD-7239: Implements request to upload bdio data to rapid scans endpoint.
  • Loading branch information
tarunsethi authored Apr 13, 2021
2 parents 7dbd369 + e2294fd commit 1b51607
Showing 1 changed file with 31 additions and 1 deletion.
32 changes: 31 additions & 1 deletion hubclient/rapid-scans-client.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,10 +17,15 @@ package hubclient
import (
"github.com/blackducksoftware/hub-client-go/hubapi"
log "github.com/sirupsen/logrus"
"strconv"
)

const (
apiDeveloperScans = "/api/developer-scans"
apiDeveloperScans = "/api/developer-scans"
headerBdMode = "X-BD-MODE"
headerBdDocumentCount = "X-BD-DOCUMENT-COUNT"
bdModeAppend = "append"
bdModeFinish = "finish"
)

func (c *Client) StartRapidScan(bdioHeaderContent string) (error, string) {
Expand All @@ -34,3 +39,28 @@ func (c *Client) StartRapidScan(bdioHeaderContent string) (error, string) {

return nil, bdioUploadEndpoint
}

func (c *Client) UploadBdioFiles(bdioUploadEndpoint string, bdioContents []string) error {
c.AddHeaderValue(headerBdMode, bdModeAppend)
c.AddHeaderValue(headerBdDocumentCount, strconv.Itoa(len(bdioContents)))

for _, bdioContent := range bdioContents {
err := c.HttpPutJSON(bdioUploadEndpoint, bdioContent, hubapi.ContentTypeRapidScan, 202)
if err != nil {
log.Errorf("Error uploading bdio files.", err)
return err
}
}

c.SetHeaderValue(headerBdMode, bdModeFinish)
err := c.HttpPutJSON(bdioUploadEndpoint, "", hubapi.ContentTypeRapidScan, 202)
if err != nil {
log.Errorf("Error uploading bdio files.", err)
return err
}

c.DeleteHeaderValue(headerBdMode)
c.DeleteHeaderValue(headerBdDocumentCount)

return nil
}

0 comments on commit 1b51607

Please sign in to comment.