• Home
• io.shiftleft
• query-primitives
• 0.9.296
• source code
• CfgNodeMethods.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

io.shiftleft.queryprimitives.steps.nodemethods.CfgNodeMethods.scala Maven / Gradle / Ivy

package io.shiftleft.queryprimitives.steps.nodemethods

import io.shiftleft.codepropertygraph.generated.nodes
import io.shiftleft.codepropertygraph.generated.nodes.NodeVisitor
import io.shiftleft.queryprimitives.steps.visitormixins.ExpressionGeneralization

class CfgNodeMethods(val node: nodes.CfgNode) extends AnyVal {
  def code: String = node.accept(CfgNodeToCode)
}

private object CfgNodeToCode extends NodeVisitor[String] with ExpressionGeneralization[String] {

  override def visit(node: nodes.Expression): String = node.code

  override def visit(node: nodes.Method): String = node.name

  override def visit(node: nodes.MethodReturn): String = node.code
}