Subversion Repositories XServices

/*

 *   Copyright 2010 Brian Rosenberger (Brutex Network)

 *

 *   Licensed 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 net.brutex.xservices.ws.impl;

import java.io.ByteArrayOutputStream;

import java.io.File;

import java.io.FileNotFoundException;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.util.List;

import javax.activation.DataHandler;

import javax.jws.WebMethod;

import javax.jws.WebParam;

import javax.jws.WebService;

import net.brutex.xservices.types.ReplacePattern;

import net.brutex.xservices.types.ReturnCode;

import net.brutex.xservices.types.ant.ArchiveResource;

import net.brutex.xservices.types.ant.AttachmentType;

import net.brutex.xservices.types.ant.FileResource;

import net.brutex.xservices.types.ant.FileSetResource;

import net.brutex.xservices.util.BrutexNamespaces;

import net.brutex.xservices.util.RunTask;

import net.brutex.xservices.ws.FileService;

import net.brutex.xservices.ws.XServicesFault;

import org.apache.cxf.aegis.type.mtom.StreamDataSource;

import org.apache.tools.ant.BuildException;

import org.apache.tools.ant.taskdefs.Basename;

import org.apache.tools.ant.taskdefs.Chmod;

import org.apache.tools.ant.taskdefs.Copy;

import org.apache.tools.ant.taskdefs.Echo;

import org.apache.tools.ant.taskdefs.LoadResource;

import org.apache.tools.ant.taskdefs.Replace;

import org.apache.tools.ant.taskdefs.optional.ReplaceRegExp;

import org.apache.tools.ant.taskdefs.optional.unix.Chgrp;

import org.apache.tools.ant.taskdefs.optional.unix.Chown;

import org.apache.tools.ant.types.FileSet;

/**

 *

 * @author Brian Rosenberger, bru@brutex.de

 */

@WebService(targetNamespace = BrutexNamespaces.WS_XSERVICES, endpointInterface = "net.brutex.xservices.ws.FileService", serviceName = "FileService")

public class FileServiceImpl implements FileService {

	/*

	 * (non-Javadoc)

	 *

	 * @see net.brutex.xservices.ws.impl.FileService#basename(java.lang.String,

	 * java.lang.String)

	 */

	public String basename(String filename, String suffix) {

		final String BASENAME_VALUE = "basename.value";

		Basename basename = new Basename();

		RunTask runner = new RunTask(basename);

		basename.setFile(new File(filename));

		if (suffix != null && !suffix.equals("")) {

			basename.setSuffix(suffix);

		}

		basename.setProperty(BASENAME_VALUE);

		ReturnCode r = runner.postTask();

		return r.getProperty(BASENAME_VALUE);

	}

	public ReturnCode replaceInFile(FileResource res, String search,

			String replace) throws XServicesFault {

		ReturnCode r = null;

		Replace rep = new Replace();

		rep.setTaskName("Replace");

		RunTask runner = new RunTask(rep);

		rep.addConfigured(res.getAntResource(rep.getProject()));

		rep.setToken(search);

		rep.setValue(replace);

		try {

			r = runner.postTask();

		} catch (BuildException e) {

			throw new XServicesFault(e);

		}

		return r;

	}

	public ReturnCode replaceInFile2(FileResource res,

			List<ReplacePattern> patternList) throws XServicesFault {

		ReturnCode r = null;

		for (ReplacePattern pat : patternList) {

			Replace rep = new Replace();

			rep.setTaskName("Replace");

			RunTask runner = new RunTask(rep);

			rep.addConfigured(res.getAntResource(rep.getProject()));

			rep.setToken(pat.search);

			rep.setValue(pat.replace);

			try {

				r = runner.postTask();

			} catch (BuildException e) {

				throw new XServicesFault(e);

			}

		}

		return r;

	}

	public ReturnCode replaceInFileRegEx(FileResource res, String search,

			String replace, String flags) throws XServicesFault {

		ReplaceRegExp rep = new ReplaceRegExp();

		rep.setTaskName("ReplaceRegExp");

		RunTask runner = new RunTask(rep);

		File infile = new File(res.uri);

		rep.setFile(infile);

		rep.setMatch(search);

		rep.setReplace(replace);

		rep.setFlags(flags);

		try {

			ReturnCode r = runner.postTask();

			return r;

		} catch (BuildException e) {

			throw new XServicesFault(e);

		}

	}

	/*

	 * (non-Javadoc)

	 *

	 * @see

	 * net.brutex.xservices.ws.impl.FileService#base64Encode(net.brutex.xservices

	 * .types.FileSetResource)

	 */

	public AttachmentType downloadFile(FileResource res) throws XServicesFault {

		InputStream is = null;

		try {

			is = res.getAntResource(null).getInputStream();

			StreamDataSource ssource = new StreamDataSource(

					"application/binary", is);

			DataHandler h = new DataHandler(ssource);

			AttachmentType t = new AttachmentType();

			t.setContent(h);

			t.setFilename(res.getAntResource(null).getName());

			return t;

		} catch (IOException e) {

			throw new XServicesFault(e);

		}

	}

	public byte[] encodeFile(FileResource res) throws XServicesFault {

		InputStream is = null;

		try {

			is = res.getAntResource(null).getInputStream();

		ByteArrayOutputStream buffer = new ByteArrayOutputStream();

		int nRead;

		byte[] data = new byte[4096];

		while ((nRead = is.read(data, 0, data.length)) != -1) {

		  buffer.write(data, 0, nRead);

		}

		buffer.flush();

		return buffer.toByteArray();

		} catch (IOException e) {

			throw new XServicesFault(e);

		}

	}

	/*

	 * (non-Javadoc)

	 *

	 * @see

	 * net.brutex.xservices.ws.impl.FileService#base64Decode(net.brutex.xservices

	 * .types.AttachmentType)

	 */

	public String uploadFile(AttachmentType file) throws XServicesFault {

		DataHandler h = file.getContent();

		File f = new File(file.getFilename());

		FileOutputStream fout;

		try {

			fout = new FileOutputStream(f);

			h.writeTo(fout);

			fout.flush();

			fout.close();

		} catch (FileNotFoundException e) {

			throw new XServicesFault(e);

		} catch (IOException e) {

			throw new XServicesFault(e);

		}

		return file.getFilename();

	}

	/*

	 * (non-Javadoc)

	 *

	 * @see

	 * net.brutex.xservices.ws.impl.FileService#copy(net.brutex.xservices.types

	 * .FileSetResource, java.lang.String, boolean, boolean, java.lang.String)

	 */

	public ReturnCode copy(FileSetResource src, String todir, boolean plm,

			boolean overwrite, String encoding) throws XServicesFault {

		Copy copy = new Copy();

		copy.setTaskName("Copy");

		RunTask runner = new RunTask(copy);

		FileSet set = src.getAntResource(copy.getProject());

		copy.add(set);

		File dst = new File(todir);

		if (dst.isDirectory()) {

			copy.setTodir(dst);

		}

		if (dst.isFile()) {

			copy.setTofile(dst);

		}

		copy.setOverwrite(overwrite);

		copy.setPreserveLastModified(plm);

		if (encoding != null && !encoding.equals("")) {

			copy.setOutputEncoding(encoding);

		} else {

			copy.setOutputEncoding(System.getProperty("file.encoding"));

		}

		return runner.postTask();

	}

	public ReturnCode copyFile(String fromFile, String tofile, boolean overwrite)

			throws XServicesFault {

		Copy copy = new Copy();

		copy.setTaskName("Copy");

		RunTask runner = new RunTask(copy);

		File f = new File(fromFile);

		if (!f.isFile())

			throw new XServicesFault("File '" + fromFile + "' not found.");

		copy.setFile(new File(fromFile));

		copy.setTofile(new File(tofile));

		copy.setOverwrite(overwrite);

		return runner.postTask();

	}

	/*

	 * (non-Javadoc)

	 *

	 * @see

	 * net.brutex.xservices.ws.impl.FileService#loadRes(net.brutex.xservices

	 * .types.FileResource, java.lang.String)

	 */

	public String loadRes(FileResource res, String encoding)

			throws XServicesFault {

		if (encoding == null || encoding.equals("")) {

			encoding = System.getProperty("file.encoding");

		}

		LoadResource lr = new LoadResource();

		lr.setTaskName("LoadResource");

		RunTask runner = new RunTask(lr);

		lr.addConfigured(res.getAntResource(lr.getProject()));

		lr.setEncoding(encoding);

		System.out.println("Using encoding: " + encoding);

		lr.setProperty("LoadResource.out");

		ReturnCode r = runner.postTask();

		return r.getProperty("LoadResource.out");

	}

	/*

	 * (non-Javadoc)

	 *

	 * @see

	 * net.brutex.xservices.ws.impl.FileService#loadResFromArchive(net.brutex

	 * .xservices.types.ArchiveResource, java.lang.String)

	 */

	public String loadResFromArchive(ArchiveResource res, String encoding) {

		if (encoding == null || encoding.equals("")) {

			encoding = System.getProperty("file.encoding");

		}

		LoadResource lr = new LoadResource();

		lr.setTaskName("LoadResource");

		RunTask runner = new RunTask(lr);

		lr.addConfigured(res.getAntResource(lr.getProject()));

		lr.setEncoding(encoding);

		System.out.println("Using encoding: " + encoding);

		lr.setProperty("LoadResource.out");

		ReturnCode r = runner.postTask();

		return r.getProperty("LoadResource.out");

	}

	/*

	 * (non-Javadoc)

	 *

	 * @see net.brutex.xservices.ws.impl.FileService#echo2file(java.lang.String,

	 * java.lang.String, java.lang.String, boolean)

	 */

	public ReturnCode echo2file(String message, String file, String encoding,

			boolean append) throws XServicesFault {

		Echo echo = new Echo();

		echo.setTaskName("toFile");

		RunTask runTask = new RunTask(echo);

		echo.addText(message);

		echo.setEncoding(encoding);

		File f = new File(file);

		try {

			if (!f.canWrite())

				throw new XServicesFault("Cannot write to file: "

						+ f.getCanonicalPath());

			echo.setFile(f);

			echo.setAppend(append);

			ReturnCode c = runTask.postTask();

			return c;

		} catch (BuildException e) {

			throw new XServicesFault("Error in echo2file.", e);

		} catch (IOException e) {

			throw new XServicesFault("Cannot write to file.", e);

		}

	}

	/*

	 * (non-Javadoc)

	 *

	 * @see

	 * net.brutex.xservices.ws.impl.FileService#changeOwner(net.brutex.xservices

	 * .types.FileSetResource, java.lang.String)

	 */

	public ReturnCode changeOwner(FileSetResource res, String owner) {

		Chown chown = new Chown();

		chown.setTaskName("Chown");

		RunTask runner = new RunTask(chown);

		chown.setOwner(owner);

		FileSet set = res.getAntResource(chown.getProject());

		chown.add(set);

		chown.setMaxParallel(300);

		return runner.postTask();

	}

	/*

	 * (non-Javadoc)

	 *

	 * @see

	 * net.brutex.xservices.ws.impl.FileService#changeGroup(net.brutex.xservices

	 * .types.FileSetResource, java.lang.String)

	 */

	public ReturnCode changeGroup(FileSetResource res, String group) {

		Chgrp chgrp = new Chgrp();

		chgrp.setTaskName("Chgrp");

		RunTask runner = new RunTask(chgrp);

		chgrp.setGroup(group);

		FileSet set = res.getAntResource(chgrp.getProject());

		chgrp.add(set);

		chgrp.setMaxParallel(300);

		return runner.postTask();

	}

	/*

	 * (non-Javadoc)

	 *

	 * @see

	 * net.brutex.xservices.ws.impl.FileService#changeMode(net.brutex.xservices

	 * .types.FileSetResource, java.lang.String)

	 */

	public ReturnCode changeMode(FileSetResource res, String perm) {

		Chmod chmod = new Chmod();

		chmod.setTaskName("Chmod");

		RunTask runner = new RunTask(chmod);

		FileSet set = res.getAntResource(chmod.getProject());

		chmod.add(set);

		chmod.setMaxParallel(300);

		chmod.setPerm(perm);

		chmod.setVerbose(true);

		return runner.postTask();

	}

}