//----------------------------BEGIN LICENSE----------------------------
/* 
 * Willow : the Open Source WorkFlow Project
 * Distributable under GNU LGPL license by gun.org
 *
 * Copyright (C) 2004-2010  huihoo.org 
 * Copyright (C) 2004-2010  ZosaTapo <dertyang@hotmail.com> 
 * 
 * ====================================================================
 * Project Homepage : http://www.huihoo.org/willow
 * Source Forge     : http://sourceforge.net/projects/huihoo
 * Mailing list     : willow@lists.sourceforge.net 
 */
//----------------------------END  LICENSE-----------------------------
package org.huihoo.willow.startup;

import org.apache.commons.digester.Digester;
import org.huihoo.commons.util.IntrospectionUtils;
import org.xml.sax.Attributes;
import org.xml.sax.SAXException;
import org.xml.sax.helpers.AttributesImpl;


public class WillowDigester extends Digester
{
	// ---------------------------------------------------------- Static Fields
	private static class SystemPropertySource
  implements IntrospectionUtils.PropertySource
	{
		public String getProperty(String key)
		{
			return System.getProperty(key);
		}
	}
	protected static IntrospectionUtils.PropertySource source[] =
		new IntrospectionUtils.PropertySource[] { new SystemPropertySource()};
	// ---------------------------------------------------------------- Methods
	/**
	 * Invoke inherited implementation after applying variable
	 * substitution to any attribute values containing variable
	 * references. 
	 */
	public void startElement(
		String namespaceURI,
		String localName,
		String qName,
		Attributes list)
		throws SAXException
	{
		list = updateAttributes(list);
		super.startElement(namespaceURI, localName, qName, list);
	}
  
	/**
	 * Invoke inherited implementation after applying variable substitution
	 * to the character data contained in the current element.
	 */
	public void endElement(String namespaceURI, String localName, String qName)
		throws SAXException
	{
		bodyText = updateBodyText(bodyText);
		super.endElement(namespaceURI, localName, qName);
	}
  
	/**
	 * Returns an attributes list which contains all the attributes
	 * passed in, with any text of form "${xxx}" in an attribute value
	 * replaced by the appropriate value from the system property.
	 */
	private Attributes updateAttributes(Attributes list)
	{
		if (list.getLength() == 0)
		{
			return list;
		}
		AttributesImpl newAttrs = new AttributesImpl(list);
		int nAttributes = newAttrs.getLength();
		for (int i = 0; i < nAttributes; ++i)
		{
			String value = newAttrs.getValue(i);
			try
			{
				String newValue =
					IntrospectionUtils.replaceProperties(value, null, source);
				if (value != newValue)
				{
					newAttrs.setValue(i, newValue);
				}
			}
			catch (Exception e)
			{
				// ignore - let the attribute have its original value
			}
		}
		return newAttrs;
	}
	/**
	 * Return a new StringBuffer containing the same contents as the
	 * input buffer, except that data of form ${varname} have been
	 * replaced by the value of that var as defined in the system property.
	 */
	private StringBuffer updateBodyText(StringBuffer bodyText)
	{
		String in = bodyText.toString();
		String out;
		try
		{
			out = IntrospectionUtils.replaceProperties(in, null, source);
		}
		catch (Exception e)
		{
			return bodyText; // return unchanged data
		}
		if (out == in)
		{
			// No substitutions required. Don't waste memory creating
			// a new buffer
			return bodyText;
		}
		else
		{
			return new StringBuffer(out);
		}
	}
}
