nitro-codegen/src/syntax/getReferencedTypes.ts

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

import { ArrayType } from './types/ArrayType.js'
import { FunctionType } from './types/FunctionType.js'
import { getTypeAs } from './types/getTypeAs.js'
import { OptionalType } from './types/OptionalType.js'
import { PromiseType } from './types/PromiseType.js'
import { RecordType } from './types/RecordType.js'
import { StructType } from './types/StructType.js'
import { TupleType } from './types/TupleType.js'
import type { Type } from './types/Type.js'
import { VariantType } from './types/VariantType.js'

export function getReferencedTypes(type: Type): Type[] {
  switch (type.kind) {
    case 'array':
      const array = getTypeAs(type, ArrayType)
      return [type, ...getReferencedTypes(array.itemType)]

    case 'function':
      const func = getTypeAs(type, FunctionType)
      return [
        type,
        ...getReferencedTypes(func.returnType),
        ...func.parameters.flatMap((t) => getReferencedTypes(t)),
      ]

    case 'optional':
      const optional = getTypeAs(type, OptionalType)
      return [type, ...getReferencedTypes(optional.wrappingType)]

    case 'promise':
      const promise = getTypeAs(type, PromiseType)
      return [type, ...getReferencedTypes(promise.resultingType)]

    case 'record':
      const record = getTypeAs(type, RecordType)
      return [
        type,
        ...getReferencedTypes(record.keyType),
        ...getReferencedTypes(record.valueType),
      ]

    case 'struct':
      const struct = getTypeAs(type, StructType)
      return [type, ...struct.properties.flatMap((p) => getReferencedTypes(p))]

    case 'tuple':
      const tuple = getTypeAs(type, TupleType)
      return [type, ...tuple.itemTypes.flatMap((t) => getReferencedTypes(t))]

    case 'variant':
      const variant = getTypeAs(type, VariantType)
      return [type, ...variant.variants.flatMap((t) => getReferencedTypes(t))]

    default:
      return [type]
  }
}