hand reason for disconnects to listeners

[jFCPlib.git] / src / net / pterodactylus / fcp / highlevel / HighLevelClient.java

diff --git a/src/net/pterodactylus/fcp/highlevel/HighLevelClient.java b/src/net/pterodactylus/fcp/highlevel/HighLevelClient.java
index 1b52e72..9fd36fd 100644 (file)
--- a/src/net/pterodactylus/fcp/highlevel/HighLevelClient.java
+++ b/src/net/pterodactylus/fcp/highlevel/HighLevelClient.java
@@ -27,8 +27,10 @@ import java.io.IOException;
 import java.net.InetAddress;
 import java.net.URL;
 import java.net.UnknownHostException;
+import java.util.ArrayList;
 import java.util.Collections;
 import java.util.HashMap;
+import java.util.List;
 import java.util.Map;
 import java.util.Map.Entry;
 import java.util.logging.Logger;
@@ -109,6 +111,9 @@ public class HighLevelClient {
        /** The FCP connection to the node. */
        private FcpConnection fcpConnection;
 
+       /** Listeners for high-level client events. */
+       private List<HighLevelClientListener> highLevelClientListeners = Collections.synchronizedList(new ArrayList<HighLevelClientListener>());
+
        /** The listener for the connection. */
        private HighLevelClientFcpListener highLevelClientFcpListener = new HighLevelClientFcpListener();
 
@@ -193,6 +198,52 @@ public class HighLevelClient {
        }
 
        //
+       // EVENT MANAGEMENT
+       //
+
+       /**
+        * Adds the given high-level client listener to list of listeners.
+        *
+        * @param highLevelClientListener
+        *            The listener to add
+        */
+       public void addHighLevelClientListener(HighLevelClientListener highLevelClientListener) {
+               highLevelClientListeners.add(highLevelClientListener);
+       }
+
+       /**
+        * Removes the given high-level client listener from the list of listeners.
+        *
+        * @param highLevelClientListener
+        *            The listener to remove
+        */
+       public void removeHighLevelClientListener(HighLevelClientListener highLevelClientListener) {
+               highLevelClientListeners.remove(highLevelClientListener);
+       }
+
+       /**
+        * Notifies all listeners that a client has connected.
+        */
+       private void fireClientConnected() {
+               for (HighLevelClientListener highLevelClientListener: highLevelClientListeners) {
+                       highLevelClientListener.clientConnected(this);
+               }
+       }
+
+       /**
+        * Notifies all listeners that a client has disconnected.
+        *
+        * @param throwable
+        *            The exception that caused the disconnect, or <code>null</code>
+        *            if there was no exception
+        */
+       private void fireClientDisconnected(Throwable throwable) {
+               for (HighLevelClientListener highLevelClientListener: highLevelClientListeners) {
+                       highLevelClientListener.clientDisconnected(this, throwable);
+               }
+       }
+
+       //
        // ACCESSORS
        //
 
@@ -231,7 +282,7 @@ public class HighLevelClient {
         * Disconnects the client from the node.
         */
        public void disconnect() {
-               fcpConnection.disconnect();
+               disconnect(null);
        }
 
        /**
@@ -386,6 +437,19 @@ public class HighLevelClient {
        }
 
        /**
+        * Disconnects the client from the node, handing the given Throwable to
+        * {@link #fireClientDisconnected(Throwable)}.
+        *
+        * @param throwable
+        *            The exception that caused the disconnect, or <code>null</code>
+        *            if there was no exception
+        */
+       private void disconnect(Throwable throwable) {
+               fcpConnection.close();
+               fireClientDisconnected(throwable);
+       }
+
+       /**
         * FCP listener for {@link HighLevelClient}.
         *
         * @author David ‘Bombe’ Roden &lt;bombe@freenetproject.org&gt;
@@ -560,14 +624,16 @@ public class HighLevelClient {
                //
 
                /**
-                * @see net.pterodactylus.fcp.FcpListener#connectionClosed(net.pterodactylus.fcp.FcpConnection)
+                * @see net.pterodactylus.fcp.FcpListener#connectionClosed(net.pterodactylus.fcp.FcpConnection,
+                *      Throwable)
                 */
                @SuppressWarnings("synthetic-access")
-               public void connectionClosed(FcpConnection fcpConnection) {
+               public void connectionClosed(FcpConnection fcpConnection, Throwable throwable) {
                        if (fcpConnection != HighLevelClient.this.fcpConnection) {
                                return;
                        }
                        cancelIdentifier(null);
+                       disconnect(throwable);
                }
 
                /**
@@ -701,6 +767,7 @@ public class HighLevelClient {
                                connectCallback.setDone();
                                connectCallback = null;
                        }
+                       fireClientConnected();
                }
 
                /**