tealrb 0.4.0 → 0.7.0

data/lib/tealrb/rewriters.rb

@@ -66,7 +66,7 @@ module TEALrb
       end
 
       def on_ivar(node)
-        insert_after(node.loc.name, '.call') unless ['@teal', '@teal_methods', '@subroutines'].include? node.source
+        insert_after(node.loc.name, '.call') unless ['@teal_methods', '@subroutines', '@scratch'].include? node.source
         super
       end
     end
@@ -82,7 +82,7 @@ module TEALrb
 
     class OpRewriter < Rewriter
       def initialize
-        @skips = 0
+        @skips = []
         super
       end
 
@@ -109,43 +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'
-      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
@@ -154,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
@@ -166,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

@@ -0,0 +1,53 @@
+# frozen_string_literal: true
+
+module TEALrb
+  class Scratch
+    def initialize
+      @open_slots = (0..256).to_a
+      @named_slots = {}
+    end
+
+    def [](key)
+      TEAL.instance << "load #{@named_slots[key]} // #{key}"
+    end
+
+    def []=(key, _value)
+      store(key)
+    end
+
+    def store(key)
+      TEAL.instance << "store #{@named_slots[key] ||= @open_slots.shift} // #{key}"
+    end
+
+    def delete(key)
+      @open_slots << @named_slots.delete(key)
+    end
+
+    def reserve(slot)
+      name = @named_slots.key(slot)
+      raise ArgumentError, "Attempted to reserve a slot (#{slot}) that corresponds to a named slot (#{name})" if name
+
+      @open_slots.delete slot
+    end
+
+    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

@@ -9,19 +9,25 @@ require_relative 'tealrb/opcodes'
 require_relative 'tealrb/opcode_modules'
 require_relative 'tealrb/rewriters'
 require_relative 'tealrb/if_block'
+require_relative 'tealrb/scratch'
 require_relative 'tealrb/contract'
 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.4.0
+  version: 0.7.0
 platform: ruby
 authors:
 - Joe Polny
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-05-16 00:00:00.000000000 Z
+date: 2022-07-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: method_source
@@ -152,6 +152,7 @@ files:
 - lib/tealrb/opcodes.rb
 - lib/tealrb/patches.rb
 - lib/tealrb/rewriters.rb
+- lib/tealrb/scratch.rb
 homepage: https://github.com/joe-p/tealrb
 licenses:
 - MIT