From 768a0ef17fac75f8a9c37cae3439a84981b1d143 Mon Sep 17 00:00:00 2001
From: Josh Holtrop <jholtrop@gmail.com>
Date: Sat, 1 May 2021 14:34:00 -0400
Subject: [PATCH] Extract class name from grammar file

---
 lib/imbecile/grammar.rb | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/lib/imbecile/grammar.rb b/lib/imbecile/grammar.rb
index 11b1d2d..ae03cad 100644
--- a/lib/imbecile/grammar.rb
+++ b/lib/imbecile/grammar.rb
@@ -16,6 +16,8 @@ module Imbecile
           # Skip comment lines.
         elsif line =~ /^\s*$/
           # Skip blank lines.
+        elsif line =~ /^\s*class\s+(\S+)$/
+          @class = $1
         elsif line =~ /^\s*token\s+(\S+)\s+(.*)$/
           name, expr = $1, $2
           unless name =~ /^[a-zA-Z_][a-zA-Z_0-9]*$/