// Copyright 2012 Citrix Systems, Inc. Licensed under the
// Apache License, Version 2.0 (the "License"); you may not use this
// file except in compliance with the License.  Citrix Systems, Inc.
// reserves all rights not expressly granted by the License.
// You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
// 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.
// 
// Automatically generated by addcopyright.py at 04/03/2012
package com.cloud.hypervisor;

import java.util.List;
import java.util.Map;

import javax.ejb.Local;
import javax.naming.ConfigurationException;

import org.apache.log4j.Logger;

import com.cloud.agent.AgentManager;
import com.cloud.agent.StartupCommandProcessor;
import com.cloud.agent.api.StartupCommand;
import com.cloud.agent.api.StartupRoutingCommand;
import com.cloud.agent.api.StartupStorageCommand;
import com.cloud.agent.manager.authn.AgentAuthnException;
import com.cloud.configuration.ConfigurationManager;
import com.cloud.configuration.ZoneConfig;
import com.cloud.configuration.dao.ConfigurationDao;
import com.cloud.dc.ClusterVO;
import com.cloud.dc.DataCenterVO;
import com.cloud.dc.DcDetailVO;
import com.cloud.dc.HostPodVO;
import com.cloud.dc.dao.ClusterDao;
import com.cloud.dc.dao.DataCenterDao;
import com.cloud.dc.dao.DcDetailsDao;
import com.cloud.dc.dao.HostPodDao;
import com.cloud.exception.ConnectionException;
import com.cloud.host.Host;
import com.cloud.host.Host.Type;
import com.cloud.host.HostVO;
import com.cloud.host.dao.HostDao;
import com.cloud.hypervisor.Hypervisor.HypervisorType;
import com.cloud.storage.Storage;
import com.cloud.utils.component.Inject;
import com.cloud.utils.exception.CloudRuntimeException;
import com.cloud.utils.net.MacAddress;
import com.cloud.utils.net.NetUtils;

/**
 * Creates a host record and supporting records such as pod and ip address
 *
 */
@Local(value=StartupCommandProcessor.class)
public class CloudZonesStartupProcessor implements StartupCommandProcessor {
    private static final Logger s_logger = Logger.getLogger(CloudZonesStartupProcessor.class);
    @Inject ClusterDao _clusterDao = null;
    @Inject ConfigurationDao _configDao = null;
    @Inject DataCenterDao _zoneDao = null;
    @Inject HostDao _hostDao = null;
    @Inject HostPodDao _podDao = null;
    @Inject DcDetailsDao _zoneDetailsDao = null;
    
    @Inject AgentManager _agentManager = null;
    @Inject ConfigurationManager _configurationManager = null;
    
    long _nodeId = -1;
   

    @Override
    public boolean configure(String name, Map<String, Object> params)
            throws ConfigurationException {
        _agentManager.registerForInitialConnects(this, false);
        if (_nodeId == -1) {
            // FIXME: We really should not do this like this. It should be done
            // at config time and is stored as a config variable.
            _nodeId = MacAddress.getMacAddress().toLong();
        }
        return true;
    }

    
    @Override
    public String getName() {
        return getClass().getName();
    }

 
    @Override
    public boolean start() {
       return true;
    }


    @Override
    public boolean stop() {
        return true;
    }

    @Override
    public boolean processInitialConnect(StartupCommand[] cmd)
            throws ConnectionException {
        StartupCommand startup = cmd[0];
        if (startup instanceof StartupRoutingCommand) {
            return processHostStartup((StartupRoutingCommand)startup);
        } else if (startup instanceof StartupStorageCommand ){
            return processStorageStartup((StartupStorageCommand)startup);
        }

        return false;
    }
    
    protected boolean processHostStartup(StartupRoutingCommand startup) throws ConnectionException{
    	/*
        boolean found = false;
        Type type = Host.Type.Routing;
        final Map<String, String> hostDetails = startup.getHostDetails();
        HostVO server = _hostDao.findByGuid(startup.getGuid());
        if (server == null) {
            server = _hostDao.findByGuid(startup.getGuidWithoutResource());
        }
        if (server != null && server.getRemoved() == null) {
            if (s_logger.isDebugEnabled()) {
                s_logger.debug("Found the host " + server.getId() + " by guid: "
                        + startup.getGuid());
            }
            found = true;

        } else {
            server = new HostVO(startup.getGuid());
        }
        server.setDetails(hostDetails);

        try {
            updateComputeHost(server, startup, type);
        } catch (AgentAuthnException e) {
            throw new ConnectionException(true, "Failed to authorize host, invalid configuration", e);
        }
        if (!found) {
            server.setHostAllocationState(Host.HostAllocationState.Enabled);
            server = _hostDao.persist(server);
        } else {
            if (!_hostDao.connect(server, _nodeId)) {
                throw new CloudRuntimeException(
                        "Agent cannot connect because the current state is "
                        + server.getStatus().toString());
            }
            s_logger.info("Old " + server.getType().toString()
                    + " host reconnected w/ id =" + server.getId());
        }
        */
        return true;
   
        
    }
    
    protected void updateComputeHost(final HostVO host, final StartupCommand startup,
            final Host.Type type) throws AgentAuthnException {

        String zoneToken = startup.getDataCenter();
        if(zoneToken == null){
        	s_logger.warn("No Zone Token passed in, cannot not find zone for the agent");
        	throw new AgentAuthnException("No Zone Token passed in, cannot not find zone for agent");
        }
        
        DataCenterVO zone = _zoneDao.findByToken(zoneToken);
        if (zone == null) {
        	zone = _zoneDao.findByName(zoneToken);
        	if(zone == null){
	            try {
	                long zoneId = Long.parseLong(zoneToken);
	                zone = _zoneDao.findById(zoneId);
	        		if (zone == null) {
	        			throw new AgentAuthnException("Could not find zone for agent with token " + zoneToken);
	        		}
	            } catch (NumberFormatException nfe) {
	                throw new AgentAuthnException("Could not find zone for agent with token " + zoneToken);
	            }
        	}
        }
    	if(s_logger.isDebugEnabled()){
    		s_logger.debug("Successfully loaded the DataCenter from the zone token passed in ");
    	}
        
        long zoneId = zone.getId();
        DcDetailVO maxHostsInZone = _zoneDetailsDao.findDetail(zoneId, ZoneConfig.MaxHosts.key());
        if(maxHostsInZone != null){
        	long maxHosts = new Long(maxHostsInZone.getValue()).longValue();
        	long currentCountOfHosts = _hostDao.countRoutingHostsByDataCenter(zoneId);
        	if(s_logger.isDebugEnabled()){
        		s_logger.debug("Number of hosts in Zone:"+ currentCountOfHosts + ", max hosts limit: "+ maxHosts);
        	}
        	if(currentCountOfHosts >= maxHosts){
        		throw new AgentAuthnException("Number of running Routing hosts in the Zone:"+ zone.getName() +" is already at the max limit:"+maxHosts+", cannot start one more host");
        	}
        }
        
        HostPodVO pod = null;
        
    	if(startup.getPrivateIpAddress() == null){
       		s_logger.warn("No private IP address passed in for the agent, cannot not find pod for agent");
    		throw new AgentAuthnException("No private IP address passed in for the agent, cannot not find pod for agent");
    	}
    	
    	if(startup.getPrivateNetmask() == null){
       		s_logger.warn("No netmask passed in for the agent, cannot not find pod for agent");
    		throw new AgentAuthnException("No netmask passed in for the agent, cannot not find pod for agent");
    	}
    	        
        if(host.getPodId() != null){
        	if(s_logger.isDebugEnabled()){
        		s_logger.debug("Pod is already created for this agent, looks like agent is reconnecting...");
        	}
        	pod = _podDao.findById(host.getPodId());
        	if(!checkCIDR(type, pod, startup.getPrivateIpAddress(),
					startup.getPrivateNetmask())){
        		pod = null;
            	if(s_logger.isDebugEnabled()){
            		s_logger.debug("Subnet of Pod does not match the subnet of the agent, not using this Pod: "+host.getPodId());
            	}        		
        	}else{
        		updatePodNetmaskIfNeeded(pod, startup.getPrivateNetmask());
        	}
        }
        
        if(pod == null){
	    	if(s_logger.isDebugEnabled()){
	    		s_logger.debug("Trying to detect the Pod to use from the agent's ip address and netmask passed in ");
	    	}
	    	
	        //deduce pod
	    	boolean podFound = false;
	        List<HostPodVO> podsInZone = _podDao.listByDataCenterId(zoneId);
			for (HostPodVO hostPod : podsInZone) {
				if (checkCIDR(type, hostPod, startup.getPrivateIpAddress(),
						startup.getPrivateNetmask())) {
					pod = hostPod;
					
					//found the default POD having the same subnet.
					updatePodNetmaskIfNeeded(pod, startup.getPrivateNetmask());
					podFound = true;
					break;
				}
			}

        
	        if (!podFound) {
	        	if(s_logger.isDebugEnabled()){
	        		s_logger.debug("Creating a new Pod since no default Pod found that matches the agent's ip address and netmask passed in ");
	        	}
	        	
	        	if(startup.getGatewayIpAddress() == null){
	           		s_logger.warn("No Gateway IP address passed in for the agent, cannot create a new pod for the agent");
	        		throw new AgentAuthnException("No Gateway IP address passed in for the agent, cannot create a new pod for the agent");
	        	}
	        	//auto-create a new pod, since pod matching the agent's ip is not found
	        	String podName = "POD-"+ (podsInZone.size()+1);
	        	try{
		        	String gateway = startup.getGatewayIpAddress();
		        	String cidr = NetUtils.getCidrFromGatewayAndNetmask(gateway, startup.getPrivateNetmask());
		            String[] cidrPair = cidr.split("\\/");
		            String cidrAddress = cidrPair[0];
		            long cidrSize = Long.parseLong(cidrPair[1]);
		        	String startIp = NetUtils.getIpRangeStartIpFromCidr(cidrAddress, cidrSize);
		        	String endIp = NetUtils.getIpRangeEndIpFromCidr(cidrAddress, cidrSize);
	        		pod = _configurationManager.createPod(-1, podName, zoneId, gateway, cidr, startIp, endIp, null, true);
	        	}catch (Exception e) {
					// no longer tolerate exception during the cluster creation phase
					throw new CloudRuntimeException("Unable to create new Pod "
							+ podName + " in Zone: "
							+ zoneId, e);
				}
	        	
	        }
        }
        final StartupRoutingCommand scc = (StartupRoutingCommand) startup;

        ClusterVO cluster = null;
        if(host.getClusterId() != null){
        	if(s_logger.isDebugEnabled()){
        		s_logger.debug("Cluster is already created for this agent, looks like agent is reconnecting...");
        	}        	
        	cluster = _clusterDao.findById(host.getClusterId());
        }
        if(cluster == null){
            //auto-create cluster - assume one host per cluster
            String clusterName = "Cluster-" + startup.getPrivateIpAddress();
            ClusterVO existingCluster = _clusterDao.findBy(clusterName, pod.getId());
            if (existingCluster != null) {
                cluster = existingCluster;
            } else {
                if(s_logger.isDebugEnabled()){
                    s_logger.debug("Creating a new Cluster for this agent with name: "+clusterName + " in Pod: " +pod.getId() +", in Zone:"+zoneId);
                }

                cluster = new ClusterVO(zoneId, pod.getId(), clusterName);
                cluster.setHypervisorType(scc.getHypervisorType().toString());
                try {
                    cluster = _clusterDao.persist(cluster);
                } catch (Exception e) {
                    // no longer tolerate exception during the cluster creation phase
                    throw new CloudRuntimeException("Unable to create cluster "
                            + clusterName + " in pod " + pod.getId() + " and data center "
                            + zoneId, e);
                }
            }
        }

    	if(s_logger.isDebugEnabled()){
    		s_logger.debug("Detected Zone: "+zoneId+", Pod: " +pod.getId() +", Cluster:" +cluster.getId());
    	}        
		host.setDataCenterId(zone.getId());
        host.setPodId(pod.getId());
        host.setClusterId(cluster.getId());
        host.setPrivateIpAddress(startup.getPrivateIpAddress());
        host.setPrivateNetmask(startup.getPrivateNetmask());
        host.setPrivateMacAddress(startup.getPrivateMacAddress());
        host.setPublicIpAddress(startup.getPublicIpAddress());
        host.setPublicMacAddress(startup.getPublicMacAddress());
        host.setPublicNetmask(startup.getPublicNetmask());
        host.setStorageIpAddress(startup.getStorageIpAddress());
        host.setStorageMacAddress(startup.getStorageMacAddress());
        host.setStorageNetmask(startup.getStorageNetmask());
        host.setVersion(startup.getVersion());
        host.setName(startup.getName());
        host.setType(type);
        host.setStorageUrl(startup.getIqn());
        host.setLastPinged(System.currentTimeMillis() >> 10);
        host.setCaps(scc.getCapabilities());
        host.setCpus(scc.getCpus());
        host.setTotalMemory(scc.getMemory());
        host.setSpeed(scc.getSpeed());
        HypervisorType hyType = scc.getHypervisorType();
        host.setHypervisorType(hyType);
        host.setHypervisorVersion(scc.getHypervisorVersion());
        
    }
    
	private boolean checkCIDR(Host.Type type, HostPodVO pod,
			String serverPrivateIP, String serverPrivateNetmask) {
		if (serverPrivateIP == null) {
			return true;
		}
		// Get the CIDR address and CIDR size
		String cidrAddress = pod.getCidrAddress();
		long cidrSize = pod.getCidrSize();

		// If the server's private IP address is not in the same subnet as the
		// pod's CIDR, return false
		String cidrSubnet = NetUtils.getCidrSubNet(cidrAddress, cidrSize);
		String serverSubnet = NetUtils.getSubNet(serverPrivateIP,
				serverPrivateNetmask);
		if (!cidrSubnet.equals(serverSubnet)) {
			return false;
		}
		return true;
	}
	
	private void updatePodNetmaskIfNeeded(HostPodVO pod, String agentNetmask){
		// If the server's private netmask is less inclusive than the pod's CIDR
		// netmask, update cidrSize of the default POD 
		//(reason: we are maintaining pods only for internal accounting.)
		long cidrSize = pod.getCidrSize();
		String cidrNetmask = NetUtils
				.getCidrSubNet("255.255.255.255", cidrSize);
		long cidrNetmaskNumeric = NetUtils.ip2Long(cidrNetmask);
		long serverNetmaskNumeric = NetUtils.ip2Long(agentNetmask);//
		if (serverNetmaskNumeric > cidrNetmaskNumeric) {
			//update pod's cidrsize
			int newCidrSize = new Long(NetUtils.getCidrSize(agentNetmask)).intValue();
			pod.setCidrSize(newCidrSize);
			_podDao.update(pod.getId(), pod);
		}		
	}
	
	
	protected boolean processStorageStartup(StartupStorageCommand startup) throws ConnectionException{
		/*
	    if (startup.getResourceType() != Storage.StorageResourceType.LOCAL_SECONDARY_STORAGE) {
            return false;
        }
	    boolean found = false;
        Type type = Host.Type.LocalSecondaryStorage;
        final Map<String, String> hostDetails = startup.getHostDetails();
        HostVO server = _hostDao.findByGuid(startup.getGuid());
        if (server == null) {
            server = _hostDao.findByGuid(startup.getGuidWithoutResource());
        }
        if (server != null && server.getRemoved() == null) {
            if (s_logger.isDebugEnabled()) {
                s_logger.debug("Found the host " + server.getId() + " by guid: "
                        + startup.getGuid());
            }
            found = true;

        } else {
            server = new HostVO(startup.getGuid());
        }
        server.setDetails(hostDetails);

        try {
            updateSecondaryHost(server, startup, type);
        } catch (AgentAuthnException e) {
            throw new ConnectionException(true, "Failed to authorize host, invalid configuration", e);
        }
        if (!found) {
            server.setHostAllocationState(Host.HostAllocationState.Enabled);
            server = _hostDao.persist(server);
        } else {
            if (!_hostDao.connect(server, _nodeId)) {
                throw new CloudRuntimeException(
                        "Agent cannot connect because the current state is "
                        + server.getStatus().toString());
            }
            s_logger.info("Old " + server.getType().toString()
                    + " host reconnected w/ id =" + server.getId());
        }
        */
        return true;
   
        
    }
	protected void updateSecondaryHost(final HostVO host, final StartupStorageCommand startup,
            final Host.Type type) throws AgentAuthnException {
	  
        String zoneToken = startup.getDataCenter();
        if(zoneToken == null){
            s_logger.warn("No Zone Token passed in, cannot not find zone for the agent");
            throw new AgentAuthnException("No Zone Token passed in, cannot not find zone for agent");
        }
        
        DataCenterVO zone = _zoneDao.findByToken(zoneToken);
        if (zone == null) {
            zone = _zoneDao.findByName(zoneToken);
            if(zone == null){
                try {
                    long zoneId = Long.parseLong(zoneToken);
                    zone = _zoneDao.findById(zoneId);
                    if (zone == null) {
                        throw new AgentAuthnException("Could not find zone for agent with token " + zoneToken);
                    }
                } catch (NumberFormatException nfe) {
                    throw new AgentAuthnException("Could not find zone for agent with token " + zoneToken);
                }
            }
        }
        if(s_logger.isDebugEnabled()){
            s_logger.debug("Successfully loaded the DataCenter from the zone token passed in ");
        }
        
        HostPodVO pod = findPod(startup, zone.getId(), Host.Type.Routing); //yes, routing
        Long podId = null;
        if (pod != null) {
            s_logger.debug("Found pod " + pod.getName() + " for the secondary storage host " + startup.getName());
            podId = pod.getId();
        }
        host.setDataCenterId(zone.getId());
        host.setPodId(podId);
        host.setClusterId(null);
        host.setPrivateIpAddress(startup.getPrivateIpAddress());
        host.setPrivateNetmask(startup.getPrivateNetmask());
        host.setPrivateMacAddress(startup.getPrivateMacAddress());
        host.setPublicIpAddress(startup.getPublicIpAddress());
        host.setPublicMacAddress(startup.getPublicMacAddress());
        host.setPublicNetmask(startup.getPublicNetmask());
        host.setStorageIpAddress(startup.getStorageIpAddress());
        host.setStorageMacAddress(startup.getStorageMacAddress());
        host.setStorageNetmask(startup.getStorageNetmask());
        host.setVersion(startup.getVersion());
        host.setName(startup.getName());
        host.setType(type);
        host.setStorageUrl(startup.getIqn());
        host.setLastPinged(System.currentTimeMillis() >> 10);
        host.setCaps(null);
        host.setCpus(null);
        host.setTotalMemory(0);
        host.setSpeed(null);
        host.setParent(startup.getParent());
        host.setTotalSize(startup.getTotalSize());
        host.setHypervisorType(HypervisorType.None);
        if (startup.getNfsShare() != null) {
            host.setStorageUrl(startup.getNfsShare());
        }

    }
	
	private HostPodVO findPod(StartupCommand startup, long zoneId, Host.Type type) {
	    HostPodVO pod = null;
	    List<HostPodVO> podsInZone = _podDao.listByDataCenterId(zoneId);
        for (HostPodVO hostPod : podsInZone) {
            if (checkCIDR(type, hostPod, startup.getPrivateIpAddress(),
                    startup.getPrivateNetmask())) {
                pod = hostPod;
                
                //found the default POD having the same subnet.
                updatePodNetmaskIfNeeded(pod, startup.getPrivateNetmask());
                
                break;
            }
        }
        return pod;

	}

}