-
Notifications
You must be signed in to change notification settings - Fork 8
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Move code taken from sbt to its own file
Since it's licensed under the Apache 2.0 license
- Loading branch information
Showing
2 changed files
with
92 additions
and
77 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,89 @@ | ||
| package com.github.sbt.sbom | ||
|
|
||
| import sbt.librarymanagement.{ ConfigurationReport, ModuleID, ModuleReport } | ||
| import sbt.{ File, OrganizationArtifactReport } | ||
|
|
||
| import scala.collection.mutable | ||
|
|
||
| /* | ||
| * taken from sbt at https://github.com/sbt/sbt/blob/1.10.x/main/src/main/scala/sbt/internal/graph/backend/SbtUpdateReport.scala | ||
| * | ||
| * Copyright 2023, Scala center | ||
| * Copyright 2011 - 2022, Lightbend, Inc. | ||
| * Copyright 2008 - 2010, Mark Harrah | ||
| * Licensed under Apache License 2.0 (see LICENSE) | ||
| */ | ||
| object SbtUpdateReport { | ||
| case class Module( | ||
| id: GraphModuleId, | ||
| license: Option[String] = None, | ||
| extraInfo: String = "", | ||
| evictedByVersion: Option[String] = None, | ||
| jarFile: Option[File] = None, | ||
| error: Option[String] = None | ||
| ) | ||
|
|
||
| private type Edge = (GraphModuleId, GraphModuleId) | ||
| private def Edge(from: GraphModuleId, to: GraphModuleId): Edge = from -> to | ||
|
|
||
| case class ModuleGraph(nodes: Seq[Module], edges: Seq[Edge]) { | ||
| lazy val modules: Map[GraphModuleId, Module] = | ||
| nodes.map(n => (n.id, n)).toMap | ||
|
|
||
| def module(id: GraphModuleId): Option[Module] = modules.get(id) | ||
|
|
||
| lazy val dependencyMap: Map[GraphModuleId, Seq[Module]] = | ||
| createMap(identity) | ||
|
|
||
| def createMap( | ||
| bindingFor: ((GraphModuleId, GraphModuleId)) => (GraphModuleId, GraphModuleId) | ||
| ): Map[GraphModuleId, Seq[Module]] = { | ||
| val m = new mutable.HashMap[GraphModuleId, mutable.Set[Module]] with mutable.MultiMap[GraphModuleId, Module] | ||
| edges.foreach { entry => | ||
| val (f, t) = bindingFor(entry) | ||
| module(t).foreach(m.addBinding(f, _)) | ||
| } | ||
| m.toMap.mapValues(_.toSeq.sortBy(_.id.idString)).toMap.withDefaultValue(Nil) | ||
| } | ||
|
|
||
| def roots: Seq[Module] = | ||
| nodes.filter(n => !edges.exists(_._2 == n.id)).sortBy(_.id.idString) | ||
| } | ||
|
|
||
| case class GraphModuleId(organization: String, name: String, version: String) { | ||
| def idString: String = organization + ":" + name + ":" + version | ||
| } | ||
| object GraphModuleId { | ||
| def apply(sbtId: ModuleID): GraphModuleId = | ||
| GraphModuleId(sbtId.organization, sbtId.name, sbtId.revision) | ||
| } | ||
|
|
||
| def fromConfigurationReport(report: ConfigurationReport, rootInfo: ModuleID): ModuleGraph = { | ||
| def moduleEdges(orgArt: OrganizationArtifactReport): Seq[(Module, Seq[Edge])] = { | ||
| val chosenVersion = orgArt.modules.find(!_.evicted).map(_.module.revision) | ||
| orgArt.modules.map(moduleEdge(chosenVersion)) | ||
| } | ||
|
|
||
| def moduleEdge(chosenVersion: Option[String])(report: ModuleReport): (Module, Seq[Edge]) = { | ||
| val evictedByVersion = if (report.evicted) chosenVersion else None | ||
| val jarFile = report.artifacts | ||
| .find(_._1.`type` == "jar") | ||
| .orElse(report.artifacts.find(_._1.extension == "jar")) | ||
| .map(_._2) | ||
| ( | ||
| Module( | ||
| id = GraphModuleId(report.module), | ||
| license = report.licenses.headOption.map(_._1), | ||
| evictedByVersion = evictedByVersion, | ||
| jarFile = jarFile, | ||
| error = report.problem | ||
| ), | ||
| report.callers.map(caller => Edge(GraphModuleId(caller.caller), GraphModuleId(report.module))) | ||
| ) | ||
| } | ||
| val (nodes, edges) = report.details.flatMap(moduleEdges).unzip | ||
| val root = Module(GraphModuleId(rootInfo)) | ||
|
|
||
| ModuleGraph(root +: nodes, edges.flatten) | ||
| } | ||
| } |