X-Git-Url: http://git.argeo.org/?a=blobdiff_plain;f=cms%2Forg.argeo.slc.support%2Fsrc%2Forg%2Fargeo%2Fslc%2Flib%2Flinux%2FScriptCall.java;fp=cms%2Forg.argeo.slc.support%2Fsrc%2Forg%2Fargeo%2Fslc%2Flib%2Flinux%2FScriptCall.java;h=755fa0cf83a18fd2701ab993223dcf39284cca5c;hb=ecc22e604e47533c79de9cecdcdeacbc752cbff1;hp=0000000000000000000000000000000000000000;hpb=e07ded4632e53f8b8869763bc1f1f4091361e76e;p=gpl%2Fargeo-slc.git

diff --git a/cms/org.argeo.slc.support/src/org/argeo/slc/lib/linux/ScriptCall.java b/cms/org.argeo.slc.support/src/org/argeo/slc/lib/linux/ScriptCall.java
new file mode 100644
index 000000000..755fa0cf8
--- /dev/null
+++ b/cms/org.argeo.slc.support/src/org/argeo/slc/lib/linux/ScriptCall.java
@@ -0,0 +1,104 @@
+package org.argeo.slc.lib.linux;
+
+import java.io.File;
+import java.io.FileOutputStream;
+import java.io.InputStream;
+import java.io.OutputStream;
+import java.util.ArrayList;
+import java.util.List;
+
+import org.apache.commons.io.FilenameUtils;
+import org.apache.commons.io.IOUtils;
+import org.argeo.slc.SlcException;
+import org.argeo.slc.core.execution.tasks.SystemCall;
+import org.springframework.beans.factory.InitializingBean;
+import org.springframework.core.io.Resource;
+
+/** Call to the interpreter of a script language. */
+public class ScriptCall extends SystemCall implements InitializingBean {
+	private Resource script;
+	private List<Object> scriptArgs = new ArrayList<Object>();
+
+	private Boolean localScriptCopy = false;
+
+	/** For use in Spring. */
+	public ScriptCall() {
+		super();
+	}
+
+	/** For use in code ({@link #init()} is called). */
+	public ScriptCall(Resource script) {
+		this.script = script;
+		init();
+	}
+
+	public void init() {
+		initInterpreter();
+		for (Object obj : scriptArgs) {
+			arg(obj.toString());
+		}
+	}
+
+	public void afterPropertiesSet() throws Exception {
+		init();
+	}
+
+	protected void initInterpreter() {
+		String ext = FilenameUtils.getExtension(script.getFilename());
+
+		if (localScriptCopy) {
+			File scriptFile = copyScript();
+			if ("sh".equals(ext))
+				arg("/bin/sh").arg(scriptFile.getAbsolutePath());
+			else if ("pl".equals(ext))
+				arg("/usr/bin/perl").arg(scriptFile.getAbsolutePath());
+			else if ("py".equals(ext))
+				arg("/usr/bin/python").arg(scriptFile.getAbsolutePath());
+			else
+				throw new SlcException(
+						"Cannot initialize script intepreter for " + script);
+		} else {
+			setStdInFile(script);
+			if ("sh".equals(ext))
+				arg("/bin/sh").arg("-s");
+			else if ("pl".equals(ext))
+				arg("/usr/bin/perl").arg("/dev/stdin");
+			else if ("py".equals(ext))
+				arg("/usr/bin/python").arg("-");
+			else
+				throw new SlcException(
+						"Cannot initialize script intepreter for " + script);
+		}
+	}
+
+	private File copyScript() {
+		InputStream in = null;
+		OutputStream out = null;
+		try {
+			File scriptFile = File.createTempFile("script", ".sh");
+			scriptFile.deleteOnExit();
+			in = script.getInputStream();
+			out = new FileOutputStream(scriptFile);
+			IOUtils.copy(in, out);
+			return scriptFile;
+		} catch (Exception e) {
+			throw new SlcException("Cannot copy " + script, e);
+		} finally {
+			IOUtils.closeQuietly(in);
+			IOUtils.closeQuietly(out);
+		}
+	}
+
+	public void setScript(Resource script) {
+		this.script = script;
+	}
+
+	public void setScriptArgs(List<Object> scriptArgs) {
+		this.scriptArgs = scriptArgs;
+	}
+
+	public void setLocalScriptCopy(Boolean localScriptCopy) {
+		this.localScriptCopy = localScriptCopy;
+	}
+
+}