feat: adding generator to js/ts/tsx and improvements for ts (#234)

* feat: adding generator to js/ts/tsx and improvements for ts

* reverting tests

queries/javascript/context.scm

@@ -10,6 +10,7 @@
   (expression_statement)
   (for_statement)
   (function_declaration)
+  (generator_function_declaration)
   (jsx_element)
   (jsx_self_closing_element)
   (lexical_declaration)

queries/tsx/context.scm

@@ -3,13 +3,11 @@
 ) @context
 
 ([
-  (arrow_function)
   (call_expression)
   (class_declaration)
   (else_clause)
   (expression_statement)
   (for_statement)
-  (function_declaration)
   (interface_declaration)
   (jsx_element)
   (jsx_self_closing_element)
@@ -19,3 +17,15 @@
   (pair)
   (while_statement)
 ] @context)
+
+(arrow_function
+  body: (_ (_) @context.end)
+) @context
+
+(function_declaration
+  body: (_ (_) @context.end)
+) @context
+
+(generator_function_declaration
+  body: (_ (_) @context.end)
+) @context

queries/typescript/context.scm

@@ -3,13 +3,11 @@
 ) @context
 
 ([
-  (arrow_function)
   (call_expression)
   (class_declaration)
   (else_clause)
   (expression_statement)
   (for_statement)
-  (function_declaration)
   (interface_declaration)
   (lexical_declaration)
   (method_definition)
@@ -17,3 +15,15 @@
   (pair)
   (while_statement)
 ] @context)
+
+(arrow_function
+  body: (_ (_) @context.end)
+) @context
+
+(function_declaration
+  body: (_ (_) @context.end)
+) @context
+
+(generator_function_declaration
+  body: (_ (_) @context.end)
+) @context

test/test.tsx

@@ -44,6 +44,19 @@ function wrapInArray(obj: string | string[]) {
 
 
 
+  }
+  return obj;
+}
+
+function* wrapInArray2(
+  obj: string | string[]):
+    string[] {
+  if (typeof obj === "string") {
+    return [obj];
+
+
+
+
   }
   return obj;
 }