Merge pull request #2 from rundeck/script-plugins

different templates for script generator

Author: ltamaster

README.md

@@ -8,17 +8,32 @@ Download the tar or zip distribution, cd to the bin directory, and run:
 
 ./rundeck-plugin-bootstrap -n MyRundeckPlugin -t java -s Notification -d /tmp
 
-A Java notification plugin will be created at /tmp/myrundeckplugin. You can cd into that directory,
-run `gradle build` and you will have an installable notification plugin that you can put in your Rundeck installation.
+A Java/Script notification plugin will be created at /tmp/myrundeckplugin. You can cd into that directory,
+run `gradle build` and you will have an installable plugin that you can put in your Rundeck installation.
 
-* Note: At this time only Notification plugins can be bootstrapped with this utility.
 
-Other examples: 
+## Existing service plugins enabled on boostrap-plugin
+
+Java Plugins:
+* ResourceModelSource
+* Notification
+* WorkflowStep
+* WorkflowNodeStep
+
+Script Plugins:
+* ResourceModelSource
+* WorkflowNodeStep
+* RemoteScriptNodeStep
+* NodeExecutor
+* FileCopier
+* NodeExecutorFileCopier: Generate both, Node Executor and File Copier service 
+
+## Other Examples: 
 
 Create a script plugin:
 
-./rundeck-plugin-bootstrap -n MyNodeExecutorPlugin -t script -s NodeExecutor /tmp
+./rundeck-plugin-bootstrap -n MyNodeExecutorPlugin -t script -s NodeExecutor -d /tmp
 
 Create a UI script plugin:
 
-./rundeck-plugin-bootstrap -n MyUIPlugin -t ui -s UI /tmp
+./rundeck-plugin-bootstrap -n MyUIPlugin -t ui -s UI -d /tmp

src/main/groovy/com/rundeck/plugin/generator/JavaPluginTemplateGenerator.groovy

@@ -24,7 +24,7 @@ class JavaPluginTemplateGenerator extends AbstractTemplateGenerator {
     private static final String TEMPLATE_BASE = "templates/java-plugin/"
     private static final String JAVA_STRUCTURE = "java-plugin.structure"
 
-    private static final List ALLOWED_TEMPLATES = ["ResourceModel","Notification","WorkflowStep","WorkflowNodeStep"]
+    private static final List ALLOWED_TEMPLATES = ["ResourceModelSource","Notification","WorkflowStep","WorkflowNodeStep"]
 
     @Override
     Map makeTemplateProperties(final String pluginName, final String providedService) {

src/main/groovy/com/rundeck/plugin/generator/ScriptPluginTemplateGenerator.groovy

@@ -24,7 +24,7 @@ class ScriptPluginTemplateGenerator extends AbstractTemplateGenerator {
     private static final String TEMPLATE_BASE = "templates/script-plugin/"
     private static final String SCRIPT_STRUCTURE = "script-plugin.structure"
 
-    private static final List<String> ALLOWED_SERVICE_TYPES = ["NodeExecutor","FileCopier","ResourceModelSource","WorkflowNodeStep","RemoteScriptNodeStep"]
+    private static final List<String> ALLOWED_SERVICE_TYPES = ["NodeExecutor","FileCopier","ResourceModelSource","WorkflowNodeStep","RemoteScriptNodeStep","NodeExecutorFileCopier"]
 
     @Override
     Map makeTemplateProperties(final String pluginName, final String providedService) {
@@ -39,7 +39,8 @@ class ScriptPluginTemplateGenerator extends AbstractTemplateGenerator {
 
     @Override
     URL getPluginStructure(final String providedService) {
-        return getClass().getClassLoader().getResource("${TEMPLATE_BASE}${SCRIPT_STRUCTURE}")
+        String path = this.getTemplatePath(providedService)
+        return getClass().getClassLoader().getResource("${TEMPLATE_BASE}${path}/${SCRIPT_STRUCTURE}")
     }
 
     @Override
@@ -56,6 +57,25 @@ class ScriptPluginTemplateGenerator extends AbstractTemplateGenerator {
 
     @Override
     String resolveTemplateName(final String providedService, final String templateName) {
-        return "${getTemplateBase()}$templateName"
+        String path = this.getTemplatePath(providedService)
+        return "${getTemplateBase()}${path}/$templateName"
+    }
+
+    String getTemplatePath(final String providedService){
+        String path=""
+        switch (providedService){
+            case ["WorkflowNodeStep","RemoteScriptNodeStep"] :
+                path="workflow"
+                break
+            case ["FileCopier","NodeExecutor","ResourceModelSource"]:
+                path=providedService.toLowerCase()
+                break
+            case "NodeExecutorFileCopier":
+                path="nodeexecutor-filecopier"
+                break
+
+        }
+
+        return path
     }
 }