Module: MilkTea::LSP::Server::ServerLifecycle

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

Constant Summary collapse

SEMANTIC_TOKENS_REFRESH_THROTTLE_MS = 
500

Instance Method Summary collapse

Instance Method Details

#handle_cancel_request(params) ⇒ Object 



161
162
163
164
165
166
167
168
169
 
# File 'lib/milk_tea/lsp/server/lifecycle.rb', line 161

def handle_cancel_request(params)
  request_id = params.is_a?(Hash) ? (params['id'] || params[:id]) : nil
  return nil if request_id.nil?

  @cancelled_requests_mutex.synchronize do
    @cancelled_request_ids << request_id
  end
  nil
end

#handle_did_change_configuration(params) ⇒ Object 



156
157
158
159
 
# File 'lib/milk_tea/lsp/server/lifecycle.rb', line 156

def handle_did_change_configuration(params)
  apply_configuration_settings(params['settings'])
  nil
end

#handle_did_change_workspace_folders(params) ⇒ Object 



171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
 
# File 'lib/milk_tea/lsp/server/lifecycle.rb', line 171

def handle_did_change_workspace_folders(params)
  event = params['event'] || {}
  added = event['added'] || []
  removed = event['removed'] || []

  removed_uris = removed.filter_map { |folder| folder.is_a?(Hash) ? folder['uri'] : nil }
  added_uris = added.filter_map { |folder| folder.is_a?(Hash) ? folder['uri'] : nil }

  if removed_uris.include?(@root_uri)
    @root_uri = added_uris.first
  elsif @root_uri.nil?
    @root_uri = added_uris.first
  end

  @workspace.workspace_root_path = uri_to_path(@root_uri)
  return nil unless @root_uri

  uris = @workspace.open_document_uris.dup
  Thread.new do
    progress = create_progress(title: 'Milk Tea LSP: Reindexing workspace', message: 'Scanning source files...')
    @workspace.index_workspace(@root_uri) { |pct, msg| progress.report(percentage: pct, message: msg) }
    @workspace.refresh_module_index_for_workspace
    doc_count = @workspace.all_documents.length
    progress.done(message: "#{doc_count} document#{doc_count == 1 ? '' : 's'} indexed")

    uris.each do |uri|
      schedule_diagnostics(uri, force: true, lint_tier: :full) unless @workspace.background_document?(uri)
    end
  end
  nil
end

#handle_exit(_params) ⇒ Object 



225
226
227
228
229
 
# File 'lib/milk_tea/lsp/server/lifecycle.rb', line 225

def handle_exit(_params)
  stop_diagnostics_workers
  @workspace.shutdown
  exit(0)
end

#handle_initialize(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
34
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
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
 
# File 'lib/milk_tea/lsp/server/lifecycle.rb', line 7

def handle_initialize(params)
  @pull_diagnostics_active = true
  if @_initialize_called
    @_indexing_thread&.join rescue nil
    @_indexing_thread = nil
    @workspace.reset
    @format_mode = :tidy
    @dependency_resolution_mode = :auto
    @platform_override = nil
    @diagnostic_report_cache.clear
    @workspace_diagnostic_cache.clear
    drain_diagnostics_queue
    @diagnostics_pending.clear
    @diagnostics_enqueued.clear
    @diagnostics_last_scheduled_hash.clear
    @diagnostics_generation.clear
    @semantic_tokens_cache.clear
    @semantic_tokens_delta_cache.clear
    @fixall_cache.clear
    @completion_session_cache.clear
    @completion_session_order.clear
    @completion_session_order_set.clear
    @completion_docs_cache.clear
    @completion_resolve_cache.clear
    @document_symbol_cache.clear
    @definition_file_token_cache.clear
    @definition_file_ast_cache.clear
    MilkTea::Types::Registry.reset!
  end
  @_initialize_called = true
  @root_uri = params['rootUri']
  @client_capabilities = params['capabilities'] || {}
  @workspace.workspace_root_path = uri_to_path(@root_uri)
  negotiate_position_encoding(params)
  apply_configuration_settings(params['initializationOptions'])
  apply_configuration_settings(params['settings'])
  {
    capabilities: {
      positionEncoding: @position_encoding,
      textDocumentSync: {
        openClose: true,
        change: 2,
        save: { includeText: true },
        willSaveWaitUntil: true
      },
      hoverProvider: true,
      definitionProvider: true,
      declarationProvider: true,
      typeDefinitionProvider: true,
      implementationProvider: true,
      referencesProvider: true,
      documentLinkProvider: {
        resolveProvider: true
      },
      documentHighlightProvider: true,
      documentSymbolProvider: true,
      documentFormattingProvider: true,
      documentRangeFormattingProvider: true,
      foldingRangeProvider: true,
      callHierarchyProvider: true,
      typeHierarchyProvider: true,
      selectionRangeProvider: true,
      linkedEditingRangeProvider: true,
      executeCommandProvider: {
        commands: ['mtc.restartServer']
      },
      documentOnTypeFormattingProvider: {
        firstTriggerCharacter: "\n",
        moreTriggerCharacter: []
      },
      signatureHelpProvider: {
        triggerCharacters: ['(', ','],
        retriggerCharacters: [',']
      },
      codeActionProvider: {
        codeActionKinds: ['quickFix', 'source.fixAll']
      },
      codeLensProvider: {
        resolveProvider: true
      },
      inlayHintProvider: true,
      semanticTokensProvider: {
        legend: {
          tokenTypes: SEMANTIC_TOKEN_TYPES,
          tokenModifiers: SEMANTIC_TOKEN_MODIFIERS
        },
        full: { delta: true },
        range: true
      },
      completionProvider: {
        triggerCharacters: ['.', '(', ' '],
        resolveProvider: true
      },
      renameProvider: { prepareProvider: true },
      workspaceSymbolProvider: true,
      diagnosticProvider: {
        interFileDependencies: true,
        workspaceDiagnostics: true
      },
      workspace: {
        workspaceFolders: {
          supported: true,
          changeNotifications: true,
        },
        fileOperations: {
          willRename: {
            filters: [{ pattern: { glob: '**/*.mt' }, scheme: 'file' }]
          }
        },
        didChangeWatchedFiles: {
          dynamicRegistration: true
        }
      }
    }
  }
end

#handle_initialized(_params) ⇒ Object 



140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
 
# File 'lib/milk_tea/lsp/server/lifecycle.rb', line 140

def handle_initialized(_params)
  pull_client_configuration if @client_capabilities&.dig('workspace', 'configuration')
  return nil unless @root_uri

  @_indexing_thread = Thread.new do
    progress = create_progress(title: 'Milk Tea LSP: Indexing workspace', message: 'Scanning source files...')
    @workspace.index_workspace(@root_uri) { |pct, msg| progress.report(percentage: pct, message: msg) }
    @workspace.refresh_module_index_for_workspace
    document_count = @workspace.all_documents.length
    progress.done(message: "#{document_count} document#{document_count == 1 ? '' : 's'} indexed")
    log_message(:info, "Milk Tea LSP ready — #{document_count} document#{document_count == 1 ? '' : 's'} indexed")
    refresh_client_semantic_tokens if document_count > 0
  end
  nil
end

#handle_shutdown(_params) ⇒ Object 



219
220
221
222
223
 
# File 'lib/milk_tea/lsp/server/lifecycle.rb', line 219

def handle_shutdown(_params)
  stop_diagnostics_workers
  @workspace.shutdown
  nil
end

#handle_will_rename_files(params) ⇒ Object 



203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
 
# File 'lib/milk_tea/lsp/server/lifecycle.rb', line 203

def handle_will_rename_files(params)
  files = params['files'] || []
  files.each do |file|
    old_uri = file['oldUri']
    new_uri = file['newUri']
    next unless old_uri && new_uri

    @workspace.rename_indexed_file(old_uri, new_uri)
    @workspace.apply_module_index_events(
      [{ 'uri' => old_uri, 'type' => 3 }, { 'uri' => new_uri, 'type' => 1 }],
      skip_open: false,
    )
  end
  nil
end

#negotiate_position_encoding(params) ⇒ Object 



124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
 
# File 'lib/milk_tea/lsp/server/lifecycle.rb', line 124

def negotiate_position_encoding(params)
  client_encodings = params.dig('capabilities', 'general', 'positionEncodings')
  if client_encodings.is_a?(Array)
    @position_encoding = if client_encodings.include?('utf-8')
                          'utf-8'
                        elsif client_encodings.include?('utf-32')
                          'utf-32'
                        else
                          'utf-16'
                        end
  else
    @position_encoding = 'utf-16'
  end
  Diagnostics.position_encoding = @position_encoding
end

#refresh_client_semantic_tokens ⇒ Object 



233
234
235
236
237
238
239
240
 
# File 'lib/milk_tea/lsp/server/lifecycle.rb', line 233

def refresh_client_semantic_tokens
  now = Process.clock_gettime(Process::CLOCK_MONOTONIC)
  @semantic_tokens_last_refresh ||= 0.0
  return if (now - @semantic_tokens_last_refresh) * 1000 < SEMANTIC_TOKENS_REFRESH_THROTTLE_MS

  @semantic_tokens_last_refresh = now
  @protocol.write_notification('workspace/semanticTokens/refresh', nil)
end