tealrb 0.5.0 → 0.8.0

data/lib/tealrb/rewriters.rb

@@ -66,8 +66,7 @@ module TEALrb
       end
 
       def on_ivar(node)
-        insert_after(node.loc.name, '.call') unless ['@teal', '@teal_methods', '@subroutines',
-                                                     '@scratch'].include? node.source
+        insert_after(node.loc.name, '.call') unless ['@teal_methods', '@subroutines', '@scratch'].include? node.source
         super
       end
     end
@@ -83,7 +82,7 @@ module TEALrb
 
     class OpRewriter < Rewriter
       def initialize
-        @skips = 0
+        @skips = []
         super
       end
 
@@ -110,44 +109,62 @@ module TEALrb
 
       OPCODE_METHODS = TEALrb::Opcodes.instance_methods.freeze
 
-      def on_const(node)
-        @skips = 1 if %w[Txna Gtxn AppArgs].include? node.loc.name.source
-        super
-      end
-
-      def on_ivar(node)
-        @skips = 2 if node.source == '@teal_methods'
-        @skips = 1 if node.source == '@scratch'
-      end
-
       def on_send(node)
-        meth_name = node.loc.selector.source.to_sym
+        meth_name = node.children[1]
 
         if OPCODE_METHODS.include? meth_name
           if meth_name[/(byte|int)cblock/]
-            @skips = node.children.size - 2
+            @skips += node.children[2..]
           else
             params = TEALrb::Opcodes.instance_method(meth_name).parameters
-            @skips = params.count { |param| param[0] == :req }
+            req_params = params.count { |param| param[0] == :req }
+            @skips += node.children[2..(1 + req_params.size)] unless req_params.zero?
           end
-        elsif %i[comment placeholder].include? meth_name
-          @skips = 1
+        elsif %i[comment placeholder rb].include?(meth_name) ||
+              (%i[[] []=].include?(meth_name) &&
+                  (
+                    %i[@scratch @teal_methods Gtxn
+                       AppArgs].include?(node.children[0].children.last) ||
+                    node.children[0].children.first&.children&.last == :Txna
+                  ))
+
+          @skips << node.children[2]
+        elsif node.children.first&.children&.last == :@scratch && meth_name[/=$/]
+          nil
+        elsif %i[@scratch Gtxn].include? node.children.first&.children&.last
+          @skips << node.children.last
         end
+
         super
       end
 
       def on_int(node)
-        wrap(node.source_range, 'int(', ')') if (@skips -= 1).negative?
+        if @skips.include? node
+          @skips.delete(node)
+        else
+          wrap(node.source_range, 'int(', ')')
+        end
+
         super
       end
 
       def on_str(node)
-        wrap(node.source_range, 'byte(', ')') if (@skips -= 1).negative?
+        if @skips.include? node
+          @skips.delete(node)
+        else
+          wrap(node.source_range, 'byte(', ')')
+        end
+
         super
       end
 
       def on_sym(node)
-        wrap(node.source_range, 'label(', ')') if node.source_range.source[/^:/] && (@skips -= 1).negative?
+        if @skips.include? node
+          @skips.delete(node)
+        elsif node.source_range.source[/^:/]
+          wrap(node.source_range, 'label(', ')')
+        end
+
         super
       end
     end
@@ -156,7 +173,7 @@ module TEALrb
       def on_if(node)
         case node.loc.keyword.source
         when 'if'
-          replace(node.loc.keyword, 'IfBlock.new(@teal, ')
+          replace(node.loc.keyword, 'IfBlock.new(')
         when 'elsif'
           replace(node.loc.keyword, 'end.elsif(')
         end
@@ -168,5 +185,30 @@ module TEALrb
         super
       end
     end
+
+    class WhileRewriter < Rewriter
+      class << self
+        attr_accessor :while_count
+      end
+
+      def initialize(*args)
+        self.class.while_count = {}
+        self.class.while_count[Thread.current] ||= 0
+        super
+      end
+
+      def while_count
+        self.class.while_count[Thread.current]
+      end
+
+      def on_while(node)
+        cond_node = node.children.first
+        replace(node.loc.keyword, ":while#{while_count}\n#{cond_node.source}\nbz :end_while#{while_count}")
+        replace(node.loc.begin, '') if node.loc.begin
+        replace(node.loc.end, "b :while#{while_count}\n:end_while#{while_count}")
+        replace(cond_node.loc.expression, '')
+        super
+      end
+    end
   end
 end

data/lib/tealrb/scratch.rb

@@ -2,14 +2,13 @@
 
 module TEALrb
   class Scratch
-    def initialize(teal)
-      @teal = teal
+    def initialize
       @open_slots = (0..256).to_a
       @named_slots = {}
     end
 
     def [](key)
-      @teal << "load #{@named_slots[key]} // #{key}"
+      TEAL.instance << "load #{@named_slots[key]} // #{key}"
     end
 
     def []=(key, _value)
@@ -17,7 +16,7 @@ module TEALrb
     end
 
     def store(key)
-      @teal << "store #{@named_slots[key] ||= @open_slots.shift} // #{key}"
+      TEAL.instance << "store #{@named_slots[key] ||= @open_slots.shift} // #{key}"
     end
 
     def delete(key)
@@ -34,5 +33,21 @@ module TEALrb
     def unreserve(slot)
       @open_slots << slot
     end
+
+    def respond_to_missing?
+      !!(meth[/=$/] || @named_slots.key?(meth.to_s))
+    end
+
+    def method_missing(meth, *args, &block)
+      super unless block.nil?
+
+      if meth[/=$/]
+        store(meth.to_s.chomp('='))
+      elsif @named_slots.key?(meth.to_s)
+        self[meth.to_s]
+      else
+        super
+      end
+    end
   end
 end

data/lib/tealrb.rb

@@ -16,13 +16,18 @@ require_relative 'tealrb/cmd_line/teal2tealrb'
 module TEALrb
   class TEAL < Array
     class << self
-      attr_accessor :current
-    end
+      attr_writer :instance
+
+      def instance
+        raise 'TEALrb does not support multi-threading' if Thread.current != Thread.main
 
-    @current = {}
+        @instance
+      end
+    end
 
-    def set_as_current
-      self.class.current[Thread.current] = self
+    def initialize(teal_array)
+      self.class.instance = self
+      super
     end
 
     def teal

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: tealrb
 version: !ruby/object:Gem::Version
-  version: 0.5.0
+  version: 0.8.0
 platform: ruby
 authors:
 - Joe Polny
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-05-22 00:00:00.000000000 Z
+date: 2022-07-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: method_source