Add FlattenGraph for GraphNode

xray/services/scan.go

@@ -3,6 +3,7 @@ package services
 import (
 	"encoding/json"
 	"github.com/jfrog/jfrog-client-go/utils/log"
+	"golang.org/x/exp/maps"
 	"net/http"
 	"strings"
 	"time"
@@ -189,6 +190,25 @@ type GraphNode struct {
 	Parent *GraphNode `json:"-"`
 }
 
+// FlattenGraph creates a map of dependencies from the given graph, and returns a flat graph of dependencies with one level.
+func FlattenGraph(graph []*GraphNode) []*GraphNode {
+	allDependencies := map[string]*GraphNode{}
+	for _, node := range graph {
+		populateUniqueDependencies(node, allDependencies)
+	}
+	return []*GraphNode{{Id: "root", Nodes: maps.Values(allDependencies)}}
+}
+
+func populateUniqueDependencies(node *GraphNode, allDependencies map[string]*GraphNode) {
+	if _, exist := allDependencies[node.Id]; exist {
+		return
+	}
+	allDependencies[node.Id] = &GraphNode{Id: node.Id}
+	for _, dependency := range node.Nodes {
+		populateUniqueDependencies(dependency, allDependencies)
+	}
+}
+
 type OtherComponentIds struct {
 	Id     string `json:"component_id,omitempty"`
 	Origin int    `json:"origin,omitempty"`

xray/services/scan_test.go

@@ -2,6 +2,8 @@ package services
 
 import (
 	"fmt"
+	"github.com/jfrog/gofrog/datastructures"
+	"github.com/stretchr/testify/assert"
 	"testing"
 )
 
@@ -47,3 +49,32 @@ func TestCreateScanGraphQueryParams(t *testing.T) {
 		})
 	}
 }
+
+func TestFlattenGraph(t *testing.T) {
+	nodeA := &GraphNode{Id: "A"}
+	nodeB := &GraphNode{Id: "B"}
+	nodeC := &GraphNode{Id: "C"}
+	nodeD := &GraphNode{Id: "D"}
+	nodeE := &GraphNode{Id: "E"}
+	nodeF := &GraphNode{Id: "F"}
+
+	// Set dependencies
+	nodeA.Nodes = []*GraphNode{nodeB, nodeC}
+	nodeB.Nodes = []*GraphNode{nodeC, nodeD}
+	nodeC.Nodes = []*GraphNode{nodeD}
+	nodeD.Nodes = []*GraphNode{nodeE, nodeF}
+	nodeF.Nodes = []*GraphNode{nodeA, nodeB, nodeC}
+
+	// Create graph
+	graph := []*GraphNode{nodeA, nodeB, nodeC}
+	flatGraph := FlattenGraph(graph)
+
+	// Check that the graph has been flattened correctly
+	assert.Equal(t, len(flatGraph[0].Nodes), 6)
+	set := datastructures.MakeSet[string]()
+	for _, node := range flatGraph[0].Nodes {
+		assert.Len(t, node.Nodes, 0)
+		assert.False(t, set.Exists(node.Id))
+		set.Add(node.Id)
+	}
+}