You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

241 lines
5.8 KiB

/**
* Copyright (c) Meta Platforms, Inc. and affiliates.
*
* This source code is licensed under the MIT license found in the
* LICENSE file in the root directory of this source tree.
*
* @format
* @flow strict
*/
'use strict';
import type {
SchemaType,
NativeModuleSchema,
NativeModuleTypeAnnotation,
Nullable,
NativeModuleMixedTypeAnnotation,
UnionTypeAnnotationMemberType,
NativeModuleUnionTypeAnnotation,
} from '../CodegenSchema.js';
const {
MissingTypeParameterGenericParserError,
MoreThanOneTypeParameterGenericParserError,
UnsupportedUnionTypeAnnotationParserError,
} = require('./errors');
import type {ParserType} from './errors';
const {
UnsupportedObjectPropertyTypeAnnotationParserError,
} = require('./errors');
const invariant = require('invariant');
import type {TypeDeclarationMap} from './utils';
const {
UnsupportedEnumDeclarationParserError,
UnsupportedGenericParserError,
} = require('./errors');
import type {Parser} from './parser';
import type {NativeModuleEnumDeclaration} from '../CodegenSchema';
function wrapModuleSchema(
nativeModuleSchema: NativeModuleSchema,
hasteModuleName: string,
): SchemaType {
return {
modules: {
[hasteModuleName]: nativeModuleSchema,
},
};
}
function unwrapNullable<+T: NativeModuleTypeAnnotation>(
x: Nullable<T>,
): [T, boolean] {
if (x.type === 'NullableTypeAnnotation') {
return [x.typeAnnotation, true];
}
return [x, false];
}
function wrapNullable<+T: NativeModuleTypeAnnotation>(
nullable: boolean,
typeAnnotation: T,
): Nullable<T> {
if (!nullable) {
return typeAnnotation;
}
return {
type: 'NullableTypeAnnotation',
typeAnnotation,
};
}
function assertGenericTypeAnnotationHasExactlyOneTypeParameter(
moduleName: string,
/**
* TODO(T108222691): Use flow-types for @babel/parser
*/
typeAnnotation: $FlowFixMe,
language: ParserType,
) {
if (typeAnnotation.typeParameters == null) {
throw new MissingTypeParameterGenericParserError(
moduleName,
typeAnnotation,
language,
);
}
const typeAnnotationType =
language === 'TypeScript'
? 'TSTypeParameterInstantiation'
: 'TypeParameterInstantiation';
invariant(
typeAnnotation.typeParameters.type === typeAnnotationType,
`assertGenericTypeAnnotationHasExactlyOneTypeParameter: Type parameters must be an AST node of type '${typeAnnotationType}'`,
);
if (typeAnnotation.typeParameters.params.length !== 1) {
throw new MoreThanOneTypeParameterGenericParserError(
moduleName,
typeAnnotation,
language,
);
}
}
function emitMixedTypeAnnotation(
nullable: boolean,
): Nullable<NativeModuleMixedTypeAnnotation> {
return wrapNullable(nullable, {
type: 'MixedTypeAnnotation',
});
}
function remapUnionTypeAnnotationMemberNames(
types: $FlowFixMe,
language: ParserType,
): UnionTypeAnnotationMemberType[] {
const remapLiteral = (item: $FlowFixMe) => {
if (language === 'Flow') {
return item.type
.replace('NumberLiteralTypeAnnotation', 'NumberTypeAnnotation')
.replace('StringLiteralTypeAnnotation', 'StringTypeAnnotation');
}
return item.literal
? item.literal.type
.replace('NumericLiteral', 'NumberTypeAnnotation')
.replace('StringLiteral', 'StringTypeAnnotation')
: 'ObjectTypeAnnotation';
};
return types
.map(remapLiteral)
.filter((value, index, self) => self.indexOf(value) === index);
}
function emitUnionTypeAnnotation(
nullable: boolean,
hasteModuleName: string,
typeAnnotation: $FlowFixMe,
language: ParserType,
): Nullable<NativeModuleUnionTypeAnnotation> {
const unionTypes = remapUnionTypeAnnotationMemberNames(
typeAnnotation.types,
language,
);
// Only support unionTypes of the same kind
if (unionTypes.length > 1) {
throw new UnsupportedUnionTypeAnnotationParserError(
hasteModuleName,
typeAnnotation,
unionTypes,
language,
);
}
return wrapNullable(nullable, {
type: 'UnionTypeAnnotation',
memberType: unionTypes[0],
});
}
function translateDefault(
hasteModuleName: string,
typeAnnotation: $FlowFixMe,
types: TypeDeclarationMap,
nullable: boolean,
parser: Parser,
): Nullable<NativeModuleEnumDeclaration> {
const maybeEnumDeclaration =
types[parser.nameForGenericTypeAnnotation(typeAnnotation)];
if (maybeEnumDeclaration && parser.isEnumDeclaration(maybeEnumDeclaration)) {
const memberType = parser.getMaybeEnumMemberType(maybeEnumDeclaration);
if (
memberType === 'NumberTypeAnnotation' ||
memberType === 'StringTypeAnnotation'
) {
return wrapNullable(nullable, {
type: 'EnumDeclaration',
memberType: memberType,
});
} else {
throw new UnsupportedEnumDeclarationParserError(
hasteModuleName,
typeAnnotation,
memberType,
parser.language(),
);
}
}
throw new UnsupportedGenericParserError(
hasteModuleName,
typeAnnotation,
parser,
);
}
function getKeyName(
propertyOrIndex: $FlowFixMe,
hasteModuleName: string,
language: ParserType,
): string {
switch (propertyOrIndex.type) {
case 'ObjectTypeProperty':
case 'TSPropertySignature':
return propertyOrIndex.key.name;
case 'ObjectTypeIndexer':
// flow index name is optional
return propertyOrIndex.id?.name ?? 'key';
case 'TSIndexSignature':
// TypeScript index name is mandatory
return propertyOrIndex.parameters[0].name;
default:
throw new UnsupportedObjectPropertyTypeAnnotationParserError(
hasteModuleName,
propertyOrIndex,
propertyOrIndex.type,
language,
);
}
}
module.exports = {
wrapModuleSchema,
unwrapNullable,
wrapNullable,
assertGenericTypeAnnotationHasExactlyOneTypeParameter,
emitMixedTypeAnnotation,
emitUnionTypeAnnotation,
getKeyName,
translateDefault,
};