Merge pull request #101 from kellyma2/lockfile

Add lockfile command to bazeldnf

Author: manuelnaranjo

cmd/BUILD.bazel

@@ -1,14 +1,16 @@
-load("@rules_go//go:def.bzl", "go_binary", "go_library")
+load("@rules_go//go:def.bzl", "go_binary", "go_library", "go_test")
 load("//bazeldnf:toolchain.bzl", "bazeldnf_toolchain")
 
 go_library(
     name = "cmd_lib",
     srcs = [
         "bazeldnf.go",
+        "config_helper.go",
         "fetch.go",
         "filter.go",
         "init.go",
         "ldd.go",
+        "lockfile.go",
         "prune.go",
         "reduce.go",
         "resolve.go",
@@ -40,6 +42,7 @@ go_library(
         "@com_github_sirupsen_logrus//:logrus",
         "@com_github_spf13_cobra//:cobra",
         "@org_golang_x_crypto//openpgp",
+        "@org_golang_x_exp//maps",
     ],
 )
 
@@ -60,3 +63,14 @@ toolchain(
     toolchain = ":host-toolchain",
     toolchain_type = "//bazeldnf:toolchain_type",
 )
+
+go_test(
+    name = "cmd_test",
+    srcs = ["config_helper_test.go"],
+    embed = [":cmd_lib"],
+    deps = [
+        "//pkg/api",
+        "//pkg/api/bazeldnf",
+        "@com_github_onsi_gomega//:gomega",
+    ],
+)

cmd/config_helper.go

@@ -0,0 +1,126 @@
+package main
+
+import (
+	"cmp"
+	"fmt"
+	"slices"
+
+	"github.com/rmohr/bazeldnf/pkg/api"
+	"github.com/rmohr/bazeldnf/pkg/api/bazeldnf"
+	"github.com/sirupsen/logrus"
+	"golang.org/x/exp/maps"
+)
+
+func keys[K cmp.Ordered, V any](m map[K]V) []K {
+	keys := maps.Keys(m)
+	slices.Sort(keys)
+	return keys
+}
+
+func toConfig(install, forceIgnored []*api.Package, targets []string, cmdline []string) (*bazeldnf.Config, error) {
+	ignored := make(map[string]bool)
+	for _, forceIgnoredPackage := range forceIgnored {
+		ignored[forceIgnoredPackage.Name] = true
+	}
+
+	allPackages := make(map[string]*bazeldnf.RPM)
+	repositories := make(map[string][]string)
+	for _, installPackage := range install {
+		repositories[installPackage.Repository.Name] = installPackage.Repository.Mirrors
+
+		deps := make([]string, 0, len(installPackage.Format.Requires.Entries))
+		for _, entry := range installPackage.Format.Requires.Entries {
+			deps = append(deps, entry.Name)
+		}
+
+		slices.Sort(deps)
+
+		allPackages[installPackage.Name] = &bazeldnf.RPM{
+			Name:         installPackage.Name,
+			SHA256:       installPackage.Checksum.Text,
+			URLs:         []string{installPackage.Location.Href},
+			Repository:   installPackage.Repository.Name,
+			Dependencies: deps,
+		}
+	}
+
+	providers := collectProviders(forceIgnored, install)
+	packageNames := keys(allPackages)
+	sortedPackages := make([]*bazeldnf.RPM, 0, len(packageNames))
+	for _, name := range packageNames {
+		pkg := allPackages[name]
+		deps, err := collectDependencies(name, pkg.Dependencies, providers, ignored)
+		if err != nil {
+			return nil, err
+		}
+
+		pkg.SetDependencies(deps)
+
+		sortedPackages = append(sortedPackages, pkg)
+	}
+
+	lockFile := bazeldnf.Config{
+		CommandLineArguments: cmdline,
+		ForceIgnored:         keys(ignored),
+		RPMs:                 sortedPackages,
+		Repositories:         repositories,
+		Targets:              targets,
+	}
+
+	return &lockFile, nil
+}
+
+func collectProviders(pkgSets ...[]*api.Package) map[string]string {
+	providers := map[string]string{}
+	for _, pkgSet := range pkgSets {
+		for _, pkg := range pkgSet {
+			for _, entry := range pkg.Format.Provides.Entries {
+				providers[entry.Name] = pkg.Name
+			}
+
+			for _, entry := range pkg.Format.Files {
+				providers[entry.Text] = pkg.Name
+			}
+		}
+	}
+
+	return providers
+}
+
+func collectDependencies(pkg string, requires []string, providers map[string]string, ignored map[string]bool) ([]string, error) {
+	depSet := make(map[string]bool)
+	for _, req := range requires {
+		if ignored[req] {
+			logrus.Debugf("Ignoring dependency %s", req)
+			continue
+		}
+		logrus.Debugf("Resolving dependency %s", req)
+		provider, ok := providers[req]
+		if !ok {
+			return nil, fmt.Errorf("could not find provider for %s", req)
+		}
+		logrus.Debugf("Found provider %s for %s", provider, req)
+		if ignored[provider] {
+			logrus.Debugf("Ignoring provider %s for %s", provider, req)
+			continue
+		}
+		depSet[provider] = true
+	}
+
+	deps := keys(depSet)
+
+	found := map[string]bool{pkg: true}
+
+	// RPMs may have circular dependencies, even depend on themselves.
+	// we need to ignore such dependencies
+	nonCyclicDeps := make([]string, 0, len(deps))
+	for _, dep := range deps {
+		if found[dep] {
+			continue
+		}
+
+		nonCyclicDeps = append(nonCyclicDeps, dep)
+	}
+
+	return nonCyclicDeps, nil
+}