package WSS4J;

import org.w3c.dom.Document;
import org.w3c.dom.Element;

import com.ibm.xml.dsig.IDResolver;

import org.apache.xpath.XPathAPI;

public class WSUIdResolver implements IDResolver
{
    public WSUIdResolver()
    {
    }

    public Element resolveID(Document doc, String id)
    {
        Element root = doc.getDocumentElement();
        String xpathString = "//*[@wsu:Id=\"" + id + "\"]";
        Element plainTextElement = null;
        try {
            plainTextElement = (Element)XPathAPI.selectSingleNode(root, xpathString);
        } catch ( Exception e ) {
            System.out.println ( "Exception in getting the plainTextElement." );
            e.printStackTrace();
        }

        return plainTextElement;
    }
}