trunk/tools/anttasks/src/org/apache/lenya/cms/ant/CopyJavaSourcesTask.java - lenya - Git at Google

 /*
  * Licensed to the Apache Software Foundation (ASF) under one or more
  *  contributor license agreements.  See the NOTICE file distributed with
  *  this work for additional information regarding copyright ownership.
  *  The ASF licenses this file to You 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.
  *
  */

 /* $Id: CopyJavaSourcesTask.java 416059 2006-06-21 18:27:27Z andreas $  */

 package org.apache.lenya.cms.ant;

 import java.io.File;
 import java.io.FileInputStream;
 import java.io.FileNotFoundException;
 import java.io.FileOutputStream;
 import java.io.FilenameFilter;
 import java.io.IOException;
 import java.io.InputStream;
 import java.io.OutputStream;
 import java.io.PrintWriter;
 import java.io.StringWriter;
 import java.util.StringTokenizer;

 import org.apache.tools.ant.BuildException;
 import org.apache.tools.ant.Project;
 import org.apache.tools.ant.Task;
 import org.apache.tools.ant.types.Path;

 /**
  * Task to copy java sources.
  */
 public class CopyJavaSourcesTask extends Task {

     private Path pubsRootDirs;
     private String javaDir;
     private String buildDir;

     /**
      * @see org.apache.tools.ant.Task#execute()
      */
     public void execute() throws BuildException {
         int numberOfDirectoriesCreated = 0;
         int numberOfFilesCopied = 0;
         TwoTuple twoTuple = new TwoTuple(numberOfDirectoriesCreated, numberOfFilesCopied);

         String translatedBuildDir = Project.translatePath(this.buildDir);
         File absoluteBuildDir = null;
         if (translatedBuildDir != null && translatedBuildDir.startsWith(File.separator)) {
             absoluteBuildDir = new File(translatedBuildDir);
         } else {
             absoluteBuildDir = new File(getProject().getBaseDir(), translatedBuildDir);
         }

         StringTokenizer st = new StringTokenizer(this.pubsRootDirs.toString(), File.pathSeparator);

         while (st.hasMoreTokens()) {
             String pubsRootDir = st.nextToken();

             File path = new File(pubsRootDir);

             if (path.isDirectory()) {
                 // In the case of a publication
                 if (new File(path, "/config/publication.xml").isFile()) {
                     copyContentOfDir(new File(path, this.javaDir),
                             absoluteBuildDir,
                             twoTuple,
                             new JavaFilenameFilter(),
                             this);
                     // In the case of a module
                 } else if (new File(path, "/config/module.xml").isFile()) {
                     copyContentOfDir(new File(path, this.javaDir),
                             absoluteBuildDir,
                             twoTuple,
                             new JavaFilenameFilter(),
                             this);
                 } else {
                     // FIXME: Look for publications defined by the file "publication.xml" or modules
                     // defined by the file "module.xml"
                     String[] pubs = path.list();
                     if (pubs != null) {
                         for (int i = 0; i < pubs.length; i++) {
                             File pubJavaDir = new File(path, new File(pubs[i], this.javaDir).toString());

                             copyContentOfDir(pubJavaDir,
                                     absoluteBuildDir,
                                     twoTuple,
                                     new JavaFilenameFilter(),
                                     this);
                         }
                     }
                 }
             } else {
                 throw new BuildException("No such directory: " + path);
             }
         }

         numberOfDirectoriesCreated = twoTuple.x;
         numberOfFilesCopied = twoTuple.y;
         log("Copying " + numberOfDirectoriesCreated + " directories to " + absoluteBuildDir,
                 Project.MSG_INFO);
         log("Copying " + numberOfFilesCopied + " files to " + absoluteBuildDir, Project.MSG_INFO);
     }

     /**
      * Copies the directory "source" into the directory "destination"
      * @param source The source directory
      * @param destination The destination directory
      * @param twoTuple The twoTuple to use
      * @param filenameFilter The filename filter to apply
      * @param client The client task
      */
     public static void copyDir(File source, File destination, TwoTuple twoTuple,
             FilenameFilter filenameFilter, Task client) {
         File actualDestination = new File(destination, source.getName());
         actualDestination.mkdirs();
         copyContentOfDir(source, actualDestination, twoTuple, filenameFilter, client);
     }

     /**
      * Copies the content of a directory into another directory
      * @param source The source directory
      * @param destination The destination directory
      * @param twoTuple The twoTuple to use
      * @param filenameFilter The filename filter to use
      * @param client The client task
      */
     public static void copyContentOfDir(File source, File destination, TwoTuple twoTuple,
             FilenameFilter filenameFilter, Task client) {
         if (source.isDirectory()) {
             String[] files;

             if (filenameFilter != null) {
                 files = source.list(filenameFilter);
             } else {
                 files = source.list();
             }

             for (int i = 0; i < files.length; i++) {
                 File file = new File(source, files[i]);

                 if (file.isFile()) {
                     copyFile(file, new File(destination, files[i]), twoTuple, client, false);
                 } else if (file.isDirectory()) {
                     copyContentOfDir(file,
                             new File(destination, files[i]),
                             twoTuple,
                             filenameFilter,
                             client);
                 } else {
                     client.log("CopyJavaSourcesTask.copyContentOfDir(): Neither file nor directory: "
                             + file,
                             Project.MSG_ERR);
                 }
             }
         }
     }

     /**
      * Copies the content of a file into another file
      * @param source The source file (not a directory!)
      * @param destination File (not a directory!)
      * @param twoTuple The twoTuple to use
      * @param client The client task
      */
     public static void copyFile(File source, File destination, TwoTuple twoTuple, Task client,
             boolean force) {
         if (source.isFile()) {
             File parentDest = new File(destination.getParent());

             if (!parentDest.exists()) {
                 parentDest.mkdirs();

                 int numberOfDirectoriesCreated = twoTuple.x;
                 numberOfDirectoriesCreated++;
                 twoTuple.x = numberOfDirectoriesCreated;
             }

             if (destination.isFile()) {
                 if (destination.lastModified() > source.lastModified() && !force) {
                     return;
                 }
             }

             byte[] buffer = new byte[1024];
             int bytesRead = -1;
             InputStream in = null;
             OutputStream out = null;

             try {
                 in = new FileInputStream(source);
                 out = new FileOutputStream(destination);

                 while ((bytesRead = in.read(buffer)) >= 0) {
                     out.write(buffer, 0, bytesRead);
                 }
             } catch (final FileNotFoundException e) {
                 StringWriter writer = new StringWriter();
                 e.printStackTrace(new PrintWriter(writer));
                 client.log("Exception caught: " + writer.toString());
             } catch (final IOException e) {
                 StringWriter writer = new StringWriter();
                 e.printStackTrace(new PrintWriter(writer));
                 client.log("Exception caught: " + writer.toString());
             } finally {
                 try {
                     if (out != null) {
                         out.close();
                     }
                     if (in != null) {
                         in.close();
                     }
                 } catch (final IOException e1) {
                     StringWriter writer = new StringWriter();
                     e1.printStackTrace(new PrintWriter(writer));
                     client.log("Exception closing stream: " + writer.toString());
                 }
             }

             int numberOfFilesCopied = twoTuple.y;
             numberOfFilesCopied++;
             twoTuple.y = numberOfFilesCopied;

         } else {
             client.log("No such file: " + source, Project.MSG_ERR);
         }
     }

     /**
      * Set the root publication root directories
      * @param _pubsRootDirs The root directories
      */
     public void setPubsRootDirs(Path _pubsRootDirs) {
         this.pubsRootDirs = _pubsRootDirs;
     }

     /**
      * Set the java directory
      * @param _javaDir The java directory
      */
     public void setJavaDir(String _javaDir) {
         this.javaDir = _javaDir;
     }

     /**
      * Set the build directory
      * @param _buildDir The build directory
      */
     public void setBuildDir(String _buildDir) {
         this.buildDir = _buildDir;
     }
 }
	/*
	* Licensed to the Apache Software Foundation (ASF) under one or more
	* contributor license agreements. See the NOTICE file distributed with
	* this work for additional information regarding copyright ownership.
	* The ASF licenses this file to You 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.
	*
	*/

	/* $Id: CopyJavaSourcesTask.java 416059 2006-06-21 18:27:27Z andreas $ */

	package org.apache.lenya.cms.ant;

	import java.io.File;
	import java.io.FileInputStream;
	import java.io.FileNotFoundException;
	import java.io.FileOutputStream;
	import java.io.FilenameFilter;
	import java.io.IOException;
	import java.io.InputStream;
	import java.io.OutputStream;
	import java.io.PrintWriter;
	import java.io.StringWriter;
	import java.util.StringTokenizer;

	import org.apache.tools.ant.BuildException;
	import org.apache.tools.ant.Project;
	import org.apache.tools.ant.Task;
	import org.apache.tools.ant.types.Path;

	/**
	* Task to copy java sources.
	*/
	public class CopyJavaSourcesTask extends Task {

	private Path pubsRootDirs;
	private String javaDir;
	private String buildDir;

	/**
	* @see org.apache.tools.ant.Task#execute()
	*/
	public void execute() throws BuildException {
	int numberOfDirectoriesCreated = 0;
	int numberOfFilesCopied = 0;
	TwoTuple twoTuple = new TwoTuple(numberOfDirectoriesCreated, numberOfFilesCopied);

	String translatedBuildDir = Project.translatePath(this.buildDir);
	File absoluteBuildDir = null;
	if (translatedBuildDir != null && translatedBuildDir.startsWith(File.separator)) {
	absoluteBuildDir = new File(translatedBuildDir);
	} else {
	absoluteBuildDir = new File(getProject().getBaseDir(), translatedBuildDir);
	}

	StringTokenizer st = new StringTokenizer(this.pubsRootDirs.toString(), File.pathSeparator);

	while (st.hasMoreTokens()) {
	String pubsRootDir = st.nextToken();

	File path = new File(pubsRootDir);

	if (path.isDirectory()) {
	// In the case of a publication
	if (new File(path, "/config/publication.xml").isFile()) {
	copyContentOfDir(new File(path, this.javaDir),
	absoluteBuildDir,
	twoTuple,
	new JavaFilenameFilter(),
	this);
	// In the case of a module
	} else if (new File(path, "/config/module.xml").isFile()) {
	copyContentOfDir(new File(path, this.javaDir),
	absoluteBuildDir,
	twoTuple,
	new JavaFilenameFilter(),
	this);
	} else {
	// FIXME: Look for publications defined by the file "publication.xml" or modules
	// defined by the file "module.xml"
	String[] pubs = path.list();
	if (pubs != null) {
	for (int i = 0; i < pubs.length; i++) {
	File pubJavaDir = new File(path, new File(pubs[i], this.javaDir).toString());

	copyContentOfDir(pubJavaDir,
	absoluteBuildDir,
	twoTuple,
	new JavaFilenameFilter(),
	this);
	}
	}
	}
	} else {
	throw new BuildException("No such directory: " + path);
	}
	}

	numberOfDirectoriesCreated = twoTuple.x;
	numberOfFilesCopied = twoTuple.y;
	log("Copying " + numberOfDirectoriesCreated + " directories to " + absoluteBuildDir,
	Project.MSG_INFO);
	log("Copying " + numberOfFilesCopied + " files to " + absoluteBuildDir, Project.MSG_INFO);
	}

	/**
	* Copies the directory "source" into the directory "destination"
	* @param source The source directory
	* @param destination The destination directory
	* @param twoTuple The twoTuple to use
	* @param filenameFilter The filename filter to apply
	* @param client The client task
	*/
	public static void copyDir(File source, File destination, TwoTuple twoTuple,
	FilenameFilter filenameFilter, Task client) {
	File actualDestination = new File(destination, source.getName());
	actualDestination.mkdirs();
	copyContentOfDir(source, actualDestination, twoTuple, filenameFilter, client);
	}

	/**
	* Copies the content of a directory into another directory
	* @param source The source directory
	* @param destination The destination directory
	* @param twoTuple The twoTuple to use
	* @param filenameFilter The filename filter to use
	* @param client The client task
	*/
	public static void copyContentOfDir(File source, File destination, TwoTuple twoTuple,
	FilenameFilter filenameFilter, Task client) {
	if (source.isDirectory()) {
	String[] files;

	if (filenameFilter != null) {
	files = source.list(filenameFilter);
	} else {
	files = source.list();
	}

	for (int i = 0; i < files.length; i++) {
	File file = new File(source, files[i]);

	if (file.isFile()) {
	copyFile(file, new File(destination, files[i]), twoTuple, client, false);
	} else if (file.isDirectory()) {
	copyContentOfDir(file,
	new File(destination, files[i]),
	twoTuple,
	filenameFilter,
	client);
	} else {
	client.log("CopyJavaSourcesTask.copyContentOfDir(): Neither file nor directory: "
	+ file,
	Project.MSG_ERR);
	}
	}
	}
	}

	/**
	* Copies the content of a file into another file
	* @param source The source file (not a directory!)
	* @param destination File (not a directory!)
	* @param twoTuple The twoTuple to use
	* @param client The client task
	*/
	public static void copyFile(File source, File destination, TwoTuple twoTuple, Task client,
	boolean force) {
	if (source.isFile()) {
	File parentDest = new File(destination.getParent());

	if (!parentDest.exists()) {
	parentDest.mkdirs();

	int numberOfDirectoriesCreated = twoTuple.x;
	numberOfDirectoriesCreated++;
	twoTuple.x = numberOfDirectoriesCreated;
	}

	if (destination.isFile()) {
	if (destination.lastModified() > source.lastModified() && !force) {
	return;
	}
	}

	byte[] buffer = new byte[1024];
	int bytesRead = -1;
	InputStream in = null;
	OutputStream out = null;

	try {
	in = new FileInputStream(source);
	out = new FileOutputStream(destination);

	while ((bytesRead = in.read(buffer)) >= 0) {
	out.write(buffer, 0, bytesRead);
	}
	} catch (final FileNotFoundException e) {
	StringWriter writer = new StringWriter();
	e.printStackTrace(new PrintWriter(writer));
	client.log("Exception caught: " + writer.toString());
	} catch (final IOException e) {
	StringWriter writer = new StringWriter();
	e.printStackTrace(new PrintWriter(writer));
	client.log("Exception caught: " + writer.toString());
	} finally {
	try {
	if (out != null) {
	out.close();
	}
	if (in != null) {
	in.close();
	}
	} catch (final IOException e1) {
	StringWriter writer = new StringWriter();
	e1.printStackTrace(new PrintWriter(writer));
	client.log("Exception closing stream: " + writer.toString());
	}
	}

	int numberOfFilesCopied = twoTuple.y;
	numberOfFilesCopied++;
	twoTuple.y = numberOfFilesCopied;

	} else {
	client.log("No such file: " + source, Project.MSG_ERR);
	}
	}

	/**
	* Set the root publication root directories
	* @param _pubsRootDirs The root directories
	*/
	public void setPubsRootDirs(Path _pubsRootDirs) {
	this.pubsRootDirs = _pubsRootDirs;
	}

	/**
	* Set the java directory
	* @param _javaDir The java directory
	*/
	public void setJavaDir(String _javaDir) {
	this.javaDir = _javaDir;
	}

	/**
	* Set the build directory
	* @param _buildDir The build directory
	*/
	public void setBuildDir(String _buildDir) {
	this.buildDir = _buildDir;
	}
	}