flow-immutable-models/src/helpers/getter.js

Generates model classes from Flow types using Immutable.js

// @flow
import getTypeAnnotationWithoutModelType from './getTypeAnnotationWithoutModelType';

export default function getter(j: Object, prop: Object) {
  const func = j.functionExpression(
    null,
    [],
    j.blockStatement([
      j.returnStatement(
        j.callExpression(
          j.memberExpression(
            j.memberExpression(j.thisExpression(), j.identifier('_state')),
            j.identifier('get')
          ),
          [j.literal(prop.key.name)]
        )
      ),
    ])
  );
  let returnType = getTypeAnnotationWithoutModelType(j, prop.value, true);
  if (prop.optional && returnType.type !== 'NullableTypeAnnotation') {
    returnType = j.nullableTypeAnnotation(returnType);
  }
  func.returnType = j.typeAnnotation(returnType);

  return j.methodDefinition('get', j.identifier(prop.key.name), func);
}