com.grosner.processor.handler.ModelViewHandler Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of DBFlow-Compiler Show documentation

DBFlow

The newest version!

package com.grosner.processor.handler;

import com.grosner.dbflow.annotation.ModelView;
import com.grosner.processor.definition.ModelViewDefinition;
import com.grosner.processor.model.ProcessorManager;
import com.squareup.javawriter.JavaWriter;

import javax.annotation.processing.RoundEnvironment;
import javax.lang.model.element.Element;
import java.io.IOException;

/**
 * Author: andrewgrosner
 * Contributors: { }
 * Description:
 */
public class ModelViewHandler extends BaseContainerHandler {

    @Override
    protected void onProcessElement(ProcessorManager processorManager, Element element) {
        ModelViewDefinition modelViewDefinition = new ModelViewDefinition(processorManager, element);

        try {
            JavaWriter javaWriter = new JavaWriter(processorManager.getProcessingEnvironment()
                    .getFiler().createSourceFile(modelViewDefinition.getSourceFileName()).openWriter());
            modelViewDefinition.write(javaWriter);

            processorManager.addModelViewDefinition(modelViewDefinition);

            javaWriter.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    @Override
    protected Class getAnnotationClass() {
        return ModelView.class;
    }
}