Added action for adding and deleting indexed scripts

jest-common/pom.xml

@@ -68,6 +68,11 @@
             <artifactId>junit</artifactId>
         </dependency>
 
+        <dependency>
+            <groupId>org.easytesting</groupId>
+            <artifactId>fest-assert</artifactId>
+        </dependency>
+
         <dependency>
             <groupId>commons-io</groupId>
             <artifactId>commons-io</artifactId>

jest-common/src/main/java/io/searchbox/indices/script/CreateIndexedScript.java

@@ -0,0 +1,107 @@
+package io.searchbox.indices.script;
+
+import com.google.gson.JsonElement;
+import com.google.gson.JsonObject;
+import io.searchbox.action.AbstractAction;
+import io.searchbox.action.GenericResultAbstractAction;
+
+import java.io.*;
+import java.net.URLEncoder;
+import java.nio.charset.Charset;
+
+import static io.searchbox.indices.script.ScriptLanguage.GROOVY;
+import static java.nio.charset.Charset.defaultCharset;
+
+public class CreateIndexedScript extends GenericResultAbstractAction {
+
+  private final String scriptName;
+  private final ScriptLanguage scriptLanguage;
+
+  public CreateIndexedScript(Builder builder) {
+    super(builder);
+    this.payload = builder.payload;
+    this.scriptName = builder.scriptName;
+    this.scriptLanguage = builder.scriptLanguage;
+    setURI(buildURI());
+  }
+
+  protected String buildURI() {
+    try {
+      return super.buildURI() + "/_scripts/" + scriptLanguage.pathParameterName + "/" + URLEncoder.encode(scriptName, CHARSET);
+    } catch (UnsupportedEncodingException e) {
+      throw new RuntimeException(e);
+    }
+  }
+
+  @Override
+  public String getRestMethodName() {
+    return "POST";
+  }
+
+  public String getScriptName() {
+    return scriptName;
+  }
+
+  public ScriptLanguage getScriptLanguage() {
+    return scriptLanguage;
+  }
+
+  public static class Builder extends AbstractAction.Builder<CreateIndexedScript, Builder> {
+
+    private String scriptName;
+    private JsonElement payload;
+    private ScriptLanguage scriptLanguage = GROOVY;
+
+    public Builder(String scriptName) {
+      this.scriptName = scriptName;
+    }
+
+    @Override
+    public CreateIndexedScript build() {
+      return new CreateIndexedScript(this);
+    }
+
+    public Builder setSource(String source) {
+      createPayload(source);
+      return this;
+    }
+
+    public Builder loadSource(File srcFile) throws IOException {
+      return loadSource(srcFile, defaultCharset());
+    }
+
+    public Builder loadSource(File srcFile, Charset encoding) throws IOException {
+      return loadSource(new FileInputStream(srcFile), encoding);
+    }
+
+    public Builder loadSource(InputStream srcStream) throws IOException {
+      return loadSource(srcStream, defaultCharset());
+    }
+
+    public Builder loadSource(InputStream srcStream, Charset encoding) throws IOException {
+      String src = readFully(srcStream, encoding);
+      createPayload(src);
+      return this;
+    }
+
+    public Builder setLanguage(ScriptLanguage language) {
+      this.scriptLanguage = language;
+      return this;
+    }
+
+    private void createPayload(String source) {
+      JsonObject jsonObject = new JsonObject();
+      jsonObject.addProperty("script", source);
+      payload = jsonObject;
+    }
+
+    private String readFully(InputStream srcStream, Charset encoding) throws IOException {
+      byte[] buf = new byte[8192];
+      StringBuilder sb = new StringBuilder();
+      for (int read; (read = srcStream.read(buf)) > 0; ) {
+        sb.append(new String(buf, 0, read, encoding));
+      }
+      return sb.toString();
+    }
+  }
+}

jest-common/src/main/java/io/searchbox/indices/script/DeleteIndexedScript.java

@@ -0,0 +1,64 @@
+package io.searchbox.indices.script;
+
+import io.searchbox.action.AbstractAction;
+import io.searchbox.action.GenericResultAbstractAction;
+
+import java.io.UnsupportedEncodingException;
+import java.net.URLEncoder;
+
+import static io.searchbox.indices.script.ScriptLanguage.GROOVY;
+
+public class DeleteIndexedScript extends GenericResultAbstractAction {
+
+  private final String scriptName;
+  private final ScriptLanguage scriptLanguage;
+
+  public DeleteIndexedScript(Builder builder) {
+    super(builder);
+    this.scriptName = builder.scriptName;
+    this.scriptLanguage = builder.scriptLanguage;
+    setURI(buildURI());
+  }
+
+  protected String buildURI() {
+    try {
+      return super.buildURI() + "/_scripts/" + scriptLanguage.pathParameterName + "/" + URLEncoder.encode(scriptName, CHARSET);
+    } catch (UnsupportedEncodingException e) {
+      throw new RuntimeException(e);
+    }
+  }
+
+  @Override
+  public String getRestMethodName() {
+    return "DELETE";
+  }
+
+  public String getScriptName() {
+    return scriptName;
+  }
+
+  public ScriptLanguage getScriptLanguage() {
+    return scriptLanguage;
+  }
+
+  public static class Builder extends AbstractAction.Builder<DeleteIndexedScript, Builder> {
+
+    private String scriptName;
+    private ScriptLanguage scriptLanguage = GROOVY;
+
+    public Builder(String scriptName) {
+      this.scriptName = scriptName;
+    }
+
+    public Builder setLanguage(ScriptLanguage scriptLanguage) {
+      this.scriptLanguage = scriptLanguage;
+      return this;
+    }
+
+    @Override
+    public DeleteIndexedScript build() {
+      return new DeleteIndexedScript(this);
+    }
+
+  }
+}

jest-common/src/main/java/io/searchbox/indices/script/ScriptLanguage.java

@@ -0,0 +1,22 @@
+package io.searchbox.indices.script;
+
+/**
+ * As described in Elasticsearch Reference
+ * <a href="https://www.elastic.co/guide/en/elasticsearch/reference/current/modules-scripting.html">
+ * https://www.elastic.co/guide/en/elasticsearch/reference/current/modules-scripting.html
+ * </a>
+ */
+public enum ScriptLanguage {
+  GROOVY("groovy"),
+  EXPRESSION("expression"),
+  MUSTACHE("mustache"),
+  MVEL("mvel"),
+  JAVASCRIPT("javascript"),
+  PYTHON("python");
+
+  public final String pathParameterName;
+
+  ScriptLanguage(String pathParameterName) {
+    this.pathParameterName = pathParameterName;
+  }
+}

jest-common/src/test/java/io/searchbox/indices/script/CreateIndexedScriptTest.java

@@ -0,0 +1,86 @@
+package io.searchbox.indices.script;
+
+import com.google.gson.Gson;
+import com.google.gson.JsonObject;
+import com.google.gson.JsonParser;
+import org.junit.Before;
+import org.junit.Test;
+
+import java.io.File;
+import java.io.FileWriter;
+import java.io.IOException;
+
+import static io.searchbox.indices.script.ScriptLanguage.GROOVY;
+import static io.searchbox.indices.script.ScriptLanguage.JAVASCRIPT;
+import static org.fest.assertions.Assertions.assertThat;
+
+public class CreateIndexedScriptTest {
+
+  private static final String A_NAME = "a_name";
+  private CreateIndexedScript script;
+  private CreateIndexedScript.Builder builder;
+  private String groovysnippet;
+
+  @Before
+  public void setUp() throws Exception {
+    builder = new CreateIndexedScript.Builder(A_NAME).setLanguage(JAVASCRIPT);
+    script = builder.build();
+    groovysnippet = "def test_a=123\n" +
+        "def test_b=\"$test_a\"\n";
+  }
+
+  @Test
+  public void defaultScriptingLanguageIsGroovy() throws Exception {
+    CreateIndexedScript script = new CreateIndexedScript.Builder(A_NAME).build();
+
+    assertThat(script.getScriptLanguage()).isEqualTo(GROOVY);
+    assertThat(script.buildURI()).contains(GROOVY.pathParameterName);
+  }
+
+  @Test
+  public void scriptingLanguageIsSetIntoPath() throws Exception {
+
+    assertThat(script.buildURI()).contains("/_scripts/" + JAVASCRIPT.pathParameterName + "/");
+  }
+
+  @Test
+  public void nameOfTheScriptIsSetIntoPath() throws Exception {
+
+    assertThat(script.buildURI()).contains("/_scripts/" + JAVASCRIPT.pathParameterName + "/" + A_NAME);
+  }
+
+  @Test
+  public void scriptSourceIsValidJsonString() throws Exception {
+    builder.setSource(groovysnippet);
+
+    script = builder.build();
+
+    JsonObject jsonPayload = parseAsGson(script.getData(new Gson()));
+    assertThat(jsonPayload.get("script")).isNotNull();
+    assertThat(jsonPayload.get("script").getAsString()).isEqualTo(groovysnippet);
+  }
+
+  @Test
+  public void fileSourceIsValidJsonString() throws Exception {
+    builder.loadSource(createTempGroovySnippetFile());
+
+    script = builder.build();
+
+    JsonObject jsonPayload = parseAsGson(script.getData(new Gson()));
+    assertThat(jsonPayload.get("script")).isNotNull();
+    assertThat(jsonPayload.get("script").getAsString()).isEqualTo(groovysnippet);
+  }
+
+  private File createTempGroovySnippetFile() throws IOException {
+    File file = File.createTempFile("test", ".groovy");
+    file.deleteOnExit();
+    FileWriter writer = new FileWriter(file);
+    writer.write(groovysnippet);
+    writer.close();
+    return file;
+  }
+
+  private JsonObject parseAsGson(String data) {
+    return new JsonParser().parse(data).getAsJsonObject();
+  }
+}

jest-common/src/test/java/io/searchbox/indices/script/DeleteIndexedScriptTest.java

@@ -0,0 +1,41 @@
+package io.searchbox.indices.script;
+
+import org.junit.Before;
+import org.junit.Test;
+
+import static io.searchbox.indices.script.ScriptLanguage.GROOVY;
+import static io.searchbox.indices.script.ScriptLanguage.JAVASCRIPT;
+import static org.fest.assertions.Assertions.assertThat;
+
+public class DeleteIndexedScriptTest {
+
+  private static final String A_NAME = "a_name";
+  private DeleteIndexedScript script;
+
+  @Before
+  public void setUp() throws Exception {
+    DeleteIndexedScript.Builder builder = new DeleteIndexedScript.Builder(A_NAME).setLanguage(JAVASCRIPT);
+    script = builder.build();
+  }
+
+  @Test
+  public void defaultScriptingLanguageIsGroovy() throws Exception {
+    DeleteIndexedScript script = new DeleteIndexedScript.Builder(A_NAME).build();
+
+    assertThat(script.getScriptLanguage()).isEqualTo(GROOVY);
+    assertThat(script.buildURI()).contains(GROOVY.pathParameterName);
+  }
+
+  @Test
+  public void scriptingLanguageIsSetIntoPath() throws Exception {
+
+    assertThat(script.buildURI()).contains("/_scripts/" + JAVASCRIPT.pathParameterName + "/");
+  }
+
+  @Test
+  public void nameOfTheScriptIsSetIntoPath() throws Exception {
+
+    assertThat(script.buildURI()).contains("/_scripts/" + JAVASCRIPT.pathParameterName + "/" + A_NAME);
+  }
+
+}

pom.xml

@@ -89,6 +89,7 @@
         <commonsLang.version>3.4</commonsLang.version>
         <commons-io.version>2.4</commons-io.version>
         <junit.version>4.12</junit.version>
+        <fest-assert.version>1.4</fest-assert.version>
 
         <plugin.surefire.version>2.18.1</plugin.surefire.version>
     </properties>
@@ -255,6 +256,12 @@
                 <version>${junit.version}</version>
                 <scope>test</scope>
             </dependency>
+            <dependency>
+                <groupId>org.easytesting</groupId>
+                <artifactId>fest-assert</artifactId>
+                <version>${fest-assert.version}</version>
+                <scope>test</scope>
+            </dependency>
 
             <dependency>
                 <groupId>commons-io</groupId>