test

Author: andyw8

lib/ruby_lsp/ruby_lsp_rails/runner_client.rb

@@ -0,0 +1,77 @@
+# typed: strict
+# frozen_string_literal: true
+
+require "json"
+require "open3"
+
+# NOTE: We should avoid printing to stderr since it causes problems. We never read the standard error pipe
+# from the client, so it will become full and eventually hang or crash.
+# Instead, return a response with an `error` key.
+
+module RubyLsp
+  module Rails
+    class RunnerClient
+      extend T::Sig
+
+      sig { void }
+      def initialize
+        stdin, stdout, stderr, wait_thread = Open3.popen3(
+          "bin/rails",
+          "runner",
+          "#{__dir__}/server_start.rb",
+        )
+        @stdin = T.let(stdin, IO)
+        @stdout = T.let(stdout, IO)
+        @stderr = T.let(stderr, IO)
+        @wait_thread = T.let(wait_thread, Process::Waiter)
+        @stdin.binmode # for Windows compatibility
+        @stdout.binmode # for Windows compatibility
+      end
+
+      sig { params(name: String).returns(T.nilable(T::Hash[Symbol, T.untyped])) }
+      def model(name)
+        make_request("model", name: name)
+      end
+
+      sig { void }
+      def shutdown
+        send_notification("shutdown")
+        sleep(0.01) while @wait_thread.alive?
+        [@stdin, @stdout, @stderr].each(&:close)
+      end
+
+      private
+
+      sig { params(request: T.untyped, params: T.untyped).returns(T.untyped) }
+      def make_request(request, params = nil)
+        send_message(request, params)
+        read_response
+      end
+
+      sig { params(request: T.untyped, params: T.untyped).void }
+      def send_message(request, params = nil)
+        json = { method: request, params: params }.compact.to_json
+
+        @stdin.write("Content-Length: #{json.length}\r\n\r\n", json)
+      end
+
+      alias_method :send_notification, :send_message
+
+      sig { returns(T.nilable(T::Hash[Symbol, T.untyped])) }
+      def read_response
+        headers = @stdout.gets("\r\n\r\n")
+        raw_response = @stdout.read(T.must(headers)[/Content-Length: (\d+)/i, 1].to_i)
+        response = JSON.parse(T.must(raw_response), symbolize_names: true)
+
+        # TODO: handle exception if JSON parse fails?
+
+        if response[:error]
+          warn("Ruby LSP Rails error: " + response[:error])
+          return
+        end
+
+        response.fetch(:result)
+      end
+    end
+  end
+end

lib/ruby_lsp/ruby_lsp_rails/server.rb

@@ -0,0 +1,70 @@
+# typed: strict
+# frozen_string_literal: true
+
+require "sorbet-runtime"
+require "json"
+
+module RubyLsp
+  module Rails
+    class Server
+      VOID = Object.new
+
+      extend T::Sig
+
+      sig { params(model_name: String).returns(T.nilable(T::Hash[Symbol, T.untyped])) }
+      def resolve_database_info_from_model(model_name)
+        const = ActiveSupport::Inflector.safe_constantize(model_name)
+        unless const && const < ActiveRecord::Base && !const.abstract_class?
+          return {
+            result: nil,
+          }
+        end
+
+        schema_file = ActiveRecord::Tasks::DatabaseTasks.schema_dump_path(const.connection.pool.db_config)
+
+        {
+          result: {
+            columns: const.columns.map { |column| [column.name, column.type] },
+            schema_file: ::Rails.root + schema_file,
+          },
+        }
+      rescue => e
+        {
+          error: e.message,
+        }
+      end
+
+      sig { void }
+      def start
+        $stdin.sync = true
+        $stdout.sync = true
+
+        running = T.let(true, T::Boolean)
+
+        while running
+          headers = $stdin.gets("\r\n\r\n")
+          request = $stdin.read(headers[/Content-Length: (\d+)/i, 1].to_i)
+
+          json = JSON.parse(request, symbolize_names: true)
+          request_method = json.fetch(:method)
+          params = json[:params]
+
+          response = case request_method
+          when "shutdown"
+            running = false
+            VOID
+          when "model"
+            resolve_database_info_from_model(params.fetch(:name))
+          else
+            VOID
+          end
+
+          next if response == VOID
+
+          json_response = response.to_json
+          $stdout.write("Content-Length: #{json_response.length}\r\n\r\n#{json_response}")
+        end
+      end
+    end
+  end
+end

lib/ruby_lsp/ruby_lsp_rails/server_start.rb

@@ -0,0 +1,5 @@
+# typed: strict
+# frozen_string_literal: true
+
+require_relative "server"
+RubyLsp::Rails::Server.new.start