nitro-codegen/lib/syntax/swift/SwiftVariant.js

The code-generator for react-native-nitro-modules.

import { indent } from '../../utils.js';
import { createFileMetadataString } from '../helpers.js';
import { getTypeAs } from '../types/getTypeAs.js';
import { OptionalType } from '../types/OptionalType.js';
function isPrimitive(type) {
    switch (type.kind) {
        case 'bigint':
        case 'boolean':
        case 'number':
        case 'string':
        case 'void':
        case 'result-wrapper':
        case 'null':
        case 'error':
            return true;
        case 'optional':
            const optional = getTypeAs(type, OptionalType);
            return isPrimitive(optional.wrappingType);
        default:
            return false;
    }
}
export function createSwiftVariant(variant) {
    const typename = variant.getAliasName('swift');
    const cases = variant.cases
        .map(([label, v]) => {
        const type = v.getCode('swift');
        return `case ${label}(${type})`;
    })
        .join('\n');
    const jsSignature = variant.variants.map((t) => t.kind).join(' | ');
    const allPrimitives = variant.variants.every((v) => isPrimitive(v));
    const enumDeclaration = allPrimitives ? 'enum' : 'indirect enum';
    const code = `
${createFileMetadataString(`${typename}.swift`)}

/**
 * An Swift enum with associated values representing a Variant/Union type.
 * JS type: \`${jsSignature}\`
 */
@frozen
public ${enumDeclaration} ${typename} {
  ${indent(cases, '  ')}
}
  `.trim();
    return {
        content: code,
        language: 'swift',
        name: `${typename}.swift`,
        platform: 'ios',
        subdirectory: [],
    };
}