io.joern.swiftsrc2cpg.passes.DependenciesPass.scala Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of swiftsrc2cpg_3 Show documentation
swiftsrc2cpg
The newest version!
package io.joern.swiftsrc2cpg.passes

import io.joern.x2cpg.X2Cpg
import io.shiftleft.codepropertygraph.generated.Cpg
import io.shiftleft.codepropertygraph.generated.nodes.Call
import io.shiftleft.codepropertygraph.generated.nodes.NewDependency
import io.shiftleft.passes.CpgPass
import io.shiftleft.semanticcpg.language.*
import org.slf4j.Logger
import org.slf4j.LoggerFactory

/** Creation of DEPENDENCY nodes from "Package.swift" files.
  */
class DependenciesPass(cpg: Cpg) extends CpgPass(cpg) {

  private val logger: Logger = LoggerFactory.getLogger(classOf[AstCreationPass])

  private val versionIds = Set("from", "branch", "revision", "exact")

  override def run(diffGraph: DiffGraphBuilder): Unit = {
    cpg.file("Package.swift").ast.isCall.nameExact("package").foreach { call =>
      call.argument.l match
        case Nil => // Do nothing
        case _ :: (pathArg: Call) :: Nil if pathArg.argument(1).code == "path" =>
          val name = X2Cpg.stripQuotes(pathArg.argument(2).code)
          val dep  = NewDependency().name(name)
          diffGraph.addNode(dep)
        case _ :: (nameArg: Call) :: (pathArg: Call) :: Nil
            if nameArg.argument(1).code == "name" && pathArg.argument(1).code == "path" =>
          val name = X2Cpg.stripQuotes(nameArg.argument(2).code)
          val path = X2Cpg.stripQuotes(pathArg.argument(2).code)
          val dep  = NewDependency().name(name).dependencyGroupId(path)
          diffGraph.addNode(dep)
        case _ :: (urlArg: Call) :: (versionArg: Call) :: Nil
            if urlArg.argument(1).code == "url" && versionIds.contains(versionArg.argument(1).code) =>
          val name    = X2Cpg.stripQuotes(urlArg.argument(2).code)
          val version = X2Cpg.stripQuotes(versionArg.argument(2).code)
          val dep     = NewDependency().name(name).version(version)
          diffGraph.addNode(dep)
        case _ :: (urlArg: Call) :: (versionRange: Call) :: Nil if urlArg.argument(1).code == "url" =>
          val name    = X2Cpg.stripQuotes(urlArg.argument(2).code)
          val version = versionRange.code.replaceAll("[\"']", "")
          val dep     = NewDependency().name(name).version(version)
          diffGraph.addNode(dep)
        case callArg =>
          logger.warn(s"Unknown dependency specification: '${callArg.mkString(", ")}'")
    }
  }

}