lsp_router 0.1.1 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c3ba53e8db9e41cea66c0d4f935517fe4f3e4e8e24a6173794e53ea409556847
-  data.tar.gz: 984ea810f99d29a70efbb798b76c5d9f161b5bb5433ab0b96c52ed015529699c
+  metadata.gz: a4434480350b472e4784e0bc3dcd003f5863eeb5ee25accbcc5491d07ecdac39
+  data.tar.gz: ba46303f99aa22018ade578cf1534ccfb7149b356b43ca262ae65c5cb729eb64
 SHA512:
-  metadata.gz: 40903c06994c73209c64535f12b417c913e59db4a7b856eae9504aac1e86102a273656e6f3a4148b3fa54e8c862855cd846e5e29d00f457b6a8e691b727ee919
-  data.tar.gz: bde44dd17b4288879bf96dee727320241ebf143d13e4e5ee8fadbf562223db6a0c5214c584847369a6ff97f20b2048192dc2faaea076fbfc7f213b68b479818c
+  metadata.gz: 4e5716f9fab94ec450991faf87af92d81544341df78a795c9506a94b7feee39add469211144762b7514f5d46b76128e4c68b21d8ef12f300dbae9c28f3bdd7e4
+  data.tar.gz: 7795648666367d4bada69bfa336c7cdbcf41f3f4039f17e819c0c79a61acbfbd2a3381c44f1b2e222f889cb9df81b32e61bf83b0e34fbfb51862b8009798c069

data/.rubocop.yml

@@ -185,5 +185,3 @@ Style/VariableInterpolation:
 
 Style/WordArray:
   Enabled: false
-
-

data/README.md

@@ -17,7 +17,7 @@ logfile '/tmp/lsp_router'
 loglevel :info
 
 server :rubocop do
-  command 'rubocop-lsp'
+  command 'rubocop --lsp'
   mode :stdio
 end
 
@@ -27,7 +27,7 @@ server :solargraph do
 end
 ```
 
-この例では、`rubocop-lsp` と `solargraph stdio` を起動して、クライアントからの処理を振り分ける。
+この例では、`rubocop --lsp` と `solargraph stdio` を起動して、クライアントからの処理を振り分ける。
 最初に各サーバーの capabilities を確認して、各サーバーにどの機能があるかを確認し、クライアントからの REQUEST は対応しているサーバーに渡す。複数のサーバーが同じ機能を持っていれば上に書いたサーバーが優先される。NOTIFICATION は全サーバーに渡す。
 
 ## Example
@@ -37,3 +37,7 @@ Emacs の Eglot の場合はこんな感じに設定するといいっぽい。
 ```
 (add-to-list 'eglot-server-programs '((ruby-mode ruby-ts-mode) . ("lsp_router" "--error=lsp_router.err" "lsp_router.conf")))
 ```
+
+## License
+
+GPLv3

data/lib/lsp_router/server_side.rb

@@ -16,21 +16,26 @@ class LspRouter
       @command = server.attr[:command]
       @stdin, @stdout, @stderr = Open3.popen3(@command)
       @capabilities = {}
+      @terminated = false
     end
 
     # @param req [Hash]
     def init(req)
       write(req)
-      data = read
-      capa = data.dig('result', 'capabilities')
-      capa.keys.grep(/Provider/).each do |key|
-        @capabilities[key.sub(/Provider\z/, '')] = true
+      while (data = read)
+        capa = data.dig('result', 'capabilities')
+        next unless capa
+        capa.keys.grep(/Provider/).each do |key|
+          @capabilities[key.sub(/Provider\z/, '')] = true
+        end
+        break
       end
       data
     end
 
     # @param data [Hash]
     def write(data)
+      return if @terminated
       log("#{name}<", data)
       json = data.to_json
       @stdin.puts "Content-Length: #{json.bytesize}\r\n\r\n"
@@ -40,8 +45,12 @@ class LspRouter
 
     # @return [Hash]
     def read
+      return nil if @terminated
       header = @stdout.gets("\r\n\r\n")
-      return unless header
+      unless header
+        @terminated = true
+        return nil
+      end
       fields = header.lines.map do |line|
         n, v = line.chomp.split(/: */, 2)
         [n.downcase, v] if n

data/lib/lsp_router/version.rb

@@ -1,3 +1,3 @@
 class LspRouter
-  VERSION = '0.1.1'
+  VERSION = '0.1.4'
 end

data/lib/lsp_router.rb

@@ -42,13 +42,21 @@ class LspRouter
   end
 
   def loop
-    req = @client.read
-    servers = @config.servers.map{LspRouter::ServerSide.new(_1, logger: @logger)}
-    primary = servers.first
+    @servers = @config.servers.map{LspRouter::ServerSide.new(_1, logger: @logger)}
+    @primary = @servers.first
+    @server_map = @servers.map.to_h{[_1.name.to_s, _1]}
+    @capa_server = {}
+
+    initial_packet
+    threads = server_loop
+    threads.push client_loop
+    threads.each(&:join)
+  end
 
+  def initial_packet
+    req = @client.read
     res = nil
-    capa_server = {}
-    servers.each do |server|
+    @servers.each do |server|
       data = server.init(req)
       if res
         res['result']['capabilities'].update(data.dig('result', 'capabilities')){|_, v, _| v}
@@ -56,40 +64,55 @@ class LspRouter
         res = data
       end
       server.capabilities.each_key do |key|
-        capa_server[key] ||= server
+        @capa_server[key] ||= server
       end
     end
     @client.write(res)
+  end
 
-    thr = Thread.new do
+  def client_loop
+    Thread.new do
       Thread.abort_on_exception = true
       while true
         data = @client.read
         break unless data
-        if data['id']
-          # request
+        if data['method'] == 'shutdown' || data['method'] == 'exit'
+          @servers.each{_1.write(data)}
+          break if data['method'] == 'exit'
+          next
+        end
+        if data['id'] # request or response
           method = data['method']&.split('/')&.last
-          server = capa_server[method] || primary
-          server.write(data)
-        else
-          # notification
-          servers.each do |server|
-            server.write(data)
+          if data['method']  # request
+            server = @capa_server[method] || @primary
+          else  # response
+            server_name, id = JSON.parse(data['id'])
+            data['id'] = id
+            server = @server_map[server_name]
           end
+          server.write(data)
+        else # notification
+          @servers.each{_1.write(data)}
         end
       end
+      @logger.info "client: end"
     end
+  end
 
-    servers.each do |server|
+  def server_loop
+    @servers.map do |server|
       Thread.new do
         Thread.abort_on_exception = true
         while true
           data = server.read
           break unless data
+          if data['id'] && data['method']  # request
+            data['id'] = [server.name, data['id']].to_json
+          end
           @client.write(data)
         end
+        @logger.info "server: [#{server.name}] end"
       end
     end
-    thr.join
   end
 end

metadata

@@ -1,14 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: lsp_router
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.4
 platform: ruby
 authors:
 - TOMITA Masahiro
-autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-12-20 00:00:00.000000000 Z
+date: 1980-01-02 00:00:00.000000000 Z
 dependencies: []
 description: routing multiple LSP server by capability
 email:
@@ -38,7 +37,6 @@ metadata:
   homepage_uri: https://gitlab.com/tmtms/lsp_router
   source_code_uri: https://gitlab.com/tmtms/lsp_router
   rubygems_mfa_required: 'true'
-post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -53,8 +51,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.0.dev
-signing_key:
+rubygems_version: 3.7.0.dev
 specification_version: 4
 summary: LSP router
 test_files: []