SerializableSaveStatementWriter.java

/*
 * Copyright 2015 Workday, Inc.
 *
 * This software is available under the MIT license.
 * Please see the LICENSE.txt file in this project.
 */

package com.workday.postman.codegen;

import com.squareup.javawriter.JavaWriter;
import com.workday.meta.MetaTypes;

import java.io.IOException;
import java.io.Serializable;
import java.util.Collection;

import javax.lang.model.element.ExecutableElement;
import javax.lang.model.element.VariableElement;

/**
 * @author nathan.taylor
 * @since 2013-12-30
 */
class SerializableSaveStatementWriter
        implements SaveStatementWriter {

    private final MetaTypes metaTypes;

    public SerializableSaveStatementWriter(MetaTypes metaTypes) {
        this.metaTypes = metaTypes;
    }

    @Override
    public boolean isApplicable(VariableElement field) {
        return metaTypes.isSubtype(field.asType(), Serializable.class);
    }

    @Override
    public void writeFieldReadStatement(VariableElement field,
                                        Collection<ExecutableElement> postCreateChildMethods,
                                        JavaWriter writer)
            throws IOException {
        writer.emitStatement("object.%s = (%s) bundle.getSerializable(\"%s\")",
                             field.getSimpleName(),
                             field.asType().toString(),
                             field.getSimpleName());
        for (ExecutableElement method : postCreateChildMethods) {
            writer.emitStatement("object.%s(object.%s)",
                                 method.getSimpleName(),
                                 field.getSimpleName());
        }
    }

    @Override
    public void writeFieldWriteStatement(VariableElement field, JavaWriter writer)
            throws IOException {
        writer.emitStatement("bundle.putSerializable(\"%s\", object.%s)",
                             field.getSimpleName(),
                             field.getSimpleName());
    }
}