Merge commit '77d3a60869e7a780c6ae069e51530e1eacece5e2'

[fanfix.git] / src / jexer / backend / MultiBackend.java

diff --git a/src/jexer/backend/MultiBackend.java b/src/jexer/backend/MultiBackend.java
index a397c65510749d101cdc007f1727453328c12914..d01b9442c0526cc9ce2a67af0be5e8f1a2a514c7 100644 (file)
--- a/src/jexer/backend/MultiBackend.java
+++ b/src/jexer/backend/MultiBackend.java
@@ -3,7 +3,7 @@
  *
  * The MIT License (MIT)
  *
- * Copyright (C) 2017 Kevin Lamonte
+ * Copyright (C) 2019 Kevin Lamonte
  *
  * Permission is hereby granted, free of charge, to any person obtaining a
  * copy of this software and associated documentation files (the "Software"),
@@ -28,16 +28,22 @@
  */
 package jexer.backend;
 
-import java.util.LinkedList;
+import java.util.ArrayList;
 import java.util.List;
 
+import jexer.event.TCommandEvent;
 import jexer.event.TInputEvent;
+import static jexer.TCommand.*;
 
 /**
  * MultiBackend mirrors its I/O to several backends.
  */
 public class MultiBackend implements Backend {
 
+    // ------------------------------------------------------------------------
+    // Variables --------------------------------------------------------------
+    // ------------------------------------------------------------------------
+
     /**
      * The screen to use.
      */
@@ -46,48 +52,47 @@ public class MultiBackend implements Backend {
     /**
      * The list of backends to use.
      */
-    private List<Backend> backends = new LinkedList<Backend>();
+    private List<Backend> backends = new ArrayList<Backend>();
 
     /**
-     * Public constructor requires one backend.  Note that this backend's
-     * screen will be replaced with a MultiScreen.
-     *
-     * @param backend the backend to add
+     * The SessionInfo to return.
      */
-    public MultiBackend(final Backend backend) {
-        backends.add(backend);
-        multiScreen = new MultiScreen(backend.getScreen());
-    }
+    private SessionInfo sessionInfo;
+
+    // ------------------------------------------------------------------------
+    // Constructors -----------------------------------------------------------
+    // ------------------------------------------------------------------------
 
     /**
-     * Add a backend to the list.
+     * Public constructor requires one backend.  Note that this backend's
+     * screen will be replaced with a MultiScreen.
      *
      * @param backend the backend to add
      */
-    public void addBackend(final Backend backend) {
+    public MultiBackend(final Backend backend) {
         backends.add(backend);
-        multiScreen.addScreen(backend.getScreen());
-    }
-
-    /**
-     * Remove a backend from the list.
-     *
-     * @param backend the backend to remove
-     */
-    public void removeBackend(final Backend backend) {
-        if (backends.size() > 1) {
-            multiScreen.removeScreen(backend.getScreen());
-            backends.remove(backend);
+        if (backend instanceof TWindowBackend) {
+            multiScreen = new MultiScreen(((TWindowBackend) backend).getOtherScreen());
+        } else {
+            multiScreen = new MultiScreen(backend.getScreen());
+        }
+        if (backend instanceof GenericBackend) {
+            ((GenericBackend) backend).abortOnDisconnect = false;
         }
+        sessionInfo = backend.getSessionInfo();
     }
 
+    // ------------------------------------------------------------------------
+    // Backend ----------------------------------------------------------------
+    // ------------------------------------------------------------------------
+
     /**
      * Getter for sessionInfo.
      *
      * @return the SessionInfo
      */
-    public final SessionInfo getSessionInfo() {
-        return backends.get(0).getSessionInfo();
+    public SessionInfo getSessionInfo() {
+        return sessionInfo;
     }
 
     /**
@@ -95,7 +100,7 @@ public class MultiBackend implements Backend {
      *
      * @return the Screen
      */
-    public final Screen getScreen() {
+    public Screen getScreen() {
         return multiScreen;
     }
 
@@ -109,6 +114,23 @@ public class MultiBackend implements Backend {
         }
     }
 
+    /**
+     * Check if there are events in the queue.
+     *
+     * @return if true, getEvents() has something to return to the application
+     */
+    public boolean hasEvents() {
+        if (backends.size() == 0) {
+            return true;
+        }
+        for (Backend backend: backends) {
+            if (backend.hasEvents()) {
+                return true;
+            }
+        }
+        return false;
+    }
+
     /**
      * Subclasses must provide an implementation to get keyboard, mouse, and
      * screen resize events.
@@ -116,8 +138,37 @@ public class MultiBackend implements Backend {
      * @param queue list to append new events to
      */
     public void getEvents(List<TInputEvent> queue) {
+        List<Backend> backendsToRemove = null;
         for (Backend backend: backends) {
-            backend.getEvents(queue);
+            if (backend.hasEvents()) {
+                backend.getEvents(queue);
+
+                // This default backend assumes a single user, and if that
+                // user becomes disconnected we should terminate the
+                // application.
+                if (queue.size() > 0) {
+                    TInputEvent event = queue.get(queue.size() - 1);
+                    if (event instanceof TCommandEvent) {
+                        TCommandEvent command = (TCommandEvent) event;
+                        if (command.equals(cmBackendDisconnect)) {
+                            if (backendsToRemove == null) {
+                                backendsToRemove = new ArrayList<Backend>();
+                            }
+                            backendsToRemove.add(backend);
+                        }
+                    }
+                }
+            }
+        }
+        if (backendsToRemove != null) {
+            for (Backend backend: backendsToRemove) {
+                multiScreen.removeScreen(backend.getScreen());
+                backends.remove(backend);
+                backend.shutdown();
+            }
+        }
+        if (backends.size() == 0) {
+            queue.add(new TCommandEvent(cmAbort));
         }
     }
 
@@ -154,4 +205,50 @@ public class MultiBackend implements Backend {
         }
     }
 
+    /**
+     * Reload backend options from System properties.
+     */
+    public void reloadOptions() {
+        for (Backend backend: backends) {
+            backend.reloadOptions();
+        }
+    }
+
+    // ------------------------------------------------------------------------
+    // MultiBackend -----------------------------------------------------------
+    // ------------------------------------------------------------------------
+
+    /**
+     * Add a backend to the list.
+     *
+     * @param backend the backend to add
+     */
+    public void addBackend(final Backend backend) {
+        backends.add(backend);
+        if (backend instanceof TWindowBackend) {
+            multiScreen.addScreen(((TWindowBackend) backend).getOtherScreen());
+        } else {
+            multiScreen.addScreen(backend.getScreen());
+        }
+        if (backend instanceof GenericBackend) {
+            ((GenericBackend) backend).abortOnDisconnect = false;
+        }
+    }
+
+    /**
+     * Remove a backend from the list.
+     *
+     * @param backend the backend to remove
+     */
+    public void removeBackend(final Backend backend) {
+        if (backends.size() > 1) {
+            if (backend instanceof TWindowBackend) {
+                multiScreen.removeScreen(((TWindowBackend) backend).getOtherScreen());
+            } else {
+                multiScreen.removeScreen(backend.getScreen());
+            }
+            backends.remove(backend);
+        }
+    }
+
 }