Fix dancing-ruby Ctrl+C stop (#735)

ruby · Oct 21, 2023 · 802b1cb · 802b1cb
1 parent e26e90e
commit 802b1cb
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/irb/easter-egg.rb b/lib/irb/easter-egg.rb
@@ -109,7 +109,9 @@ def render_frame(i)
           end
         end
       when :dancing
-        begin
+        STDOUT.cooked do
+          interrupted = false
+          prev_trap = trap("SIGINT") { interrupted = true }
           canvas = Canvas.new(Reline.get_screen_size)
           Reline::IOGate.set_winch_handler do
             canvas = Canvas.new(Reline.get_screen_size)
@@ -125,10 +127,12 @@ def render_frame(i)
             buff[0, 20] = "\e[0mPress Ctrl+C to stop\e[31m\e[1m"
             print "\e[H" + buff
             sleep 0.05
+            break if interrupted
           end
         rescue Interrupt
         ensure
           print "\e[0m\e[?1049l"
+          trap("SIGINT", prev_trap)
         end
       end
     end