Graph Construction : kcl mod graph to show the dependency graph of KCL package

pkg/client/client.go

@@ -9,6 +9,7 @@ import (
 	"reflect"
 	"strings"
 
+	"github.com/BurntSushi/toml"
 	"github.com/dominikbraun/graph"
 	ocispec "github.com/opencontainers/image-spec/specs-go/v1"
 	"github.com/otiai10/copy"
@@ -852,6 +853,42 @@ func (c *KpmClient) DownloadFromGit(dep *pkg.Git, localPath string) (string, err
 	return localPath, err
 }
 
+func (c *KpmClient) ParseKclModFile(kclPkg *pkg.KclPkg) (map[string][]string, error) {
+	// Get path to kcl.mod file
+	modFilePath := kclPkg.ModFile.GetModFilePath()
+
+	// Read the content of the kcl.mod file
+	modFileBytes, err := os.ReadFile(modFilePath)
+	if err != nil {
+		return nil, err
+	}
+
+	// Normalize line endings for Windows systems
+	modFileContent := strings.ReplaceAll(string(modFileBytes), "\r\n", "\n")
+
+	// Parse the TOML content
+	var modFileData map[string]interface{}
+	if err := toml.Unmarshal([]byte(modFileContent), &modFileData); err != nil {
+		return nil, err
+	}
+
+	// Extract dependency information
+	dependencies := make(map[string][]string)
+	if deps, ok := modFileData["dependencies"].(map[string]interface{}); ok {
+		for dep, versions := range deps {
+			if v, ok := versions.([]interface{}); ok {
+				for _, ver := range v {
+					if version, ok := ver.(string); ok {
+						dependencies[dep] = append(dependencies[dep], version)
+					}
+				}
+			}
+		}
+	}
+
+	return dependencies, nil
+}
+
 // DownloadFromOci will download the dependency from the oci repository.
 func (c *KpmClient) DownloadFromOci(dep *pkg.Oci, localPath string) (string, error) {
 	ociClient, err := oci.NewOciClient(dep.Reg, dep.Repo, &c.settings)

pkg/client/dependency_graph.go

@@ -0,0 +1,57 @@
+package client
+
+import pkg "kcl-lang.io/kpm/pkg/package"
+
+// Construct dependency graph
+type DependencyGraph map[string][]string
+
+// Function to construct dependency graph by parsing kcl.mod file
+func ConstructDependencyGraphFromModFile(kpmClient *KpmClient, kclPkg *pkg.KclPkg) (DependencyGraph, error) {
+	dependencies, err := kpmClient.ParseKclModFile(kclPkg)
+	if err != nil {
+		return nil, err
+	}
+	return ConstructDependencyGraph(dependencies), nil
+}
+
+// Function to construct dependency graph from dependency map
+func ConstructDependencyGraph(dependencies map[string][]string) DependencyGraph {
+	graph := make(DependencyGraph)
+	for dependency, versions := range dependencies {
+		graph[dependency] = versions
+	}
+	return graph
+}
+
+// Traverse dependency graph using depth-first search (DFS)
+func DFS(graph DependencyGraph, dependency string, visited map[string]bool, result []string) []string {
+	// Mark current dependency as visited
+	visited[dependency] = true
+
+	// Add current dependency to result
+	result = append(result, dependency)
+
+	// Recursively traverse dependencies
+	for _, dep := range graph[dependency] {
+		if !visited[dep] {
+			result = DFS(graph, dep, visited, result)
+		}
+	}
+
+	return result
+}
+
+// Output dependencies in the same format as go mod graph
+func OutputDependencies(graph DependencyGraph) []string {
+	result := make([]string, 0)
+	visited := make(map[string]bool)
+
+	// Traverse each dependency using DFS
+	for dependency := range graph {
+		if !visited[dependency] {
+			result = DFS(graph, dependency, visited, result)
+		}
+	}
+
+	return result
+}