neh 0.0.24 → 0.0.25

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: acfcef0ee91e2a3a87f1618116449112811a1db27d457f4b822618e73823935c
-  data.tar.gz: 8573e2ac999dd18e5a5529793f548420d69ab70d8c98f8e29195354011c43222
+  metadata.gz: 550819e91c6cd6dc8c57c7fa1cff4698652e77333a7bac1379697ebef5a7b314
+  data.tar.gz: 25526b39861ddd276647f446aacea145743757a254613d0245236a79237c157a
 SHA512:
-  metadata.gz: d90e5b71167430ef4a4eb85c3a3024f207757a28593c71bfe398230e724cbf378755deea51921f73136031d1bbca6b7b03f7629baca83bc4fe711ca301fd51b3
-  data.tar.gz: 7918d5ffaa197c6c36d0e38d95efb1cf906bf4d8fed5c00e218cfac5e66c47dc55a690b088808ab3b10a0bc301783594f766216ba1fabd9ab5091455d315ea7a
+  metadata.gz: 670d5fa4c6d5188462e66ec1de7f7d9449d1f568762c0da7618b3f37603520d7386c0b4998991ff8237690e59e836c26262d5d76ec41158662b3714e61dcd781
+  data.tar.gz: 6b00589c49832231863c3e2282b9a5e45cd89c24889c11a13dbb6a04c7eb5e4bd1b9ce7157bd682041d03dbe0776eb7edb35d6c2c3b9319050a0a4674cdc3cd7

data/lib/neh/cli/o.rb

@@ -1,12 +1,149 @@
 # frozen_string_literal: true
 
-module Neh::Cli
-  class O < Base
-    private
+require 'securerandom'
+require 'net/http'
+require 'async'
+require 'async/io/stream'
+require 'async/http/endpoint'
+require 'async/websocket/client'
+require 'faraday'
+require 'pry'
+require 'active_support/all'
 
-    def close(connection)
-      puts ''
-      super
+class Neh::Cli::O # rubocop:disable Metrics/ClassLength
+  attr_reader :uuid
+
+  def initialize(*args, options:)
+    @message = args.join(' ')
+    @options = options
+    @message_pool = {}
+    @expected_sequence_number = 1
+    @uuid = SecureRandom.uuid
+  end
+
+  def execute
+    Async::WebSocket::Client.connect(endpoint, headers: { 'Authorization' => "Bearer #{token}" }) do |connection|
+      while (message = connection.read)
+        parsed_message =
+          JSON.parse(message, symbolize_names: true)
+
+        on_receive(connection, parsed_message)
+      end
     end
   end
+
+  private
+
+  def token
+    neh_personal_access_token = ENV.fetch('NEH_PERSONAL_ACCESS_TOKEN', nil)
+
+    puts 'Please set the environment variable NEH_PERSONAL_ACCESS_TOKEN.' if neh_personal_access_token.blank?
+
+    neh_personal_access_token
+  end
+
+  def on_receive(connection, body)
+    if body[:type]
+      handle_action_cable_messages(connection, body)
+    else
+      handle_broadcasted_messages(connection, body[:message])
+    end
+  end
+
+  def handle_action_cable_messages(connection, message)
+    type = message[:type]
+
+    case type
+    when 'welcome'
+      subscribe(connection)
+    when 'confirm_subscription'
+      on_subscribed
+    when 'ping'
+      # do nothing
+    when 'disconnect'
+      puts "Connection has been disconnected. Reason: #{message[:reason]}"
+      close(connection)
+    else
+      puts "Unknown message type: #{type}. Closing connection."
+      close(connection)
+    end
+  end
+
+  def process_message_in_order
+    while @message_pool.key?(@expected_sequence_number)
+      print @message_pool.delete(@expected_sequence_number)
+      @expected_sequence_number += 1
+    end
+  end
+
+  def handle_broadcasted_messages(connection, message)
+    type = message[:type]
+    body = message[:body]
+
+    case type
+    when 'output'
+      sequence_number = message[:sequence_number]
+      @message_pool[sequence_number] = body
+      process_message_in_order
+    when 'error'
+      print body
+    when 'worker_done'
+      close(connection)
+    end
+  end
+
+  def close(connection)
+    connection.close
+  end
+
+  def subscribe(connection)
+    content =
+      {
+        command: :subscribe,
+        identifier: {
+          channel: 'LargeLanguageModelQueryChannel',
+          uuid:
+        }.to_json
+      }
+    connection.write(content.to_json)
+    connection.flush
+  end
+
+  def on_subscribed
+    response = http_connection.post("/api/neh/o") do |req|
+      req.body = {
+        message: @message,
+        uuid:,
+        token:
+      }.to_json
+
+      req.headers['Content-Type'] = 'application/json'
+    end
+
+    body = JSON.parse response.body
+    message = body['message']
+    puts message if message.present?
+  end
+
+  def http_connection
+    Faraday.new(url: http_url) do |faraday|
+      faraday.adapter Faraday.default_adapter
+      faraday.headers['Authorization'] = "Bearer #{token}"
+    end
+  end
+
+  def endpoint
+    Async::HTTP::Endpoint.parse(
+      ws_url,
+      alpn_protocols: Async::HTTP::Protocol::HTTP11.names #=> ["http/1.1"]
+    )
+  end
+
+  def http_url
+    ENV['WORKING_ON_LOCALHOST'].present? ? "http://localhost:6060/cable?uuid=#{uuid}" : "https://yoryo-app.onrender.com/cable?uuid=#{uuid}"
+  end
+
+  def ws_url
+    ENV['WORKING_ON_LOCALHOST'].present? ? "ws://localhost:6060/cable?uuid=#{uuid}" : "wss://yoryo-app.onrender.com/cable?uuid=#{uuid}" # rubocop:disable Metrics/LineLength
+  end
 end

data/lib/neh/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Neh
-  VERSION = "0.0.24"
+  VERSION = "0.0.25"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: neh
 version: !ruby/object:Gem::Version
-  version: 0.0.24
+  version: 0.0.25
 platform: ruby
 authors:
 - Atsushi Ishida
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-08-29 00:00:00.000000000 Z
+date: 2024-09-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport