/*
 * Copyright (c) 1998-2011 Caucho Technology -- all rights reserved
 *
 * This file is part of Resin(R) Open Source
 *
 * Each copy or derived work must preserve the copyright notice and this
 * notice unmodified.
 *
 * Resin Open Source 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 of the License, or
 * (at your option) any later version.
 *
 * Resin Open Source 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, or any warranty
 * of NON-INFRINGEMENT.  See the GNU General Public License for more
 * details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Resin Open Source; if not, write to the
 *   Free SoftwareFoundation, Inc.
 *   59 Temple Place, Suite 330
 *   Boston, MA 02111-1307  USA
 *
 * @author Scott Ferguson
 */

package com.caucho.soap.wsdl;

import com.caucho.log.Log;
import com.caucho.util.L10N;
import com.caucho.vfs.Vfs;

import javax.jws.WebService;

import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Unmarshaller;

import javax.xml.ws.WebServiceException;

import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.util.logging.Level;
import java.util.logging.Logger;

/**
 * WSDL Parser
 */
public class WSDLParser {
  private final static Logger log = Log.open(WSDLParser.class);
  private final static L10N L = new L10N(WSDLParser.class);
  private static JAXBContext _context = null;

  public static WSDLDefinitions parse(Class api)
    throws WebServiceException
  {
    WebService service = (WebService) api.getAnnotation(WebService.class);

    if (service == null) {
      if (log.isLoggable(Level.FINER))
        log.finer("No @WebService found on " + api);

      return null;
    }

    String wsdlLocation = null;

    if (! "".equals(service.wsdlLocation())) {
      wsdlLocation = service.wsdlLocation();
    } 
    else if (! "".equals(service.endpointInterface())) {
      try {
        ClassLoader loader = api.getClassLoader();
        Class ei = loader.loadClass(service.endpointInterface());

        service = (WebService) api.getAnnotation(WebService.class);

        if (service == null) {
          if (log.isLoggable(Level.FINER))
            log.finer("No @WebService found on " + api);

          return null;
        }

        if (! "".equals(service.wsdlLocation()))
          wsdlLocation = service.wsdlLocation();
      }
      catch (ClassNotFoundException e) {
      }
    }

    if (wsdlLocation == null) {
      if (log.isLoggable(Level.FINER))
        log.finer("No WSDL location found on " + api);

      return null;
    }

    return parse(wsdlLocation);
  }

  public static WSDLDefinitions parse(URL wsdlLocation)
    throws WebServiceException
  {
    return parse(wsdlLocation.toString());
  }

  public static WSDLDefinitions parse(String wsdlLocation)
    throws WebServiceException
  {
    InputStream is = null;

    if (log.isLoggable(Level.FINER))
      log.finer("Getting WSDL: " + wsdlLocation);

    try {
      is = Vfs.openRead(wsdlLocation);

      return parse(is);
    }
    catch (IOException e) {
      throw new WebServiceException(L.l("Unable to download WSDL: {0}", 
                                        wsdlLocation), e);
    }
    catch (JAXBException e) {
      throw new WebServiceException(L.l("Unable to parse WSDL: {0}", 
                                        wsdlLocation), e);
    }
    finally {
      try {
        if (is != null)
          is.close();
      }
      catch (IOException e) {
      }
    }
  }

  /**
   * Starts a WSDL element.
   */
  public static WSDLDefinitions parse(InputStream is)
    throws IOException, JAXBException
  {
    try {
      if (_context == null) {
        _context = JAXBContext.newInstance("com.caucho.soap.wsdl:" +
                                           "com.caucho.xml.schema");
      }

      Unmarshaller unmarshaller = _context.createUnmarshaller();
      return (WSDLDefinitions) unmarshaller.unmarshal(is);
    } 
    catch (JAXBException e) {
      throw e;
    } 
    catch (Exception e) {
      throw new JAXBException(e);
    }
  }
}