/*
* Copyright 2013 Red Hat, Inc. and/or its affiliates.
*
* Licensed under the Eclipse Public License version 1.0, available at
* http://www.eclipse.org/legal/epl-v10.html
*/
package org.jboss.forge.addon.javaee.jpa.ui;
import java.io.FileNotFoundException;
import java.util.ArrayList;
import java.util.List;
import javax.inject.Inject;
import javax.persistence.Entity;
import javax.persistence.GenerationType;
import org.jboss.forge.addon.javaee.jpa.JPAFacet;
import org.jboss.forge.addon.javaee.jpa.PersistenceOperations;
import org.jboss.forge.addon.javaee.jpa.ui.setup.JPASetupWizard;
import org.jboss.forge.addon.javaee.ui.AbstractJavaEECommand;
import org.jboss.forge.addon.parser.java.facets.JavaSourceFacet;
import org.jboss.forge.addon.parser.java.resources.JavaResource;
import org.jboss.forge.addon.parser.java.resources.JavaResourceVisitor;
import org.jboss.forge.addon.projects.Project;
import org.jboss.forge.addon.projects.facets.MetadataFacet;
import org.jboss.forge.addon.resource.DirectoryResource;
import org.jboss.forge.addon.resource.FileResource;
import org.jboss.forge.addon.resource.visit.VisitContext;
import org.jboss.forge.addon.ui.command.PrerequisiteCommandsProvider;
import org.jboss.forge.addon.ui.command.UICommand;
import org.jboss.forge.addon.ui.context.UIBuilder;
import org.jboss.forge.addon.ui.context.UIContext;
import org.jboss.forge.addon.ui.context.UIExecutionContext;
import org.jboss.forge.addon.ui.context.UISelection;
import org.jboss.forge.addon.ui.hints.InputType;
import org.jboss.forge.addon.ui.input.UIInput;
import org.jboss.forge.addon.ui.input.UISelectOne;
import org.jboss.forge.addon.ui.metadata.WithAttributes;
import org.jboss.forge.addon.ui.result.Result;
import org.jboss.forge.addon.ui.result.Results;
import org.jboss.forge.addon.ui.util.Categories;
import org.jboss.forge.addon.ui.util.Metadata;
import org.jboss.forge.parser.java.JavaSource;
/**
* @author <a href="mailto:lincolnbaxter@gmail.com">Lincoln Baxter, III</a>
*/
public class NewEntityCommand extends AbstractJavaEECommand implements PrerequisiteCommandsProvider
{
@Inject
@WithAttributes(label = "Entity name", description = "The simple name of the generated class", required = true)
private UIInput<String> named;
@Inject
@WithAttributes(label = "Target package", type = InputType.JAVA_PACKAGE_PICKER)
private UIInput<String> targetPackage;
@Inject
@WithAttributes(label = "ID Column Generation Strategy", defaultValue = "AUTO")
private UISelectOne<GenerationType> idStrategy;
@Inject
@WithAttributes(label = "Target Directory", required = true)
private UIInput<DirectoryResource> targetLocation;
@Inject
@WithAttributes(label = "Table Name")
private UIInput<String> tableName;
@Inject
private PersistenceOperations persistenceOperations;
@Override
public Metadata getMetadata(UIContext context)
{
return Metadata.from(super.getMetadata(context), getClass()).name("JPA: New Entity")
.description("Create a new JPA Entity")
.category(Categories.create(super.getMetadata(context).getCategory(), "JPA"));
}
@Override
public void initializeUI(UIBuilder builder) throws Exception
{
idStrategy.setDefaultValue(GenerationType.AUTO);
Project project = getSelectedProject(builder.getUIContext());
if (project == null)
{
UISelection<FileResource<?>> currentSelection = builder.getUIContext().getInitialSelection();
if (!currentSelection.isEmpty())
{
FileResource<?> resource = currentSelection.get();
if (resource instanceof DirectoryResource)
{
targetLocation.setDefaultValue((DirectoryResource) resource);
}
else
{
targetLocation.setDefaultValue(resource.getParent());
}
}
}
else if (project.hasFacet(JavaSourceFacet.class))
{
JavaSourceFacet facet = project.getFacet(JavaSourceFacet.class);
targetLocation.setDefaultValue(facet.getSourceDirectory()).setEnabled(false);
targetPackage.setValue(calculateModelPackage(project));
}
builder.add(targetLocation);
builder.add(targetPackage).add(named).add(idStrategy).add(tableName);
}
/**
* @param project
* @return
*/
private String calculateModelPackage(Project project)
{
final String[] value = new String[1];
project.getFacet(JavaSourceFacet.class).visitJavaSources(new JavaResourceVisitor()
{
@Override
public void visit(VisitContext context, JavaResource javaResource)
{
try
{
JavaSource<?> javaSource = javaResource.getJavaSource();
if (javaSource.hasAnnotation(Entity.class))
{
value[0] = javaSource.getPackage();
}
}
catch (FileNotFoundException ignore)
{
}
}
});
if (value[0] == null)
{
value[0] = project.getFacet(MetadataFacet.class).getTopLevelPackage() + ".model";
}
return value[0];
}
@Override
public Result execute(UIExecutionContext context) throws Exception
{
String entityName = named.getValue();
String entityPackage = targetPackage.getValue();
GenerationType idStrategyChosen = idStrategy.getValue();
if (idStrategyChosen == null)
{
idStrategyChosen = GenerationType.AUTO;
}
DirectoryResource targetDir = targetLocation.getValue();
Project project = getSelectedProject(context);
JavaResource javaResource;
if (project == null)
{
javaResource = persistenceOperations.newEntity(targetDir, entityName, entityPackage, idStrategyChosen,
tableName.getValue());
}
else
{
javaResource = persistenceOperations.newEntity(project, entityName, entityPackage, idStrategyChosen,
tableName.getValue());
}
context.getUIContext().setSelection(javaResource);
return Results.success("Entity " + javaResource + " created");
}
@Override
protected boolean isProjectRequired()
{
return false;
}
@Override
public List<Class<? extends UICommand>> getPrerequisiteCommands(UIContext context)
{
List<Class<? extends UICommand>> setup = new ArrayList<>();
Project project = getSelectedProject(context);
if (project != null)
{
if (!project.hasFacet(JPAFacet.class))
{
setup.add(JPASetupWizard.class);
}
}
return setup;
}
}