Now body of each syntax type is getting extracted perfectly

Author: preeesha

rocket-chatter-ingestion-server/src/process/prepare/processor/core/treeNode.ts

@@ -2,12 +2,10 @@ export type TreeNodeSourceLocation = {
 	start: {
 		line: number
 		column: number
-		index: number
 	}
 	end: {
 		line: number
 		column: number
-		index: number
 	}
 }
 

rocket-chatter-ingestion-server/src/process/prepare/processor/file.ts

@@ -86,7 +86,18 @@ export class FileProcessor implements IFileProcessor {
 		}
 
 		// Add the nodes to the global reference
+		const fileContentLines = fileContent.split("\n")
+
 		for (const treeNode of treeNodes) {
+			if (!treeNode.body) {
+				const startLine = treeNode.location.start.line - 1
+				const endLine = treeNode.location.end.line + 1
+				treeNode.body = fileContentLines
+					.slice(startLine, endLine)
+					.join("\n")
+					.trim()
+			}
+
 			const dbNode = DBNode.fromTreeNode(treeNode)
 			nodesRef[dbNode.id] = dbNode
 		}

rocket-chatter-ingestion-server/src/process/prepare/processor/syntax/classes.ts

@@ -1,30 +1,21 @@
 import { namedTypes } from "ast-types"
 
-import { print } from "recast"
 import { TreeNode } from "../core/treeNode"
 import { TypeArgument } from "../core/typeArgument"
 import { Functions } from "./functions"
 
 export namespace Classes {
 	export function Handle(n: namedTypes.ClassDeclaration) {
-		const node = new TreeNode(
-			n.id?.name.toString() ?? "",
-			"Class",
-			n.body.body.map((e) => print(e).code).join("\n"),
-			"",
-			{
-				start: {
-					line: n.loc?.start.line ?? 0,
-					column: n.loc?.start.column ?? 0,
-					index: (n as any).start ?? 0,
-				},
-				end: {
-					line: n.loc?.end.line ?? 0,
-					column: n.loc?.end.column ?? 0,
-					index: (n as any).end ?? 0,
-				},
-			}
-		)
+		const node = new TreeNode(n.id?.name.toString() ?? "", "Class", "", "", {
+			start: {
+				line: n.body.loc?.start.line ?? 0,
+				column: n.body.loc?.start.column ?? 0,
+			},
+			end: {
+				line: n.body.loc?.end.line ?? 0,
+				column: n.body.loc?.end.column ?? 0,
+			},
+		})
 
 		// Check for type parameters
 		const typeParameters: string[] = []

rocket-chatter-ingestion-server/src/process/prepare/processor/syntax/enums.ts

@@ -1,4 +1,5 @@
 import { namedTypes } from "ast-types"
+
 import { TreeNode } from "../core/treeNode"
 import { Classes } from "./classes"
 import { Functions } from "./functions"
@@ -9,12 +10,10 @@ export namespace Enums {
 			start: {
 				line: n.loc?.start.line ?? 0,
 				column: n.loc?.start.column ?? 0,
-				index: (n as any).start ?? 0,
 			},
 			end: {
 				line: n.loc?.end.line ?? 0,
 				column: n.loc?.end.column ?? 0,
-				index: (n as any).end ?? 0,
 			},
 		})
 

rocket-chatter-ingestion-server/src/process/prepare/processor/syntax/functions.ts

@@ -1,30 +1,22 @@
 import { namedTypes } from "ast-types"
-import { print } from "recast"
+
 import { TreeNode } from "../core/treeNode"
 import { TypeAnnotation } from "../core/typeAnnotation"
 import { TypeArgument } from "../core/typeArgument"
 import { Classes } from "./classes"
 
 export namespace Functions {
 	export function Handle(n: namedTypes.FunctionDeclaration) {
-		const node = new TreeNode(
-			n.id?.name.toString() ?? "",
-			"Function",
-			n.body.body.map((e) => print(e).code).join("\n"),
-			"",
-			{
-				start: {
-					line: n.loc?.start.line ?? 0,
-					column: n.loc?.start.column ?? 0,
-					index: (n as any).start ?? 0,
-				},
-				end: {
-					line: n.loc?.end.line ?? 0,
-					column: n.loc?.end.column ?? 0,
-					index: (n as any).end ?? 0,
-				},
-			}
-		)
+		const node = new TreeNode(n.id?.name.toString() ?? "", "Function", "", "", {
+			start: {
+				line: n.body.loc?.start.line ?? 0,
+				column: n.body.loc?.start.column ?? 0,
+			},
+			end: {
+				line: n.body.loc?.end.line ?? 0,
+				column: n.body.loc?.end.column ?? 0,
+			},
+		})
 
 		// Check for type parameters
 		const typeParameters: string[] = []

rocket-chatter-ingestion-server/src/process/prepare/processor/syntax/interface.ts

@@ -1,5 +1,5 @@
 import { namedTypes } from "ast-types"
-import { print } from "recast"
+
 import { TreeNode } from "../core/treeNode"
 import { TypeAnnotation } from "../core/typeAnnotation"
 import { TypeArgument } from "../core/typeArgument"
@@ -9,18 +9,16 @@ export namespace Interface {
 		const node = new TreeNode(
 			n.id?.name.toString() ?? "",
 			"Interface",
-			(n.body as any).body.map((e: any) => print(e).code).join("\n"),
+			"",
 			"",
 			{
 				start: {
 					line: n.loc?.start.line ?? 0,
 					column: n.loc?.start.column ?? 0,
-					index: (n as any).start ?? 0,
 				},
 				end: {
 					line: n.loc?.end.line ?? 0,
 					column: n.loc?.end.column ?? 0,
-					index: (n as any).end ?? 0,
 				},
 			}
 		)
@@ -32,7 +30,7 @@ export namespace Interface {
 		}
 
 		// Check for extensions
-		for (const e of n.extends) {
+		for (const e of n.extends ?? []) {
 			const name = (e as any).expression.name
 			if (typeParameters.includes(name)) continue
 

rocket-chatter-ingestion-server/src/process/prepare/processor/syntax/namespaces.ts

@@ -1,4 +1,5 @@
-import { namedTypes } from "ast-types/gen/namedTypes"
+import { namedTypes } from "ast-types"
+
 import { TreeNode } from "../core/treeNode"
 import { Enums } from "./enums"
 import { Functions } from "./functions"
@@ -16,12 +17,10 @@ export namespace Namespaces {
 				start: {
 					line: n.loc?.start.line ?? 0,
 					column: n.loc?.start.column ?? 0,
-					index: (n as any).start ?? 0,
 				},
 				end: {
 					line: n.loc?.end.line ?? 0,
 					column: n.loc?.end.column ?? 0,
-					index: (n as any).end ?? 0,
 				},
 			}
 		)

rocket-chatter-ingestion-server/src/process/prepare/processor/syntax/typeAlias.ts

@@ -1,5 +1,5 @@
 import { namedTypes } from "ast-types"
-import { writeFileSync } from "fs"
+
 import { TreeNode } from "../core/treeNode"
 import { TypeAnnotation } from "../core/typeAnnotation"
 
@@ -14,19 +14,14 @@ export namespace TypeAlias {
 				start: {
 					line: n.loc?.start.line ?? 0,
 					column: n.loc?.start.column ?? 0,
-					index: (n as any).start ?? 0,
 				},
 				end: {
 					line: n.loc?.end.line ?? 0,
 					column: n.loc?.end.column ?? 0,
-					index: (n as any).end ?? 0,
 				},
 			}
 		)
 
-		// extract body of the type alias
-		writeFileSync("test.json", JSON.stringify(n, null, 2))
-
 		// Check for type parameters
 		const typeParameters: string[] = []
 		for (const p of n.typeParameters?.params ?? []) {

rocket-chatter-ingestion-server/src/process/prepare/processor/syntax/variableDeclarations.ts

@@ -1,4 +1,5 @@
 import { namedTypes } from "ast-types"
+
 import { TreeNode } from "../core/treeNode"
 import { Classes } from "./classes"
 import { Functions } from "./functions"
@@ -14,12 +15,10 @@ export namespace VariableDeclarations {
 					start: {
 						line: n.loc?.start.line ?? 0,
 						column: n.loc?.start.column ?? 0,
-						index: (n as any).start ?? 0,
 					},
 					end: {
 						line: n.loc?.end.line ?? 0,
 						column: n.loc?.end.column ?? 0,
-						index: (n as any).end ?? 0,
 					},
 				})