homura-runtime 0.3.5 → 0.3.7

data/lib/homura/runtime/async_registry.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require 'set'
+require "set"
 
 module HomuraRuntime
   class AsyncRegistry
@@ -22,11 +22,15 @@ module HomuraRuntime
       end
 
       def taint_return(class_name, method_name, return_class_name)
-        (@registry.taint_returns[class_name] ||= {})[method_name] = return_class_name
+        (@registry.taint_returns[class_name] ||= {})[
+          method_name
+        ] = return_class_name
       end
 
       def async_accessor(lvar_name, accessor_name, class_name)
-        @registry.async_accessors[[lvar_name.to_sym, accessor_name.to_sym]] = class_name
+        @registry.async_accessors[
+          [lvar_name.to_sym, accessor_name.to_sym]
+        ] = class_name
       end
 
       def async_helper(method_name, class_name)
@@ -75,28 +79,46 @@ module HomuraRuntime
         Gem.loaded_specs.each_value do |spec|
           next if spec.full_gem_path.nil?
 
-          lib_dir = File.join(spec.full_gem_path, 'lib')
+          lib_dir = File.join(spec.full_gem_path, "lib")
           next unless Dir.exist?(lib_dir)
 
-          Dir.glob(File.join(lib_dir, '**', '*.rb')).each do |path|
-            next unless File.read(path).include?('register_async_source')
-
-            require_path = path.sub(Regexp.new("^#{Regexp.escape(lib_dir)}/"), '').sub(/\.rb\z/, '')
-            begin
-              require require_path
-              loaded += 1
-              puts "[auto-await] loaded async source from #{spec.name}: #{require_path}" if debug
-            rescue LoadError, StandardError => e
-              warn "[auto-await] Warning: failed to load async source from #{spec.name}/#{require_path}: #{e.message}" if debug
+          Dir
+            .glob(File.join(lib_dir, "**", "*.rb"))
+            .each do |path|
+              next unless File.read(path).include?("register_async_source")
+
+              require_path =
+                path.sub(Regexp.new("^#{Regexp.escape(lib_dir)}/"), "").sub(
+                  /\.rb\z/,
+                  ""
+                )
+              begin
+                require require_path
+                loaded += 1
+                if debug
+                  puts "[auto-await] loaded async source from #{spec.name}: #{require_path}"
+                end
+              rescue LoadError, StandardError => e
+                if debug
+                  warn "[auto-await] Warning: failed to load async source from #{spec.name}/#{require_path}: #{e.message}"
+                end
+              end
             end
-          end
         end
 
-        puts "[auto-await] auto-loaded #{loaded} async source file(s)" if debug && loaded.positive?
+        if debug && loaded.positive?
+          puts "[auto-await] auto-loaded #{loaded} async source file(s)"
+        end
       end
     end
 
-    attr_reader :async_classes, :async_methods, :async_factories, :taint_returns, :async_accessors, :async_helpers, :helper_factories
+    attr_reader :async_classes,
+                :async_methods,
+                :async_factories,
+                :taint_returns,
+                :async_accessors,
+                :async_helpers,
+                :helper_factories
 
     def initialize
       @async_classes = {}
@@ -113,7 +135,10 @@ module HomuraRuntime
       methods = @async_methods[class_name]
       return true if methods&.include?(method_name)
       except = @async_classes[class_name]
-      return true if except && !except.include?(method_name.to_s) && !except.include?(method_name.to_sym)
+      if except && !except.include?(method_name.to_s) &&
+           !except.include?(method_name.to_sym)
+        return true
+      end
       false
     end
 
@@ -126,10 +151,8 @@ module HomuraRuntime
     end
 
     def tainted_class?(class_name)
-      @async_classes.key?(class_name) ||
-        @async_methods.key?(class_name) ||
-        @async_factories.key?(class_name) ||
-        @taint_returns.key?(class_name)
+      @async_classes.key?(class_name) || @async_methods.key?(class_name) ||
+        @async_factories.key?(class_name) || @taint_returns.key?(class_name)
     end
   end
 end
@@ -138,86 +161,100 @@ end
 # Each binding declares which methods return Promises so the
 # build-time analyzer can insert .__await__ automatically.
 HomuraRuntime::AsyncRegistry.register_async_source do
-  async_method 'Cloudflare::D1Database', :execute
-  async_method 'Cloudflare::D1Database', :get_first_row
-  async_method 'Cloudflare::D1Database', :execute_insert
-  async_method 'Cloudflare::D1Database', :execute_batch
-  taint_return 'Cloudflare::D1Database', :prepare, 'Cloudflare::D1Statement'
-  taint_return 'Cloudflare::D1Database', :[], 'Cloudflare::D1Statement'
-
-  async_method 'Cloudflare::D1Statement', :all
-  async_method 'Cloudflare::D1Statement', :first
-  async_method 'Cloudflare::D1Statement', :run
+  async_method "Cloudflare::D1Database", :execute
+  async_method "Cloudflare::D1Database", :get_first_row
+  async_method "Cloudflare::D1Database", :execute_insert
+  async_method "Cloudflare::D1Database", :execute_batch
+  taint_return "Cloudflare::D1Database", :prepare, "Cloudflare::D1Statement"
+  taint_return "Cloudflare::D1Database", :[], "Cloudflare::D1Statement"
+
+  async_method "Cloudflare::D1Statement", :all
+  async_method "Cloudflare::D1Statement", :first
+  async_method "Cloudflare::D1Statement", :run
   # `bind` returns a new D1Statement for further chaining. Tainting the
   # return preserves the type so the auto-await pass keeps chaining
   # `.run` / `.all` / `.first` on the bound statement (otherwise
   # `db.prepare(sql).bind(...).run` drops await on `.run` and
   # `flatten_meta` ends up receiving a JS Promise instead of the
   # metadata Hash).
-  taint_return 'Cloudflare::D1Statement', :bind, 'Cloudflare::D1Statement'
-
-  async_method 'Cloudflare::KVNamespace', :get
-  async_method 'Cloudflare::KVNamespace', :get_with_metadata
-  async_method 'Cloudflare::KVNamespace', :put
-  async_method 'Cloudflare::KVNamespace', :delete
-  async_method 'Cloudflare::KVNamespace', :list
-
-  async_method 'Cloudflare::R2Bucket', :get
-  async_method 'Cloudflare::R2Bucket', :get_binary
-  async_method 'Cloudflare::R2Bucket', :put
-  async_method 'Cloudflare::R2Bucket', :delete
-  async_method 'Cloudflare::R2Bucket', :list
-  async_method 'Cloudflare::R2Bucket', :head
-
-  async_method 'Cloudflare::AI', :run
-  taint_return 'Cloudflare::AI', :run_stream, 'Cloudflare::AI::Stream'
-  async_method 'Cloudflare::AI::Binding', :run
-  taint_return 'Cloudflare::AI::Binding', :run_stream, 'Cloudflare::AI::Stream'
-
-  async_method 'Cloudflare::Cache', :match
-  async_method 'Cloudflare::Cache', :put
-  async_method 'Cloudflare::Cache', :delete
-
-  async_factory 'Cloudflare::Email', :new
-  async_method 'Cloudflare::Email', :send
-
-  async_method 'Cloudflare::Queue', :send
-  async_method 'Cloudflare::Queue', :send_batch
-
-  async_factory 'Cloudflare::DurableObjectNamespace', :new
-  taint_return 'Cloudflare::DurableObjectNamespace', :get, 'Cloudflare::DurableObjectStub'
-  taint_return 'Cloudflare::DurableObjectNamespace', :get_by_name, 'Cloudflare::DurableObjectStub'
-  taint_return 'Cloudflare::DurableObjectState', :storage, 'Cloudflare::DurableObjectStorage'
-  async_method 'Cloudflare::DurableObjectStub', :fetch
-  async_method 'Cloudflare::DurableObjectStub', :request
-  async_method 'Cloudflare::DurableObjectStub', :get
-  async_method 'Cloudflare::DurableObjectStub', :post
-  async_method 'Cloudflare::DurableObjectStub', :put
-  async_method 'Cloudflare::DurableObjectStub', :delete
-
-  async_method 'Cloudflare::DurableObjectStorage', :get
-  async_method 'Cloudflare::DurableObjectStorage', :put
-  async_method 'Cloudflare::DurableObjectStorage', :delete
-  async_method 'Cloudflare::DurableObjectStorage', :list
-  async_method 'Cloudflare::DurableObjectStorage', :transaction
-
-  async_method 'Cloudflare::HTTP', :fetch
-
-  async_method 'Faraday::Connection', :get
-  async_method 'Faraday::Connection', :post
-  async_method 'Faraday::Connection', :put
-  async_method 'Faraday::Connection', :delete
-  async_method 'Faraday::Connection', :patch
-  async_method 'Faraday::Connection', :head
-
-  helper_factory :d1, 'Cloudflare::D1Database'
-  helper_factory :db, 'Cloudflare::D1Database'
-  helper_factory :kv, 'Cloudflare::KVNamespace'
-  helper_factory :bucket, 'Cloudflare::R2Bucket'
-  helper_factory :ai, 'Cloudflare::AI::Binding'
-  helper_factory :send_email, 'Cloudflare::Email'
-  helper_factory :jobs_queue, 'Cloudflare::Queue'
-  helper_factory :jobs_dlq, 'Cloudflare::Queue'
-  helper_factory :do_counter, 'Cloudflare::DurableObjectNamespace'
-  helper_factory :durable_object, 'Cloudflare::DurableObjectStub'
+  taint_return "Cloudflare::D1Statement", :bind, "Cloudflare::D1Statement"
+
+  async_method "Cloudflare::KVNamespace", :get
+  async_method "Cloudflare::KVNamespace", :get_with_metadata
+  async_method "Cloudflare::KVNamespace", :put
+  async_method "Cloudflare::KVNamespace", :delete
+  async_method "Cloudflare::KVNamespace", :list
+
+  async_method "Cloudflare::R2Bucket", :get
+  async_method "Cloudflare::R2Bucket", :get_binary
+  async_method "Cloudflare::R2Bucket", :put
+  async_method "Cloudflare::R2Bucket", :delete
+  async_method "Cloudflare::R2Bucket", :list
+  async_method "Cloudflare::R2Bucket", :head
+
+  async_method "Cloudflare::AI", :run
+  async_method "Cloudflare::AI", :speak
+  async_method "Cloudflare::AI", :speak_data_url
+  taint_return "Cloudflare::AI", :run_stream, "Cloudflare::AI::Stream"
+  async_method "Cloudflare::AI::Binding", :run
+  async_method "Cloudflare::AI::Binding", :chat
+  async_method "Cloudflare::AI::Binding", :chat_text
+  async_method "Cloudflare::AI::Binding", :transcribe
+  async_method "Cloudflare::AI::Binding", :transcribe_text
+  async_method "Cloudflare::AI::Binding", :speak
+  async_method "Cloudflare::AI::Binding", :speak_data_url
+  taint_return "Cloudflare::AI::Binding", :run_stream, "Cloudflare::AI::Stream"
+
+  async_method "Cloudflare::Cache", :match
+  async_method "Cloudflare::Cache", :put
+  async_method "Cloudflare::Cache", :delete
+
+  async_factory "Cloudflare::Email", :new
+  async_method "Cloudflare::Email", :send
+
+  async_method "Cloudflare::Queue", :send
+  async_method "Cloudflare::Queue", :send_batch
+
+  async_factory "Cloudflare::DurableObjectNamespace", :new
+  taint_return "Cloudflare::DurableObjectNamespace",
+               :get,
+               "Cloudflare::DurableObjectStub"
+  taint_return "Cloudflare::DurableObjectNamespace",
+               :get_by_name,
+               "Cloudflare::DurableObjectStub"
+  taint_return "Cloudflare::DurableObjectState",
+               :storage,
+               "Cloudflare::DurableObjectStorage"
+  async_method "Cloudflare::DurableObjectStub", :fetch
+  async_method "Cloudflare::DurableObjectStub", :request
+  async_method "Cloudflare::DurableObjectStub", :get
+  async_method "Cloudflare::DurableObjectStub", :post
+  async_method "Cloudflare::DurableObjectStub", :put
+  async_method "Cloudflare::DurableObjectStub", :delete
+
+  async_method "Cloudflare::DurableObjectStorage", :get
+  async_method "Cloudflare::DurableObjectStorage", :put
+  async_method "Cloudflare::DurableObjectStorage", :delete
+  async_method "Cloudflare::DurableObjectStorage", :list
+  async_method "Cloudflare::DurableObjectStorage", :transaction
+
+  async_method "Cloudflare::HTTP", :fetch
+
+  async_method "Faraday::Connection", :get
+  async_method "Faraday::Connection", :post
+  async_method "Faraday::Connection", :put
+  async_method "Faraday::Connection", :delete
+  async_method "Faraday::Connection", :patch
+  async_method "Faraday::Connection", :head
+
+  helper_factory :d1, "Cloudflare::D1Database"
+  helper_factory :db, "Cloudflare::D1Database"
+  helper_factory :kv, "Cloudflare::KVNamespace"
+  helper_factory :bucket, "Cloudflare::R2Bucket"
+  helper_factory :ai, "Cloudflare::AI::Binding"
+  helper_factory :send_email, "Cloudflare::Email"
+  helper_factory :jobs_queue, "Cloudflare::Queue"
+  helper_factory :jobs_dlq, "Cloudflare::Queue"
+  helper_factory :do_counter, "Cloudflare::DurableObjectNamespace"
+  helper_factory :durable_object, "Cloudflare::DurableObjectStub"
 end

data/lib/homura/runtime/auto_await/analyzer.rb

@@ -1,8 +1,8 @@
 # frozen_string_literal: true
 
-require 'parser/current'
-require 'parser/source/tree_rewriter'
-require 'set'
+require "parser/current"
+require "parser/source/tree_rewriter"
+require "set"
 
 module HomuraRuntime
   module AutoAwait
@@ -16,7 +16,7 @@ module HomuraRuntime
         @async_local_methods = Set.new
       end
 
-      def process(source, filename = '(auto-await)')
+      def process(source, filename = "(auto-await)")
         buffer = Parser::Source::Buffer.new(filename)
         buffer.source = source
         parser = Parser::CurrentRuby.new
@@ -52,7 +52,9 @@ module HomuraRuntime
         # Bottom-up traversal: process children first so helper-factory
         # sends (e.g. bare +db+) populate @env before their parent
         # send (e.g. +db.execute(...)+) is checked by should_await?.
-        node.children.each { |child| process_node(child) if child.is_a?(Parser::AST::Node) }
+        node.children.each do |child|
+          process_node(child) if child.is_a?(Parser::AST::Node)
+        end
         case node.type
         when :lvasgn
           process_lvasgn(node)
@@ -85,13 +87,16 @@ module HomuraRuntime
         before_awaits = @await_nodes.length
         @env = @registry.helper_factories.dup
 
-        node.children[1..-1].each { |child| process_node(child) if child.is_a?(Parser::AST::Node) }
+        node.children[1..-1].each do |child|
+          process_node(child) if child.is_a?(Parser::AST::Node)
+        end
 
         body = node.children[2]
         return_cls = infer_class(body)
         @method_returns[method_name] = return_cls if return_cls
         body_source = node.loc.expression&.source.to_s
-        if @await_nodes.length > before_awaits || body_source.include?('.__await__')
+        if @await_nodes.length > before_awaits ||
+             body_source.include?(".__await__")
           @async_local_methods << method_name
         end
 
@@ -115,7 +120,8 @@ module HomuraRuntime
 
       def process_send(node)
         receiver, method_name = *node
-        if receiver.nil? && (factory_cls = @registry.helper_factories[method_name])
+        if receiver.nil? &&
+             (factory_cls = @registry.helper_factories[method_name])
           @env[method_name] = factory_cls
         end
         if should_await?(node)
@@ -144,7 +150,9 @@ module HomuraRuntime
         when :send
           receiver, method_name = *node
           if receiver.nil?
-            return @method_returns[method_name] if @method_returns.key?(method_name)
+            if @method_returns.key?(method_name)
+              return @method_returns[method_name]
+            end
             return @env[method_name] if @env.key?(method_name)
           end
           infer_send_class(node)
@@ -175,7 +183,8 @@ module HomuraRuntime
             key_node = node.children[2]
             if key_node&.type == :str
               key = key_node.children[0]
-              mapped = @registry.async_accessors[[env_name(receiver), key.to_sym]]
+              mapped =
+                @registry.async_accessors[[env_name(receiver), key.to_sym]]
               return mapped if mapped
             end
           end
@@ -189,7 +198,9 @@ module HomuraRuntime
             return ret if ret
           end
         else
-          return @method_returns[method_name] if @method_returns.key?(method_name)
+          if @method_returns.key?(method_name)
+            return @method_returns[method_name]
+          end
         end
         nil
       end
@@ -198,14 +209,17 @@ module HomuraRuntime
         return {} unless durable_object_define_call?(call_node)
         return {} unless args_node&.type == :args
 
-        arg_names = args_node.children.filter_map do |arg|
-          next unless arg&.type == :arg
-          arg.children[0]
-        end
+        arg_names =
+          args_node.children.filter_map do |arg|
+            next unless arg&.type == :arg
+            arg.children[0]
+          end
         return {} if arg_names.empty?
 
-        bindings = { arg_names[0] => 'Cloudflare::DurableObjectState' }
-        bindings[arg_names[1]] = 'Cloudflare::DurableObjectRequest' if arg_names.length > 1
+        bindings = { arg_names[0] => "Cloudflare::DurableObjectState" }
+        bindings[
+          arg_names[1]
+        ] = "Cloudflare::DurableObjectRequest" if arg_names.length > 1
         bindings
       end
 
@@ -213,7 +227,8 @@ module HomuraRuntime
         return false unless call_node&.type == :send
 
         receiver, method_name = *call_node
-        method_name == :define && const_path(receiver) == 'Cloudflare::DurableObject'
+        method_name == :define &&
+          const_path(receiver) == "Cloudflare::DurableObject"
       end
 
       def infer_index_class(node)
@@ -273,7 +288,7 @@ module HomuraRuntime
           parts.unshift(n.children[1])
           n = n.children[0]
         end
-        parts.join('::')
+        parts.join("::")
       end
 
       def debug(msg)

data/lib/homura/runtime/auto_await/transformer.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-require 'parser/source/tree_rewriter'
+require "parser/source/tree_rewriter"
 
 module HomuraRuntime
   module AutoAwait