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

Go to download

Show more of this group Show more artifacts with this name
Show all versions of query-primitives Show documentation

query-primitives

There is a newer version: 0.9.322

package io.shiftleft.queryprimitives.steps.nodemethods

import io.shiftleft.codepropertygraph.generated.nodes.NodeVisitor
import io.shiftleft.codepropertygraph.generated.{EdgeTypes, nodes}
import io.shiftleft.Implicits.JavaIteratorDeco
import io.shiftleft.queryprimitives.steps.visitormixins.ExpressionGeneralization
import org.apache.tinkerpop.gremlin.structure.Direction

class WithinMethodMethods(val node: nodes.WithinMethod) extends AnyVal {
  def method: nodes.Method = node.accept(WithinMethodToMethod)
}

private object WithinMethodToMethod extends NodeVisitor[nodes.Method] with ExpressionGeneralization[nodes.Method] {

  override def visit(node: nodes.Method): nodes.Method = node.asInstanceOf[nodes.Method]

  override def visit(node: nodes.MethodParameterIn): nodes.Method =
    node.vertices(Direction.IN, EdgeTypes.AST).nextChecked.asInstanceOf[nodes.Method]

  override def visit(node: nodes.MethodParameterOut): nodes.Method =
    node.vertices(Direction.IN, EdgeTypes.AST).nextChecked.asInstanceOf[nodes.Method]

  override def visit(node: nodes.MethodReturn): nodes.Method =
    node.vertices(Direction.IN, EdgeTypes.AST).nextChecked.asInstanceOf[nodes.Method]

  override def visit(node: nodes.ImplicitCall): nodes.Method =
    node.vertices(Direction.IN, EdgeTypes.AST).nextChecked.asInstanceOf[nodes.Method]

  override def visit(node: nodes.Expression): nodes.Method =
    node.vertices(Direction.IN, EdgeTypes.CONTAINS).nextChecked.asInstanceOf[nodes.Method]
}