/* DocumentFunction.java -- 
   Copyright (C) 2004 Free Software Foundation, Inc.

This file is part of GNU Classpath.

GNU Classpath is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2, or (at your option)
any later version.

GNU Classpath is distributed in the hope that it will be useful, but
WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
General Public License for more details.

You should have received a copy of the GNU General Public License
along with GNU Classpath; see the file COPYING.  If not, write to the
Free Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
02111-1307 USA.

Linking this library statically or dynamically with other modules is
making a combined work based on this library.  Thus, the terms and
conditions of the GNU General Public License cover the whole
combination.

As a special exception, the copyright holders of this library give you
permission to link this library with independent modules to produce an
executable, regardless of the license terms of these independent
modules, and to copy and distribute the resulting executable under
terms of your choice, provided that you also meet, for each linked
independent module, the terms and conditions of the license of that
module.  An independent module is a module which is not derived from
or based on this library.  If you modify this library, you may extend
this exception to your version of the library, but you are not
obligated to do so.  If you do not wish to do so, delete this
exception statement from your version. */

package gnu.xml.transform;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;
import java.util.TreeSet;
import javax.xml.transform.TransformerException;
import javax.xml.transform.dom.DOMSource;
import javax.xml.xpath.XPathFunction;
import javax.xml.xpath.XPathFunctionException;
import org.w3c.dom.Node;
import gnu.xml.xpath.Constant;
import gnu.xml.xpath.Expr;
import gnu.xml.xpath.Function;
import gnu.xml.xpath.IdFunction;

/**
 * The XSLT <code>document()</code>function.
 *
 * @author <a href='mailto:dog@gnu.org'>Chris Burdess</a>
 */
final class DocumentFunction
  extends Expr
  implements Function, XPathFunction
{

  final Stylesheet stylesheet;
  final Node base;
  List args;
  List values;

  DocumentFunction(Stylesheet stylesheet, Node base)
  {
    this.stylesheet = stylesheet;
    this.base = base;
  }

  public Object evaluate(List args)
    throws XPathFunctionException
  {
    values = args;
    return evaluate(null, 1, 1);
  }

  public void setArguments(List args)
  {
    this.args = args;
  }
  
  public Object evaluate(Node context, int pos, int len)
  {
    int arity = args.size();
    if (values == null)
      {
        values = new ArrayList(arity);
        for (int i = 0; i < arity; i++)
          {
            Expr arg = (Expr) args.get(i);
            values.add(arg.evaluate(context, pos, len));
          }
      }
    Object ret;
    switch (arity)
      {
      case 1:
        Object arg = values.get(0);
        if (arg instanceof Collection)
          {
            Collection ns = (Collection) arg;
            Collection acc = new TreeSet();
            for (Iterator i = ns.iterator(); i.hasNext(); )
              {
                Node node = (Node) i.next();
                String s = Expr.stringValue(node);
                acc.addAll(document(s, node.getBaseURI()));
              }
            ret = acc;
          }
        else
          {
            String s = Expr._string(context, arg);
            ret = document(s, base.getBaseURI());
          }
        break;
      case 2:
        Object arg1 = values.get(0);
        Object arg2 = values.get(1);
        if (!(arg2 instanceof Collection))
          {
            throw new RuntimeException("second argument is not a node-set");
          }
        Collection arg2ns = (Collection) arg2;
        String base2 = arg2ns.isEmpty() ? null :
          ((Node) arg2ns.iterator().next()).getBaseURI();
        if (arg1 instanceof Collection)
          {
            Collection arg1ns = (Collection) arg1;
            Collection acc = new TreeSet();
            for (Iterator i = arg1ns.iterator(); i.hasNext(); )
              {
                Node node = (Node) i.next();
                String s = Expr.stringValue(node);
                acc.addAll(document(s, base2));
              }
            ret = acc;
          }
        else
          {
            String s = Expr._string(context, arg1);
            ret = document(s, base2);
          }
        break;
      default:
        throw new RuntimeException("invalid arity");
      }
    values = null;
    return ret;
  }

  /**
   * The XSL <code>document</code> function.
   * @see XSLT 12.1
   * @param uri the URI from which to retrieve nodes
   * @param base the base URI for relative URIs
   */
  Collection document(String uri, String base)
  {
    if ("".equals(uri) || uri == null)
      {
        uri = this.base.getBaseURI();
      }
    
    // Get fragment
    Expr fragment = null;
    int hi = uri.indexOf('#');
    if (hi != -1)
      {
        String f = uri.substring(hi + 1);
        uri = uri.substring(0, hi);
        // TODO handle xpointer() here
        // this only handles IDs
        fragment = new IdFunction(new Constant(f));
      }

    // Get document source
    try
      {
        DOMSource source;
        XSLURIResolver resolver = stylesheet.factory.resolver;
        synchronized (resolver)
          {
            if (stylesheet.transformer != null)
              {
                resolver.setUserResolver(stylesheet.transformer.uriResolver);
                resolver.setUserListener(stylesheet.transformer.errorListener);
              }
            source = resolver.resolveDOM(null, base, uri);
          }
        Node node = source.getNode();
        if (fragment == null)
          {
            return Collections.singleton(node);
          }
        else
          {
            Object ret = fragment.evaluate(node, 1, 1);
            if (!(ret instanceof Collection))
              {
                // XXX Report error?
                return Collections.EMPTY_SET;
              }
            return (Collection) ret;
          }
      }
    catch (TransformerException e)
      {
        String msg = "can't open " + uri;
        if (base != null)
          {
            msg += " with base " + base;
          }
        throw new RuntimeException(msg);
      }
  }

  public Expr clone(Object context)
  {
    Stylesheet s = stylesheet;
    if (context instanceof Stylesheet)
      {
        s = (Stylesheet) context;
      }
    DocumentFunction f = new DocumentFunction(s, base);
    int len = args.size();
    List args2 = new ArrayList(len);
    for (int i = 0; i < len; i++)
      {
        args2.add(((Expr) args.get(i)).clone(context));
      }
    f.setArguments(args2);
    return f;
  }
  
}