|
| 1 | +package cmd |
| 2 | + |
| 3 | +import ( |
| 4 | + "bytes" |
| 5 | + "crypto/rand" |
| 6 | + "fmt" |
| 7 | + "os" |
| 8 | + |
| 9 | + cdx "github.com/CycloneDX/cyclonedx-go" |
| 10 | + "github.com/spf13/cobra" |
| 11 | +) |
| 12 | + |
| 13 | +var ( |
| 14 | + MergeStructure string |
| 15 | + MergeGroup string |
| 16 | + MergeName string |
| 17 | + MergeVersion string |
| 18 | + MergeRootComponentMode string |
| 19 | + MergeInputFiles []string |
| 20 | + MergePurl string |
| 21 | + MergeOutfile string |
| 22 | +) |
| 23 | + |
| 24 | +var mergeBomsCmd = &cobra.Command{ |
| 25 | + Use: "merge-boms", |
| 26 | + Short: "Merge multiple CycloneDX BOMs into a single BOM", |
| 27 | + Run: func(cmd *cobra.Command, args []string) { |
| 28 | + // 1. Read all BOM objects from input files |
| 29 | + var boms []*cdx.BOM |
| 30 | + for _, path := range MergeInputFiles { |
| 31 | + data, err := os.ReadFile(path) |
| 32 | + if err != nil { |
| 33 | + fmt.Printf("Error reading file %s: %v\n", path, err) |
| 34 | + os.Exit(1) |
| 35 | + } |
| 36 | + bom := readBomFromBytes(data) |
| 37 | + boms = append(boms, bom) |
| 38 | + } |
| 39 | + |
| 40 | + // 2. Extract roots, components, dependencies |
| 41 | + var roots []*cdx.Component |
| 42 | + var allComponents []*cdx.Component |
| 43 | + var allDependencies []*cdx.Dependency |
| 44 | + |
| 45 | + for _, bom := range boms { |
| 46 | + if bom.Metadata != nil && bom.Metadata.Component != nil { |
| 47 | + roots = append(roots, bom.Metadata.Component) |
| 48 | + } |
| 49 | + if bom.Components != nil { |
| 50 | + for _, comp := range *bom.Components { |
| 51 | + allComponents = append(allComponents, &comp) |
| 52 | + } |
| 53 | + } |
| 54 | + if bom.Dependencies != nil { |
| 55 | + for _, dep := range *bom.Dependencies { |
| 56 | + allDependencies = append(allDependencies, &dep) |
| 57 | + } |
| 58 | + } |
| 59 | + } |
| 60 | + |
| 61 | + // Prepare deduplication map for FLAT mode |
| 62 | + componentMap := make(map[string]*cdx.Component) // key: bom-ref |
| 63 | + for _, comp := range allComponents { |
| 64 | + if comp.BOMRef != "" { |
| 65 | + componentMap[comp.BOMRef] = comp |
| 66 | + } |
| 67 | + } |
| 68 | + |
| 69 | + // 3. Merge logic |
| 70 | + // 3.1 Metadata: create new BOM object, set root from flags |
| 71 | + mergedBOM := &cdx.BOM{ |
| 72 | + BOMFormat: "CycloneDX", |
| 73 | + SpecVersion: cdx.SpecVersion1_6, |
| 74 | + SerialNumber: generateSerialNumber(), |
| 75 | + Version: 1, |
| 76 | + Metadata: &cdx.Metadata{ |
| 77 | + Component: &cdx.Component{ |
| 78 | + Type: cdx.ComponentTypeApplication, |
| 79 | + Name: MergeName, |
| 80 | + Group: MergeGroup, |
| 81 | + Version: MergeVersion, |
| 82 | + }, |
| 83 | + }, |
| 84 | + } |
| 85 | + |
| 86 | + // Set merged root component BOMRef and PackageURL |
| 87 | + mergedRootRef := setMergedRootComponent( |
| 88 | + mergedBOM.Metadata.Component, |
| 89 | + MergeGroup, MergeName, MergeVersion, MergePurl, |
| 90 | + ) |
| 91 | + |
| 92 | + // 3.2 Components: FLAT or HIERARCHICAL |
| 93 | + if MergeStructure == "FLAT" { |
| 94 | + flatComponents := mergeFlatComponents(componentMap) |
| 95 | + mergedBOM.Components = &flatComponents |
| 96 | + } else { |
| 97 | + hierComponents := mergeHierarchicalComponents(roots, boms) |
| 98 | + mergedBOM.Components = &hierComponents |
| 99 | + } |
| 100 | + |
| 101 | + // 3.3 Dependencies: merge according to root component merge mode |
| 102 | + var mergedDependencies []cdx.Dependency |
| 103 | + switch MergeRootComponentMode { |
| 104 | + case "PRESERVE_UNDER_NEW_ROOT": |
| 105 | + mergedDependencies = mergeDependenciesPreserve(roots, allDependencies, mergedRootRef) |
| 106 | + case "FLATTEN_UNDER_NEW_ROOT": |
| 107 | + mergedDependencies = mergeDependenciesFlatten(roots, allDependencies, mergedRootRef) |
| 108 | + default: |
| 109 | + fmt.Printf("Unknown MergeRootComponentMode: %s\n", MergeRootComponentMode) |
| 110 | + os.Exit(1) |
| 111 | + } |
| 112 | + if len(mergedDependencies) > 0 { |
| 113 | + mergedBOM.Dependencies = &mergedDependencies |
| 114 | + } |
| 115 | + |
| 116 | + // 4. Output |
| 117 | + buf := new(bytes.Buffer) |
| 118 | + if err := cdx.NewBOMEncoder(buf, cdx.BOMFileFormatJSON).Encode(mergedBOM); err != nil { |
| 119 | + fmt.Printf("Error encoding merged BOM: %v\n", err) |
| 120 | + os.Exit(1) |
| 121 | + } |
| 122 | + if MergeOutfile == "" || MergeOutfile == "-" { |
| 123 | + _, err := os.Stdout.Write(buf.Bytes()) |
| 124 | + if err != nil { |
| 125 | + fmt.Printf("Error writing to stdout: %v\n", err) |
| 126 | + os.Exit(1) |
| 127 | + } |
| 128 | + } else { |
| 129 | + if err := os.WriteFile(MergeOutfile, buf.Bytes(), 0644); err != nil { |
| 130 | + fmt.Printf("Error writing to file %s: %v\n", MergeOutfile, err) |
| 131 | + os.Exit(1) |
| 132 | + } |
| 133 | + } |
| 134 | + |
| 135 | + }, |
| 136 | +} |
| 137 | + |
| 138 | +func init() { |
| 139 | + mergeBomsCmd.Flags().StringVar(&MergeStructure, "structure", "FLAT", "Structure of the merged BOM (FLAT, HIERARCHICAL)") |
| 140 | + mergeBomsCmd.Flags().StringVar(&MergeGroup, "group", "", "New root component group") |
| 141 | + mergeBomsCmd.Flags().StringVar(&MergeName, "name", "", "New root component name") |
| 142 | + mergeBomsCmd.Flags().StringVar(&MergeVersion, "version", "", "New root component version") |
| 143 | + mergeBomsCmd.Flags().StringVar(&MergeRootComponentMode, "root-component-merge-mode", "PRESERVE_UNDER_NEW_ROOT", "Root component merge mode (PRESERVE_UNDER_NEW_ROOT, FLATTEN_UNDER_NEW_ROOT)") |
| 144 | + mergeBomsCmd.Flags().StringSliceVar(&MergeInputFiles, "input-files", nil, "Input file paths for BOMs to merge") |
| 145 | + mergeBomsCmd.Flags().StringVar(&MergePurl, "purl", "", "Set bom-ref and purl for the root merged component") |
| 146 | + mergeBomsCmd.Flags().StringVar(&MergeOutfile, "outfile", "", "Output file path to write merged BOM (default: stdout)") |
| 147 | + |
| 148 | + bomUtils.AddCommand(mergeBomsCmd) |
| 149 | +} |
| 150 | + |
| 151 | +// generateSerialNumber creates a UUID-style serial number for the merged BOM |
| 152 | +func generateSerialNumber() string { |
| 153 | + b := make([]byte, 16) |
| 154 | + _, err := rand.Read(b) |
| 155 | + if err != nil { |
| 156 | + // Fallback to a simple timestamp-based serial number if random generation fails |
| 157 | + return fmt.Sprintf("urn:uuid:merged-bom-%d", len(b)) |
| 158 | + } |
| 159 | + // Format as UUID v4 |
| 160 | + b[6] = (b[6] & 0x0f) | 0x40 // Version 4 |
| 161 | + b[8] = (b[8] & 0x3f) | 0x80 // Variant bits |
| 162 | + return fmt.Sprintf("urn:uuid:%x-%x-%x-%x-%x", b[0:4], b[4:6], b[6:8], b[8:10], b[10:16]) |
| 163 | +} |
0 commit comments