debug 1.3.3 → 1.5.0

data/lib/debug/server_dap.rb

@@ -1,10 +1,68 @@
 # frozen_string_literal: true
 
 require 'json'
+require 'irb/completion'
+require 'tmpdir'
+require 'fileutils'
 
 module DEBUGGER__
   module UI_DAP
-    SHOW_PROTOCOL = ENV['RUBY_DEBUG_DAP_SHOW_PROTOCOL'] == '1'
+    SHOW_PROTOCOL = ENV['DEBUG_DAP_SHOW_PROTOCOL'] == '1' || ENV['RUBY_DEBUG_DAP_SHOW_PROTOCOL'] == '1'
+
+    def self.setup debug_port
+      if File.directory? '.vscode'
+        dir = Dir.pwd
+      else
+        dir = Dir.mktmpdir("ruby-debug-vscode-")
+        tempdir = true
+      end
+
+      at_exit do
+        CONFIG[:skip_path] = [//] # skip all
+        FileUtils.rm_rf dir if tempdir
+      end
+
+      key = rand.to_s
+
+      Dir.chdir(dir) do
+        Dir.mkdir('.vscode') if tempdir
+
+        # vscode-rdbg 0.0.9 or later is needed
+        open('.vscode/rdbg_autoattach.json', 'w') do |f|
+          f.puts JSON.pretty_generate({
+            type: "rdbg",
+            name: "Attach with rdbg",
+            request: "attach",
+            rdbgPath: File.expand_path('../../exe/rdbg', __dir__),
+            debugPort: debug_port,
+            localfs: true,
+            autoAttach: key,
+          })
+        end
+      end
+
+      cmds = ['code', "#{dir}/"]
+      cmdline = cmds.join(' ')
+      ssh_cmdline = "code --remote ssh-remote+[SSH hostname] #{dir}/"
+
+      STDERR.puts "Launching: #{cmdline}"
+      env = ENV.delete_if{|k, h| /RUBY/ =~ k}.to_h
+      env['RUBY_DEBUG_AUTOATTACH'] = key
+
+      unless system(env, *cmds)
+        DEBUGGER__.warn <<~MESSAGE
+        Can not invoke the command.
+        Use the command-line on your terminal (with modification if you need).
+
+          #{cmdline}
+
+        If your application is running on a SSH remote host, please try:
+
+          #{ssh_cmdline}
+
+        MESSAGE
+      end
+    end
 
     def show_protocol dir, msg
       if SHOW_PROTOCOL
@@ -12,9 +70,20 @@ module DEBUGGER__
       end
     end
 
+    @local_fs = false
+
+    def self.local_fs
+      @local_fs
+    end
+
+    def self.local_fs_set
+      @local_fs = true
+    end
+
     def dap_setup bytes
       CONFIG.set_config no_color: true
       @seq = 0
+      UI_DAP.local_fs_set if self.kind_of?(UI_UnixDomainServer)
 
       show_protocol :>, bytes
       req = JSON.load(bytes)
@@ -44,11 +113,12 @@ module DEBUGGER__
              ],
              supportsExceptionFilterOptions: true,
              supportsStepBack: true,
+             supportsEvaluateForHovers: true,
+             supportsCompletionsRequest: true,
 
              ## Will be supported
              # supportsExceptionOptions: true,
              # supportsHitConditionalBreakpoints:
-             # supportsEvaluateForHovers:
              # supportsSetVariable: true,
              # supportSuspendDebuggee:
              # supportsLogPoints:
@@ -58,7 +128,6 @@ module DEBUGGER__
 
              ## Possible?
              # supportsRestartFrame:
-             # supportsCompletionsRequest:
              # completionTriggerCharacters:
              # supportsModulesRequest:
              # additionalModuleColumns:
@@ -86,23 +155,23 @@ module DEBUGGER__
     def send **kw
       kw[:seq] = @seq += 1
       str = JSON.dump(kw)
+      @sock.write "Content-Length: #{str.bytesize}\r\n\r\n#{str}"
       show_protocol '<', str
-      @sock.write "Content-Length: #{str.size}\r\n\r\n#{str}"
     end
 
-    def send_response req, success: true, **kw
+    def send_response req, success: true, message: nil, **kw
       if kw.empty?
         send type: 'response',
              command: req['command'],
              request_seq: req['seq'],
              success: success,
-             message: success ? 'Success' : 'Failed'
+             message: message || (success ? 'Success' : 'Failed')
       else
         send type: 'response',
              command: req['command'],
              request_seq: req['seq'],
              success: success,
-             message: success ? 'Success' : 'Failed',
+             message: message || (success ? 'Success' : 'Failed'),
              body: kw
       end
     end
@@ -119,29 +188,27 @@ module DEBUGGER__
     end
 
     def recv_request
-      begin
-        r = IO.select([@sock])
-
-        @session.process_group.sync do
-          raise RetryBecauseCantRead unless IO.select([@sock], nil, nil, 0)
-
-          case header = @sock.gets
-          when /Content-Length: (\d+)/
-            b = @sock.read(2)
-            raise b.inspect unless b == "\r\n"
-
-            l = @sock.read(s = $1.to_i)
-            show_protocol :>, l
-            JSON.load(l)
-          when nil
-            nil
-          else
-            raise "unrecognized line: #{l} (#{l.size} bytes)"
-          end
+      r = IO.select([@sock])
+
+      @session.process_group.sync do
+        raise RetryBecauseCantRead unless IO.select([@sock], nil, nil, 0)
+
+        case header = @sock.gets
+        when /Content-Length: (\d+)/
+          b = @sock.read(2)
+          raise b.inspect unless b == "\r\n"
+
+          l = @sock.read(s = $1.to_i)
+          show_protocol :>, l
+          JSON.load(l)
+        when nil
+          nil
+        else
+          raise "unrecognized line: #{l} (#{l.size} bytes)"
         end
-      rescue RetryBecauseCantRead
-        retry
       end
+    rescue RetryBecauseCantRead
+      retry
     end
 
     def process
@@ -155,15 +222,18 @@ module DEBUGGER__
         when 'launch'
           send_response req
           @is_attach = false
+          UI_DAP.local_fs_set if req.dig('arguments', 'localfs')
         when 'attach'
           send_response req
-          Process.kill(:SIGURG, Process.pid)
+          Process.kill(UI_ServerBase::TRAP_SIGNAL, Process.pid)
           @is_attach = true
+          UI_DAP.local_fs_set if req.dig('arguments', 'localfs')
         when 'setBreakpoints'
           path = args.dig('source', 'path')
-          bp_args = args['breakpoints']
+          SESSION.clear_line_breakpoints path
+
           bps = []
-          bp_args.each{|bp|
+          args['breakpoints'].each{|bp|
             line = bp['line']
             if cond = bp['condition']
               bps << SESSION.add_line_breakpoint(path, line, cond: cond)
@@ -222,20 +292,41 @@ module DEBUGGER__
           @q_msg << 'c'
           send_response req, allThreadsContinued: true
         when 'next'
-          @q_msg << 'n'
-          send_response req
+          begin
+            @session.check_postmortem
+            @q_msg << 'n'
+            send_response req
+          rescue PostmortemError
+            send_response req,
+                          success: false, message: 'postmortem mode',
+                          result: "'Next' is not supported while postmortem mode"
+          end
         when 'stepIn'
-          @q_msg << 's'
-          send_response req
+          begin
+            @session.check_postmortem
+            @q_msg << 's'
+            send_response req
+          rescue PostmortemError
+            send_response req,
+                          success: false, message: 'postmortem mode',
+                          result: "'stepIn' is not supported while postmortem mode"
+          end
         when 'stepOut'
-          @q_msg << 'fin'
-          send_response req
+          begin
+            @session.check_postmortem
+            @q_msg << 'fin'
+            send_response req
+          rescue PostmortemError
+            send_response req,
+                          success: false, message: 'postmortem mode',
+                          result: "'stepOut' is not supported while postmortem mode"
+          end
         when 'terminate'
           send_response req
           exit
         when 'pause'
           send_response req
-          Process.kill(:SIGURG, Process.pid)
+          Process.kill(UI_ServerBase::TRAP_SIGNAL, Process.pid)
         when 'reverseContinue'
           send_response req,
                         success: false, message: 'cancelled',
@@ -255,13 +346,16 @@ module DEBUGGER__
              'scopes',
              'variables',
              'evaluate',
-             'source'
+             'source',
+             'completions'
           @q_msg << req
 
         else
           raise "Unknown request: #{req.inspect}"
         end
       end
+    ensure
+      send_event :terminated unless @sock.closed?
     end
 
     ## called by the SESSION thread
@@ -361,7 +455,6 @@ module DEBUGGER__
           case ref[0]
           when :globals
             vars = global_variables.map do |name|
-              File.write('/tmp/x', "#{name}\n")
               gv = 'Not implemented yet...'
               {
                 name: name,
@@ -402,11 +495,13 @@ module DEBUGGER__
         end
       when 'evaluate'
         frame_id = req.dig('arguments', 'frameId')
+        context = req.dig('arguments', 'context')
+
         if @frame_map[frame_id]
           tid, fid = @frame_map[frame_id]
           expr = req.dig('arguments', 'expression')
           if tc = find_waiting_tc(tid)
-            tc << [:dap, :evaluate, req, fid, expr]
+            tc << [:dap, :evaluate, req, fid, expr, context]
           else
             fail_response req
           end
@@ -416,11 +511,26 @@ module DEBUGGER__
       when 'source'
         ref = req.dig('arguments', 'sourceReference')
         if src = @src_map[ref]
-          @ui.respond req, content: src.join
+          @ui.respond req, content: src.join("\n")
         else
           fail_response req, message: 'not found...'
         end
         return :retry
+
+      when 'completions'
+        frame_id = req.dig('arguments', 'frameId')
+        tid, fid = @frame_map[frame_id]
+
+        if tc = find_waiting_tc(tid)
+          text = req.dig('arguments', 'text')
+          line = req.dig('arguments', 'line')
+          if col  = req.dig('arguments', 'column')
+            text = text.split(/\n/)[line.to_i - 1][0...(col.to_i - 1)]
+          end
+          tc << [:dap, :completions, req, fid, text]
+        else
+          fail_response req
+        end
       else
         raise "Unknown DAP request: #{req.inspect}"
       end
@@ -458,8 +568,15 @@ module DEBUGGER__
         register_vars result[:variables], tid
         @ui.respond req, result
       when :evaluate
-        tid = result.delete :tid
-        register_var result, tid
+        message = result.delete :message
+        if message
+          @ui.respond req, success: false, message: message
+        else
+          tid = result.delete :tid
+          register_var result, tid
+          @ui.respond req, result
+        end
+      when :completions
         @ui.respond req, result
       else
         raise "unsupported: #{args.inspect}"
@@ -491,9 +608,13 @@ module DEBUGGER__
       case type
       when :backtrace
         event! :dap_result, :backtrace, req, {
-          stackFrames: @target_frames.map.{|frame|
+          stackFrames: @target_frames.map{|frame|
             path = frame.realpath || frame.path
-            ref = frame.file_lines unless path && File.exist?(path)
+            source_name = path ? File.basename(path) : frame.location.to_s
+
+            if !UI_DAP.local_fs || !(path && File.exist?(path))
+              ref = frame.file_lines
+            end
 
             {
               # id: ??? # filled by SESSION
@@ -501,7 +622,7 @@ module DEBUGGER__
               line: frame.location.lineno,
               column: 1,
               source: {
-                name: File.basename(frame.path),
+                name: source_name,
                 path: path,
                 sourceReference: ref,
               },
@@ -510,7 +631,7 @@ module DEBUGGER__
         }
       when :scopes
         fid = args.shift
-        frame = @target_frames[fid]
+        frame = get_frame(fid)
 
         lnum =
           if frame.binding
@@ -538,26 +659,12 @@ module DEBUGGER__
         }]
       when :scope
         fid = args.shift
-        frame = @target_frames[fid]
-        if b = frame.binding
-          vars = b.local_variables.map{|name|
-            v = b.local_variable_get(name)
-            variable(name, v)
-          }
-          vars.unshift variable('%raised', frame.raised_exception) if frame.has_raised_exception
-          vars.unshift variable('%return', frame.return_value) if frame.has_return_value
-          vars.unshift variable('%self', b.receiver)
-        elsif lvars = frame.local_variables
-          vars = lvars.map{|var, val|
-            variable(var, val)
-          }
-        else
-          vars = [variable('%self', frame.self)]
-          vars.push variable('%raised', frame.raised_exception) if frame.has_raised_exception
-          vars.push variable('%return', frame.return_value) if frame.has_return_value
+        frame = get_frame(fid)
+        vars = collect_locals(frame).map do |var, val|
+          variable(var, val)
         end
-        event! :dap_result, :scope, req, variables: vars, tid: self.id
 
+        event! :dap_result, :scope, req, variables: vars, tid: self.id
       when :variable
         vid = args.shift
         obj = @var_map[vid]
@@ -575,7 +682,7 @@ module DEBUGGER__
             case obj
             when Hash
               vars = obj.map{|k, v|
-                variable(DEBUGGER__.short_inspect(k), v)
+                variable(DEBUGGER__.safe_inspect(k), v,)
               }
             when Struct
               vars = obj.members.map{|m|
@@ -607,32 +714,121 @@ module DEBUGGER__
         event! :dap_result, :variable, req, variables: (vars || []), tid: self.id
 
       when :evaluate
-        fid, expr = args
-        frame = @target_frames[fid]
+        fid, expr, context = args
+        frame = get_frame(fid)
+        message = nil
 
-        if frame && (b = frame.binding)
-          begin
-            result = b.eval(expr.to_s, '(DEBUG CONSOLE)')
-          rescue Exception => e
-            result = e
+        if frame && (b = frame.eval_binding)
+          special_local_variables frame do |name, var|
+            b.local_variable_set(name, var) if /\%/ !~ name
+          end
+
+          case context
+          when 'repl', 'watch'
+            begin
+              result = b.eval(expr.to_s, '(DEBUG CONSOLE)')
+            rescue Exception => e
+              result = e
+            end
+
+          when 'hover'
+            case expr
+            when /\A\@\S/
+              begin
+                result = b.receiver.instance_variable_get(expr)
+              rescue NameError
+                message = "Error: Not defined instance variable: #{expr.inspect}"
+              end
+            when /\A\$\S/
+              global_variables.each{|gvar|
+                if gvar.to_s == expr
+                  result = eval(gvar.to_s)
+                  break false
+                end
+              } and (message = "Error: Not defined global variable: #{expr.inspect}")
+            when /(\A((::[A-Z]|[A-Z])\w*)+)/
+              unless result = search_const(b, $1)
+                message = "Error: Not defined constants: #{expr.inspect}"
+              end
+            else
+              begin
+                result = b.local_variable_get(expr)
+              rescue NameError
+                # try to check method
+                if b.receiver.respond_to? expr, include_all: true
+                  result = b.receiver.method(expr)
+                else
+                  message = "Error: Can not evaluate: #{expr.inspect}"
+                end
+              end
+            end
+          else
+            message = "Error: unknown context: #{context}"
           end
         else
-          result = 'can not evaluate on this frame...'
+          result = 'Error: Can not evaluate on this frame'
+        end
+
+        event! :dap_result, :evaluate, req, message: message, tid: self.id, **evaluate_result(result)
+
+      when :completions
+        fid, text = args
+        frame = get_frame(fid)
+
+        if (b = frame&.binding) && word = text&.split(/[\s\{]/)&.last
+          words = IRB::InputCompletor::retrieve_completion_data(word, bind: b).compact
         end
-        event! :dap_result, :evaluate, req, tid: self.id, **evaluate_result(result)
+
+        event! :dap_result, :completions, req, targets: (words || []).map{|phrase|
+          if /\b([_a-zA-Z]\w*[!\?]?)\z/ =~ phrase
+            w = $1
+          else
+            w = phrase
+          end
+
+          begin
+            v = b.local_variable_get(w)
+            phrase += " (variable:#{DEBUGGER__.safe_inspect(v)})"
+          rescue NameError
+          end
+
+          {
+            label: phrase,
+            text: w,
+          }
+        }
+
       else
         raise "Unknown req: #{args.inspect}"
       end
     end
 
+    def search_const b, expr
+      cs = expr.delete_prefix('::').split('::')
+      [Object, *b.eval('Module.nesting')].reverse_each{|mod|
+        if cs.all?{|c|
+             if mod.const_defined?(c)
+               mod = mod.const_get(c)
+             else
+               false
+             end
+           }
+          # if-body
+          return mod
+        end
+      }
+      false
+    end
+
     def evaluate_result r
       v = variable nil, r
-      v.delete(:name)
-      v[:result] = DEBUGGER__.short_inspect(r)
+      v.delete :name
+      v.delete :value
+      v[:result] = DEBUGGER__.safe_inspect(r)
       v
     end
 
-    def variable_ name, obj, indexedVariables: 0, namedVariables: 0, use_short: true
+    def variable_ name, obj, indexedVariables: 0, namedVariables: 0
       if indexedVariables > 0 || namedVariables > 0
         vid = @var_map.size + 1
         @var_map[vid] = obj
@@ -643,7 +839,7 @@ module DEBUGGER__
       ivnum = obj.instance_variables.size
 
       { name: name,
-        value: DEBUGGER__.short_inspect(obj, use_short),
+        value: DEBUGGER__.safe_inspect(obj),
         type: obj.class.name || obj.class.to_s,
         variablesReference: vid,
         indexedVariables: indexedVariables,
@@ -658,7 +854,7 @@ module DEBUGGER__
       when Hash
         variable_ name, obj, namedVariables: obj.size
       when String
-        variable_ name, obj, use_short: false, namedVariables: 3 # #to_str, #length, #encoding
+        variable_ name, obj, namedVariables: 3 # #to_str, #length, #encoding
       when Struct
         variable_ name, obj, namedVariables: obj.size
       when Class, Module