Parser + boilerplate for Among Us

3 files changed, 111 insertions, 0 deletions

diff --git a/src/among_us/among_us.cr b/src/among_us/among_us.cr
new file mode 100644
index 0000000..4b4f3b8
--- /dev/null
+++ b/src/among_us/among_us.cr
@@ -0,0 +1,29 @@
+module AmongUs
+
+  enum Command
+    SUS
+    VENTED
+    SUSSY
+    ELECTRICAL
+    WHO
+    WHERE
+  end
+
+  enum Color
+    RED
+    BLUE
+    PURPLE
+    GREEN
+    YELLOW
+    CYAN
+    BLACK
+    WHITE
+    BROWN
+    LIME
+    PINK
+    ORANGE
+  end
+
+  alias Instruction = Command | Color
+end
+
diff --git a/src/among_us/parser.cr b/src/among_us/parser.cr
new file mode 100644
index 0000000..84894bd
--- /dev/null
+++ b/src/among_us/parser.cr
@@ -0,0 +1,48 @@
+require "../util.cr"
+require "./among_us.cr"
+
+struct AmongUs::Parser
+  def initialize(code_io : IO)
+    @raw_code = code_io.gets_to_end
+    code_io.close
+  end
+
+  def parse : Array(Instruction)
+    tokens = @raw_code.split
+    instructions = [] of Instruction
+    color_seen = false
+    sus_safe = false
+
+    tokens.each do |token|
+      case token.upcase
+      when "WHO?"
+        instructions << Command::WHO
+      when "WHERE?"
+        instructions << Command::WHERE
+      else
+        i = parse_instruction(token)
+        if i.nil?
+          raise Util::ParserError.new("Failed to parse command/color '#{token}'")
+        end
+        instructions << i
+      end
+
+      color_seen = true if instructions[-1].is_a?(Color)
+      if instructions[-1] == Command::SUS && !sus_safe
+        unless color_seen
+          raise Util::ParserError.new("Provided program uses SUS command, but no colors are provided before first SUS")
+        end
+        sus_safe = true
+      end
+    end
+
+    return instructions
+  end
+
+  private def parse_instruction(str : String) : Instruction?
+    i = Command.parse?(str)
+    return i unless i.nil?
+    Color.parse?(str)
+  end
+end
+
diff --git a/src/among_us/program.cr b/src/among_us/program.cr
new file mode 100644
index 0000000..a54f24b
--- /dev/null
+++ b/src/among_us/program.cr
@@ -0,0 +1,34 @@
+require "big"
+
+struct AmongUs::Program < Flint::Program
+  def interpret : Nil
+    code = Parser.new(@source_io).parse
+    jumps = find_jumps(code)
+
+    stack = [] of BigInt
+    acc1 = BigInt.new
+    acc2 = BigInt.new
+    color = code.find &.is_a?(Color) 
+
+    puts code
+    puts jumps
+  end
+
+  private def find_jumps(code : Array(Instruction)) : Hash(Int32, Int32)
+    jumps = {} of Int32 => Int32
+    stack = [] of Int32
+
+    code.each_index do |i|
+      case code[i]
+      when Command::WHO
+        stack << i
+      when Command::WHERE
+        jumps[i] = stack.pop
+        jumps[jumps[i]] = i
+      end
+    end
+
+    return jumps
+  end
+end
+