1 files changed, 52 insertions, 0 deletions
diff --git a/src/main/java/org/elasticsearch/search/fetch/FieldsParseElement.java b/src/main/java/org/elasticsearch/search/fetch/FieldsParseElement.java
new file mode 100644
index 0000000..140d60d
--- /dev/null
+++ b/src/main/java/org/elasticsearch/search/fetch/FieldsParseElement.java
@@ -0,0 +1,52 @@
+/*
+ * Licensed to Elasticsearch under one or more contributor
+ * license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright
+ * ownership. Elasticsearch licenses this file to you under
+ * the Apache License, Version 2.0 (the "License"); you may
+ * not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *    http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+package org.elasticsearch.search.fetch;
+
+import org.elasticsearch.common.xcontent.XContentParser;
+import org.elasticsearch.script.SearchScript;
+import org.elasticsearch.search.SearchParseElement;
+import org.elasticsearch.search.fetch.script.ScriptFieldsContext;
+import org.elasticsearch.search.internal.SearchContext;
+
+/**
+ *
+ */
+public class FieldsParseElement implements SearchParseElement {
+
+    @Override
+    public void parse(XContentParser parser, SearchContext context) throws Exception {
+        XContentParser.Token token = parser.currentToken();
+        if (token == XContentParser.Token.START_ARRAY) {
+            boolean added = false;
+            while ((token = parser.nextToken()) != XContentParser.Token.END_ARRAY) {
+                String name = parser.text();
+                added = true;
+                context.fieldNames().add(name);
+            }
+            if (!added) {
+                context.emptyFieldNames();
+            }
+        } else if (token == XContentParser.Token.VALUE_STRING) {
+            String name = parser.text();
+            context.fieldNames().add(name);
+
+        }
+    }
+}