Skip to content

Commit d561aca

Browse files
authored
use generated Cpg class (#1770)
also to minify the diff for the flatgraph port
1 parent 7bf9c82 commit d561aca

File tree

10 files changed

+15
-15
lines changed

10 files changed

+15
-15
lines changed

README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -77,7 +77,7 @@ You can run `./regenerate-test-cpgs.sh` to update them, but this requires the pr
7777
sbt semanticcpg/console
7878
```
7979
```scala
80-
import io.shiftleft.codepropertygraph.Cpg
80+
import io.shiftleft.codepropertygraph.generated.Cpg
8181
import io.shiftleft.semanticcpg.language._
8282
val cpg = io.shiftleft.codepropertygraph.cpgloading.CpgLoader.load("./resources/testcode/cpgs/hello-shiftleft-0.0.5/cpg.bin.zip")
8383
```

codepropertygraph/src/main/scala/io/shiftleft/OverflowDbTestInstance.scala

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
package io.shiftleft
22

3-
import io.shiftleft.codepropertygraph.Cpg
3+
import io.shiftleft.codepropertygraph.generated.Cpg
44
import overflowdb.Graph
55

66
object OverflowDbTestInstance {

codepropertygraph/src/main/scala/io/shiftleft/codepropertygraph/cpgloading/CpgLoader.scala

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
package io.shiftleft.codepropertygraph.cpgloading
22

33
import better.files.File
4-
import io.shiftleft.codepropertygraph.Cpg
4+
import io.shiftleft.codepropertygraph.generated.Cpg
55
import io.shiftleft.codepropertygraph.generated.PropertyNames
66
import org.slf4j.{Logger, LoggerFactory}
77

codepropertygraph/src/main/scala/io/shiftleft/codepropertygraph/cpgloading/ProtoCpgLoader.scala

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
package io.shiftleft.codepropertygraph.cpgloading
22

33
import com.google.protobuf.GeneratedMessageV3
4-
import io.shiftleft.codepropertygraph.Cpg
4+
import io.shiftleft.codepropertygraph.generated.Cpg
55
import io.shiftleft.proto.cpg.Cpg.CpgStruct.Edge
66
import io.shiftleft.proto.cpg.Cpg.CpgStruct.Edge.EdgeType
77
import io.shiftleft.proto.cpg.Cpg.{CpgOverlay, CpgStruct}

codepropertygraph/src/main/scala/io/shiftleft/codepropertygraph/cpgloading/ProtoToCpg.scala

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
package io.shiftleft.codepropertygraph.cpgloading
22

3-
import io.shiftleft.codepropertygraph.Cpg
3+
import io.shiftleft.codepropertygraph.generated.Cpg
44
import io.shiftleft.proto.cpg.Cpg.CpgStruct.Edge.EdgeType
55
import io.shiftleft.proto.cpg.Cpg.CpgStruct.{Edge, Node}
66
import io.shiftleft.proto.cpg.Cpg.PropertyValue

codepropertygraph/src/main/scala/io/shiftleft/passes/CpgPass.scala

+1-1
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ package io.shiftleft.passes
22

33
import com.google.protobuf.GeneratedMessageV3
44
import io.shiftleft.SerializedCpg
5-
import io.shiftleft.codepropertygraph.Cpg
5+
import io.shiftleft.codepropertygraph.generated.Cpg
66
import org.slf4j.{Logger, LoggerFactory, MDC}
77
import overflowdb.BatchedUpdate
88

codepropertygraph/src/main/scala/io/shiftleft/passes/ParallelCpgPass.scala

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
package io.shiftleft.passes
22
import io.shiftleft.SerializedCpg
3-
import io.shiftleft.codepropertygraph.Cpg
3+
import io.shiftleft.codepropertygraph.generated.Cpg
44
import io.shiftleft.utils.ExecutionContextProvider
55
import org.slf4j.MDC
66

codepropertygraph/src/test/scala/io/shiftleft/passes/CpgPassNewTests.scala

+3-3
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ package io.shiftleft.passes
22

33
import better.files.File
44
import io.shiftleft.SerializedCpg
5-
import io.shiftleft.codepropertygraph.Cpg
5+
import io.shiftleft.codepropertygraph.generated.Cpg
66
import io.shiftleft.codepropertygraph.generated.nodes.NewFile
77
import org.scalatest.matchers.should.Matchers
88
import org.scalatest.wordspec.AnyWordSpec
@@ -14,7 +14,7 @@ class CpgPassNewTests extends AnyWordSpec with Matchers {
1414

1515
private object Fixture {
1616
def apply(keyPool: Option[KeyPool] = None)(f: (Cpg, CpgPassBase) => Unit): Unit = {
17-
val cpg = Cpg.emptyCpg
17+
val cpg = Cpg.empty
1818
class MyPass(cpg: Cpg) extends SimpleCpgPass(cpg, "MyPass", keyPool) {
1919
override def run(builder: DiffGraphBuilder): Unit = {
2020
val builder2 = new DiffGraphBuilder
@@ -52,7 +52,7 @@ class CpgPassNewTests extends AnyWordSpec with Matchers {
5252
}
5353

5454
"fail for schema violations" in {
55-
val cpg = Cpg.emptyCpg
55+
val cpg = Cpg.empty
5656
val pass = new SimpleCpgPass(cpg, "pass1") {
5757
override def run(dst: DiffGraphBuilder): Unit = {
5858
val file1 = NewFile().name("foo")

codepropertygraph/src/test/scala/io/shiftleft/passes/ParallelCpgPassNewTests.scala

+4-4
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@ package io.shiftleft.passes
22

33
import better.files.File
44
import io.shiftleft.SerializedCpg
5-
import io.shiftleft.codepropertygraph.Cpg
5+
import io.shiftleft.codepropertygraph.generated.Cpg
66
import io.shiftleft.codepropertygraph.generated.Properties
77
import io.shiftleft.codepropertygraph.generated.nodes.{NewCall, NewFile}
88
import org.scalatest.matchers.should.Matchers
@@ -16,7 +16,7 @@ class ParallelCpgPassNewTests extends AnyWordSpec with Matchers {
1616

1717
private object Fixture {
1818
def apply(keyPools: Option[Iterator[KeyPool]] = None)(f: (Cpg, CpgPassBase) => Unit): Unit = {
19-
val cpg = Cpg.emptyCpg
19+
val cpg = Cpg.empty
2020
val pool = keyPools.flatMap(_.nextOption())
2121
class MyPass(cpg: Cpg) extends ConcurrentWriterCpgPass[String](cpg, "MyPass", pool) {
2222
override def generateParts(): Array[String] = Array("foo", "bar")
@@ -56,7 +56,7 @@ class ParallelCpgPassNewTests extends AnyWordSpec with Matchers {
5656
}
5757

5858
"fail for schema violations" in {
59-
val cpg = Cpg.emptyCpg
59+
val cpg = Cpg.empty
6060
val pass = new ConcurrentWriterCpgPass[String](cpg, "pass2") {
6161
override def generateParts() = Array("a", "b")
6262
override def runOnPart(diffGraph: DiffGraphBuilder, part: String): Unit =
@@ -83,7 +83,7 @@ class ParallelCpgPassNewTests extends AnyWordSpec with Matchers {
8383
}
8484

8585
"add NewNodes that are referenced in different parts only once" in {
86-
val cpg = Cpg.emptyCpg
86+
val cpg = Cpg.empty
8787
val pass = new ConcurrentWriterCpgPass[String](cpg, "pass2") {
8888
val call1 = NewCall().name("call1")
8989
val call2 = NewCall().name("call2")

samples/pass/src/main/scala/io/shiftleft/passes/mypass/SamplePass.scala

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
package io.shiftleft.mypass
22

33
import io.shiftleft.SerializedCpg
4-
import io.shiftleft.codepropertygraph.Cpg
4+
import io.shiftleft.codepropertygraph.generated.Cpg
55
import io.shiftleft.codepropertygraph.cpgloading.{CpgLoader, CpgLoaderConfig}
66
import io.shiftleft.codepropertygraph.generated.nodes.NewFile
77
import io.shiftleft.passes.SimpleCpgPass

0 commit comments

Comments
 (0)