refined-steep-server 0.1.0

data/sig/external/steep.rbs

@@ -0,0 +1,376 @@
+module Steep
+  module PathHelper
+    def self.to_pathname: (String uri) -> Pathname?
+    def self.to_uri: (Pathname path) -> URI::Generic
+  end
+
+  module Server
+    module LSPFormatter
+      def self.format_hover_content: (untyped content) -> untyped
+      def self.format_completion_docs: (untyped item) -> String?
+      def self.declaration_summary: (untyped decl) -> String?
+      def self.markup_content: (?String? string) { () -> String? } -> untyped
+    end
+  end
+
+  class Source
+    def self.parse: (String content, path: Pathname, factory: untyped) -> Source
+    def without_unrelated_defs: (line: Integer, column: Integer) -> Source
+  end
+
+  class Typing
+    class UnknownNodeError < StandardError
+    end
+  end
+
+  class Project
+    attr_reader targets: Array[Target]
+    attr_reader steepfile_path: Pathname
+    attr_reader base_dir: Pathname
+    attr_accessor global_options: untyped
+
+    def initialize: (steepfile_path: Pathname, ?base_dir: Pathname?) -> void
+    def relative_path: (Pathname path) -> Pathname
+    def absolute_path: (Pathname path) -> Pathname
+    def target_for_source_path: (Pathname path) -> Target?
+    def target_for_signature_path: (Pathname path) -> Target?
+
+    class Target
+      attr_reader name: Symbol
+      def possible_source_file?: (Pathname path) -> untyped
+      def code_diagnostics_config: () -> Hash[untyped, untyped]?
+      def new_env_loader: () -> untyped
+      def implicitly_returns_nil: () -> bool
+    end
+
+    class DSL
+      def self.parse: (Project project, String code, ?filename: String) -> void
+    end
+  end
+
+  module Services
+    class TypeCheckService
+      attr_reader project: Project
+      attr_reader source_files: Hash[Pathname, SourceFile]
+      attr_reader signature_services: Hash[Symbol, SignatureService]
+      attr_reader signature_validation_diagnostics: untyped
+
+      def initialize: (project: Project) -> void
+      def update: (changes: Hash[Pathname, Array[ContentChange]]) -> void
+      def typecheck_source: (path: Pathname, target: Project::Target) -> Array[untyped]?
+      def validate_signature: (path: Pathname, target: Project::Target) -> Array[untyped]?
+
+      class SourceFile
+        attr_reader path: Pathname
+        attr_reader content: String
+        attr_reader node: untyped
+        attr_reader typing: untyped
+        attr_reader errors: Array[untyped]?
+        attr_reader ignores: untyped
+
+        def diagnostics: () -> Array[untyped]?
+      end
+    end
+
+    class SignatureService
+      def current_subtyping: () -> untyped
+      def latest_rbs_index: () -> untyped
+      def latest_env: () -> untyped
+      def env_rbs_paths: () -> Set[Pathname]
+    end
+
+    class ContentChange
+      attr_reader range: [Position, Position]?
+      attr_reader text: String
+
+      def initialize: (?range: [Position, Position]?, text: String) -> void
+
+      class Position
+        attr_reader line: Integer
+        attr_reader column: Integer
+
+        def initialize: (line: Integer, column: Integer) -> void
+      end
+    end
+
+    class PathAssignment
+      def self.all: () -> PathAssignment
+    end
+
+    class FileLoader
+      def initialize: (base_dir: Pathname) -> void
+      def each_path_in_target: (Project::Target target, ?Array[String] command_line_args) { (Pathname) -> void } -> void
+    end
+
+    module HoverProvider
+      def self.content_for: (service: TypeCheckService, path: Pathname, line: Integer, column: Integer) -> untyped
+    end
+
+    class GotoService
+      def initialize: (type_check: TypeCheckService, assignment: PathAssignment) -> void
+      def definition: (path: Pathname, line: Integer, column: Integer) -> Array[untyped]
+      def implementation: (path: Pathname, line: Integer, column: Integer) -> Array[untyped]
+      def type_definition: (path: Pathname, line: Integer, column: Integer) -> Array[untyped]
+    end
+
+    class SignatureHelpProvider
+      def initialize: (source: Source, subtyping: untyped) -> void
+      def run: (line: Integer, column: Integer) -> [Array[untyped], Integer]?
+    end
+
+    class CompletionProvider
+      def initialize: (source_text: String, path: Pathname, subtyping: untyped) -> void
+      def run: (line: Integer, column: Integer) -> Array[untyped]
+
+      class LocalVariableItem
+        attr_reader identifier: Symbol
+        attr_reader type: untyped
+        attr_reader range: untyped
+      end
+
+      class ConstantItem
+        attr_reader identifier: Symbol
+        attr_reader range: untyped
+        attr_reader env: untyped
+        attr_reader full_name: untyped
+        def class?: () -> bool
+        def module?: () -> bool
+        def decl: () -> untyped
+        def comments: () -> Array[untyped]
+        def deprecated?: () -> bool
+      end
+
+      class SimpleMethodNameItem
+        attr_reader identifier: Symbol
+        attr_reader method_name: untyped
+        attr_reader range: untyped
+        attr_reader deprecated: bool
+      end
+
+      class ComplexMethodNameItem
+        attr_reader identifier: Symbol
+        attr_reader method_names: Array[untyped]
+        attr_reader range: untyped
+      end
+
+      class GeneratedMethodNameItem
+        attr_reader identifier: Symbol
+        attr_reader range: untyped
+      end
+
+      class InstanceVariableItem
+        attr_reader identifier: Symbol
+        attr_reader type: untyped
+        attr_reader range: untyped
+      end
+
+      class KeywordArgumentItem
+        attr_reader identifier: Symbol
+        attr_reader range: untyped
+      end
+
+      class TypeNameItem
+        attr_reader absolute_type_name: untyped
+        attr_reader relative_type_name: untyped
+        attr_reader range: untyped
+        attr_reader env: untyped
+      end
+
+      class TextItem
+        attr_reader label: String
+        attr_reader text: String
+        attr_reader help_text: String?
+        attr_reader range: untyped
+      end
+    end
+  end
+
+  module Diagnostic
+    class LSPFormatter
+      def initialize: (?Hash[untyped, untyped]? config, ?default_severity: Symbol) -> void
+      def format: (untyped diagnostic) -> untyped
+    end
+  end
+
+  module Index
+    class SignatureSymbolProvider
+      attr_reader indexes: Hash[untyped, untyped]
+
+      def initialize: (project: Project, assignment: Services::PathAssignment) -> void
+      def query_symbol: (String query) -> Array[untyped]
+    end
+  end
+end
+
+module LanguageServer
+  module Protocol
+    module Interface
+      # All interface classes return hash-like objects
+      class InitializeResult
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class ServerCapabilities
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class TextDocumentSyncOptions
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class SaveOptions
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class CompletionOptions
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class SignatureHelpOptions
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class Hover
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class MarkupContent
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class Range
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class Position
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class CompletionList
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class CompletionItem
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class CompletionItemLabelDetails
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class TextEdit
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class SignatureHelp
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class SignatureInformation
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class ParameterInformation
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class SymbolInformation
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class ShowMessageParams
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class LogMessageParams
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class PublishDiagnosticsParams
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class ProgressParams
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class WorkDoneProgressBegin
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+
+      class WorkDoneProgressCreateParams
+        def initialize: (**untyped) -> void
+        def to_hash: () -> Hash[Symbol, untyped]
+      end
+    end
+
+    module Constant
+      module TextDocumentSyncKind
+        INCREMENTAL: Integer
+      end
+
+      module MessageType
+        LOG: Integer
+        INFO: Integer
+        WARNING: Integer
+        ERROR: Integer
+      end
+
+      module ErrorCodes
+        REQUEST_CANCELLED: Integer
+        INTERNAL_ERROR: Integer
+        REQUEST_FAILED: Integer
+      end
+
+      module CompletionItemKind
+        TEXT: Integer
+        FUNCTION: Integer
+        VARIABLE: Integer
+        CLASS: Integer
+        INTERFACE: Integer
+        FIELD: Integer
+        CONSTANT: Integer
+        SNIPPET: Integer
+      end
+
+      module InsertTextFormat
+        SNIPPET: Integer
+      end
+
+      module MarkupKind
+        MARKDOWN: String
+      end
+    end
+
+    module Transport
+    end
+  end
+end
+
+module Parser
+  class SyntaxError < StandardError
+  end
+end

data/sig/refined/steep/server.rbs

@@ -0,0 +1,8 @@
+module Refined
+  module Steep
+    module Server
+      VERSION: String
+      # See the writing guide of rbs: https://github.com/ruby/rbs#guides
+    end
+  end
+end

data/spec/refined/steep/server/base_server_spec.rb

@@ -0,0 +1,116 @@
+# frozen_string_literal: true
+
+RSpec.describe Refined::Steep::Server::BaseServer do
+  def build_message(method:, id: nil, params: {})
+    msg = { method: method, params: params }
+    msg[:id] = id if id
+    msg
+  end
+
+  def encode_message(hash)
+    json = JSON.generate(hash)
+    "Content-Length: #{json.bytesize}\r\n\r\n#{json}"
+  end
+
+  def create_server(messages)
+    raw = messages.map { |m| encode_message(m) }.join
+    reader = StringIO.new(raw)
+    writer = StringIO.new
+
+    server = TestServer.new(reader: reader, writer: writer)
+    [server, writer]
+  end
+
+  # Concrete subclass for testing
+  before do
+    stub_const("TestServer", Class.new(Refined::Steep::Server::BaseServer) do
+      attr_reader :processed_messages
+
+      def initialize(**options)
+        @processed_messages = []
+        super
+      end
+
+      def process_message(message)
+        @processed_messages << message
+      end
+
+      private
+
+      def shutdown
+        # no-op
+      end
+    end)
+  end
+
+  describe "#start" do
+    it "processes initialize message synchronously" do
+      messages = [build_message(method: "initialize", id: 1, params: { capabilities: {} })]
+      server, = create_server(messages)
+
+      server.start
+
+      expect(server.processed_messages.size).to eq(1)
+      expect(server.processed_messages[0][:method]).to eq("initialize")
+    end
+
+    it "processes initialized notification synchronously" do
+      messages = [build_message(method: "initialized")]
+      server, = create_server(messages)
+
+      server.start
+
+      expect(server.processed_messages.size).to eq(1)
+      expect(server.processed_messages[0][:method]).to eq("initialized")
+    end
+
+    it "queues other messages for worker thread" do
+      messages = [build_message(method: "textDocument/hover", id: 1)]
+      server, = create_server(messages)
+
+      server.start
+      # Give the worker thread time to process
+      sleep 0.1
+
+      expect(server.processed_messages.size).to eq(1)
+      expect(server.processed_messages[0][:method]).to eq("textDocument/hover")
+    end
+
+    it "handles shutdown and exit sequence" do
+      messages = [
+        build_message(method: "shutdown", id: 1),
+        build_message(method: "exit"),
+      ]
+      server, writer = create_server(messages)
+
+      expect { server.start }.to raise_error(SystemExit) do |error|
+        expect(error.status).to eq(0)
+      end
+
+      # Verify shutdown response was written
+      output = writer.string
+      expect(output).to include('"id":1')
+      expect(output).to include('"result":null')
+    end
+  end
+
+  describe "#send_message" do
+    it "sends result message through outgoing queue" do
+      messages = [
+        build_message(method: "initialize", id: 1, params: { capabilities: {} }),
+      ]
+      server, writer = create_server(messages)
+
+      # Override process_message to send a result
+      allow(server).to receive(:process_message) do |msg|
+        server.send(:send_message, Refined::Steep::Server::Result.new(id: msg[:id], response: { capabilities: {} }))
+      end
+
+      server.start
+      sleep 0.1
+
+      output = writer.string
+      expect(output).to include('"capabilities"')
+    end
+  end
+end

data/spec/refined/steep/server/io_spec.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+RSpec.describe Refined::Steep::Server::MessageReader do
+  it "reads JSON-RPC messages from IO" do
+    json = '{"method":"initialize","params":{}}'
+    raw = "Content-Length: #{json.bytesize}\r\n\r\n#{json}"
+    io = StringIO.new(raw)
+
+    reader = Refined::Steep::Server::MessageReader.new(io)
+    messages = []
+    reader.each_message { |msg| messages << msg }
+
+    expect(messages.size).to eq(1)
+    expect(messages[0][:method]).to eq("initialize")
+  end
+
+  it "reads multiple messages" do
+    messages_data = [
+      { method: "initialize", params: {} },
+      { method: "initialized", params: {} },
+    ]
+
+    raw = messages_data.map do |data|
+      json = JSON.generate(data)
+      "Content-Length: #{json.bytesize}\r\n\r\n#{json}"
+    end.join
+
+    io = StringIO.new(raw)
+    reader = Refined::Steep::Server::MessageReader.new(io)
+    messages = []
+    reader.each_message { |msg| messages << msg }
+
+    expect(messages.size).to eq(2)
+    expect(messages[0][:method]).to eq("initialize")
+    expect(messages[1][:method]).to eq("initialized")
+  end
+end
+
+RSpec.describe Refined::Steep::Server::MessageWriter do
+  it "writes JSON-RPC messages to IO" do
+    io = StringIO.new
+    writer = Refined::Steep::Server::MessageWriter.new(io)
+
+    writer.write({ id: 1, result: nil })
+
+    output = io.string
+    expect(output).to match(/Content-Length: \d+\r\n\r\n/)
+    json_part = output.sub(/Content-Length: \d+\r\n\r\n/, "")
+    parsed = JSON.parse(json_part, symbolize_names: true)
+    expect(parsed[:jsonrpc]).to eq("2.0")
+    expect(parsed[:id]).to eq(1)
+    expect(parsed[:result]).to be_nil
+  end
+end