added remediation

christianprando · Dec 6, 2019 · 879a897 · 879a897
1 parent cb44460
commit 879a897
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 1 deletion.
diff --git a/hubapi/component-api.go b/hubapi/component-api.go
@@ -111,3 +111,17 @@ type ComponentRequest struct {
 	ApprovalStatus      string   `json:"approvalStatus"`
 	Type                string   `json:"type"`
 }
+
+type ComponentRemediation struct {
+	FixesPreviousVulnerabilities *RemediationInfo `json:"fixesPreviousVulnerabilities,omitempty"`
+	NoVulnerabilities            *RemediationInfo `json:"noVulnerabilities,omitempty"`
+	LatestAfterCurrent           *RemediationInfo `json:"latestAfterCurrent,omitempty"`
+	Meta                         Meta            `json:"_meta"`
+}
+
+type RemediationInfo struct {
+	Name               string    `json:"name"`
+	ComponentVersion   string    `json:"componentVersion"`
+	ReleasedOn         time.Time `json:"releasedOn"`
+	VulnerabilityCount int       `json:"vulnerabilityCount"`
+}
diff --git a/hubclient/component-client.go b/hubclient/component-client.go
@@ -19,7 +19,10 @@ import (
 	log "github.com/sirupsen/logrus"
 )
 
-const apiComponents = "/api/components"
+const (
+	apiComponents  = "/api/components"
+	remediatingApi = "/remediating"
+)
 
 func (c *Client) ListComponents(options *hubapi.GetListOptions) (*hubapi.ComponentList, error) {
 	componentURL := c.baseURL + apiComponents
@@ -96,3 +99,15 @@ func (c *Client) GetComponentVersion(link hubapi.ResourceLink) (*hubapi.Componen
 
 	return &componentVersion, nil
 }
+
+func (c *Client) GetComponentVersionRemediation(componentVersionHref string) (*hubapi.ComponentRemediation, error) {
+	var componentRemediation hubapi.ComponentRemediation
+
+	err := c.HttpGetJSON(componentVersionHref+remediatingApi, &componentRemediation, 200)
+
+	if err != nil {
+		return nil, AnnotateHubClientError(err, "Error trying to retrieve component remediation advice")
+	}
+
+	return &componentRemediation, nil
+}