/*
    * (C) Copyright 1989-2011 Trend Micro, Inc.
    * All Rights Reserved.
    *
    * This program is an unpublished copyrighted work which is proprietary
    * to Trend Micro, Inc. and contains confidential information that is not
    * to be reproduced or disclosed to any other person or entity without
    * prior written consent from Trend Micro, Inc. in each and every instance.
    *
   * WARNING:  Unauthorized reproduction of this program as well as
   * unauthorized preparation of derivative works based upon the
   * program or distribution of copies by sale, rental, lease or
   * lending are violations of federal copyright laws and state trade
   * secret laws, punishable by civil and criminal penalties.
   */
  
  package com.trendmicro.grid.acl.l0;
  
  import com.trendmicro.grid.acl.client.AbstractClient;
  import com.trendmicro.grid.acl.l0.datatypes.NameList;
  import com.trendmicro.grid.acl.l0.datatypes.UsageStatistics;
  import com.trendmicro.grid.acl.l0.datatypes.UsageStatisticsCollection;
  import org.springframework.beans.factory.annotation.Autowired;
  import org.springframework.stereotype.Service;
  
  import javax.annotation.PostConstruct;
  import javax.annotation.Resource;
  import javax.ws.rs.*;
  import java.util.Collection;
  import java.util.LinkedHashSet;
  import java.util.Set;
  
  /**
   * Implements a REST service that provides the values offered by the server statistics service.
   *
   * @author Juergen_Kellerer, 2011-04-12
   * @version 1.0
   */
  @Service
  @Path("/internal/server-statistics")
  @Produces({"application/xml", "application/json"})
  public class ServerStatisticsRestService implements Level0RestService {
  
  	@Resource
  	InternalCacheControlService cacheControlService;
  	@Resource
  	ServerStatisticsService localStatisticsService;
  	@Autowired(required = false)
  	Collection<AbstractClient> soapClients;
  
  	AbstractClient<ServerStatisticsService> statisticsServiceClient;
  
  	@PostConstruct
  	@SuppressWarnings("unchecked")
  	void selectServiceClient() {
  		if (soapClients != null) {
  			Class<ServerStatisticsService> eif = ServerStatisticsService.class;
  			for (AbstractClient soapClient : soapClients)
  				if (eif.equals(soapClient.getEndpointInterface()))
  					statisticsServiceClient = (AbstractClient<ServerStatisticsService>) soapClient;
  
  			if (statisticsServiceClient == null) {
  				throw new IllegalStateException("No soap client was found that satisfies " +
  						"the endpoint interface " + eif);
  			}
  		}
  	}
  
  	ServerStatisticsService selectStatisticsService(String nodeName) {
  		RestUtil.assertIsNotCrossSiteScriptingVulnerable(nodeName);
  
  		if (nodeName == null || nodeName.isEmpty() || statisticsServiceClient == null ||
  				"localhost".equalsIgnoreCase(nodeName) || nodeName.startsWith("127.0"))
  			return localStatisticsService;
  		else {
  			if (extractNetworkAddresses(
  					cacheControlService.getLocalClusterNodeAddressListInternal()).contains(nodeName))
  				return localStatisticsService;
  
  			Set<String> validNodeNames = extractNetworkAddresses(
  					cacheControlService.getCacheClusterNodeAddressListInternal());
  			if (!validNodeNames.contains(nodeName)) {
  				throw new IllegalArgumentException("Specified a node name of '" + nodeName +
  						"' that is not contained within " + validNodeNames);
  			}
  
  			return statisticsServiceClient.getRemotePort(nodeName);
  		}
  	}
  
  	Set<String> extractNetworkAddresses(Collection<String> sourceAddresses) {
  		Set<String> addresses = new LinkedHashSet<String>(sourceAddresses.size());
  		for (String sourceAddress : sourceAddresses) {
  			int idx = sourceAddress.indexOf('@');
  			if (idx == -1)
  				continue;
  			addresses.add(sourceAddress.substring(idx + 1));
  		}
  		return addresses;
 	}
 
 	/**
 	 * Returns the node names (=node-addresses) of all ACL nodes within a cache cluster.
 	 *
 	 * @return the node names (=node-addresses) of all ACL nodes within a cache cluster.
 	 */
 	@GET
 	@Path("/nodeNames")
 	public NameList getNodeNames() {
 		Set<String> addresses = extractNetworkAddresses(cacheControlService.getCacheClusterNodeAddressListInternal());
 		return new NameList(addresses);
 	}
 
 	/**
 	 * Delegates to {@link ServerStatisticsService#getOverallUsageStatistics()}.
 	 * <p/>
 	 * The rest service optionally accepts to specify a node name allowing to retrieve the value from
 	 * another node than the one it connected to.
 	 * (This allows cross site scripting on this particular service,
 	 * which is needed to display multiple node's data on a single page)
 	 *
 	 * @param nodeName the optional name of the node (defaults to 'localhost').
 	 * @return the delegate's return value.
 	 */
 	@GET
 	@Path("/overallUsageStatistics/{nodeName}")
 	public UsageStatistics getOverallUsageStatistics(
 			@DefaultValue("localhost") @PathParam("nodeName") String nodeName) {
 		return selectStatisticsService(nodeName).getOverallUsageStatistics();
 	}
 
 	/**
 	 * Delegates to {@link ServerStatisticsService#getCurrentUsageStatistics()}.
 	 * <p/>
 	 * The rest service optionally accepts to specify a node name allowing to retrieve the value from
 	 * another node than the one it connected to.
 	 * (This allows cross site scripting on this particular service,
 	 * which is needed to display multiple node's data on a single page)
 	 *
 	 * @param nodeName the optional name of the node (defaults to 'localhost').
 	 * @return the delegate's return value.
 	 */
 	@GET
 	@Path("/currentUsageStatistics/{nodeName}")
 	public UsageStatistics getCurrentUsageStatistics(
 			@DefaultValue("localhost") @PathParam("nodeName") String nodeName) {
 		return selectStatisticsService(nodeName).getCurrentUsageStatistics();
 	}
 
 	/**
 	 * Delegates to {@link ServerStatisticsService#getCollectedUsageStatistics()}.
 	 * <p/>
 	 * The rest service optionally accepts to specify a node name allowing to retrieve the value from
 	 * another node than the one it connected to.
 	 * (This allows cross site scripting on this particular service,
 	 * which is needed to display multiple node's data on a single page)
 	 *
 	 * @param nodeName the optional name of the node (defaults to 'localhost').
 	 * @return the delegate's return value.
 	 */
 	@GET
 	@Path("/collectedUsageStatistics/{nodeName}")
 	public UsageStatisticsCollection getCollectedUsageStatistics(
 			@DefaultValue("localhost") @PathParam("nodeName") String nodeName) {
 		return selectStatisticsService(nodeName).getCollectedUsageStatistics();
 	}
 }