GenerateDtosMenuAction.java

package de.jigp.plugin.actions.dto;

import com.intellij.openapi.actionSystem.DataContext;
import com.intellij.psi.PsiClass;
import de.jigp.plugin.GeneratorPluginContext;
import de.jigp.plugin.actions.menu.DetermineTargetClassChooser;
import de.jigp.plugin.actions.menu.HandleAnnotatedClassAction;

public class GenerateDtosMenuAction extends HandleAnnotatedClassAction {

    protected void handleClass(DataContext dataContext, PsiClass annotatedClass, String targetClassSuffix) {
        new DtoGenerator(dataContext, annotatedClass, targetClassSuffix).build();
    }

    protected DetermineTargetClassChooser getTargetClassChooser(DataContext dataContext) {
        return new DtoTargetClassChooser(dataContext);
    }

    protected String getAnnotationName() {
        return GeneratorPluginContext.getConfiguration().dtoAnnotation;
    }


}