tealrb 0.1.0

data/lib/tealrb/rewriters.rb

@@ -0,0 +1,149 @@
+# frozen_string_literal: true
+
+module TEALrb
+  module Rewriters
+    class Rewriter < Parser::TreeRewriter
+      include RuboCop::AST::Traversal
+    end
+
+    class MethodRewriter < Rewriter
+      def on_def(node)
+        replace node.source_range, node.body.source
+        super
+      end
+
+      def on_block(node)
+        replace node.source_range, node.body.source
+      end
+
+      def on_send(node)
+        remove node.loc.selector if node.loc.selector.source == 'subroutine' || node.loc.selector.source == 'teal'
+
+        # @teal_methods[:name] = ->(*args) { ... } becomes ->(*args) { ... }
+        if ['@teal_methods', '@subroutines'].include? node.children[0]&.source
+          replace node.source_range, node.children[3].body.source
+        end
+
+        super
+      end
+    end
+
+    class AssignRewriter < Rewriter
+      def on_lvasgn(node)
+        wrap(node.children[1].source_range, '-> { ', ' }')
+        super
+      end
+
+      def on_lvar(node)
+        insert_after(node.loc.name, '.call')
+        super
+      end
+
+      def on_ivasgn(node)
+        wrap(node.children[1].source_range, '-> { ', ' }')
+        super
+      end
+
+      def on_ivar(node)
+        insert_after(node.loc.name, '.call') unless ['@teal', '@teal_methods', '@subroutines'].include? node.source
+        super
+      end
+    end
+
+    class ComparisonRewriter < Rewriter
+      def on_send(node)
+        if TEALrb::Opcodes::BINARY_OPCODE_METHOD_MAPPING.keys.map(&:to_s).include? node.loc.selector.source
+          wrap(node.source_range, '(', ')')
+        end
+        super
+      end
+    end
+
+    class OpRewriter < Rewriter
+      def initialize
+        @skips = 0
+        super
+      end
+
+      def on_and(node)
+        wrap(node.children[1].source_range, '(', ')')
+
+        op_loc = node.loc.operator
+        op_loc = op_loc.resize(3) if op_loc.resize(3).source == '&& '
+
+        replace(op_loc, '.boolean_and')
+
+        super
+      end
+
+      def on_or(node)
+        wrap(node.children[1].source_range, '(', ')')
+
+        op_loc = node.loc.operator
+        op_loc = op_loc.resize(3) if op_loc.resize(3).source == '|| '
+
+        replace(op_loc, '.boolean_or')
+        super
+      end
+
+      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
+
+        if OPCODE_METHODS.include? meth_name
+          if meth_name[/(byte|int)cblock/]
+            @skips = node.children.size - 2
+          else
+            params = TEALrb::Opcodes.instance_method(meth_name).parameters
+            @skips = params.count { |param| param[0] == :req }
+          end
+        elsif %i[comment placeholder].include? meth_name
+          @skips = 1
+        end
+        super
+      end
+
+      def on_int(node)
+        wrap(node.source_range, 'int(', ')') if (@skips -= 1).negative?
+        super
+      end
+
+      def on_str(node)
+        wrap(node.source_range, 'byte(', ')') if (@skips -= 1).negative?
+        super
+      end
+
+      def on_sym(node)
+        wrap(node.source_range, 'label(', ')') if node.source_range.source[/^:/] && (@skips -= 1).negative?
+        super
+      end
+    end
+
+    class IfRewriter < Rewriter
+      def on_if(node)
+        case node.loc.keyword.source
+        when 'if'
+          replace(node.loc.keyword, 'IfBlock.new(@teal, ')
+        when 'elsif'
+          replace(node.loc.keyword, 'end.elsif(')
+        end
+
+        cond_expr = node.children.first.source_range
+        replace(cond_expr, "#{cond_expr.source} ) do")
+
+        replace(node.loc.else, 'end.else do') if node.loc.else && node.loc.else.source == 'else'
+        super
+      end
+    end
+  end
+end

data/lib/tealrb.rb

@@ -0,0 +1,51 @@
+# frozen_string_literal: true
+
+require 'method_source'
+require 'rubocop'
+
+require_relative 'tealrb/constants'
+require_relative 'tealrb/abi'
+require_relative 'tealrb/opcodes'
+require_relative 'tealrb/opcode_modules'
+require_relative 'tealrb/rewriters'
+require_relative 'tealrb/if_block'
+require_relative 'tealrb/contract'
+require_relative 'tealrb/cmd_line/teal2tealrb'
+
+module TEALrb
+  class TEAL < Array
+    class << self
+      attr_accessor :current
+    end
+
+    @current = {}
+
+    def set_as_current
+      self.class.current[Thread.current] = self
+    end
+
+    def teal
+      flatten
+    end
+
+    def boolean_and(_other)
+      self << '&&'
+    end
+
+    def boolean_or(_other)
+      self << '||'
+    end
+
+    TEALrb::Opcodes::BINARY_OPCODE_METHOD_MAPPING.each do |meth, opcode|
+      define_method(meth) do |other|
+        ExtendedOpcodes.send(opcode, self, other)
+      end
+    end
+
+    TEALrb::Opcodes::UNARY_OPCODE_METHOD_MAPPING.each do |meth, opcode|
+      define_method(meth) do
+        ExtendedOpcodes.send(opcode, self)
+      end
+    end
+  end
+end

metadata

@@ -0,0 +1,152 @@
+--- !ruby/object:Gem::Specification
+name: tealrb
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Joe Polny
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2022-05-10 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: method_source
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rubocop
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: redcarpet
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rubocop-minitest
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: yard
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: 
+email: joepolny+dev@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/tealrb.rb
+- lib/tealrb/abi.rb
+- lib/tealrb/cmd_line/teal2tealrb.rb
+- lib/tealrb/constants.rb
+- lib/tealrb/contract.rb
+- lib/tealrb/if_block.rb
+- lib/tealrb/opcode_modules.rb
+- lib/tealrb/opcodes.rb
+- lib/tealrb/patches.rb
+- lib/tealrb/rewriters.rb
+homepage: https://github.com/joe-p/tealrb
+licenses:
+- MIT
+metadata:
+  source_code_uri: https://github.com/joe-p/tealrb
+  rubygems_mfa_required: 'true'
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - "~>"
+    - !ruby/object:Gem::Version
+      version: '2.7'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.1.6
+signing_key: 
+specification_version: 4
+summary: A DSL for building Algorand applications
+test_files: []