weinre.server/src/com/phonegap/weinre/server/ConnectionManager.java

/*
 * weinre is available under *either* the terms of the modified BSD license *or* the
 * MIT License (2008). See http://opensource.org/licenses/alphabetical for full text.
 * 
 * Copyright (c) 2010, 2011 IBM Corporation
 */

package com.phonegap.weinre.server;

import java.util.ArrayList;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 
 */
public class ConnectionManager {

    static final public ConnectionManager $ = new ConnectionManager();
    
    private Map<String,Client> clientMap;
    private Map<String,Target> targetMap;
    private boolean            listening = false; 

    /**
     * 
     */
    private ConnectionManager() {
        super();
        
        clientMap = Collections.synchronizedMap(new HashMap<String,Client>());
        targetMap = Collections.synchronizedMap(new HashMap<String,Target>());
    }

    /**
     * 
     */
    public void startChannelListener() {
        if (listening) return;
        listening = true;
        
        ChannelManager.$.addEventListener(new ChannelManagerEventListener() {
            
            @Override
            public void channelRegistered(Channel channel) {
            }

            @Override
            public void channelDeregistered(Channel channel) {
                Connector connector = channel.getConnector();
                if (null == connector) return;

                if (connector.isClient()) ConnectionManager.$._closeClient((Client) connector);
                if (connector.isTarget()) ConnectionManager.$._closeTarget((Target) connector);
            }
        });
    }
    
    /**
     * 
     */
    public void addClient(Client client) {
        clientMap.put(client.getName(), client);
        _sendAllClientsEvent("WeinreClientEvents", "clientRegistered", client.getDescription());
    }

    /**
     * 
     */
    public void addTarget(Target target) {
        targetMap.put(target.getName(), target);
        _sendAllClientsEvent("WeinreClientEvents", "targetRegistered", target.getDescription());
    }

    /**
     * 
     */
    private void _removeClient(Client client) {
        _sendAllClientsEvent("WeinreClientEvents", "clientUnregistered", client.getName());
        clientMap.remove(client.getName());
    }

    /**
     * 
     */
    private void _removeTarget(Target target) {
        _sendAllClientsEvent("WeinreClientEvents", "targetUnregistered", target.getName());
        targetMap.remove(target.getName());
    }

    /**
     * 
     */
    public Client getClient(String id) {
        return clientMap.get(id);
    }

    /**
     * 
     */
    public Target getTarget(String id) {
        return targetMap.get(id);
    }

    /**
     * 
     */
    public List<Client> getClients() {
        List<Client> result = new ArrayList<Client>();
        
        result.addAll(clientMap.values());
        
        return result;
    }

    /**
     * 
     */
    public List<Target> getTargets() {
        List<Target> result = new ArrayList<Target>();
        
        result.addAll(targetMap.values());
        
        return result;
    }

    /**
     * 
     */
    public void connect(Client client, Target target) {
        if ((client == null) || (target == null)) return;

        Target connectedTarget = client.getConnectedTarget();
        if (connectedTarget == target) return;

        disconnect(client, connectedTarget);
        
        client._connect(target);
        target._connect(client);
        
        _sendConnectionCreatedEvent(client, target);
    }

    /**
     * 
     */
    public void disconnect(Client client, Target target) {
        if ((client == null) || (target == null)) return;

        client._disconnect(target);
        target._disconnect(client);

        _sendConnectionDestroyedEvent(client, target);
    }

    /*
     * 
     */
    private void _sendConnectionCreatedEvent(Client client, Target target) {
        _sendConnectionEvent(client, target, "connectionCreated");
    }
    
    /*
     * 
     */
    private void _sendConnectionDestroyedEvent(Client client, Target target) {
        _sendConnectionEvent(client, target, "connectionDestroyed");
    }
    
    /*
     * 
     */
    private void _sendConnectionEvent(Client client, Target target, String message) {
        String clientName = client.getChannel().getName();
        String targetName = target.getChannel().getName();

        _sendAllClientsEvent("WeinreClientEvents", message, clientName, targetName);
        target.getChannel().sendEvent("WeinreTargetEvents", message, clientName, targetName);
    }
    
    /*
     * 
     */
    private void _sendAllClientsEvent(String intfName, String message, Object... args) {
        for (Client aClient: getClients()) {
            aClient.getChannel().sendEvent(intfName, message, args);
        }
    }

    /**
     * 
     */
    protected void closeConnector(Connector connector) {
        if (null == connector) return;
        
        if (connector.isClient()) _closeClient((Client)connector);
        if (connector.isTarget()) _closeTarget((Target)connector);
    }

    /**
     * 
     */
    private void _closeClient(Client client) {
        if (null == client) return;

        Target target = client.getConnectedTarget();
        if (null != target) {
            disconnect(client, target);
        }
        
        _removeClient(client);
    }
    
    /**
     * 
     */
    private void _closeTarget(Target target) {
        if (null == target) return;

        List<Client> clients = target.getConnectedClients();
        
        for (Client client: clients) {
            disconnect(client, target);
        }
        
        _removeTarget(target);
    }
    
}