From 8ce1437b742266e5276c27ab7cc21df6f2bb62a1 Mon Sep 17 00:00:00 2001
From: Jeremy Kaplan <jdkaplan@metagram.net>
Date: Thu, 5 Dec 2019 23:06:33 -0800
Subject: [PATCH] Move user input to run method

---
 2019/day5/intcode.rb | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/2019/day5/intcode.rb b/2019/day5/intcode.rb
index f60ee22..4826b1f 100644
--- a/2019/day5/intcode.rb
+++ b/2019/day5/intcode.rb
@@ -33,11 +33,7 @@ module Intcode
       @initial_state = mem
     end
 
-    def on_input(&block)
-      @input = block
-    end
-
-    def run
+    def run(&user_input)
       @mem = Memory.new(@initial_state.clone)
       ip = 0
       loop do
@@ -68,7 +64,7 @@ module Intcode
           r1 = mem[ip + 1]
           raise StandardError, "unexpected mode: #{mode[0]}" if mode[0] != 0
 
-          mem[r1] = input
+          mem[r1] = user_input.nil? ? input : user_input.call
           ip += 2
         when 4
           r1 = mem[ip + 1]