Module: MilkTea::LSP::Server::ServerCodeLens

Included in:
MilkTea::LSP::Server
Defined in:
lib/milk_tea/lsp/server/code_lens.rb

Instance Method Summary collapse

Instance Method Details

#handle_code_lens(params) ⇒ Object 



7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
 
# File 'lib/milk_tea/lsp/server/code_lens.rb', line 7

def handle_code_lens(params)
  uri = params.dig("textDocument", "uri")
  return [] unless uri

  symbols = @workspace.get_symbols(uri)
  return [] if symbols.empty?

  function_symbols = symbols.select { |s| s[:kind].to_s == "function" && s[:line] }
  function_symbols.map do |sym|
    line = sym[:line] - 1
    {
      range: {
        start: { line: line, character: 0 },
        end: { line: line, character: 0 },
      },
      data: {
        uri: uri,
        name: sym[:name],
        line: sym[:line],
        column: sym[:column],
      },
    }
  end
rescue StandardError => e
  warn "Error in codeLens handler: #{e.message}"
  []
end

#handle_code_lens_resolve(params) ⇒ Object 



35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
 
# File 'lib/milk_tea/lsp/server/code_lens.rb', line 35

def handle_code_lens_resolve(params)
  data = params["data"] || params[:data]
  return params unless data.is_a?(Hash) || data.respond_to?(:[])

  name = data["name"] || data[:name]
  return params if name.to_s.empty?

  uri = data["uri"] || data[:uri]
  count = if uri
    facts = @workspace.get_facts(uri)
    if facts && module_level_name?(facts, name)
      level_kind = module_level_name?(facts, name)
      if level_kind == :method
        refs = static_method_code_lens_count(uri, name, data, facts)
        refs || module_level_reference_locations(uri, name, facts, include_declaration: true).length
      else
        module_level_reference_locations(uri, name, facts, include_declaration: true).length
      end
    else
      ast = @workspace.get_ast(uri)
      if ast && module_level_ast_name?(ast, name)
        refs = module_level_reference_locations(uri, name, nil, include_declaration: true)
        refs.length
      else
        refs = @workspace.find_all_references(name)
        refs ? refs.length : 0
      end
    end
  else
    refs = @workspace.find_all_references(name)
    refs ? refs.length : 0
  end

  params["command"] = {
    "title" => "#{count} reference#{count == 1 ? "" : "s"}",
    "command" => "",
  }

  params
rescue StandardError => e
  warn "Error in codeLens/resolve handler: #{e.message}"
  params
end

#static_method_code_lens_count(uri, name, data, facts) ⇒ Object 



79
80
81
82
83
84
85
86
87
88
89
90
91
 
# File 'lib/milk_tea/lsp/server/code_lens.rb', line 79

def static_method_code_lens_count(uri, name, data, facts)
  line = data["line"] || data[:line]
  column = data["column"] || data[:column]
  return nil unless line && column

  token = @workspace.find_token_at(uri, line - 1, column - 1)
  return nil unless token&.type == :identifier

  target = resolve_static_type_reference_target(uri, token, facts)
  return nil unless target

  static_type_method_references(target, include_declaration: true).length
end