charai 0.1.0

data/lib/charai/spline_deceleration.rb

@@ -0,0 +1,129 @@
+module Charai
+  # ref: https://android.googlesource.com/platform/frameworks/base/+/master/core/java/android/widget/OverScroller.java
+  # Converted using ChatGPT 4o
+  #
+  # Usage:
+  # deceleration = SplineDeceleration.new(200)
+  # loop do
+  #   delta = deceleration.calc
+  #   sleep 0.016 # 60fps
+  #   break if delta.zero?
+  # end
+
+  class SplineDeceleration
+    DECELERATION_RATE = Math.log(0.78) / Math.log(0.9)
+    INFLEXION = 0.35
+    START_TENSION = 0.5
+    END_TENSION = 1.0
+    P1 = START_TENSION * INFLEXION
+    P2 = 1.0 - END_TENSION * (1.0 - INFLEXION)
+    NB_SAMPLES = 100
+
+    attr_reader :current_velocity, :current_position
+
+    def initialize(initial_velocity)
+      @initial_velocity = initial_velocity
+      @current_velocity = initial_velocity
+      @physical_coeff = 1000
+      @elapsed_time = 0 # 累積時間を管理
+      @duration = calculate_duration
+      @distance = calculate_distance
+      @previous_position = 0 # 前回の位置を保持
+      @current_position = 0
+
+      # スプラインテーブルを構築
+      @spline_position = Array.new(NB_SAMPLES + 1)
+      @spline_time = Array.new(NB_SAMPLES + 1)
+      build_spline_tables
+    end
+
+    def calculate_distance
+      l = Math.log(INFLEXION * @initial_velocity.abs / @physical_coeff)
+      decel_minus_one = DECELERATION_RATE - 1.0
+      @physical_coeff * Math.exp(DECELERATION_RATE / decel_minus_one * l)
+    end
+
+    def calculate_duration
+      l = Math.log(INFLEXION * @initial_velocity.abs / @physical_coeff)
+      decel_minus_one = DECELERATION_RATE - 1.0
+      (1000.0 * Math.exp(l / decel_minus_one)).to_i
+    end
+
+    def calc
+      @elapsed_time += 16 # 毎回16ms経過
+
+      return 0 if @elapsed_time > @duration
+
+      t = @elapsed_time.to_f / @duration
+      distance_coef = interpolate_spline_position(t)
+      @current_velocity = interpolate_spline_velocity(t)
+
+      # 現在の位置を計算
+      @current_position = (distance_coef * @distance).to_i
+
+      # 16ms前の位置からのdeltaを計算
+      delta_position = @current_position - @previous_position
+
+      # 現在の位置を次回のために保存
+      @previous_position = @current_position
+
+      delta_position
+    end
+
+    private
+
+    def build_spline_tables
+      (0...NB_SAMPLES).each do |i|
+        alpha = i.to_f / NB_SAMPLES
+        # x 方向のスプライン補間
+        x_min = 0.0
+        x_max = 1.0
+        while true
+          x = (x_min + x_max) / 2.0
+          coef = 3.0 * x * (1.0 - x)
+          tx = coef * ((1.0 - x) * P1 + x * P2) + x**3
+          break if (tx - alpha).abs < 1e-5
+          tx > alpha ? x_max = x : x_min = x
+        end
+        @spline_position[i] = coef * ((1.0 - x) * START_TENSION + x) + x**3
+
+        # y 方向のスプライン補間
+        y_min = 0.0
+        y_max = 1.0
+        while true
+          y = (y_min + y_max) / 2.0
+          coef = 3.0 * y * (1.0 - y)
+          dy = coef * ((1.0 - y) * START_TENSION + y) + y**3
+          break if (dy - alpha).abs < 1e-5
+          dy > alpha ? y_max = y : y_min = y
+        end
+        @spline_time[i] = coef * ((1.0 - y) * P1 + y * P2) + y**3
+      end
+      @spline_position[NB_SAMPLES] = @spline_time[NB_SAMPLES] = 1.0
+    end
+
+    def interpolate_spline_position(t)
+      index = (NB_SAMPLES * t).to_i
+      return 1.0 if index >= NB_SAMPLES
+
+      t_inf = index.to_f / NB_SAMPLES
+      t_sup = (index + 1).to_f / NB_SAMPLES
+      d_inf = @spline_position[index]
+      d_sup = @spline_position[index + 1]
+      velocity_coef = (d_sup - d_inf) / (t_sup - t_inf)
+      d_inf + (t - t_inf) * velocity_coef
+    end
+
+    def interpolate_spline_velocity(t)
+      index = (NB_SAMPLES * t).to_i
+      return 0.0 if index >= NB_SAMPLES
+
+      t_inf = index.to_f / NB_SAMPLES
+      t_sup = (index + 1).to_f / NB_SAMPLES
+      d_inf = @spline_position[index]
+      d_sup = @spline_position[index + 1]
+      velocity_coef = (d_sup - d_inf) / (t_sup - t_inf)
+      velocity_coef * @distance / @duration * 1000.0
+    end
+  end
+end

data/lib/charai/util.rb

@@ -0,0 +1,11 @@
+module Charai
+  class Util
+    def self.macos?
+      RUBY_PLATFORM =~ /darwin/
+    end
+
+    def self.linux?
+      RUBY_PLATFORM =~ /linux/
+    end
+  end
+end

data/lib/charai/version.rb

@@ -0,0 +1,5 @@
+# frozen_string_literal: true
+
+module Charai
+  VERSION = "0.1.0"
+end

data/lib/charai/web_socket.rb

@@ -0,0 +1,156 @@
+require 'openssl'
+require 'socket'
+require 'websocket/driver'
+
+module Charai
+  # ref: https://github.com/rails/rails/blob/master/actioncable/lib/action_cable/connection/client_socket.rb
+  # ref: https://github.com/cavalle/chrome_remote/blob/master/lib/chrome_remote/web_socket_client.rb
+  class WebSocket
+    class DriverImpl # providing #url, #write(string)
+      class SecureSocketFactory
+        def initialize(host, port)
+          @host = host
+          @port = port || 443
+        end
+
+        def create
+          tcp_socket = TCPSocket.new(@host, @port)
+          OpenSSL::SSL::SSLSocket.new(tcp_socket).tap(&:connect)
+        end
+      end
+
+      def initialize(url)
+        @url = url
+
+        endpoint = URI.parse(url)
+        @socket =
+          if endpoint.scheme == 'wss'
+            SecureSocketFactory.new(endpoint.host, endpoint.port).create
+          else
+            TCPSocket.new(endpoint.host, endpoint.port)
+          end
+      end
+
+      attr_reader :url
+
+      def write(data)
+        @socket.write(data)
+      rescue Errno::EPIPE
+        raise EOFError.new('already closed')
+      rescue Errno::ECONNRESET
+        raise EOFError.new('closed by remote')
+      end
+
+      def readpartial(maxlen = 1024)
+        @socket.readpartial(maxlen)
+      rescue Errno::ECONNRESET
+        raise EOFError.new('closed by remote')
+      end
+
+      def dispose
+        @socket.close
+      end
+    end
+
+    STATE_CONNECTING = 0
+    STATE_OPENED = 1
+    STATE_CLOSING = 2
+    STATE_CLOSED = 3
+
+    def initialize(url:, max_payload_size: 256 * 1024 * 1024)
+      @impl = DriverImpl.new(url)
+      @driver = ::WebSocket::Driver.client(@impl, max_length: max_payload_size)
+
+      setup
+      @driver.start
+
+      Thread.new do
+        wait_for_data until @ready_state >= STATE_CLOSING
+      rescue EOFError
+        # Browser was gone.
+        # We have nothing todo. Just finish polling.
+        if @ready_state < STATE_CLOSING
+          handle_on_close(reason: 'Going Away', code: 1001)
+        end
+      rescue IOError
+        # connection closed. Just ignore it.
+        raise if @ready_state < STATE_CLOSING
+      end
+    end
+
+    private def setup
+      @ready_state = STATE_CONNECTING
+      @driver.on(:open) do
+        @ready_state = STATE_OPENED
+        handle_on_open
+      end
+      @driver.on(:close) do |event|
+        @ready_state = STATE_CLOSED
+        handle_on_close(reason: event.reason, code: event.code)
+      end
+      @driver.on(:error) do |event|
+        unless handle_on_error(error_message: event.message)
+          raise event.message
+        end
+      end
+      @driver.on(:message) do |event|
+        handle_on_message(event.data)
+      end
+    end
+
+    private def wait_for_data
+      @driver.parse(@impl.readpartial)
+    end
+
+    # @param message [String]
+    def send_text(message)
+      return if @ready_state >= STATE_CLOSING
+      @driver.text(message)
+    end
+
+    def close(code: 1000, reason: "")
+      return if @ready_state >= STATE_CLOSING
+      @ready_state = STATE_CLOSING
+      @driver.close(reason, code)
+      @impl.dispose
+    end
+
+    def on_open(&block)
+      @on_open = block
+    end
+
+    # @param block [Proc(reason: String, code: Numeric)]
+    def on_close(&block)
+      @on_close = block
+    end
+
+    # @param block [Proc(error_message: String)]
+    def on_error(&block)
+      @on_error = block
+    end
+
+    def on_message(&block)
+      @on_message = block
+    end
+
+    private def handle_on_open
+      @on_open&.call
+    end
+
+    private def handle_on_close(reason:, code:)
+      @on_close&.call(reason, code)
+    end
+
+    private def handle_on_error(error_message:)
+      return false if @on_error.nil?
+
+      @on_error.call(error_message)
+    end
+
+    private def handle_on_message(data)
+      return if @ready_state != STATE_OPENED
+
+      @on_message&.call(data)
+    end
+  end
+end

data/lib/charai.rb

@@ -0,0 +1,18 @@
+require 'charai/version'
+
+require 'capybara'
+require 'concurrent'
+
+require 'charai/action_queue'
+require 'charai/agent'
+require 'charai/browser'
+require 'charai/browser_launcher'
+require 'charai/browser_process'
+require 'charai/browsing_context'
+require 'charai/driver'
+require 'charai/input_tool'
+require 'charai/openai_chat'
+require 'charai/openai_configuration'
+require 'charai/spline_deceleration'
+require 'charai/util'
+require 'charai/web_socket'

metadata

@@ -0,0 +1,109 @@
+--- !ruby/object:Gem::Specification
+name: charai
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- YusukeIwaki
+autorequire:
+bindir: exe
+cert_chain: []
+date: 2024-10-25 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: capybara
+  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: concurrent-ruby
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.1.6
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.1.6
+- !ruby/object:Gem::Dependency
+  name: websocket-driver
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.6.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 0.6.1
+description: Prototype impl for Kaigi on Rails 2024 presentation.
+email:
+- q7w8e9w8q7w8e9@yahoo.co.jp
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".rspec"
+- ".rubocop.yml"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- charai.gemspec
+- lib/charai.rb
+- lib/charai/action_queue.rb
+- lib/charai/agent.rb
+- lib/charai/browser.rb
+- lib/charai/browser_launcher.rb
+- lib/charai/browser_process.rb
+- lib/charai/browsing_context.rb
+- lib/charai/driver.rb
+- lib/charai/input_tool.rb
+- lib/charai/openai_chat.rb
+- lib/charai/openai_configuration.rb
+- lib/charai/spline_deceleration.rb
+- lib/charai/util.rb
+- lib/charai/version.rb
+- lib/charai/web_socket.rb
+homepage: https://github.com/YusukeIwaki/charai
+licenses:
+- MIT
+metadata:
+  homepage_uri: https://github.com/YusukeIwaki/charai
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 3.2.0
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.5.3
+signing_key:
+specification_version: 4
+summary: charai(Chat + Ruby + AI) Capybara driver
+test_files: []