tealrb 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ccb46efa84c6dfdf2378ab562eaf2345d947f0a9af9850c05220acfa1b540bc3
-  data.tar.gz: d7bfaa9d68b2fe60adeeb229f54f087a56b73319f0d3f6f8bf8d673b8a103258
+  metadata.gz: 232757505472510465b62384efb3502fd68bb6589cb543e5bdad87b8f29f82a0
+  data.tar.gz: d6f98b807101a75092f86ff80749dc28cb42b20474706436fbdc2ba356211659
 SHA512:
-  metadata.gz: '0805c1e920c22e3d0fc64975135850a08147b3a3c9f1fb9390868ed60220d504f3fd20c89ea4ecdc96f71068e75b0748f62e502155335007ae0d3530fcbd4af4'
-  data.tar.gz: ec0abbd56e219ea3082200d9fb5bab2cd6ead37957af0c7dccc359e6c94064967cdac907b776bad0d07d9f1ecfb22fa46c37a84322777dfeb1d7fde0d1df57a2
+  metadata.gz: 5d6f8f0e315cc9e045a4279e7d1a79a8a6d033218d830a93d20b72ef4bb7fb6f239007479763fcd99c89f46a0fee17478a47c67450cf560ca2c9c77a8e511a2b
+  data.tar.gz: 89dfcbe94801a5fb0ad54c496472f6442fdb4126804fbe817ce5d5d3b4e32aec2360eae93d6fb72acac736c98c548761fe1f8eaa9b632ed8a9d6593005aa2ca7

data/lib/tealrb/cmd_line/teal2tealrb.rb

@@ -11,15 +11,9 @@ module TEALrb
           line.strip!
           next if line[/^#/]
 
-          if line[%r{^//}]
-            line = %(comment("#{line.sub(%r{//}, '')}"))
-            output.puts line
-            next
-          elsif line[%r{//.*}]
-            comment = line[%r{(?<=//).*}]
-            line.sub!(%r{//.*}, '')
-            inline_comment = %(comment("#{comment}", inline: true))
-          end
+          comment = line[%r{//.*}]
+          comment&.sub!('//', '# //')
+          line.sub!(%r{//.*}, '')
 
           opcode = line.split.first
 
@@ -55,8 +49,11 @@ module TEALrb
             line = "#{opcode} #{args.join(', ')}"
           end
 
-          output.puts line
-          output.puts inline_comment if inline_comment
+          if line.empty?
+            output.puts comment
+          else
+            output.puts "#{line} #{comment}"
+          end
         end
 
         output.string

data/lib/tealrb/contract.rb

@@ -70,6 +70,12 @@ module TEALrb
     def initialize
       @teal = TEAL.new ["#pragma version #{self.class.version}"]
 
+      self.class.subroutines.each_key do |name|
+        define_singleton_method(name) do |*_args|
+          callsub(name)
+        end
+      end
+
       self.class.subroutines.each do |name, blk|
         define_subroutine name, blk
       end
@@ -112,6 +118,10 @@ module TEALrb
     def define_subroutine(name, definition)
       @teal.set_as_current
 
+      define_singleton_method(name) do |*_args|
+        callsub(name)
+      end
+
       @teal << 'b main' unless @teal.include? 'b main'
 
       label(name) # add teal label
@@ -134,10 +144,6 @@ module TEALrb
       new_source = "#{pre_string.string}#{new_source}retsub"
       eval_tealrb(new_source, debug_context: "subroutine: #{name}")
 
-      define_singleton_method(name) do |*_args|
-        callsub(name)
-      end
-
       nil
     end
 
@@ -145,11 +151,12 @@ module TEALrb
     # @param content [String] content of the comment
     # @param inline [Boolean] whether the comment should be on the previous TEAL line
     def comment(content, inline: false)
+      content = " #{content}" unless content[0] == ' '
       if inline
         last_line = @teal.pop
-        @teal << "#{last_line} // #{content}"
+        @teal << "#{last_line} //#{content}"
       else
-        @teal << "// #{content}"
+        @teal << "//#{content}"
       end
     end
 
@@ -184,7 +191,7 @@ module TEALrb
 
     def rewrite_with_rewriter(string, rewriter)
       process_source = RuboCop::ProcessedSource.new(string, RUBY_VERSION[/\d\.\d/].to_f)
-      rewriter.new.rewrite(process_source.buffer, process_source.ast)
+      rewriter.new.rewrite(process_source)
     end
 
     def rewrite(string, method_rewriter: false)
@@ -194,7 +201,7 @@ module TEALrb
         puts ''
       end
 
-      [ComparisonRewriter, IfRewriter, OpRewriter, AssignRewriter].each do |rw|
+      [CommentRewriter, ComparisonRewriter, IfRewriter, OpRewriter, AssignRewriter].each do |rw|
         string = rewrite_with_rewriter(string, rw)
       end
 

data/lib/tealrb/opcode_modules.rb

@@ -33,7 +33,7 @@ module TEALrb
         private
 
         def txn_type_int(type)
-          TEALrb::TEAL.current[Thread.current] << "int '#{type}'"
+          TEALrb::TEAL.current[Thread.current] << "int #{type}"
         end
       end
 

data/lib/tealrb/rewriters.rb

@@ -4,6 +4,27 @@ module TEALrb
   module Rewriters
     class Rewriter < Parser::TreeRewriter
       include RuboCop::AST::Traversal
+
+      def rewrite(processed_source)
+        @comments = processed_source.comments
+        super(processed_source.buffer, processed_source.ast)
+      end
+    end
+
+    class CommentRewriter < Rewriter
+      def process(_)
+        @comments.each do |comment|
+          line = @source_rewriter.source_buffer.source_lines[comment.loc.expression.line - 1].strip
+
+          next unless comment.inline? && comment.text[%r{^#\s*//}]
+
+          inline_comment = line[/^#/].nil?
+          content = comment.text.sub('# ', '')
+          content.sub!('//', '')
+
+          replace comment.loc.expression, "; comment(%q(#{content}), inline: #{inline_comment})"
+        end
+      end
     end
 
     class MethodRewriter < Rewriter

metadata

@@ -1,113 +1,127 @@
 --- !ruby/object:Gem::Specification
 name: tealrb
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Joe Polny
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2022-05-10 00:00:00.000000000 Z
+date: 2022-05-12 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'
+        version: '1.0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '1.0'
 - !ruby/object:Gem::Dependency
   name: rubocop
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '1.27'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '1.27'
 - !ruby/object:Gem::Dependency
   name: minitest
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '5.15'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: '5.15'
 - !ruby/object:Gem::Dependency
   name: pry
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.14.1
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.14.1
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 13.0.1
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 13.0.1
 - !ruby/object:Gem::Dependency
   name: redcarpet
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 3.5.1
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 3.5.1
 - !ruby/object:Gem::Dependency
   name: rubocop-minitest
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.19.1
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.19.1
 - !ruby/object:Gem::Dependency
   name: yard
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.9.27
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.9.27
 description: 
 email: joepolny+dev@gmail.com
 executables: []
@@ -136,7 +150,7 @@ 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