nrepl-lazuli 0.4.0 → 0.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8c38f3cd330d07e579fff5971c656a12c4ee9f9a1488aba33d8183dd80b1236c
-  data.tar.gz: a63be29edddccc449e4a5bbdd9f80a05e2fb2fa0b3ca0b271ee50830da521118
+  metadata.gz: e7d68153dc3d474b045d5856d097c51670cfeb3ae688396a45c5cd16d117232a
+  data.tar.gz: 5e65794b9839a3f7d9c0aa341f3675c58e1a25290afa0f57f532cb1cb23a8be2
 SHA512:
-  metadata.gz: 2d75413ea25c513b6107daacd4f4128649c7696fc170f8536d8f636a166ce5fcb9ba675a4c218de9b9ec906f488c7cc9dfbdd48afbce0e828677d08c0bc96283
-  data.tar.gz: a3bb98a1bc7d9ba46d9bbc0cd1ea3b30dfa186da9444a80acf03015c427430962d4bf3f14dfc0d695b7050fe8caf632b14c1bac0a92edbe65ee9fd0f04a803c5
+  metadata.gz: 14c80200837368c174425e48735927e8c4e0be8214bbd5c1171e1338f19ac3b225b02a0631c9a269f19bc5846d7eb540944a96edf75ffd922cf8938923803f4a
+  data.tar.gz: e66a6400f737902c281c0c2c09c0d584ebbc5bd47a8d1262fd3b46db6afdbd32a57d1a5acfc220baa37bdf2fa9c02425d6f4f76acff4f16ecc7fca85bc5e910a

data/lib/nrepl-lazuli/bencode.rb

@@ -0,0 +1,96 @@
+class BEncode
+  def self.encode(data)
+    case data
+    when Integer
+      "i#{data}e"
+    when String
+      "#{data.bytesize}:#{data}"
+    when Array
+      "l#{data.map { |e| encode(e) }.join}e"
+    when Hash
+      "d#{data.map { |k, v| encode(k) + encode(v) }.join}e"
+    else
+      raise ArgumentError, "Cannot BEncode type: #{data.class}"
+    end
+  end
+
+  def initialize(stream)
+    @stream =
+      if stream.kind_of?(IO) || stream.kind_of?(StringIO)
+        stream
+      elsif stream.respond_to? :string
+        StringIO.new stream.string
+      elsif stream.respond_to? :to_s
+        StringIO.new stream.to_s
+      end
+    @encoding = @stream.external_encoding || Encoding::default_external
+  end
+
+  def parse!
+    case peek
+      when ?i then parse_integer!
+      when ?l then parse_list!
+      when ?d then parse_dict!
+      when ?0 .. ?9 then parse_string!
+    end
+  end
+
+  def eos?
+    @stream.eof?
+  end
+
+  private
+
+  def parse_integer!
+    @stream.getc
+    num = @stream.gets("e") or raise ArgumentError
+    num.chop.to_i
+  end
+
+  def parse_list!
+    @stream.getc
+    ary = []
+    ary.push(parse!) until peek == ?e
+    @stream.getc
+    ary
+  end
+
+  def parse_dict!
+    @stream.getc
+    hsh = {}
+    until peek == ?e
+      key = parse!
+
+      unless key.is_a? String or key.is_a? Integer
+        raise ArgumentError, "key must be a string or number"
+      end
+
+      val = parse!
+
+      hsh.store(key.to_s, val)
+    end
+    @stream.getc
+    hsh
+  end
+
+  def parse_string!
+    num = @stream.gets(":") or
+      raise ArgumentError, "invalid string length (no colon)"
+
+    begin
+      length = num.chop.to_i
+      return "" if length == 0 # Workaround for Rubinius bug
+      str = @stream.read(length).force_encoding(@encoding)
+    rescue => e
+      raise ArgumentError, "invalid string length #{e}"
+    end
+
+    str
+  end
+
+  def peek
+    c = @stream.getc
+    @stream.ungetc(c)
+    c
+  end
+end

data/lib/nrepl-lazuli/client.rb

@@ -0,0 +1,66 @@
+# frozen_string_literal: true
+
+require_relative 'bencode'
+require 'timeout'
+
+module NREPL
+  class Client
+    attr_reader :host, :port
+
+    def initialize(port: DEFAULT_PORT, host: DEFAULT_HOST)
+      @port       = port
+      @host       = host
+      @socket = TCPSocket.new(@host, @port)
+      @bencode = BEncode.new(@socket)
+      @last_id = 0
+    end
+
+    def stop
+      unless closed?
+        @socket.close
+        @socket = nil
+        @bencode = nil
+      end
+    end
+
+    def rpc(msg)
+      id = msg["id"] || msg[:id]
+      if(!id)
+        @last_id += 1
+        id = "autogen_#@last_id"
+        msg = msg.merge('id' => id)
+      end
+
+      write(msg)
+      Timeout.timeout(10) do
+        loop do
+          ret = read
+          if(ret["id"] == id)
+            break ret
+          end
+        end
+      end
+    end
+
+    def write(msg)
+      @socket.write(BEncode.encode(msg))
+      @socket.flush
+    end
+
+    def read
+      @bencode.parse!
+    end
+
+    def register_session
+      write('op' => 'clone')
+      msg = read!
+
+      raise ArgumentError, "failed to create session" unless msg['new_session']
+      msg['new_session']
+    end
+
+    def closed?
+      @socket.nil?
+    end
+  end
+end

data/lib/nrepl-lazuli/connection.rb

@@ -3,12 +3,16 @@ module NREPL
     @@debug_counter = 0
 
     def initialize(
-      input, debug: false, out: input,
+      input,
+      server:,
+      debug: false,
+      out: input,
       binding: nil,
       bindings: {},
       bindings_by_id: {}
     )
       @debug = debug
+      @server = server
       @in = input
       @out = out
       @pending_evals = {}
@@ -135,6 +139,10 @@ module NREPL
           'rows' => rows_bindings.keys.sort,
           'op' => msg['op']
         }))
+      when 'set_trace'
+        disable = msg['trace'] == 0
+        disable ? @server.call_trace.disable : @server.call_trace.enable
+        send_msg(response_for(msg, { 'op' => msg['op'], 'status' => ['done'] }))
       else
         send_msg(response_for(msg, {
           'op' => msg['op'],

data/lib/nrepl-lazuli/server.rb

@@ -10,7 +10,7 @@ require_relative 'fake_stdout'
 
 module NREPL
   class Server
-    attr_reader :debug, :port, :host
+    attr_reader :debug, :port, :host, :call_trace
     alias debug? debug
 
     def self.spawn(args = {})
@@ -35,7 +35,8 @@ module NREPL
       debug: false,
       binding: nil,
       pwd: Dir.pwd,
-      tracing: true
+      tracing: true,
+      loader: nil
     )
       @port  = port
       @pwd = pwd
@@ -56,6 +57,7 @@ module NREPL
         }
       }
       @tracing = tracing
+      @loader = loader
       Thread.current[:nrepl_server] = self
       NREPL.class_variable_set(:@@connections, @connections)
       NREPL.class_variable_set(:@@bindings, @bindings)
@@ -76,7 +78,7 @@ module NREPL
       @old_out, @old_err = $stdout, $stderr
       $stdout = FakeStdout.new(@connections, STDOUT, "out")
       $stderr = FakeStdout.new(@connections, STDERR, "err")
-      auto_create_bindings! if @tracing
+      auto_create_bindings!
 
       Signal.trap("INT") { stop }
       Signal.trap("TERM") { stop }
@@ -87,7 +89,8 @@ module NREPL
         Thread.start(@socket.accept) do |client|
           connection = Connection.new(
             client,
-            debug: debug?,
+            server: self,
+            debug: @debug,
             binding: @binding,
             bindings_by_id: @bindings_by_id,
             bindings: @bindings
@@ -105,7 +108,7 @@ module NREPL
 
     def auto_create_bindings!
       dir_regex = Regexp.new("^#{Regexp.escape(@pwd)}")
-      @call_trace = TracePoint.new(:class, :call) do |tp|
+      trace_proc = proc do |tp|
         path = tp.path
         next if tp.path =~ /^(\<|.eval)/
         path = File.join(@pwd, path) if File.dirname(path) == '.'
@@ -115,6 +118,13 @@ module NREPL
         @bindings[path] ||= {}
         @bindings[path][tp.lineno-1] ||= {}
         @bindings[path][tp.lineno-1][id] = b
+        path
+      end
+      @class_trace = TracePoint.new(:class, &trace_proc)
+      @class_trace.enable
+
+      @call_trace = TracePoint.new(:call) do |tp|
+        path = trace_proc.call(tp)
         if path =~ dir_regex
           @connections.each do |connection|
             connection.send_msg(
@@ -126,7 +136,23 @@ module NREPL
           end
         end
       end
-      @call_trace.enable
+      @call_trace.enable if @tracing
+
+      if @loader
+        need_to_pause = true
+        original_trace = nil
+        @loader.on_load do
+          next unless need_to_pause
+          need_to_pause = false
+          original_trace = call_trace.enabled?
+          @call_trace.disable
+          Thread.new do
+            sleep 5
+            @call_trace.enable if original_trace
+            need_to_pause = true
+          end
+        end
+      end
 
       @ex_trace = TracePoint.new(:raise) do |tp|
         e = tp.raised_exception

data/lib/nrepl-lazuli.rb

@@ -1,4 +1,5 @@
 # frozen_string_literal: true
+require 'set'
 
 module NREPL
   VERSION = '0.1.0'

metadata

@@ -1,14 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: nrepl-lazuli
 version: !ruby/object:Gem::Version
-  version: 0.4.0
+  version: 0.6.0
 platform: ruby
 authors:
 - Maurício Szabo
-autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-11-13 00:00:00.000000000 Z
+date: 2025-05-06 00:00:00.000000000 Z
 dependencies: []
 description: A Ruby nREPL server, made to be used with Lazuli plug-in (but can be
   used with any nREPL client too)
@@ -18,6 +17,8 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/nrepl-lazuli.rb
+- lib/nrepl-lazuli/bencode.rb
+- lib/nrepl-lazuli/client.rb
 - lib/nrepl-lazuli/connection.rb
 - lib/nrepl-lazuli/fake_stdout.rb
 - lib/nrepl-lazuli/server.rb
@@ -26,7 +27,6 @@ licenses:
 - MIT
 metadata:
   source_code_uri: https://gitlab.com/clj-editors/nrepl-lazuli
-post_install_message: 
 rdoc_options: []
 require_paths:
 - lib
@@ -41,8 +41,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.11
-signing_key: 
+rubygems_version: 3.6.2
 specification_version: 4
 summary: A Ruby nREPL server
 test_files: []