package com.erggroup.mass.ant;

import java.io.File;
import org.apache.tools.ant.*;
import org.apache.tools.ant.types.*;
import org.apache.tools.ant.taskdefs.*;
import java.util.ArrayList;
import java.util.Iterator;

/** 
 * Custom ANT task that will take 
 * @param dir = as a base directory 
 * @param includes = as a fileNamePattern that should be found in the base dir
 * @param webDeploymentName = defined inside <jbossnet> tag
 * 
 * This task will search for  "@jboss-net.xml-schema" token and replace it with
 *  @jboss-net.xml-schema urn="webDeploymentName:ClassName"
 *
 *  <wsdetails dir="" webDeploymentName="">
 * 		<wsinclude name="filePattern" />
 * 		....
 * 	</wsdetails>
 * 
 *	<wsdetails	dir="" includes="pattern1,pattern2" webDeploymentName="" />
 *
 */
public class WSReplaceElement extends Task
{
	private String token = "@jboss-net.xml-schema";
	
	private String webDeploymentName;
	private File dir; 	
	private String includes;
	private ArrayList includeElements = new ArrayList();
	
	
	public void execute()
	{

		if ( includeElements.size() == 0) 
		{
			if ( dir == null || includes == null || webDeploymentName == null) 
			{
				throw new BuildException(" \'dir\' and \'includes\' and \'webDeploymentName\' attribute must be provided");	
			}
		} 
		else if ( includeElements.size() > 0 && dir == null || webDeploymentName == null ) 
		{
			throw new BuildException(" \'dir\' and \'webDeploymentName\' attribute must be provided and at least one <wsinlcude>");	
		} 
			
		// returns a list of all include file Names
		FileSet fileSet = new FileSet();
		fileSet.setDir(this.dir);
		//if <wsinclude> is used				
		if (includeElements.size() > 0)
		{   
			this.includes = "";
			Iterator i = includeElements.iterator();
			int index = 0;
			while(i.hasNext())
			{
				this.includes += ((WSInclude)i.next()).getName(); 
				if (index < includeElements.size()-1 )
				{
					this.includes += ",";
				}
				index ++;
			}
		}
		fileSet.setIncludes(this.includes);
		DirectoryScanner scanner = fileSet.getDirectoryScanner(getProject());
		String[] files = scanner.getIncludedFiles();;
		
		// for each file name create replaceValue and replace specified token 
		for (int i=0; i<files.length; i++)
		{
			Replace replace = new Replace();
			replace.setProject(getProject());
			replace.setToken(this.token);
			replace.setDir(this.dir);
			
			String fileName = files[i];
			replace.setIncludes(files[i]);
			fileName = fileName.substring(fileName.lastIndexOf("\\")+1, fileName.length() );
			fileName = fileName.substring(0, fileName.lastIndexOf(".java") );
			String replaceValue = "@jboss-net.xml-schema urn=\"" + this.webDeploymentName + ":" + fileName + "\"";
			replace.setValue(replaceValue);
			replace.execute();			
		}
	}

	public WSInclude createWsinclude()
	{
		WSInclude element = new WSInclude();
		includeElements.add(element);
		return element;
	}



	public void setWebDeploymentName(String webDeploymentName)
	{
		this.webDeploymentName = webDeploymentName;
	}

	public void setIncludes(String includes)
	{
		this.includes = includes;
	}
	
	public void setDir(File dir)
	{
		this.dir = dir;
	}
	
}