@react-native/codegen 0.72.5 → 0.73.0-nightly-20230603-fd9e295be

package/lib/generators/components/GenerateEventEmitterCpp.js

@@ -10,9 +10,11 @@
 
 'use strict';
 
-const _require = require('./CppHelpers.js'),
+const _require = require('./CppHelpers'),
   generateEventStructName = _require.generateEventStructName;
-const FileTemplate = ({events, libraryName}) => `
+const _require2 = require('../Utils'),
+  indent = _require2.indent;
+const FileTemplate = ({events, libraryName, extraIncludes}) => `
 /**
  * This code was generated by [react-native-codegen](https://www.npmjs.com/package/react-native-codegen).
  *
@@ -23,12 +25,11 @@ const FileTemplate = ({events, libraryName}) => `
  */
 
 #include <react/renderer/components/${libraryName}/EventEmitters.h>
+${[...extraIncludes].join('\n')}
 
 namespace facebook {
 namespace react {
-
 ${events}
-
 } // namespace react
 } // namespace facebook
 `;
@@ -39,16 +40,16 @@ const ComponentTemplate = ({
   dispatchEventName,
   implementation,
 }) => {
-  const capture = implementation.includes('event')
-    ? 'event=std::move(event)'
+  const capture = implementation.includes('$event')
+    ? '$event=std::move($event)'
     : '';
   return `
-void ${className}EventEmitter::${eventName}(${structName} event) const {
+void ${className}EventEmitter::${eventName}(${structName} $event) const {
   dispatchEvent("${dispatchEventName}", [${capture}](jsi::Runtime &runtime) {
     ${implementation}
   });
 }
-`.trim();
+`;
 };
 const BasicComponentTemplate = ({className, eventName, dispatchEventName}) =>
   `
@@ -56,84 +57,261 @@ void ${className}EventEmitter::${eventName}() const {
   dispatchEvent("${dispatchEventName}");
 }
 `.trim();
-function generateSetter(variableName, propertyName, propertyParts) {
-  const trailingPeriod = propertyParts.length === 0 ? '' : '.';
-  const eventChain = `event.${propertyParts.join(
-    '.',
-  )}${trailingPeriod}${propertyName});`;
-  return `${variableName}.setProperty(runtime, "${propertyName}", ${eventChain}`;
+function generateSetter(
+  variableName,
+  propertyName,
+  propertyParts,
+  usingEvent,
+  valueMapper = value => value,
+) {
+  const eventChain = usingEvent
+    ? `$event.${[...propertyParts, propertyName].join('.')}`
+    : [propertyParts, propertyName].join('.');
+  return `${variableName}.setProperty(runtime, "${propertyName}", ${valueMapper(
+    eventChain,
+  )});`;
+}
+function generateObjectSetter(
+  variableName,
+  propertyName,
+  propertyParts,
+  typeAnnotation,
+  extraIncludes,
+  usingEvent,
+) {
+  return `
+{
+  auto ${propertyName} = jsi::Object(runtime);
+  ${indent(
+    generateSetters(
+      propertyName,
+      typeAnnotation.properties,
+      propertyParts.concat([propertyName]),
+      extraIncludes,
+      usingEvent,
+    ),
+    2,
+  )}
+  ${variableName}.setProperty(runtime, "${propertyName}", ${propertyName});
+}
+`.trim();
 }
-function generateEnumSetter(variableName, propertyName, propertyParts) {
-  const trailingPeriod = propertyParts.length === 0 ? '' : '.';
-  const eventChain = `event.${propertyParts.join(
-    '.',
-  )}${trailingPeriod}${propertyName})`;
-  return `${variableName}.setProperty(runtime, "${propertyName}", toString(${eventChain});`;
+function setValueAtIndex(
+  propertyName,
+  indexVariable,
+  loopLocalVariable,
+  mappingFunction = value => value,
+) {
+  return `${propertyName}.setValueAtIndex(runtime, ${indexVariable}++, ${mappingFunction(
+    loopLocalVariable,
+  )});`;
 }
-function generateSetters(parentPropertyName, properties, propertyParts) {
+function generateArraySetter(
+  variableName,
+  propertyName,
+  propertyParts,
+  elementType,
+  extraIncludes,
+  usingEvent,
+) {
+  const eventChain = usingEvent
+    ? `$event.${[...propertyParts, propertyName].join('.')}`
+    : [propertyParts, propertyName].join('.');
+  const indexVar = `${propertyName}Index`;
+  const innerLoopVar = `${propertyName}Value`;
+  return `
+    auto ${propertyName} = jsi::Array(runtime, ${eventChain}.size());
+    size_t ${indexVar} = 0;
+    for (auto ${innerLoopVar} : ${eventChain}) {
+      ${handleArrayElementType(
+        elementType,
+        propertyName,
+        indexVar,
+        innerLoopVar,
+        propertyParts,
+        extraIncludes,
+        usingEvent,
+      )}
+    }
+    ${variableName}.setProperty(runtime, "${propertyName}", ${propertyName});
+  `;
+}
+function handleArrayElementType(
+  elementType,
+  propertyName,
+  indexVariable,
+  loopLocalVariable,
+  propertyParts,
+  extraIncludes,
+  usingEvent,
+) {
+  switch (elementType.type) {
+    case 'BooleanTypeAnnotation':
+      return setValueAtIndex(
+        propertyName,
+        indexVariable,
+        loopLocalVariable,
+        val => `(bool)${val}`,
+      );
+    case 'StringTypeAnnotation':
+    case 'Int32TypeAnnotation':
+    case 'DoubleTypeAnnotation':
+    case 'FloatTypeAnnotation':
+      return setValueAtIndex(propertyName, indexVariable, loopLocalVariable);
+    case 'MixedTypeAnnotation':
+      return setValueAtIndex(
+        propertyName,
+        indexVariable,
+        loopLocalVariable,
+        val => `jsi::valueFromDynamic(runtime, ${val})`,
+      );
+    case 'StringEnumTypeAnnotation':
+      return setValueAtIndex(
+        propertyName,
+        indexVariable,
+        loopLocalVariable,
+        val => `toString(${val})`,
+      );
+    case 'ObjectTypeAnnotation':
+      return convertObjectTypeArray(
+        propertyName,
+        indexVariable,
+        loopLocalVariable,
+        propertyParts,
+        elementType,
+        extraIncludes,
+      );
+    case 'ArrayTypeAnnotation':
+      return convertArrayTypeArray(
+        propertyName,
+        indexVariable,
+        loopLocalVariable,
+        propertyParts,
+        elementType,
+        extraIncludes,
+        usingEvent,
+      );
+    default:
+      throw new Error(
+        `Received invalid elementType for array ${elementType.type}`,
+      );
+  }
+}
+function convertObjectTypeArray(
+  propertyName,
+  indexVariable,
+  loopLocalVariable,
+  propertyParts,
+  objectTypeAnnotation,
+  extraIncludes,
+) {
+  return `auto ${propertyName}Object = jsi::Object(runtime);
+      ${generateSetters(
+        `${propertyName}Object`,
+        objectTypeAnnotation.properties,
+        [].concat([loopLocalVariable]),
+        extraIncludes,
+        false,
+      )}
+      ${setValueAtIndex(propertyName, indexVariable, `${propertyName}Object`)}`;
+}
+function convertArrayTypeArray(
+  propertyName,
+  indexVariable,
+  loopLocalVariable,
+  propertyParts,
+  eventTypeAnnotation,
+  extraIncludes,
+  usingEvent,
+) {
+  if (eventTypeAnnotation.type !== 'ArrayTypeAnnotation') {
+    throw new Error(
+      `Inconsistent eventTypeAnnotation received. Expected type = 'ArrayTypeAnnotation'; received = ${eventTypeAnnotation.type}`,
+    );
+  }
+  return `auto ${propertyName}Array = jsi::Array(runtime, ${loopLocalVariable}.size());
+      size_t ${indexVariable}Internal = 0;
+      for (auto ${loopLocalVariable}Internal : ${loopLocalVariable}) {
+        ${handleArrayElementType(
+          eventTypeAnnotation.elementType,
+          `${propertyName}Array`,
+          `${indexVariable}Internal`,
+          `${loopLocalVariable}Internal`,
+          propertyParts,
+          extraIncludes,
+          usingEvent,
+        )}
+      }
+      ${setValueAtIndex(propertyName, indexVariable, `${propertyName}Array`)}`;
+}
+function generateSetters(
+  parentPropertyName,
+  properties,
+  propertyParts,
+  extraIncludes,
+  usingEvent = true,
+) {
   const propSetters = properties
     .map(eventProperty => {
       const typeAnnotation = eventProperty.typeAnnotation;
       switch (typeAnnotation.type) {
         case 'BooleanTypeAnnotation':
-          return generateSetter(
-            parentPropertyName,
-            eventProperty.name,
-            propertyParts,
-          );
         case 'StringTypeAnnotation':
+        case 'Int32TypeAnnotation':
+        case 'DoubleTypeAnnotation':
+        case 'FloatTypeAnnotation':
           return generateSetter(
             parentPropertyName,
             eventProperty.name,
             propertyParts,
+            usingEvent,
           );
-        case 'Int32TypeAnnotation':
+        case 'MixedTypeAnnotation':
+          extraIncludes.add('#include <jsi/JSIDynamic.h>');
           return generateSetter(
             parentPropertyName,
             eventProperty.name,
             propertyParts,
+            usingEvent,
+            prop => `jsi::valueFromDynamic(runtime, ${prop})`,
           );
-        case 'DoubleTypeAnnotation':
+        case 'StringEnumTypeAnnotation':
           return generateSetter(
             parentPropertyName,
             eventProperty.name,
             propertyParts,
+            usingEvent,
+            prop => `toString(${prop})`,
           );
-        case 'FloatTypeAnnotation':
-          return generateSetter(
+        case 'ObjectTypeAnnotation':
+          return generateObjectSetter(
             parentPropertyName,
             eventProperty.name,
             propertyParts,
+            typeAnnotation,
+            extraIncludes,
+            usingEvent,
           );
-        case 'StringEnumTypeAnnotation':
-          return generateEnumSetter(
+        case 'ArrayTypeAnnotation':
+          return generateArraySetter(
             parentPropertyName,
             eventProperty.name,
             propertyParts,
+            typeAnnotation.elementType,
+            extraIncludes,
+            usingEvent,
           );
-        case 'ObjectTypeAnnotation':
-          const propertyName = eventProperty.name;
-          return `
-            {
-              auto ${propertyName} = jsi::Object(runtime);
-              ${generateSetters(
-                propertyName,
-                typeAnnotation.properties,
-                propertyParts.concat([propertyName]),
-              )}
-
-              ${parentPropertyName}.setProperty(runtime, "${propertyName}", ${propertyName});
-            }
-          `.trim();
         default:
           typeAnnotation.type;
-          throw new Error('Received invalid event property type');
+          throw new Error(
+            `Received invalid event property type ${typeAnnotation.type}`,
+          );
       }
     })
     .join('\n');
   return propSetters;
 }
-function generateEvent(componentName, event) {
+function generateEvent(componentName, event, extraIncludes) {
   // This is a gross hack necessary because native code is sending
   // events named things like topChange to JS which is then converted back to
   // call the onChange prop. We should be consistent throughout the system.
@@ -145,9 +323,14 @@ function generateEvent(componentName, event) {
   )}`;
   if (event.typeAnnotation.argument) {
     const implementation = `
-    auto payload = jsi::Object(runtime);
-    ${generateSetters('payload', event.typeAnnotation.argument.properties, [])}
-    return payload;
+    auto $payload = jsi::Object(runtime);
+    ${generateSetters(
+      '$payload',
+      event.typeAnnotation.argument.properties,
+      [],
+      extraIncludes,
+    )}
+    return $payload;
   `.trim();
     if (!event.name.startsWith('on')) {
       throw new Error('Expected the event name to start with `on`');
@@ -183,20 +366,20 @@ module.exports = {
       })
       .filter(Boolean)
       .reduce((acc, components) => Object.assign(acc, components), {});
-    const fileName = 'EventEmitters.cpp';
+    const extraIncludes = new Set();
     const componentEmitters = Object.keys(moduleComponents)
       .map(componentName => {
         const component = moduleComponents[componentName];
         return component.events
-          .map(event => {
-            return generateEvent(componentName, event);
-          })
+          .map(event => generateEvent(componentName, event, extraIncludes))
           .join('\n');
       })
       .join('\n');
+    const fileName = 'EventEmitters.cpp';
     const replacedTemplate = FileTemplate({
       libraryName,
       events: componentEmitters,
+      extraIncludes,
     });
     return new Map([[fileName, replacedTemplate]]);
   },