Add API to get a specific class in the application.

Signed-off-by: Rahul Krishna <[email protected]>

Author: rahlk

src/analysis/java/JavaAnalysis.ts

@@ -173,10 +173,29 @@ export class JavaAnalysis {
     }, {} as Record<string, types.JTypeType>);
   }
 
+  /**
+   * Get all methods in the application.
+   * @returns {Promise<Record<string, Record<string, types.JCallableType>>>} A promise that resolves to a record of
+   * method names and their corresponding {@link JCallableType} objects
+   * 
+   * @notes This method retrieves all methods from the symbol table and returns them as a record. The returned
+   *       record contains class names as keys and their corresponding {@link JCallableType} objects as values.
+   *       Each {@link JCallableType} object contains information about the method's parameters, return type, and
+   *       other relevant details.
+   */
   public async getAllMethods(): Promise<Record<string, Record<string, types.JCallableType>>> {
     return Object.entries(await this.getAllClasses()).reduce((allMethods, [key, value]) => {
       allMethods[key] = value.callable_declarations;
       return allMethods;
     }, {} as Record<string, Record<string, types.JCallableType>>);
   }
+
+  public async getClassByQualifiedName(qualifiedName: string): Promise<types.JTypeType> {
+    const allClasses = await this.getAllClasses();
+    if (allClasses[qualifiedName]) {
+      return allClasses[qualifiedName];
+    }
+    else
+      throw new Error(`Class ${qualifiedName} not found in the application.`);
+  }
 }

test/unit/analysis/java/JavaAnalysis.test.ts

@@ -1,3 +1,4 @@
+import { JType } from "../../../../src/models/java/";
 import { daytraderJavaAnalysis } from "../../../conftest";
 import { expect, test } from "bun:test";
 
@@ -19,7 +20,15 @@ test("Must get all classes in a Java application", async () => {
   expect(await daytraderJavaAnalysis.getAllClasses()).toBeDefined();
 });
 
-test("Must get all methods in the application", async () => {
-  const allMethods = await daytraderJavaAnalysis.getAllMethods();
-  expect(allMethods).toBeDefined();
+test("Must get a specific class the application", async () => {
+  const tradeDirectObject = await daytraderJavaAnalysis.getClassByQualifiedName("com.ibm.websphere.samples.daytrader.impl.direct.TradeDirect");
+  console.log(tradeDirectObject);
+  expect(async () => JType.parse(tradeDirectObject)).not.toThrow(); 
 });
+
+test("Must get all methods in the application", () => {
+  return daytraderJavaAnalysis.getAllMethods().then((methods) => {
+    expect(methods).toBeDefined()
+  });
+});
+