FORGE-1176: Set default value

forge · Sep 24, 2013 · 2f31829 · 2f31829
1 parent da14948
commit 2f31829
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/...c/main/java/org/jboss/forge/addon/javaee/rest/generator/EntityBasedResourceGenerator.java b/...c/main/java/org/jboss/forge/addon/javaee/rest/generator/EntityBasedResourceGenerator.java
@@ -62,7 +62,7 @@ public List<JavaClass> generateFrom(RestGenerationContext context) throws Except
       map.put("orderClause", orderClause);
       map.put("resourcePath", resourcePath);
 
-      String output = processor.processTemplate(map, "org/jboss/forge/rest/Endpoint.jv");
+      String output = processor.processTemplate(map, "org/jboss/forge/addon/javaee/rest/Endpoint.jv");
       JavaClass resource = JavaParser.parse(JavaClass.class, output);
       resource.addImport(entity.getQualifiedName());
       resource.setPackage(context.getTargetPackageName());

diff --git a/...impl/src/main/java/org/jboss/forge/addon/javaee/rest/ui/RestEndpointFromEntityWizard.java b/...impl/src/main/java/org/jboss/forge/addon/javaee/rest/ui/RestEndpointFromEntityWizard.java
@@ -19,6 +19,7 @@
 import org.jboss.forge.addon.convert.Converter;
 import org.jboss.forge.addon.javaee.jpa.PersistenceFacet;
 import org.jboss.forge.addon.javaee.rest.generation.RestResourceGenerator;
+import org.jboss.forge.addon.javaee.rest.generator.EntityBasedResourceGenerator;
 import org.jboss.forge.addon.javaee.rest.generator.RestGenerationContextImpl;
 import org.jboss.forge.addon.javaee.ui.AbstractJavaEECommand;
 import org.jboss.forge.addon.parser.java.facets.JavaSourceFacet;
@@ -72,6 +73,9 @@ public class RestEndpointFromEntityWizard extends AbstractJavaEECommand
    @WithAttributes(label = "Overwrite existing classes?", enabled = false, defaultValue = "false")
    private UIInput<Boolean> overwrite;
 
+   @Inject
+   private EntityBasedResourceGenerator defaultResourceGenerator;
+
    @Override
    public UICommandMetadata getMetadata(UIContext context)
    {
@@ -111,6 +115,7 @@ public String convert(JavaClass source)
       packageName.setDefaultValue(javaSourceFacet.getBasePackage() + ".rest");
 
       contentType.setValueChoices(Arrays.asList(MediaType.APPLICATION_XML, MediaType.APPLICATION_JSON));
+      generator.setDefaultValue(defaultResourceGenerator);
       if (context.getProvider().isGUI())
       {
          generator.setItemLabelConverter(new Converter<RestResourceGenerator, String>()