added stamp function

pdf/src/main/java/org/expath/exist/pdf/ExistExpathPdfModule.java

@@ -28,6 +28,7 @@
 import org.exist.xquery.FunctionDef;
 import org.expath.exist.pdf.formControls.GetTextFieldsFunction;
 import org.expath.exist.pdf.formControls.SetTextFieldsFunction;
+import org.expath.exist.pdf.stamp.StampFunction;
 
 import ro.kuberam.libs.java.pdf.ModuleDescription;
 
@@ -45,7 +46,8 @@ public class ExistExpathPdfModule extends AbstractInternalModule {
 
 	private final static FunctionDef[] functions = {
 			new FunctionDef(GetTextFieldsFunction.signature, GetTextFieldsFunction.class),
-			new FunctionDef(SetTextFieldsFunction.signature, SetTextFieldsFunction.class) };
+			new FunctionDef(SetTextFieldsFunction.signature, SetTextFieldsFunction.class),
+			new FunctionDef(StampFunction.signature, StampFunction.class) };
 
 	public ExistExpathPdfModule(Map<String, List<? extends Object>> parameters) throws Exception {
 		super(functions, parameters);

pdf/src/main/java/org/expath/exist/pdf/stamp/StampFunction.java

@@ -0,0 +1,111 @@
+/*
+ *  Copyright (C) 2014
+ *
+ *  This program is free software; you can redistribute it and/or
+ *  modify it under the terms of the GNU Lesser General Public License
+ *  as published by the Free Software Foundation; either version 2
+ *  of the License, or (at your option) any later version.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU Lesser General Public License for more details.
+ *
+ *  You should have received a copy of the GNU Lesser General Public
+ *  License along with this library; if not, write to the Free Software
+ *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+ *
+ *  $Id$
+ */
+
+package org.expath.exist.pdf.stamp;
+
+/**
+ * Implements the pdf:set-text-fields() function for eXist.
+ * 
+ * @author Claudius Teodorescu <claudius.teodorescu@gmail.com>
+ */
+
+import java.io.ByteArrayInputStream;
+import java.io.ByteArrayOutputStream;
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Map.Entry;
+
+import org.apache.log4j.Logger;
+import org.exist.dom.QName;
+import org.exist.xquery.BasicFunction;
+import org.exist.xquery.Cardinality;
+import org.exist.xquery.FunctionSignature;
+import org.exist.xquery.XPathException;
+import org.exist.xquery.XQueryContext;
+import org.exist.xquery.functions.map.AbstractMapType;
+import org.exist.xquery.value.AtomicValue;
+import org.exist.xquery.value.Base64BinaryValueType;
+import org.exist.xquery.value.BinaryValue;
+import org.exist.xquery.value.BinaryValueFromInputStream;
+import org.exist.xquery.value.FunctionParameterSequenceType;
+import org.exist.xquery.value.FunctionReturnSequenceType;
+import org.exist.xquery.value.Sequence;
+import org.exist.xquery.value.SequenceType;
+import org.exist.xquery.value.Type;
+import org.exist.xquery.value.ValueSequence;
+import org.expath.exist.pdf.ExistExpathPdfModule;
+
+import ro.kuberam.libs.java.pdf.stamp.Stamper;
+
+public class StampFunction extends BasicFunction {
+
+	private final static Logger log = Logger.getLogger(Stamper.class);
+
+	public final static FunctionSignature signature = new FunctionSignature(new QName("stamp",
+			ExistExpathPdfModule.NAMESPACE_URI, ExistExpathPdfModule.PREFIX),
+			"Set a stamp on a PDF document.",
+				new SequenceType[] { new FunctionParameterSequenceType("contents", Type.BASE64_BINARY,
+					Cardinality.ZERO_OR_ONE, "PDF contents where to get the text fields from."),
+				new FunctionParameterSequenceType("stamp", Type.STRING, Cardinality.EXACTLY_ONE,
+						"the stamp text"),
+				new FunctionParameterSequenceType("stamp-selector", Type.STRING, Cardinality.EXACTLY_ONE,
+						"the selector for the stamp"),
+				new FunctionParameterSequenceType("stamp-style", Type.STRING, Cardinality.EXACTLY_ONE,
+						"the style for the stamp as CSS text")},
+			new FunctionReturnSequenceType(Type.BASE64_BINARY, Cardinality.ZERO_OR_ONE,
+					"The updated document."));
+
+	public StampFunction(XQueryContext context, FunctionSignature signature) {
+		super(context, signature);
+	}
+
+	@Override
+	public Sequence eval(Sequence[] args, Sequence contextSequence) throws XPathException {
+		byte[] binary = (byte[]) ((BinaryValue) args[0].itemAt(0)).toJavaObject(byte[].class);
+		String stampText = (String) args[1].getStringValue();
+		String stampSelector = (String) args[2].getStringValue();
+		String stampStyle = (String) args[3].getStringValue();
+		//AbstractMapType fieldsXqueryMap = (AbstractMapType) args[2].itemAt(0);
+		//Map<String, String> fieldsMap = new HashMap<String, String>();
+
+		/*for (final Entry<AtomicValue, Sequence> entry : fieldsXqueryMap) {
+			fieldsMap.put(entry.getKey().getStringValue(), entry.getValue().getStringValue());
+		}*/
+
+		Sequence result = new ValueSequence();
+
+		ByteArrayOutputStream resultAsStreamResult = new ByteArrayOutputStream();
+
+		try {
+			BinaryValue data = BinaryValueFromInputStream.getInstance(context, new Base64BinaryValueType(),
+					new ByteArrayInputStream(binary));
+
+			resultAsStreamResult = Stamper.run(data.getInputStream(), stampText, stampSelector, stampStyle);
+
+			result.add(BinaryValueFromInputStream.getInstance(context, new Base64BinaryValueType(),
+					new ByteArrayInputStream(resultAsStreamResult.toByteArray())));
+
+		} catch (Exception ex) {
+			throw new XPathException(ex.getMessage());
+		}
+
+		return result;
+	}
+}