fix the parent-child issue parsing

Author: tawsifkamal

src/codegen/sdk/core/parser.py

@@ -136,7 +136,7 @@ def parse_ts_statements(self, node: TSNode, file_node_id: NodeId, ctx: CodebaseC
             # =====[ Type Alias Declarations ]=====
             elif child.type == "type_alias_declaration":
                 if import_node := find_import_node(child):
-                    statements.append(TSImportStatement(import_node, file_node_id, ctx, parent, len(statements)))
+                    statements.append(TSImportStatement(child, file_node_id, ctx, parent, len(statements), source_node=import_node))
                 else:
                     statements.append(SymbolStatement(child, file_node_id, ctx, parent, len(statements)))
 
@@ -169,7 +169,7 @@ def parse_ts_statements(self, node: TSNode, file_node_id: NodeId, ctx: CodebaseC
                 if function_node := find_first_function_descendant(child):
                     statements.append(SymbolStatement(child, file_node_id, ctx, parent, len(statements), function_node))
                 elif import_node := find_import_node(child):
-                    statements.append(TSImportStatement(import_node, file_node_id, ctx, parent, len(statements)))
+                    statements.append(TSImportStatement(child, file_node_id, ctx, parent, len(statements), source_node=import_node))
                 else:
                     statements.append(
                         TSAssignmentStatement.from_assignment(
@@ -180,7 +180,7 @@ def parse_ts_statements(self, node: TSNode, file_node_id: NodeId, ctx: CodebaseC
                 statements.append(TSAttribute(child, file_node_id, ctx, parent, pos=len(statements)))
             elif child.type == "expression_statement":
                 if import_node := find_import_node(child):
-                    statements.append(TSImportStatement(import_node, file_node_id, ctx, parent, pos=len(statements)))
+                    statements.append(TSImportStatement(child, file_node_id, ctx, parent, pos=len(statements), source_node=import_node))
                     continue
 
                 for var in child.named_children:

src/codegen/sdk/typescript/import_resolution.py

@@ -451,7 +451,10 @@ def from_dynamic_import_statement(cls, import_call_node: TSNode, module_node: TS
                 return imports
 
             # If import statement is a variable declaration, capture the variable scoping keyword (const, let, var, etc)
-            statement_node = import_statement_node.parent if import_statement_node.type in ["variable_declarator", "assignment_expression"] else import_statement_node
+            if import_statement_node.type == "lexical_declaration":
+                statement_node = import_statement_node
+            else:
+                statement_node = import_statement_node.parent if import_statement_node.type in ["variable_declarator", "assignment_expression"] else import_statement_node
 
             # ==== [ Named dynamic import ] ====
             if name_node.type == "property_identifier":

src/codegen/sdk/typescript/statements/import_statement.py

@@ -35,9 +35,9 @@ def __init__(self, ts_node: TSNode, file_node_id: NodeId, ctx: CodebaseContext,
         imports = []
         if ts_node.type == "import_statement":
             imports.extend(TSImport.from_import_statement(ts_node, file_node_id, ctx, self))
-        elif ts_node.type == "call_expression":
-            import_call_node = ts_node.child_by_field_name("function")
-            arguments = ts_node.child_by_field_name("arguments")
+        elif ts_node.type in ["call_expression", "lexical_declaration", "expression_statement", "type_alias_declaration"]:
+            import_call_node = source_node.child_by_field_name("function")
+            arguments = source_node.child_by_field_name("arguments")
             imports.extend(TSImport.from_dynamic_import_statement(import_call_node, arguments, file_node_id, ctx, self))
         elif ts_node.type == "export_statement":
             imports.extend(TSImport.from_export_statement(source_node, file_node_id, ctx, self))