package com.zhangjun.xml;

import java.io.PrintStream;
import java.io.StringWriter;
import java.util.Enumeration;
import java.util.Hashtable;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.transform.Source;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;
import javax.xml.transform.stream.StreamSource;
import org.w3c.dom.Document;

public class TransXML
{
  public String trans(String xslname, String xmlname, Hashtable hs)
  {
    StreamSource xsl = new StreamSource(xslname);
    StreamSource xmlData = new StreamSource(xmlname);
    return go_trans(xmlData, xsl, hs);
  }

  public String trans(String xslname, Document dom, Hashtable hs)
  {
    StreamSource xsl = new StreamSource(xslname);
    DOMSource xmlData = new DOMSource(dom);
    return go_trans(xmlData, xsl, hs);
  }

  private String go_trans(Source xmlSource, Source xslSource, Hashtable hs)
  {
    try
    {
      DocumentBuilderFactory dFactory = DocumentBuilderFactory.newInstance();
      dFactory.setNamespaceAware(true);
      DocumentBuilder dBuilder = dFactory.newDocumentBuilder();

      TransformerFactory tFactory = TransformerFactory.newInstance();
      Transformer transformer = tFactory.newTransformer(xslSource);
      StringWriter wt = new StringWriter();
      StreamResult streamResult = new StreamResult();
      streamResult.setWriter(wt);
      transformer.setOutputProperty("encoding", "GB2312");
      Enumeration hs_value = hs.elements();
      for (Enumeration hs_name = hs.keys(); hs_name.hasMoreElements(); )
      {
        String para_name = (String)hs_name.nextElement();
        String para_value = (String)hs_value.nextElement();
        if (para_value != null) transformer.setParameter(para_name, para_value);
      }
      transformer.transform(xmlSource, streamResult);
      return wt.toString();
    }
    catch (Exception e)
    {
      System.out.println("trans error:" + e);
      e.printStackTrace(); }
    return "error";
  }
}