Subversion Repositories XServices

Rev

Rev 60 | Rev 64 | Go to most recent revision | Blame | Compare with Previous | Last modification | View Log | Download | RSS feed

/*
 *   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.BufferedInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.UnsupportedEncodingException;
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.AntProperty;
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.ResourceInterface;
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.commons.codec.binary.Base64;
import org.apache.commons.codec.binary.Base64InputStream;
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
        @WebMethod(operationName = "basename")
        public ReturnCode basename(@WebParam(name = "file") String filename,
                        @WebParam(name = "suffix") String suffix) {
                return basename(new File(filename), suffix);
        }

        @WebMethod(operationName = "replaceInFile")
        public ReturnCode replaceInFile(
                        @WebParam(name = "file") FileResource res,
                        @WebParam(name = "search") String search,
                        @WebParam(name = "replace") String replace) {
                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);
                r = runner.postTask();
                return r;
        }
        
        @WebMethod(operationName = "replaceInFile2")
        public ReturnCode replaceInFile2(
                        @WebParam(name = "file") FileResource res,
                        @WebParam(name = "patternList") List<ReplacePattern> patternList) {
                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);
                        r = runner.postTask();
                }
                return r;
        }
        
        @WebMethod(operationName = "replaceInFileRegEx")
        public ReturnCode replaceInFileRegEx(
                        @WebParam(name = "file") FileResource res,
                        @WebParam(name = "search") String search,
                        @WebParam(name = "replace") String replace,
                        @WebParam(name = "flags") String flags) {               
                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);
                ReturnCode r = runner.postTask();               
                return r;
        }
        
        
        /*
         * (non-Javadoc)
         * 
         * @see
         * net.brutex.xservices.ws.impl.FileService#base64Encode(net.brutex.xservices
         * .types.FileSetResource)
         */
        @Override
        @WebMethod(operationName = "downloadFile")
        public AttachmentType downloadFile(@WebParam(name = "file") FileResource res) {
                InputStream is = null;
                try {
                        is = res.getAntResource(null).getInputStream();
                } catch (IOException e) {
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                }
                DataHandler h = new DataHandler(new StreamDataSource(
                                "application/binary", is));
                AttachmentType t = new AttachmentType();
                t.setContent(h);
                t.setFilename(res.getAntResource(null).getName());
                return t;
        }

        /*
         * (non-Javadoc)
         * 
         * @see
         * net.brutex.xservices.ws.impl.FileService#base64Decode(net.brutex.xservices
         * .types.AttachmentType)
         */
        @Override
        @WebMethod(operationName = "uploadFile")
        public String uploadFile(@WebParam(name = "file") AttachmentType file) {
                DataHandler h =  file.getContent();
                File f = new File(file.getFilename());
                FileOutputStream fout;
                try {
                        fout = new FileOutputStream(f);
                        /*
                         * InputStream in = h.getInputStream();
                        int b;
                        while( (b=in.read())!= -1 ) {
                                fout.write(b);
                        }
                        */
                        h.writeTo(fout);
                        fout.flush();
                        fout.close();
                        //in.close();
                } catch (FileNotFoundException e) {
                        throw new BuildException(e);
                } catch (IOException e) {
                        throw new BuildException(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
        @WebMethod(operationName = "copy")
        public ReturnCode copy(@WebParam(name = "fileset") FileSetResource src,
                        @WebParam(name = "todir") String todir,
                        @WebParam(name = "preservelastmodified") boolean plm,
                        @WebParam(name = "overwrite") boolean overwrite,
                        @WebParam(name = "encoding") String encoding) throws XServicesFault {
                return copy(src, new File(todir), plm, overwrite, encoding);
        }

        /*
         * (non-Javadoc)
         * 
         * @see
         * net.brutex.xservices.ws.impl.FileService#loadRes(net.brutex.xservices
         * .types.FileResource, java.lang.String)
         */
        @Override
        @WebMethod(operationName = "loadResource")
        public ReturnCode loadRes(@WebParam(name = "resource") FileResource res,
                        @WebParam(name = "encoding") String encoding) {
                if (encoding == null || encoding.equals("")) {
                        encoding = System.getProperty("file.encoding");
                }
                return loadResource(res, encoding);
        }

        /*
         * (non-Javadoc)
         * 
         * @see
         * net.brutex.xservices.ws.impl.FileService#loadResFromArchive(net.brutex
         * .xservices.types.ArchiveResource, java.lang.String)
         */
        @Override
        @WebMethod(operationName = "loadResourceFromArchive")
        public ReturnCode loadResFromArchive(
                        @WebParam(name = "archiveresource") ArchiveResource res,
                        @WebParam(name = "encoding") String encoding) {
                if (encoding == null || encoding.equals("")) {
                        encoding = System.getProperty("file.encoding");
                }
                return loadResource(res, encoding);
        }

        /*
         * (non-Javadoc)
         * 
         * @see net.brutex.xservices.ws.impl.FileService#echo2file(java.lang.String,
         * java.lang.String, java.lang.String, boolean)
         */
        @Override
        @WebMethod(operationName = "echoToFile")
        public ReturnCode echo2file(@WebParam(name = "message") String message,
                        @WebParam(name = "file") String file,
                        @WebParam(name = "encoding") String encoding,
                        @WebParam(name = "append") boolean append) {
                return echo(message, new File(file), encoding, append);
        }

        /*
         * (non-Javadoc)
         * 
         * @see
         * net.brutex.xservices.ws.impl.FileService#changeOwner(net.brutex.xservices
         * .types.FileSetResource, java.lang.String)
         */
        @Override
        @WebMethod(operationName = "changeOwner")
        public ReturnCode changeOwner(
                        @WebParam(name = "fileset") FileSetResource res,
                        @WebParam(name = "owner") String owner) {
                return chown(res, owner);
        }

        /*
         * (non-Javadoc)
         * 
         * @see
         * net.brutex.xservices.ws.impl.FileService#changeGroup(net.brutex.xservices
         * .types.FileSetResource, java.lang.String)
         */
        @Override
        @WebMethod(operationName = "changeGroup")
        public ReturnCode changeGroup(
                        @WebParam(name = "fileset") FileSetResource res,
                        @WebParam(name = "group") String group) {
                return chgrp(res, group);
        }

        /*
         * (non-Javadoc)
         * 
         * @see
         * net.brutex.xservices.ws.impl.FileService#changeMode(net.brutex.xservices
         * .types.FileSetResource, java.lang.String)
         */
        @Override
        @WebMethod(operationName = "changeMode")
        public ReturnCode changeMode(
                        @WebParam(name = "fileset") FileSetResource res,
                        @WebParam(name = "permissions") String perm) {
                return chmod(res, perm);
        }

        @WebMethod(exclude = true)
        private ReturnCode basename(File file, String suffix) {
                Basename basename = new Basename();
                RunTask runner = new RunTask(basename);
                basename.setFile(file);
                if (suffix != null && !suffix.equals("")) {
                        basename.setSuffix(suffix);
                }
                basename.setProperty("basename.value");
                return runner.postTask();
        }

        @WebMethod(exclude = true)
        private ReturnCode loadResource(ResourceInterface src, String encoding) {
                LoadResource lr = new LoadResource();
                lr.setTaskName("LoadResource");
                RunTask runner = new RunTask(lr);
                lr.addConfigured(src.getAntResource(lr.getProject()));
                lr.setEncoding(encoding);
                System.out.println("Using encoding: " + encoding);
                lr.setProperty("LoadResource.out");
                return runner.postTask();
        }

        @WebMethod(exclude = true)
        private ReturnCode echo(String msg, File file, String encoding,
                        boolean append) {
                Echo echo = new Echo();
                echo.setTaskName("toFile");
                RunTask runTask = new RunTask(echo);
                echo.addText(msg);
                echo.setEncoding(encoding);
                echo.setFile(file);
                echo.setAppend(append);
                return runTask.postTask();
        }

        @WebMethod(exclude = true)
        private ReturnCode copy(FileSetResource src, File dst,
                        boolean preservelastmodified, boolean overwrite, String encoding) {
                Copy copy = new Copy();
                copy.setTaskName("Copy");
                RunTask runner = new RunTask(copy);
                FileSet set = src.getAntResource(copy.getProject());
                copy.add(set);

                if (dst.isDirectory()) {
                        copy.setTodir(dst);
                }
                if (dst.isFile()) {
                        copy.setTofile(dst);
                }
                copy.setOverwrite(overwrite);
                copy.setPreserveLastModified(preservelastmodified);
                if (encoding != null && !encoding.equals("")) {
                        copy.setOutputEncoding(encoding);
                } else {
                        copy.setOutputEncoding(System.getProperty("file.encoding"));
                }

                return runner.postTask();
        }

        @WebMethod(exclude = true)
        private ReturnCode chown(FileSetResource src, String owner) {
                Chown chown = new Chown();
                chown.setTaskName("Chown");
                RunTask runner = new RunTask(chown);
                chown.setOwner(owner);
                FileSet set = src.getAntResource(chown.getProject());
                chown.add(set);
                chown.setMaxParallel(300);
                return runner.postTask();
        }

        @WebMethod(exclude = true)
        private ReturnCode chgrp(FileSetResource src, String group) {
                Chgrp chgrp = new Chgrp();
                chgrp.setTaskName("Chgrp");
                RunTask runner = new RunTask(chgrp);
                chgrp.setGroup(group);
                FileSet set = src.getAntResource(chgrp.getProject());
                chgrp.add(set);
                chgrp.setMaxParallel(300);
                return runner.postTask();
        }

        @WebMethod(exclude = true)
        private ReturnCode chmod(FileSetResource src, String perm) {
                Chmod chmod = new Chmod();
                chmod.setTaskName("Chmod");
                RunTask runner = new RunTask(chmod);
                FileSet set = src.getAntResource(chmod.getProject());
                chmod.add(set);
                chmod.setMaxParallel(300);
                chmod.setPerm(perm);
                chmod.setVerbose(true);
                return runner.postTask();
        }

}