From 6adfa1dca65618dc4986a80fd795a544b31bc6b9 Mon Sep 17 00:00:00 2001
From: Pasi Kallinen <paxed@alt.org>
Date: Tue, 2 Mar 2010 17:42:24 +0000
Subject: [PATCH] Accept spectate commands when doing catchup-playback for
 ttyrecs.  From Darshan Shaligram

git-svn-id: svn://katsu.triplehelix.org/dgamelaunch/trunk@529 db0b04b0-f4d1-0310-9a6d-de3e77497b0e
---
 ttyplay.c | 143 +++++++++++++++++++++++++++++++++---------------------
 ttyplay.h |   3 +-
 2 files changed, 89 insertions(+), 57 deletions(-)

diff --git a/ttyplay.c b/ttyplay.c
index e13e169..35347ee 100644
--- a/ttyplay.c
+++ b/ttyplay.c
@@ -122,11 +122,84 @@ ttynowait (struct timeval prev, struct timeval cur, double speed)
   return 0;                     /* Speed isn't important. */
 }
 
+int
+kbhit(void)
+{
+    int i = 0;
+    nodelay(stdscr, TRUE);
+    timeout(0);
+    i = wgetch(stdscr);
+    nodelay(stdscr, FALSE);
+
+    if (i == -1)
+        i = 0;
+    else
+        ungetch(i);
+    return (i);
+}
+
+int
+ttyplay_keyboard_action(int c)
+{
+    struct termios t;
+    switch (c)
+    {
+    case 'q':
+        return READ_QUIT;
+    case 'r':
+        if (term_resizex > 0 && term_resizey > 0) {
+            printf ("\033[8;%d;%dt", term_resizey, term_resizex);
+            return READ_RESTART;
+        }
+        break;
+    case 's':
+        switch (stripped)
+        {
+	case NO_GRAPHICS: populate_gfx_array ((stripped = DEC_GRAPHICS)); break;
+	case DEC_GRAPHICS: populate_gfx_array ((stripped = IBM_GRAPHICS)); break;
+	case IBM_GRAPHICS: populate_gfx_array ((stripped = NO_GRAPHICS)); break;
+        }
+        return READ_RESTART;
+
+    case 'm':
+        tcgetattr (0, &t);
+        if (!loggedin)
+        {
+            initcurses();
+            loginprompt(1);
+        }
+        if (loggedin)
+        {
+            initcurses ();
+            domailuser (chosen_name);
+        }
+        endwin ();
+        tcsetattr (0, TCSANOW, &t);
+        return READ_RESTART;
+    case '?':
+        tcgetattr (0, &t);
+        initcurses();
+        (void) runmenuloop(dgl_find_menu("watchmenu_help"));
+        endwin ();
+        tcsetattr (0, TCSANOW, &t);
+        return READ_RESTART;
+    }
+    return (READ_DATA);
+}
+
 int
 ttyread (FILE * fp, Header * h, char **buf, int pread)
 {
   long offset;
 
+  if (kbhit())
+      {
+	  const int c = wgetch(stdscr);
+	  const int action = ttyplay_keyboard_action(c);
+	  if (action != READ_DATA)
+	      return (action);
+      }
+
   /* do this BEFORE header read, hlen bug */
   offset = ftell (fp);
 
@@ -171,9 +244,8 @@ ttypread (FILE * fp, Header * h, char **buf, int pread)
   struct timeval origw = { 0, 100000 };
   int counter = 0;
   fd_set readfs;
-  struct termios t;
   int doread = 0;
-  static int tried_resize = 0;
+  int action = READ_DATA;
 
 #ifdef HAVE_KQUEUE
   if (kq == -1)
@@ -188,7 +260,7 @@ ttypread (FILE * fp, Header * h, char **buf, int pread)
   /*
    * Read persistently just like tail -f.
    */
-  while (ttyread (fp, h, buf, 1) == READ_EOF)
+  while ((action = ttyread (fp, h, buf, 1)) == READ_EOF)
     {
       fflush(stdout);
       clearerr (fp);
@@ -249,55 +321,13 @@ ttypread (FILE * fp, Header * h, char **buf, int pread)
           char c;
           read (STDIN_FILENO, &c, 1); /* drain the character */
 
-          switch (c)
-            {
-            case 'q':
-              return READ_EOF;
-              break;
-	    case 'r':
-		if (term_resizex > 0 && term_resizey > 0) {
-		    printf ("\033[8;%d;%dt", term_resizey, term_resizex);
-		    return READ_RESTART;
-		}
-		break;
-	    case 's':
-	      switch (stripped)
-	      {
-		case NO_GRAPHICS: populate_gfx_array ((stripped = DEC_GRAPHICS)); break;
-		case DEC_GRAPHICS: populate_gfx_array ((stripped = IBM_GRAPHICS)); break;
-		case IBM_GRAPHICS: populate_gfx_array ((stripped = NO_GRAPHICS)); break;
-	      }
-	      return READ_RESTART;
-	      break;
-
-            case 'm':
-	      tcgetattr (0, &t);
-	      if (!loggedin)
-	      {
-		initcurses();
-		loginprompt(1);
-	      }
-              if (loggedin)
-	      {
-		initcurses ();
-		domailuser (chosen_name);
-	      }
-              endwin ();
-	      tcsetattr (0, TCSANOW, &t);
-              return READ_RESTART;
-              break;
-            case '?':
-		tcgetattr (0, &t);
-		initcurses();
-		(void) runmenuloop(dgl_find_menu("watchmenu_help"));
-		endwin ();
-		tcsetattr (0, TCSANOW, &t);
-		return READ_RESTART;
-		break;
-            }
+	  action = ttyplay_keyboard_action(c);
+	  if (action != READ_DATA)
+	      return action;
+
         }
     }
-  return READ_DATA;
+  return (action);
 }
 
 void
@@ -469,15 +499,16 @@ void
 ttypeek (FILE * fp, double speed)
 {
   int r;
-
   do
   {
     setvbuf (fp, NULL, _IOFBF, 0);
-    ttyplay (fp, 0, ttyread, ttywrite, ttynowait, find_seek_offset_clrscr (fp));
-    clearerr (fp);
-    setvbuf (fp, NULL, _IONBF, 0);
-    fflush (stdout);
-    r = ttyplay (fp, speed, ttypread, ttywrite, ttynowait, -1);
+    r = ttyplay(fp, 0, ttyread, ttywrite, ttynowait, find_seek_offset_clrscr(fp));
+    if (r == READ_EOF) {
+	clearerr (fp);
+        setvbuf (fp, NULL, _IONBF, 0);
+        fflush (stdout);
+        r = ttyplay (fp, speed, ttypread, ttywrite, ttynowait, -1);
+    }
   } while (r == READ_RESTART);
 }
 
diff --git a/ttyplay.h b/ttyplay.h
index 27e1a6c..a734cad 100644
--- a/ttyplay.h
+++ b/ttyplay.h
@@ -13,7 +13,8 @@ typedef void (*WriteFunc) (char *buf, int len);
 
 /* Return values for ReadFunc (and ProcessFunc) */
 #define READ_DATA	0 /* Data */
-#define READ_EOF	1 /* Normal EOF or user aborted */
+#define READ_EOF	1 /* Normal EOF */
 #define READ_RESTART	2 /* Screen must be redrawn (after simplemail) */
+#define READ_QUIT	3 /* User aborted */
 
 #endif /* !INCLUDED_ttyplay_h */
-- 
2.47.3