Add support for Windows PowerShell

Add a new LanguageOption for PowerShell Updated TemplateUtil to support PowerShell Strings and Dictionaries Created a new template file for generating PowerShell scripts Created a new CodeTemplateBuilderTest to validate proper code generation

Author: mattpresson

scriptgen-core/src/main/java/com/h3xstream/scriptgen/LanguageOption.java

@@ -14,8 +14,9 @@ public class LanguageOption {
  public static final LanguageOption RUBY_NET_HTTP = new LanguageOption("Ruby (Net::HTTP)", "ruby", BASE_TPL_CLASSPATH + "ruby_nethttp.tpl", SyntaxConstants.SYNTAX_STYLE_RUBY , "rb");
  public static final LanguageOption PERL_LWP = new LanguageOption("Perl (LWP)", "perl", BASE_TPL_CLASSPATH + "perl_lwp.tpl", SyntaxConstants.SYNTAX_STYLE_PERL, "pl");
  public static final LanguageOption PHP_CURL = new LanguageOption("PHP (cURL)", "php", BASE_TPL_CLASSPATH + "php_curl.tpl", SyntaxConstants.SYNTAX_STYLE_PHP, "php");
+public static final LanguageOption POWERSHELL = new LanguageOption("PowerShell", "powershell", BASE_TPL_CLASSPATH + "psh_webrequest.tpl", SyntaxConstants.SYNTAX_STYLE_UNIX_SHELL, "ps1");
 
- public static final LanguageOption[] values = {PYTHON_REQUEST, RUBY_NET_HTTP, PERL_LWP, PHP_CURL};
+ public static final LanguageOption[] values = {PYTHON_REQUEST, RUBY_NET_HTTP, PERL_LWP, PHP_CURL, POWERSHELL};
 
  //Properties of each language
  private final String title;

scriptgen-core/src/main/java/com/h3xstream/scriptgen/template/TemplateUtil.java

@@ -4,13 +4,13 @@
 
 public class TemplateUtil {
 
- private String buildMap(Map<String,String> map,String start,String end,String entrySeparator) {
+ private String buildMap(Map<String,String> map,String start,String end,String keyValueSeparator, String entrySeparator) {
  StringBuilder buffer = new StringBuilder(start);
 
  int i=0;
  for(Map.Entry<String,String> e : map.entrySet()) {
- if(i++ != 0) buffer.append(",");
- buffer.append("\""+ pythonStr(e.getKey()) +entrySeparator + pythonStr(e.getValue())+"\"");
+ if(i++ != 0) buffer.append( entrySeparator );
+ buffer.append("\""+ pythonStr(e.getKey()) + keyValueSeparator + pythonStr(e.getValue())+"\"");
  }
 
  buffer.append(end);
@@ -54,7 +54,7 @@ else if( 32 <= ch && ch <= 126) { //Digits, alpha and most specials
  ///Python util method
 
  public String pythonDict(Map<String,String> map) {
- return buildMap(map,"{","}","\":\"");
+ return buildMap(map,"{","}","\":\"",",");
  }
 
  public String pythonStr(String value) {
@@ -64,7 +64,7 @@ public String pythonStr(String value) {
  //Ruby
 
  public String rubyMap(Map<String,String> map) {
- return buildMap(map,"{","}","\"=>\"");
+ return buildMap(map,"{","}","\"=>\"",",");
  }
 
  public String rubyStr(String value) {
@@ -74,7 +74,7 @@ public String rubyStr(String value) {
  //Perl
 
  public String perlMap(Map<String,String> map) {
- return buildMap(map,"{","}","\"=>\"");
+ return buildMap(map,"{","}","\"=>\"",",");
  }
 
  public String perlStr(String value) {
@@ -88,12 +88,12 @@ public String phpStr(String value) {
  }
 
  public String phpMap(Map<String,String> map) {
- return buildMap(map,"array(",")","\"=>\"");
+ return buildMap(map,"array(",")","\"=>\"",",");
  }
 
  public String phpHeadersList(Map<String,String> map) {
 
- return buildMap(map,"array(",")",": ");
+ return buildMap(map,"array(",")",": ",",");
  }
 
  public String phpUrlEncode(String value) {
@@ -114,4 +114,14 @@ public String phpCookies(Map<String,String> cookies) {
  }
  return str.toString();
  }
+
+// PowerShell
+ public String powershellStr(String value) {
+ return genericString(value);
+ }
+
+ public String powershellDict(Map<String,String> map) {
+ return buildMap(map,"@{","}","\"=\"","; ");
+ }
+
 }

scriptgen-core/src/main/resources/com/h3xstream/scriptgen/templates/psh_webrequest.tpl

@@ -0,0 +1,29 @@
+$uri = "${req.url}"
+$method = "${req.method}"
+
+<#if req.headers??>
+$headers = ${util.powershellDict(req.headers)}
+$headers.Remove("Proxy-Connection")
+</#if>
+
+<#if req.cookies??>
+$cc = New-Object System.Net.CookieContainer
+<#list req.cookies?keys as c>
+$cc.Add( $(New-Object Uri( $uri )), $(New-Object System.Net.Cookie("${util.powershellStr(c)}", "${util.powershellStr(req.cookies[c])}")) )
+</#list>
+$session = New-Object Microsoft.PowerShell.Commands.WebRequestSession 
+$session.Cookies = $cc 
+</#if>
+
+<#if req.parametersGet??>
+$paramsGet = ${util.powershellDict(req.parametersGet)}
+</#if>
+
+<#if req.parametersPost??>
+$paramsPost = ${util.powershellDict(req.parametersPost)}
+</#if>
+
+$response = Invoke-WebRequest -Method $method -Uri $uri<#if req.headers??> -Headers $headers</#if><#if req.cookies??> -WebSession $session</#if><#if req.parametersPost??> -Body $paramsPost</#if><#if req.parametersGet??> -Body $paramsGet</#if>
+
+Write-Host "Status code: $($response.StatusCode)" 
+Write-host "Response body: $($response.Content)"

scriptgen-core/src/test/java/com/h3xstream/scriptgen/template/CodeTemplateBuilderPowershellTest.java

@@ -0,0 +1,24 @@
+package com.h3xstream.scriptgen.template;
+
+import com.h3xstream.scriptgen.model.HttpRequestInfo;
+import com.h3xstream.scriptgen.HttpRequestInfoFixtures;
+import org.testng.annotations.Test;
+
+import static org.testng.Assert.assertTrue;
+
+public class CodeTemplateBuilderPowershellTest extends CodeTemplateBuilderBaseTest {
+
+
+ HttpRequestInfo reqGet = HttpRequestInfoFixtures.getGetRequest();
+ HttpRequestInfo reqPost = HttpRequestInfoFixtures.getPostRequest();
+
+ @Test
+ public void testGetTemplate() throws Exception {
+ testTemplateContains("com/h3xstream/scriptgen/templates/psh_webrequest.tpl","-Body $paramsGet",reqGet);
+ }
+
+ @Test
+ public void testPostTemplate() throws Exception {
+ testTemplateContains("com/h3xstream/scriptgen/templates/psh_webrequest.tpl","-Body $paramsPost",reqPost);
+ }
+}