Codegen 105: add typeAlias prop in parsers (#37334)

Summary:
Part of Codegen Issue: #34872
> Add a typeAlias: string property to the Parser class. Implement it in the Flow parser so that it returns TypeAlias and in
> the TypeScriptParser so that it returns TSTypeAliasDeclaration. Replace the case in the switch  in the [parsers/flow/utils.js](https://github.com/facebook/react-native/blob/e133100721939108b0f28dfa9f60ac627c804018/packages/react-native-codegen/src/parsers/flow/utils.js#L57) and [parsers/typescript/utils.js](https://github.com/facebook/react-native/blob/e133100721939108b0f28dfa9f60ac627c804018/packages/react-native-codegen/src/parsers/typescript/utils.js#L60) with this prop.

bypass-github-export-checks

## Changelog:

<!-- Help reviewers and the release process by writing your own changelog entry.

Pick one each for the category and type tags:

[ANDROID|GENERAL|IOS|INTERNAL] [BREAKING|ADDED|CHANGED|DEPRECATED|REMOVED|FIXED|SECURITY] - Message

For more details, see:
https://reactnative.dev/contributing/changelogs-in-pull-requests
-->
[Internal][Added]: Add typeAlias property in parsers

Pull Request resolved: #37334

Test Plan: ```yarn test```

Reviewed By: dmytrorykun

Differential Revision: D45691915

Pulled By: cipolleschi

fbshipit-source-id: efe5c87afc3193f6a35325190a67f98d64426ab3

Author: tarunrajput

packages/react-native-codegen/src/parsers/__tests__/parsers-test.js

@@ -305,6 +305,12 @@ describe('FlowParser', () => {
       expect(parser.isOptionalProperty(property)).toEqual(true);
     });
   });
+
+  describe('typeAlias', () => {
+    it('returns typeAlias Property', () => {
+      expect(parser.typeAlias).toEqual('TypeAlias');
+    });
+  });
 });
 
 describe('TypeScriptParser', () => {
@@ -582,4 +588,10 @@ describe('TypeScriptParser', () => {
       expect(parser.isOptionalProperty(property)).toEqual(false);
     });
   });
+
+  describe('typeAlias', () => {
+    it('returns typeAlias Property', () => {
+      expect(parser.typeAlias).toEqual('TSTypeAliasDeclaration');
+    });
+  });
 });

packages/react-native-codegen/src/parsers/flow/modules/index.js

@@ -20,6 +20,7 @@ import type {
 } from '../../../CodegenSchema';
 
 import type {Parser} from '../../parser';
+const {resolveTypeAnnotation} = require('../utils');
 import type {ParserErrorCapturer, TypeDeclarationMap} from '../../utils';
 
 const {
@@ -59,7 +60,7 @@ function translateTypeAnnotation(
   parser: Parser,
 ): Nullable<NativeModuleTypeAnnotation> {
   const {nullable, typeAnnotation, typeResolutionStatus} =
-    parser.getResolvedTypeAnnotation(flowTypeAnnotation, types);
+    resolveTypeAnnotation(flowTypeAnnotation, types, parser);
 
   switch (typeAnnotation.type) {
     case 'GenericTypeAnnotation': {

packages/react-native-codegen/src/parsers/flow/parser.js

@@ -22,13 +22,20 @@ import type {
   NativeModuleEnumMap,
 } from '../../CodegenSchema';
 import type {ParserType} from '../errors';
-import type {GetSchemaInfoFN, GetTypeAnnotationFN, Parser} from '../parser';
+import type {
+  GetSchemaInfoFN,
+  GetTypeAnnotationFN,
+  Parser,
+  ResolveTypeAnnotationFN,
+} from '../parser';
 import type {
   ParserErrorCapturer,
   TypeDeclarationMap,
   PropAST,
   TypeResolutionStatus,
 } from '../utils';
+
+const {resolveTypeAnnotation} = require('./utils');
 const invariant = require('invariant');
 
 const {
@@ -55,6 +62,7 @@ const {
 
 class FlowParser implements Parser {
   typeParameterInstantiation: string = 'TypeParameterInstantiation';
+  typeAlias: string = 'TypeAlias';
 
   isProperty(property: $FlowFixMe): boolean {
     return property.type === 'ObjectTypeProperty';
@@ -429,6 +437,10 @@ class FlowParser implements Parser {
       typeResolutionStatus,
     };
   }
+
+  getResolveTypeAnnotationFN(): ResolveTypeAnnotationFN {
+    return resolveTypeAnnotation;
+  }
 }
 
 module.exports = {

packages/react-native-codegen/src/parsers/flow/utils.js

@@ -10,7 +10,81 @@
 
 'use strict';
 
-import type {TypeDeclarationMap, ASTNode} from '../utils';
+import type {TypeResolutionStatus, TypeDeclarationMap, ASTNode} from '../utils';
+import type {Parser} from '../../parsers/parser';
+
+const invariant = require('invariant');
+
+function resolveTypeAnnotation(
+  // TODO(T71778680): This is an Flow TypeAnnotation. Flow-type this
+  typeAnnotation: $FlowFixMe,
+  types: TypeDeclarationMap,
+  parser: Parser,
+): {
+  nullable: boolean,
+  typeAnnotation: $FlowFixMe,
+  typeResolutionStatus: TypeResolutionStatus,
+} {
+  invariant(
+    typeAnnotation != null,
+    'resolveTypeAnnotation(): typeAnnotation cannot be null',
+  );
+
+  let node = typeAnnotation;
+  let nullable = false;
+  let typeResolutionStatus: TypeResolutionStatus = {
+    successful: false,
+  };
+
+  for (;;) {
+    if (node.type === 'NullableTypeAnnotation') {
+      nullable = true;
+      node = node.typeAnnotation;
+      continue;
+    }
+
+    if (node.type !== 'GenericTypeAnnotation') {
+      break;
+    }
+
+    const resolvedTypeAnnotation = types[node.id.name];
+    if (resolvedTypeAnnotation == null) {
+      break;
+    }
+
+    switch (resolvedTypeAnnotation.type) {
+      case parser.typeAlias: {
+        typeResolutionStatus = {
+          successful: true,
+          type: 'alias',
+          name: node.id.name,
+        };
+        node = resolvedTypeAnnotation.right;
+        break;
+      }
+      case 'EnumDeclaration': {
+        typeResolutionStatus = {
+          successful: true,
+          type: 'enum',
+          name: node.id.name,
+        };
+        node = resolvedTypeAnnotation.body;
+        break;
+      }
+      default: {
+        throw new TypeError(
+          `A non GenericTypeAnnotation must be a type declaration ('${parser.typeAlias}') or enum ('EnumDeclaration'). Instead, got the unsupported ${resolvedTypeAnnotation.type}.`,
+        );
+      }
+    }
+  }
+
+  return {
+    nullable: nullable,
+    typeAnnotation: node,
+    typeResolutionStatus,
+  };
+}
 
 function getValueFromTypes(value: ASTNode, types: TypeDeclarationMap): ASTNode {
   if (value.type === 'GenericTypeAnnotation' && types[value.id.name]) {
@@ -21,4 +95,5 @@ function getValueFromTypes(value: ASTNode, types: TypeDeclarationMap): ASTNode {
 
 module.exports = {
   getValueFromTypes,
+  resolveTypeAnnotation,
 };

packages/react-native-codegen/src/parsers/parser.js

@@ -63,6 +63,16 @@ export type BuildSchemaFN<T> = (
   parser: Parser,
 ) => ?NamedShape<T>;
 
+export type ResolveTypeAnnotationFN = (
+  typeAnnotation: $FlowFixMe,
+  types: TypeDeclarationMap,
+  parser: Parser,
+) => {
+  nullable: boolean,
+  typeAnnotation: $FlowFixMe,
+  typeResolutionStatus: TypeResolutionStatus,
+};
+
 /**
  * This is the main interface for Parsers of various languages.
  * It exposes all the methods that contain language-specific logic.
@@ -73,6 +83,11 @@ export interface Parser {
    */
   typeParameterInstantiation: string;
 
+  /**
+   * TypeAlias property of the Parser
+   */
+  typeAlias: string;
+
   /**
    * Given a declaration, it returns true if it is a property
    */
@@ -328,4 +343,6 @@ export interface Parser {
     typeAnnotation: $FlowFixMe,
     typeResolutionStatus: TypeResolutionStatus,
   };
+
+  getResolveTypeAnnotationFN(): ResolveTypeAnnotationFN;
 }

packages/react-native-codegen/src/parsers/parserMock.js

@@ -10,7 +10,12 @@
 
 'use strict';
 
-import type {GetSchemaInfoFN, GetTypeAnnotationFN, Parser} from './parser';
+import type {
+  GetSchemaInfoFN,
+  GetTypeAnnotationFN,
+  Parser,
+  ResolveTypeAnnotationFN,
+} from './parser';
 import type {ParserType} from './errors';
 import type {
   UnionTypeAnnotationMemberType,
@@ -55,6 +60,7 @@ const schemaMock = {
 
 export class MockedParser implements Parser {
   typeParameterInstantiation: string = 'TypeParameterInstantiation';
+  typeAlias: string = 'TypeAlias';
 
   isProperty(property: $FlowFixMe): boolean {
     return property.type === 'ObjectTypeProperty';
@@ -280,6 +286,16 @@ export class MockedParser implements Parser {
     };
   }
 
+  getResolveTypeAnnotationFN(): ResolveTypeAnnotationFN {
+    return () => {
+      return {
+        nullable: false,
+        typeAnnotation: null,
+        typeResolutionStatus: {successful: false},
+      };
+    };
+  }
+
   getResolvedTypeAnnotation(
     typeAnnotation: $FlowFixMe,
     types: TypeDeclarationMap,

packages/react-native-codegen/src/parsers/parsers-commons.js

@@ -336,9 +336,11 @@ function buildPropertySchema(
         : property.typeAnnotation;
   }
 
-  ({nullable, typeAnnotation: value} = parser.getResolvedTypeAnnotation(
+  const resolveTypeAnnotationFN = parser.getResolveTypeAnnotationFN();
+  ({nullable, typeAnnotation: value} = resolveTypeAnnotationFN(
     value,
     types,
+    parser,
   ));
 
   throwIfModuleTypeIsUnsupported(

packages/react-native-codegen/src/parsers/typescript/modules/index.js

@@ -29,6 +29,7 @@ const {flattenIntersectionType} = require('../parseTopLevelType');
 const {flattenProperties} = require('../components/componentsUtils');
 
 const {parseObjectProperty} = require('../../parsers-commons');
+const {resolveTypeAnnotation} = require('../utils');
 
 const {
   emitArrayType,
@@ -189,6 +190,7 @@ function translateTypeAnnotation(
 ): Nullable<NativeModuleTypeAnnotation> {
   const {nullable, typeAnnotation, typeResolutionStatus} =
     parser.getResolvedTypeAnnotation(typeScriptTypeAnnotation, types);
+  resolveTypeAnnotation(typeScriptTypeAnnotation, types, parser);
 
   switch (typeAnnotation.type) {
     case 'TSArrayType': {

packages/react-native-codegen/src/parsers/typescript/parser.js

@@ -22,7 +22,12 @@ import type {
   NativeModuleEnumMap,
 } from '../../CodegenSchema';
 import type {ParserType} from '../errors';
-import type {GetSchemaInfoFN, GetTypeAnnotationFN, Parser} from '../parser';
+import type {
+  GetSchemaInfoFN,
+  GetTypeAnnotationFN,
+  Parser,
+  ResolveTypeAnnotationFN,
+} from '../parser';
 import type {
   ParserErrorCapturer,
   TypeDeclarationMap,
@@ -46,7 +51,7 @@ const {
   getSchemaInfo,
   getTypeAnnotation,
 } = require('./components/componentsUtils');
-
+const {resolveTypeAnnotation} = require('./utils');
 const fs = require('fs');
 
 const {
@@ -55,6 +60,7 @@ const {
 
 class TypeScriptParser implements Parser {
   typeParameterInstantiation: string = 'TSTypeParameterInstantiation';
+  typeAlias: string = 'TSTypeAliasDeclaration';
 
   isProperty(property: $FlowFixMe): boolean {
     return property.type === 'TSPropertySignature';
@@ -437,6 +443,10 @@ class TypeScriptParser implements Parser {
       typeResolutionStatus,
     };
   }
+
+  getResolveTypeAnnotationFN(): ResolveTypeAnnotationFN {
+    return resolveTypeAnnotation;
+  }
 }
 
 module.exports = {