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

Generates model classes from Flow types using Immutable.js

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

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

  return j.methodDefinition('method', j.identifier(`set${capitalize(propName)}`), func);
}