io.joern.pysrc2cpg.ImportsPass.scala Maven / Gradle / Ivy
package io.joern.pysrc2cpg
import io.joern.x2cpg.passes.frontend.XImportsPass
import io.shiftleft.codepropertygraph.Cpg
import io.shiftleft.codepropertygraph.generated.nodes.*
import io.shiftleft.semanticcpg.language._
import io.shiftleft.semanticcpg.language.operatorextension.OpNodes.Assignment
class ImportsPass(cpg: Cpg) extends XImportsPass(cpg) {
override protected val importCallName: String = "import"
override protected def importCallToPart(x: Call): Iterator[(Call, Assignment)] = x.inAssignment.map(y => (x, y))
override def importedEntityFromCall(call: Call): String = {
call.argument.code.l match {
case List("", what) => what.split('.')(0)
case List(where, what) => s"$where.$what"
case List("", what, _) => what
case List(where, what, _) => s"$where.$what"
case _ => ""
}
}
}