add getAnnotatedElementProperties fn to parser

facebook · Mar 16, 2023 · 7464edc · 7464edc
1 parent 4360da4
commit 7464edc
Show file tree

Hide file tree

Showing 5 changed files with 24 additions and 4 deletions.
diff --git a/packages/react-native-codegen/src/parsers/flow/parser.js b/packages/react-native-codegen/src/parsers/flow/parser.js
@@ -295,6 +295,10 @@ class FlowParser implements Parser {
   functionTypeAnnotation(propertyValueType: string): boolean {
     return propertyValueType === 'FunctionTypeAnnotation';
   }
+
+  getAnnotatedElementProperties(annotatedElement: $FlowFixMe): $FlowFixMe {
+    return annotatedElement.right.properties;
+  }
 }
 
 module.exports = {

diff --git a/packages/react-native-codegen/src/parsers/parser.js b/packages/react-native-codegen/src/parsers/parser.js
@@ -211,4 +211,11 @@ export interface Parser {
    * @returns: a boolean specifying if the property is a function type annotation.
    */
   functionTypeAnnotation(propertyValueType: string): boolean;
+
+  /**
+   * Given a annotatedElement, it returns the properties of annotated element.
+   * @parameter annotatedElement: the annotated element.
+   * @returns: the properties of annotated element.
+   */
+  getAnnotatedElementProperties(annotatedElement: $FlowFixMe): $FlowFixMe;
 }
diff --git a/packages/react-native-codegen/src/parsers/parserMock.js b/packages/react-native-codegen/src/parsers/parserMock.js
@@ -213,4 +213,8 @@ export class MockedParser implements Parser {
   functionTypeAnnotation(propertyValueType: string): boolean {
     return propertyValueType === 'FunctionTypeAnnotation';
   }
+
+  getAnnotatedElementProperties(annotatedElement: $FlowFixMe): $FlowFixMe {
+    return annotatedElement.right.properties;
+  }
 }
diff --git a/packages/react-native-codegen/src/parsers/parsers-primitives.js b/packages/react-native-codegen/src/parsers/parsers-primitives.js
@@ -494,7 +494,8 @@ function emitPartial(
     typeAnnotation,
     types,
   );
-  const annotatedElementProperties = annotatedElement.right.properties;
+  const annotatedElementProperties =
+    parser.getAnnotatedElementProperties(annotatedElement);
 
   const partialProperties = parser.computePartialProperties(
     annotatedElementProperties,
@@ -520,10 +521,10 @@ function emitCommonTypes(
   nullable: boolean,
   parser: Parser,
 ): $FlowFixMe {
-  const typeAnnotationType =
-    const typeAnnotationType = parser.nameForGenericTypeAnnotation()
+  const genericTypeAnnotationName =
+    parser.nameForGenericTypeAnnotation(typeAnnotation);
 
-  switch (typeAnnotationType) {
+  switch (genericTypeAnnotationName) {
     case 'Stringish': {
       return emitStringish(nullable);
     }

diff --git a/packages/react-native-codegen/src/parsers/typescript/parser.js b/packages/react-native-codegen/src/parsers/typescript/parser.js
@@ -284,6 +284,10 @@ class TypeScriptParser implements Parser {
       propertyValueType === 'TSMethodSignature'
     );
   }
+
+  getAnnotatedElementProperties(annotatedElement: $FlowFixMe): $FlowFixMe {
+    return annotatedElement.typeAnnotation.members;
+  }
 }
 
 module.exports = {