com.autonomousapps.internal.graph.graphs.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dependency-analysis-gradle-plugin Show documentation

Analyzes dependency usage in Android and JVM projects

There is a newer version: 2.12.0

Show newest version

@file:Suppress("UnstableApiUsage")

package com.autonomousapps.internal.graph

import com.autonomousapps.model.Coordinates
import com.autonomousapps.model.IncludedBuildCoordinates
import com.autonomousapps.model.ProjectCoordinates
import com.google.common.graph.ElementOrder
import com.google.common.graph.Graph
import com.google.common.graph.GraphBuilder
import com.google.common.graph.ImmutableGraph

internal operator fun  Graph.plus(other: Graph): Graph where T : Any {
  val builder = newGraphBuilder()

  nodes().forEach { builder.addNode(it) }
  edges().forEach { edge -> builder.putEdge(edge.source(), edge.target()) }
  other.nodes().forEach { builder.addNode(it) }
  other.edges().forEach { edge -> builder.putEdge(edge.source(), edge.target()) }

  return builder.build()
}

/**
 * Flattens a graph by stripping out the
 * [GradleVariantIdentification][com.autonomousapps.model.GradleVariantIdentification], which essentially combines nodes
 * to different variants of a module into a single node. This simplifies reporting on certain scenarios where the module
 * itself is the unit of analysis, rather than the variant.
 */
internal fun Graph.stripVariants(buildPath: String): Graph {
  val builder = newGraphBuilder()

  nodes().forEach { builder.addNode(it.maybeProjectCoordinates(buildPath).flatten()) }
  edges().forEach { edge ->
    val source = edge.source().maybeProjectCoordinates(buildPath).flatten()
    val target = edge.target().maybeProjectCoordinates(buildPath).flatten()

    // In the un-flattened graphs, self-loops are sort of possible because nodes with different capabilities are
    // different.
    if (source != target) {
      builder.putEdge(source, target)
    }
  }

  return builder.build()
}

internal fun  newGraphBuilder(): ImmutableGraph.Builder {
  return GraphBuilder.directed()
    .allowsSelfLoops(false)
    .incidentEdgeOrder(ElementOrder.stable())
    .immutable()
}

/**
 * Might transform [this][Coordinates] into [ProjectCoordinates], if it is an [IncludedBuildCoordinates] that is from
 * "this" build (with buildPath == [buildPath]).
 */
internal fun Coordinates.maybeProjectCoordinates(buildPath: String): Coordinates {
  return if (this is IncludedBuildCoordinates && isForBuild(buildPath)) resolvedProject else this
}

private fun Coordinates.flatten(): Coordinates {
  return if (this is ProjectCoordinates) flatten() else this
}