com.nhaarman.ellie.internal.codegen.step.ModelAdapterStep Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of compiler Show documentation
Compile-time active record ORM for Android.
The newest version!
/*
 * Copyright (C) 2014 Michael Pardo
 * Copyright (C) 2014 Niek Haarman
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.nhaarman.ellie.internal.codegen.step;

import com.nhaarman.ellie.annotation.Column;
import com.nhaarman.ellie.annotation.GetterFor;
import com.nhaarman.ellie.annotation.SetterFor;
import com.nhaarman.ellie.annotation.Table;
import com.nhaarman.ellie.internal.codegen.Registry;
import com.nhaarman.ellie.internal.codegen.element.ColumnElement;
import com.nhaarman.ellie.internal.codegen.element.ModelElement;
import com.nhaarman.ellie.internal.codegen.validator.ColumnValidator;
import com.nhaarman.ellie.internal.codegen.validator.GetterForValidator;
import com.nhaarman.ellie.internal.codegen.validator.ModelAdapterValidator;
import com.nhaarman.ellie.internal.codegen.validator.SetterForValidator;
import com.nhaarman.ellie.internal.codegen.writer.ModelAdapterWriter;
import com.nhaarman.ellie.internal.codegen.writer.SourceWriter;

import java.io.IOException;
import java.io.Writer;
import java.util.List;
import java.util.Set;

import javax.annotation.processing.Filer;
import javax.annotation.processing.RoundEnvironment;
import javax.lang.model.element.Element;
import javax.lang.model.element.ExecutableElement;
import javax.lang.model.element.TypeElement;
import javax.lang.model.element.VariableElement;
import javax.lang.model.util.Elements;
import javax.tools.JavaFileObject;

@SuppressWarnings("rawtypes")
public class ModelAdapterStep implements ProcessingStep {

    private final Registry mRegistry;
    private final Elements mElements;
    private final Filer mFiler;
    private final ModelAdapterValidator mModelAdapterValidator;
    private final ColumnValidator mColumnValidator;
    private final GetterForValidator mGetterForValidator;
    private final SetterForValidator mSetterForValidator;
    private final SourceWriter mSourceWriter;

    public ModelAdapterStep(final Registry registry) {
        mRegistry = registry;
        mElements = registry.getElements();
        mFiler = registry.getFiler();
        mModelAdapterValidator = new ModelAdapterValidator(registry);
        mGetterForValidator = new GetterForValidator(registry);
        mSetterForValidator = new SetterForValidator(registry);
        mColumnValidator = new ColumnValidator(registry);
        mSourceWriter = new ModelAdapterWriter(registry);
    }

    @Override
    public void process(final Set annotations, final RoundEnvironment roundEnv) {
        Set tableElements = roundEnv.getElementsAnnotatedWith(Table.class);
        for (Element tableElement : tableElements) {
            if (mModelAdapterValidator.validate(tableElement.getEnclosingElement(), tableElement)) {
                mRegistry.addModelElement(new ModelElement((TypeElement) tableElement));

                addColumnElements((TypeElement) tableElement);
                addAccessors((TypeElement) tableElement);

                try {
                    String name = mSourceWriter.createSourceName(tableElement);
                    JavaFileObject object = mFiler.createSourceFile(name, tableElement);
                    Writer writer = object.openWriter();
                    mSourceWriter.writeSource(writer, tableElement);
                    writer.flush();
                    writer.close();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        }
    }

    private void addColumnElements(final TypeElement element) {
        final List members = mElements.getAllMembers(element);

        for (Element member : members) {
            if (member.getAnnotation(Column.class) != null && mColumnValidator.validate(element, member)) {
                mRegistry.addColumnElement(new ColumnElement(mRegistry, element, (VariableElement) member));
            }
        }
    }

    private void addAccessors(final TypeElement element) {
        final List members = mElements.getAllMembers(element);

        for (Element member : members) {
            if (member.getAnnotation(GetterFor.class) != null && mGetterForValidator.validate(element, member)) {
                addGetter(element, member);
            } else if (member.getAnnotation(SetterFor.class) != null && mSetterForValidator.validate(element, member)) {
                addSetter(element, member);
            }
        }
    }

    private void addGetter(final TypeElement element, final Element member) {
        Set columnElements = mRegistry.getColumnElements(element);
        for (ColumnElement columnElement : columnElements) {
            if (columnElement.getColumnName().equals(member.getAnnotation(GetterFor.class).value())) {
                columnElement.setGetter((ExecutableElement) member);
            }
        }
    }

    private void addSetter(final TypeElement element, final Element member) {
        Set columnElements = mRegistry.getColumnElements(element);
        for (ColumnElement columnElement : columnElements) {
            if (columnElement.getColumnName().equals(member.getAnnotation(SetterFor.class).value())) {
                columnElement.setSetter((ExecutableElement) member);
            }
        }
    }
}