/*

 Copyright (c) 2006, University of Tromsø

 All rights reserved.

 Redistribution and use in source and binary forms, with or without 

 modification, are permitted provided that the following conditions are met:

  * Redistributions of source code must retain the above copyright notice, this list 

    of conditions and the following disclaimer.

  * Redistributions in binary form must reproduce the above copyright notice, this 

    list of conditions and the following disclaimer in the documentation and/or other 

    materials provided with the distribution.

  * Neither the name of the University of Tromsø nor the names of its contributors may 

    be used to endorse or promote products derived from this software without specific 

    prior written permission.

 THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY 

 EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 

 OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT 

 SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, 

 INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED 

 TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR 

 BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 

 CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN 

 ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH 

 DAMAGE.

 */

 package argos.bangbang.axis;

 import java.lang.reflect.Method;

 import java.util.ArrayList;

 import java.util.List;

 import java.util.logging.Level;

 import java.util.logging.Logger;

 import javax.management.Notification;

 import org.apache.axis.AxisFault;

 import org.apache.axis.handlers.soap.SOAPService;

 import org.apache.axis.server.AxisServer;

 import org.apache.axis.transport.http.AdminServlet;

 import org.mortbay.jetty.servlet.Context;

 import org.mortbay.jetty.servlet.ServletHolder;

 import argos.annotation.Description;

 import argos.annotation.Impact;

 import argos.annotation.Init;

 import argos.annotation.NotificationHandler;

 import argos.annotation.RemoveInstrumentation;

 import argos.annotation.ServiceMeta;

 import argos.bangbang.jetty6.Jetty;

 import argos.bangbang.webservice.annotation.WebMethod;

 import argos.deploy.ComponentManager;

 import argos.metadata.ComponentMetaInfo;

 import argos.metadata.ServiceMetaInfo;

 import argos.naming.NamingService;

 import axis.dynamic.AxisServlet2;

 import axis.dynamic.SoapReceiver;

 import axis.dynamic.SoapServerConfiguration;

 @Description("Bang bang SOAP service.")

 public class Axis {

         private static final Logger logger = Logger.getLogger(Axis.class.getName());

         public static AxisServer axisServer;

         private SoapServerConfiguration soapServerConfiguration;

         private List<String> added;

         @ServiceMeta public ServiceMetaInfo meta;

         @Init

         @Impact(Impact.ACTION)

         @Description("Starts Axis")

         public void init() throws Exception {

                 added = new ArrayList<String>();

                 soapServerConfiguration = new SoapServerConfiguration();

                 soapServerConfiguration.setDisableServicesList(false); 

                 soapServerConfiguration.setServicesPath("/axis");

                 axisServer = new AxisServer(soapServerConfiguration);

                 axisServer.setName("AxisServlet");

                 Jetty jetty = (Jetty) NamingService.getInstance().getComponentByName("!!Jetty6");

                 Context con = new Context(jetty.getJettyServer(), "/axis");

                 con.addServlet(new ServletHolder(AxisServlet2.class), "/AxisServlet");

                 con.addServlet(new ServletHolder(AxisServlet2.class), "/services/*");

                 con.addServlet(new ServletHolder(AdminServlet.class), "/AdminServlet");

                 con.setAttribute("AxisEngine", axisServer);

                 jetty.addContext(con);

         }

         @NotificationHandler

         @RemoveInstrumentation

         public void handler(Notification not) {

                 if(handle(not)) {

                         ServiceMetaInfo temp = (ServiceMetaInfo) not.getUserData();

                         if(temp.dependsOn(meta)) {

                                 if(not.getType().equals(ComponentManager.SERVICE_STARTED)) {

                                         for(ComponentMetaInfo tempMeta : ((ServiceMetaInfo) not.getUserData()).getComponentMetaInfo()) {

                                                 addComponent(tempMeta);        

                                         }

                                 }

                                 else if(not.getType().equals(ComponentManager.SERVICE_STOPPED)) {

                                         for(ComponentMetaInfo tempMeta : ((ServiceMetaInfo) not.getUserData()).getComponentMetaInfo()) {

                                                 removeComponent(tempMeta);        

                                         }

                                 }

                         }

                 }

         }

         @Impact(Impact.ACTION)

         @Description("Removes a soap (component) service.")

         public void addComponent(ComponentMetaInfo componentMeta) {

                 Object component = NamingService.getInstance().getComponentByMeta(componentMeta);

                 String name = componentMeta.getName();

                 Class<?> clazz = component.getClass();

                 List<String> methods = new ArrayList<String>();

                 for(Method method : clazz.getMethods()) {

                         if(method.isAnnotationPresent(WebMethod.class)) {

                                 methods.add(method.getName());

                         }

                 }

                 if(methods.isEmpty()) {

                         return;

                 }

                 String[] m = new String[methods.size()];

                 for(int i = 0; i < methods.size(); i++) {

                         m[i] = methods.get(i);

                 }

                 SoapReceiver publisher = new SoapReceiver();

                 publisher.setServiceName(name);

                 publisher.setTarget(component); 

                 publisher.setAllowedMethods(m);

                 publisher.setServerConfiguration(soapServerConfiguration);

                 try {

                         SOAPService service = publisher.createService();

                         soapServerConfiguration.deployService(name, service);

                         added.add(name);

                         logger.info("Deployed SOAP service " + name + " with operations " + 

                                         service.getServiceDescription().getAllowedMethods());

                 }

                 catch (AxisFault e) {

                         logger.log(Level.SEVERE, "Unable to create SOAP service " + name + ": " + 

                                         e.getMessage(), e);

                 }

         }

         @Description("Removes a soap (component) service.")

         @Impact(Impact.ACTION)

         public void removeComponent(ComponentMetaInfo componentMeta) {

                 if(added.remove(componentMeta.getName())) {

                         soapServerConfiguration.undeployService(componentMeta.getName());

                         logger.info("Undeployed SOAP service " + componentMeta.getName());

                 }

         }

         private boolean handle(Notification not) {

                 return (not.getType().equals(ComponentManager.SERVICE_STARTED) ||

                                 not.getType().equals(ComponentManager.SERVICE_STOPPED));

         }

 }