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.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.ArchiveResource;
import net.brutex.xservices.types.AttachmentType;
import net.brutex.xservices.types.FileResource;
import net.brutex.xservices.types.FileSetResource;
import net.brutex.xservices.types.ReplacePattern;
import net.brutex.xservices.types.ReturnCode;
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)
         */
        @Override
        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)
         */
        @Override
        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);
                }
        }

        /*
         * (non-Javadoc)
         * 
         * @see
         * net.brutex.xservices.ws.impl.FileService#base64Decode(net.brutex.xservices
         * .types.AttachmentType)
         */
        @Override
        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)
         */
        @Override
        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();
        }

        @Override
        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)
         */
        @Override
        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)
         */
        @Override
        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)
         */
        @Override
        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)
         */
        @Override
        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)
         */
        @Override
        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)
         */
        @Override
        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();
        }

}