@react-native/codegen
Version:
Code generation tools for React Native
215 lines (200 loc) • 7.22 kB
JavaScript
/**
* 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
*/
'use strict';
const _require = require('../Utils'),
createAliasResolver = _require.createAliasResolver,
getModules = _require.getModules;
const _require2 = require('./header/serializeStruct'),
serializeStruct = _require2.serializeStruct;
const _require3 = require('./serializeEventEmitter'),
EventEmitterHeaderTemplate = _require3.EventEmitterHeaderTemplate;
const _require4 = require('./serializeMethod'),
serializeMethod = _require4.serializeMethod;
const _require5 = require('./source/serializeModule'),
serializeModuleSource = _require5.serializeModuleSource;
const _require6 = require('./StructCollector'),
StructCollector = _require6.StructCollector;
const ModuleDeclarationTemplate = ({
hasteModuleName,
structDeclarations,
eventEmitters,
protocolMethods,
}) => `${structDeclarations}
@protocol ${hasteModuleName}Spec <RCTBridgeModule, RCTTurboModule>
${protocolMethods}
@end
@interface ${hasteModuleName}SpecBase : NSObject {
@protected
facebook::react::EventEmitterCallback _eventEmitterCallback;
}
- (void)setEventEmitterCallback:(EventEmitterCallbackWrapper *)eventEmitterCallbackWrapper;
${eventEmitters}
@end
namespace facebook::react {
/**
* ObjC++ class for module '${hasteModuleName}'
*/
class JSI_EXPORT ${hasteModuleName}SpecJSI : public ObjCTurboModule {
public:
${hasteModuleName}SpecJSI(const ObjCTurboModule::InitParams ¶ms);
};
} // namespace facebook::react`;
const HeaderFileTemplate = ({
headerFileName,
moduleDeclarations,
structInlineMethods,
assumeNonnull,
}) => {
const headerFileNameWithNoExt = headerFileName.replace(/\.h$/, '');
return (
`/**
* This code was generated by [react-native-codegen](https://www.npmjs.com/package/react-native-codegen).
*
* Do not edit this file as changes may cause incorrect behavior and will be lost
* once the code is regenerated.
*
* ${'@'}generated by codegen project: GenerateModuleObjCpp
*
* We create an umbrella header (and corresponding implementation) here since
* Cxx compilation in BUCK has a limitation: source-code producing genrule()s
* must have a single output. More files => more genrule()s => slower builds.
*/
// Avoid multiple includes of ${headerFileNameWithNoExt} symbols
` +
(assumeNonnull ? '\nNS_ASSUME_NONNULL_BEGIN\n' : '') +
moduleDeclarations +
'\n' +
structInlineMethods +
(assumeNonnull ? '\nNS_ASSUME_NONNULL_END\n' : '\n') +
`
'\n'
);
};
const SourceFileTemplate = ({headerFileName, moduleImplementations}) => `/**
* This code was generated by [react-native-codegen](https://www.npmjs.com/package/react-native-codegen).
*
* Do not edit this file as changes may cause incorrect behavior and will be lost
* once the code is regenerated.
*
* ${'@'}generated by codegen project: GenerateModuleObjCpp
*
* We create an umbrella header (and corresponding implementation) here since
* Cxx compilation in BUCK has a limitation: source-code producing genrule()s
* must have a single output. More files => more genrule()s => slower builds.
*/
${moduleImplementations}
`;
module.exports = {
generate(libraryName, schema, packageName, assumeNonnull, headerPrefix) {
const nativeModules = getModules(schema);
const moduleDeclarations = [];
const structInlineMethods = [];
const moduleImplementations = [];
const hasteModuleNames = Object.keys(nativeModules).sort();
for (const hasteModuleName of hasteModuleNames) {
const _nativeModules$hasteM = nativeModules[hasteModuleName],
aliasMap = _nativeModules$hasteM.aliasMap,
excludedPlatforms = _nativeModules$hasteM.excludedPlatforms,
spec = _nativeModules$hasteM.spec;
if (excludedPlatforms != null && excludedPlatforms.includes('iOS')) {
continue;
}
const resolveAlias = createAliasResolver(aliasMap);
const structCollector = new StructCollector();
const methodSerializations = [];
const serializeProperty = property => {
methodSerializations.push(
...serializeMethod(
hasteModuleName,
property,
structCollector,
resolveAlias,
),
);
};
/**
* Note: As we serialize NativeModule methods, we insert structs into
* StructCollector, as we encounter them.
*/
spec.methods
.filter(property => property.name !== 'getConstants')
.forEach(serializeProperty);
spec.methods
.filter(property => property.name === 'getConstants')
.forEach(serializeProperty);
const generatedStructs = structCollector.getAllStructs();
const structStrs = [];
const methodStrs = [];
for (const struct of generatedStructs) {
const _serializeStruct = serializeStruct(hasteModuleName, struct),
methods = _serializeStruct.methods,
declaration = _serializeStruct.declaration;
structStrs.push(declaration);
methodStrs.push(methods);
}
moduleDeclarations.push(
ModuleDeclarationTemplate({
hasteModuleName: hasteModuleName,
structDeclarations: structStrs.join('\n'),
eventEmitters: spec.eventEmitters
.map(eventEmitter => EventEmitterHeaderTemplate(eventEmitter))
.join('\n'),
protocolMethods: methodSerializations
.map(({protocolMethod}) => protocolMethod)
.join('\n'),
}),
);
structInlineMethods.push(methodStrs.join('\n'));
moduleImplementations.push(
serializeModuleSource(
hasteModuleName,
generatedStructs,
hasteModuleName,
spec.eventEmitters,
methodSerializations.filter(
({selector}) => selector !== '@selector(constantsToExport)',
),
),
);
}
const headerFileName = `${libraryName}.h`;
const headerFile = HeaderFileTemplate({
headerFileName,
moduleDeclarations: moduleDeclarations.join('\n'),
structInlineMethods: structInlineMethods.join('\n'),
assumeNonnull,
});
const sourceFileName = `${libraryName}-generated.mm`;
const sourceFile = SourceFileTemplate({
headerFileName,
moduleImplementations: moduleImplementations.join('\n'),
});
return new Map([
[headerFileName, headerFile],
[sourceFileName, sourceFile],
]);
},
};