/**

  * Copyright 2005-2011 The Kuali Foundation

  *

  * Licensed under the Educational Community License, Version 2.0 (the "License");

  * you may not use this file except in compliance with the License.

  * You may obtain a copy of the License at

  *

  * http://www.opensource.org/licenses/ecl2.php

  *

  * Unless required by applicable law or agreed to in writing, software

  * distributed under the License is distributed on an "AS IS" BASIS,

  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

  * See the License for the specific language governing permissions and

  * limitations under the License.

  */

 package org.kuali.rice.ksb.messaging.web;

 import org.apache.commons.lang.StringUtils;

 import org.apache.log4j.Logger;

 import org.apache.struts.action.ActionForm;

 import org.apache.struts.action.ActionForward;

 import org.apache.struts.action.ActionMapping;

 import org.apache.struts.action.ActionMessages;

 import org.kuali.rice.core.api.config.CoreConfigHelper;

 import org.kuali.rice.core.api.config.property.ConfigContext;

 import org.kuali.rice.core.api.util.RiceUtilities;

 import org.kuali.rice.ksb.api.KsbApiServiceLocator;

 import org.kuali.rice.ksb.api.bus.Endpoint;

 import org.kuali.rice.ksb.api.bus.ServiceBus;

 import org.kuali.rice.ksb.api.bus.ServiceConfiguration;

 import org.kuali.rice.ksb.api.registry.ServiceInfo;

 import org.kuali.rice.ksb.api.registry.ServiceRegistry;

 import javax.servlet.ServletException;

 import javax.servlet.http.HttpServletRequest;

 import javax.servlet.http.HttpServletResponse;

 import javax.xml.namespace.QName;

 import java.io.IOException;

 import java.util.ArrayList;

 import java.util.List;

 import java.util.Map;

 /**

  * Struts action for interacting with the queue of messages.

  *

  * @author Kuali Rice Team (rice.collab@kuali.org)

  */

 public class ServiceRegistryAction extends KSBAction {

         private static final String REMOVED_APPLICATION_ID_PARAM = "removedApplicationId";

     private static final Logger LOG = Logger.getLogger(ServiceRegistryAction.class);

     public ActionForward start(ActionMapping mapping, ActionForm form, HttpServletRequest request,

             HttpServletResponse response) throws IOException, ServletException {

         return mapping.findForward("basic");

     }

     public ActionForward refreshServiceRegistry(ActionMapping mapping, ActionForm form, HttpServletRequest request,

             HttpServletResponse response) throws IOException, ServletException {

             // TODO is this what really constitutes a "refresh" of the service registry?

             KsbApiServiceLocator.getServiceBus().synchronize();

             return mapping.findForward("basic");

     }

         /**

      * Enable deletion of localhost service registry entries.

      */

     public ActionForward deleteLocalhostEntries(ActionMapping mapping, ActionForm form, HttpServletRequest request,

                 HttpServletResponse response) throws IOException, ServletException {

             ServiceRegistry serviceRegistry = KsbApiServiceLocator.getServiceRegistry();

             List<ServiceInfo> serviceInfos = serviceRegistry.getAllOnlineServices();

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

             for (ServiceInfo serviceInfo : serviceInfos) {

                     if (serviceInfo.getServerIpAddress().equals("localhost")) {

                             serviceEndpointsToDelete.add(serviceInfo.getServiceId());

                     }

             }

             serviceRegistry.removeServiceEndpoints(serviceEndpointsToDelete);

             KsbApiServiceLocator.getServiceBus().synchronize();

                 return mapping.findForward("basic");

     }

     public ActionForward deleteApplicationIdEntries(ActionMapping mapping, ActionForm form, HttpServletRequest request,

             HttpServletResponse response) throws IOException, ServletException {

         final String applicationId = request.getParameter(REMOVED_APPLICATION_ID_PARAM);

         if(StringUtils.isNotBlank(applicationId)) {

             ServiceRegistry serviceRegistry = KsbApiServiceLocator.getServiceRegistry();

             List<ServiceInfo> serviceInfos = serviceRegistry.getAllOnlineServices();

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

             for (ServiceInfo serviceInfo : serviceInfos) {

                 if (serviceInfo.getApplicationId().equals(applicationId)) {

                     serviceEndpointsToDelete.add(serviceInfo.getServiceId());

                 }

             }

             serviceRegistry.removeServiceEndpoints(serviceEndpointsToDelete);

             KsbApiServiceLocator.getServiceBus().synchronize();

         } else {

             LOG.info("No rows were deleted from the KRSB_SVC_DEF_T table because the application ID was null or blank.");

         }

         return mapping.findForward("basic");

     }

     public ActionMessages establishRequiredState(HttpServletRequest request, ActionForm actionForm) throws Exception {

         ServiceRegistryForm form = (ServiceRegistryForm)actionForm;

         form.setMyIpAddress(RiceUtilities.getIpNumber());

         form.setMyApplicationId(CoreConfigHelper.getApplicationId());

         form.setDevMode(ConfigContext.getCurrentContextConfig().getDevMode());

         ServiceBus serviceBus = KsbApiServiceLocator.getServiceBus();

         form.setMyInstanceId(serviceBus.getInstanceId());

         form.setPublishedServices(getPublishedServices(serviceBus));

         ServiceRegistry serviceRegistry = KsbApiServiceLocator.getServiceRegistry();

         form.setGlobalRegistryServices(getGlobalRegistryServices(serviceRegistry));

         return null;

     }

     private List<ServiceConfiguration> getPublishedServices(ServiceBus serviceBus) {

             Map<QName, Endpoint> localEndpoints = serviceBus.getLocalEndpoints();

             List<ServiceConfiguration> publishedServices = new ArrayList<ServiceConfiguration>();

             for (Endpoint endpoint : localEndpoints.values()) {

                     publishedServices.add(endpoint.getServiceConfiguration());

             }

             return publishedServices;

     }

     private List<ServiceInfo> getGlobalRegistryServices(ServiceRegistry serviceRegistry) {

             return serviceRegistry.getAllServices();

     }

 }