Prebuffer some data before outputing to player/file.

This is a simple way to check if stream server is
dead/has problems before launching a player/file output.

1 files changed, 7 insertions, 0 deletions

diff --git a/src/livestreamer/cli.py b/src/livestreamer/cli.py
index 5ce15c5..9dc1098 100644
--- a/src/livestreamer/cli.py
+++ b/src/livestreamer/cli.py
@@ -105,6 +105,11 @@ def output_stream(stream, args):
     except livestreamer.StreamError as err:
         exit(("Could not open stream - {0}").format(err))
 
+    try:
+        prebuffer = fd.read(8192)
+    except IOError:
+        exit("Failed to read data from stream")
+
     if args.output:
         if args.output == "-":
             out = stdout
@@ -134,6 +139,8 @@ def output_stream(stream, args):
         import msvcrt
         msvcrt.setmode(out.fileno(), os.O_BINARY)
 
+    out.write(prebuffer)
+
     try:
         write_stream(fd, out, progress)
     except KeyboardInterrupt: