Refactor

mattbrictson · Apr 2, 2024 · e9e496f · e9e496f
1 parent 3479fcd
commit e9e496f
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/lib/mighty_test/console.rb b/lib/mighty_test/console.rb
@@ -16,9 +16,10 @@ def clear
     end
 
     def wait_for_keypress
-      return stdin.getc unless stdin.respond_to?(:raw) && tty?
-
-      stdin.raw(intr: true) { stdin.getc }
+      with_raw_input do
+        sleep if stdin.eof?
+        stdin.getc
+      end
     end
 
     def play_sound(name, wait: false)
@@ -54,5 +55,11 @@ def play_sound(name, wait: false)
     def tty?
       $stdout.respond_to?(:tty?) && $stdout.tty?
     end
+
+    def with_raw_input(&)
+      return yield unless stdin.respond_to?(:raw) && tty?
+
+      stdin.raw(intr: true, &)
+    end
   end
 end