flow-immutable-models/lib/helpers/getTypeAnnotationWithoutModelType.js.flow

Generates model classes from Flow types using Immutable.js

// @flow

import { withoutModelTypeSuffix } from './withoutModelTypeSuffix';

export default function getTypeAnnotationWithoutModelType(
  j: Object,
  value: Object,
  convertToImmutable: boolean = false
): Object {
  const type: string = value.type;

  switch (type) {
    case 'GenericTypeAnnotation': {
      if (value.id.type === 'Identifier') {
        const nameWithoutSuffix = withoutModelTypeSuffix(value.id.name);
        let typeId;
        if (nameWithoutSuffix === 'Array' && convertToImmutable) {
          typeId = j.qualifiedTypeIdentifier(j.identifier('Immutable'), j.identifier('List'));
        } else {
          typeId = j.identifier(nameWithoutSuffix);
        }
        return j.genericTypeAnnotation(
          typeId,
          value.typeParameters
            ? j.typeParameterInstantiation(
                value.typeParameters.params.map(typeParam =>
                  getTypeAnnotationWithoutModelType(j, typeParam)
                )
              )
            : null
        );
      }
      if (value.id.type === 'QualifiedTypeIdentifier') {
        return j.genericTypeAnnotation(
          j.qualifiedTypeIdentifier(
            value.id.qualification,
            j.identifier(withoutModelTypeSuffix(value.id.id.name))
          ),
          j.typeParameterInstantiation(
            value.typeParameters.params.map(typeParam =>
              getTypeAnnotationWithoutModelType(j, typeParam)
            )
          )
        );
      }
      return value;
    }
    case 'NullableTypeAnnotation':
      return j.nullableTypeAnnotation(
        getTypeAnnotationWithoutModelType(j, value.typeAnnotation, convertToImmutable)
      );
    case 'ObjectTypeAnnotation':
      if (value.indexers.length > 0) {
        const propIndexer = value.indexers[0];
        return j.genericTypeAnnotation(
          j.qualifiedTypeIdentifier(j.identifier('Immutable'), j.identifier('Map')),
          j.typeParameterInstantiation([
            getTypeAnnotationWithoutModelType(j, propIndexer.key),
            getTypeAnnotationWithoutModelType(j, propIndexer.value, true),
          ])
        );
      }
      return value;
    default:
      return value;
  }
}