Allow result and exception handler to be changed

1 files changed, 6 insertions, 1 deletions

diff --git a/src/main/TweetShell.java b/src/main/TweetShell.java
index 533939b..b318544 100644
--- a/src/main/TweetShell.java
+++ b/src/main/TweetShell.java
@@ -36,7 +36,12 @@ public class TweetShell implements TwitterApi.PinSupplier {
         if (stream_cached == null) {
             OAuthRequester requester = (OAuthRequester) getApi().getRequester();
             StreamHandler handler = new StreamHandler();
-            stream_cached = new StreamImpl(handler, handler, requester);
+            stream_cached = new StreamImpl(requester);
+            StreamImpl streamObserver = (StreamImpl) stream_cached;
+            streamObserver.setExceptionListener(handler);
+            // TODO: wire up DataWriter and add a command ("target"?) to set a
+            // point to save results
+            streamObserver.setResultListener(handler);
         }
         return stream_cached;
     }