Export individual representations (#44)

Signed-off-by: Prabhu Subramanian <[email protected]>

build.sbt

@@ -1,6 +1,6 @@
 name                     := "chen"
 ThisBuild / organization := "io.appthreat"
-ThisBuild / version      := "1.1.0"
+ThisBuild / version      := "1.1.1"
 ThisBuild / scalaVersion := "3.3.1"
 
 val cpgVersion = "1.4.22"

codemeta.json

@@ -7,7 +7,7 @@
   "downloadUrl": "https://github.com/AppThreat/chen",
   "issueTracker": "https://github.com/AppThreat/chen/issues",
   "name": "chen",
-  "version": "1.1.0",
+  "version": "1.1.1",
   "description": "Code Hierarchy Exploration Net (chen) is an advanced exploration toolkit for your application source code and its dependency hierarchy.",
   "applicationCategory": "code-analysis",
   "keywords": [

meta.yaml

@@ -1,4 +1,4 @@
-{% set version = "1.1.0" %}
+{% set version = "1.1.1" %}
 
 package:
   name: chen

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "appthreat-chen"
-version = "1.1.0"
+version = "1.1.1"
 description = "Code Hierarchy Exploration Net (chen)"
 authors = ["Team AppThreat <[email protected]>"]
 license = "Apache-2.0"

semanticcpg/src/main/scala/io/shiftleft/semanticcpg/language/types/structure/MethodTraversal.scala

@@ -1,5 +1,6 @@
 package io.shiftleft.semanticcpg.language.types.structure
 
+import better.files.File
 import io.shiftleft.codepropertygraph.generated.*
 import io.shiftleft.codepropertygraph.generated.nodes.*
 import io.shiftleft.semanticcpg.language.*
@@ -264,4 +265,30 @@ class MethodTraversal(val traversal: Iterator[Method]) extends AnyVal:
     end dot
 
     def dot: ExportResult = dot(null)
+
+    def exportAllRepr(dotCfgDir: String = null): Unit =
+        var pathToUse = getOrCreateExportPath(dotCfgDir)
+        traversal
+            .foreach { method =>
+                val methodName     = method.name
+                val methodFullName = method.fullName
+                val filename       = method.location.filename
+                val methodHash     = Fingerprinting.calculate_hash(methodFullName)
+                File(
+                  pathToUse,
+                  s"${methodName}-${sanitizeFilename(filename)}-${methodHash.slice(0, 8)}-ast.dot"
+                ).write(method.dotAst.head)
+                File(
+                  pathToUse,
+                  s"${methodName}-${sanitizeFilename(filename)}-${methodHash.slice(0, 8)}-cdg.dot"
+                ).write(method.dotCdg.head)
+                File(
+                  pathToUse,
+                  s"${methodName}-${sanitizeFilename(filename)}-${methodHash.slice(0, 8)}-cfg.dot"
+                ).write(method.dotCfg.head)
+            }
+    end exportAllRepr
+
+    def exportAllRepr: Unit = exportAllRepr(null)
+
 end MethodTraversal