tealrb 0.7.0 → 0.8.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 94ca888a944ccee0517d5e543498570af448a9eb3ea1ba74d3d3261ffc4b86f0
-  data.tar.gz: 54ac7826a7af43fb82fa17cf010a2ce869511546067f7a151181b86d3445b0bf
+  metadata.gz: 2480b69400b8916b30ebd58c19b804190a6390780ad335006ac0364ea9d4ad4e
+  data.tar.gz: 6cf7f7f79217b1938f71bf07348923a7d31d3d83d76a4219f43e58bb1db2e3bc
 SHA512:
-  metadata.gz: 4f986ae5a4e3c37460142951923c3908ec95b58d0d91f695a49a07f8e8eee062499d496b185e1f2eafec8ed2020ca9d83a4c8c50c97531a8a67e26057e7bb794
-  data.tar.gz: c73f78d20f1c76ccc41e42450f713b14dc42049d8d44ee1498c59e38cda3e7699e4f1e655932f5b86c68d707d1a1abba7d59fc5c92244552236bb2a472a3d4c2
+  metadata.gz: c8220e4aadbdcbe9a992ada723431281665edda6a19c07771978b5a8b4ae9cf987a208ed69e9c96f4360a6a3ace9762dc930e4bab99574b2114effc0ffd1060e
+  data.tar.gz: 47790d4ee2171f4c2e95473f7c7cc6674d9634804c8c0dac2eeafff368df073166d9dfd371efc24026e1c94f4c333b1904af9fa4a959992116df7859dc314875

data/lib/tealrb/contract.rb

@@ -11,7 +11,8 @@ module TEALrb
     attr_reader :teal
 
     class << self
-      attr_accessor :subroutines, :version, :teal_methods, :abi_method_hash, :abi_description, :debug
+      attr_accessor :subroutines, :version, :teal_methods, :abi_method_hash, :abi_description, :debug,
+                    :disable_abi_routing
 
       private
 
@@ -22,6 +23,7 @@ module TEALrb
         klass.abi_description = ABI::ABIDescription.new
         klass.abi_method_hash = {}
         klass.debug = false
+        klass.disable_abi_routing = false
         super
       end
     end
@@ -164,14 +166,28 @@ module TEALrb
       nil
     end
 
-    # transpiles #main
+    # transpiles #main and routes abi methods. To disable abi routing, set `@disable_abi_routing` to true in your
+    # Contract subclass
     def compile
       TEAL.instance << 'main:' if TEAL.instance.include? 'b main'
-      eval_tealrb(rewrite(method(:main).source, method_rewriter: true), debug_context: 'main')
+      route_abi_methods unless self.class.disable_abi_routing
+      eval_tealrb(rewrite(method(:main).source, method_rewriter: true), debug_context: 'main') if respond_to? :main
     end
 
     private
 
+    def route_abi_methods
+      self.class.abi_description.methods.each do |meth|
+        signature = "#{meth[:name]}(#{meth[:args].map{ _1[:type]}.join(',')})#{meth[:returns][:type]}"
+        selector = OpenSSL::Digest.new('SHA512-256').hexdigest(signature)[..7]
+
+        IfBlock.new(AppArgs[0] == byte(selector)) do
+          callsub(meth[:name])
+          approve
+        end
+      end
+    end
+
     def generate_method_source(name, definition)
       new_source = rewrite(definition.source, method_rewriter: true)
 

metadata

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