1 use languageserver_types::{
2 self, CreateFile, DocumentChangeOperation, DocumentChanges, InsertTextFormat, Location,
3 Position, Range, RenameFile, ResourceOp, SymbolKind, TextDocumentEdit, TextDocumentIdentifier,
4 TextDocumentItem, TextDocumentPositionParams, Url, VersionedTextDocumentIdentifier,
8 CompletionItem, CompletionItemKind, FileId, FilePosition, FileRange, FileSystemEdit,
9 InsertText, NavigationTarget, SourceChange, SourceFileEdit,
10 LineCol, LineIndex, translate_offset_with_edit
12 use ra_syntax::{SyntaxKind, TextRange, TextUnit};
13 use ra_text_edit::{AtomTextEdit, TextEdit};
15 use crate::{req, server_world::ServerWorld, Result};
19 fn conv(self) -> Self::Output;
25 fn conv_with(self, ctx: &Self::Ctx) -> Self::Output;
28 pub trait TryConvWith {
31 fn try_conv_with(self, ctx: &Self::Ctx) -> Result<Self::Output>;
34 impl Conv for SyntaxKind {
35 type Output = SymbolKind;
37 fn conv(self) -> <Self as Conv>::Output {
39 SyntaxKind::FN_DEF => SymbolKind::Function,
40 SyntaxKind::STRUCT_DEF => SymbolKind::Struct,
41 SyntaxKind::ENUM_DEF => SymbolKind::Enum,
42 SyntaxKind::TRAIT_DEF => SymbolKind::Interface,
43 SyntaxKind::MODULE => SymbolKind::Module,
44 SyntaxKind::TYPE_DEF => SymbolKind::TypeParameter,
45 SyntaxKind::STATIC_DEF => SymbolKind::Constant,
46 SyntaxKind::CONST_DEF => SymbolKind::Constant,
47 SyntaxKind::IMPL_BLOCK => SymbolKind::Object,
48 _ => SymbolKind::Variable,
53 impl Conv for CompletionItemKind {
54 type Output = ::languageserver_types::CompletionItemKind;
56 fn conv(self) -> <Self as Conv>::Output {
57 use languageserver_types::CompletionItemKind::*;
59 CompletionItemKind::Keyword => Keyword,
60 CompletionItemKind::Snippet => Snippet,
61 CompletionItemKind::Module => Module,
62 CompletionItemKind::Function => Function,
63 CompletionItemKind::Struct => Struct,
64 CompletionItemKind::Enum => Enum,
65 CompletionItemKind::EnumVariant => EnumMember,
66 CompletionItemKind::Binding => Variable,
67 CompletionItemKind::Field => Field,
72 impl Conv for CompletionItem {
73 type Output = ::languageserver_types::CompletionItem;
75 fn conv(self) -> <Self as Conv>::Output {
76 let mut res = ::languageserver_types::CompletionItem {
77 label: self.label().to_string(),
78 filter_text: Some(self.lookup().to_string()),
79 kind: self.kind().map(|it| it.conv()),
82 match self.insert_text() {
83 InsertText::PlainText { text } => {
84 res.insert_text = Some(text);
85 res.insert_text_format = Some(InsertTextFormat::PlainText);
87 InsertText::Snippet { text } => {
88 res.insert_text = Some(text);
89 res.insert_text_format = Some(InsertTextFormat::Snippet);
96 impl ConvWith for Position {
98 type Output = TextUnit;
100 fn conv_with(self, line_index: &LineIndex) -> TextUnit {
101 let line_col = LineCol {
102 line: self.line as u32,
103 col_utf16: self.character as u32,
105 line_index.offset(line_col)
109 impl ConvWith for TextUnit {
110 type Ctx = LineIndex;
111 type Output = Position;
113 fn conv_with(self, line_index: &LineIndex) -> Position {
114 let line_col = line_index.line_col(self);
115 Position::new(u64::from(line_col.line), u64::from(line_col.col_utf16))
119 impl ConvWith for TextRange {
120 type Ctx = LineIndex;
123 fn conv_with(self, line_index: &LineIndex) -> Range {
125 self.start().conv_with(line_index),
126 self.end().conv_with(line_index),
131 impl ConvWith for Range {
132 type Ctx = LineIndex;
133 type Output = TextRange;
135 fn conv_with(self, line_index: &LineIndex) -> TextRange {
137 self.start.conv_with(line_index),
138 self.end.conv_with(line_index),
143 impl ConvWith for TextEdit {
144 type Ctx = LineIndex;
145 type Output = Vec<languageserver_types::TextEdit>;
147 fn conv_with(self, line_index: &LineIndex) -> Vec<languageserver_types::TextEdit> {
150 .map_conv_with(line_index)
155 impl<'a> ConvWith for &'a AtomTextEdit {
156 type Ctx = LineIndex;
157 type Output = languageserver_types::TextEdit;
159 fn conv_with(self, line_index: &LineIndex) -> languageserver_types::TextEdit {
160 languageserver_types::TextEdit {
161 range: self.delete.conv_with(line_index),
162 new_text: self.insert.clone(),
167 impl<T: ConvWith> ConvWith for Option<T> {
168 type Ctx = <T as ConvWith>::Ctx;
169 type Output = Option<<T as ConvWith>::Output>;
170 fn conv_with(self, ctx: &Self::Ctx) -> Self::Output {
171 self.map(|x| ConvWith::conv_with(x, ctx))
175 impl<'a> TryConvWith for &'a Url {
176 type Ctx = ServerWorld;
177 type Output = FileId;
178 fn try_conv_with(self, world: &ServerWorld) -> Result<FileId> {
179 world.uri_to_file_id(self)
183 impl TryConvWith for FileId {
184 type Ctx = ServerWorld;
186 fn try_conv_with(self, world: &ServerWorld) -> Result<Url> {
187 world.file_id_to_uri(self)
191 impl<'a> TryConvWith for &'a TextDocumentItem {
192 type Ctx = ServerWorld;
193 type Output = FileId;
194 fn try_conv_with(self, world: &ServerWorld) -> Result<FileId> {
195 self.uri.try_conv_with(world)
199 impl<'a> TryConvWith for &'a VersionedTextDocumentIdentifier {
200 type Ctx = ServerWorld;
201 type Output = FileId;
202 fn try_conv_with(self, world: &ServerWorld) -> Result<FileId> {
203 self.uri.try_conv_with(world)
207 impl<'a> TryConvWith for &'a TextDocumentIdentifier {
208 type Ctx = ServerWorld;
209 type Output = FileId;
210 fn try_conv_with(self, world: &ServerWorld) -> Result<FileId> {
211 world.uri_to_file_id(&self.uri)
215 impl<'a> TryConvWith for &'a TextDocumentPositionParams {
216 type Ctx = ServerWorld;
217 type Output = FilePosition;
218 fn try_conv_with(self, world: &ServerWorld) -> Result<FilePosition> {
219 let file_id = self.text_document.try_conv_with(world)?;
220 let line_index = world.analysis().file_line_index(file_id);
221 let offset = self.position.conv_with(&line_index);
222 Ok(FilePosition { file_id, offset })
226 impl<'a> TryConvWith for (&'a TextDocumentIdentifier, Range) {
227 type Ctx = ServerWorld;
228 type Output = FileRange;
229 fn try_conv_with(self, world: &ServerWorld) -> Result<FileRange> {
230 let file_id = self.0.try_conv_with(world)?;
231 let line_index = world.analysis().file_line_index(file_id);
232 let range = self.1.conv_with(&line_index);
233 Ok(FileRange { file_id, range })
237 impl<T: TryConvWith> TryConvWith for Vec<T> {
238 type Ctx = <T as TryConvWith>::Ctx;
239 type Output = Vec<<T as TryConvWith>::Output>;
240 fn try_conv_with(self, ctx: &Self::Ctx) -> Result<Self::Output> {
241 let mut res = Vec::with_capacity(self.len());
243 res.push(item.try_conv_with(ctx)?);
249 impl TryConvWith for SourceChange {
250 type Ctx = ServerWorld;
251 type Output = req::SourceChange;
252 fn try_conv_with(self, world: &ServerWorld) -> Result<req::SourceChange> {
253 let cursor_position = match self.cursor_position {
256 let line_index = world.analysis().file_line_index(pos.file_id);
260 .find(|it| it.file_id == pos.file_id)
262 let line_col = match edit {
263 Some(edit) => translate_offset_with_edit(&*line_index, pos.offset, edit),
264 None => line_index.line_col(pos.offset),
267 Position::new(u64::from(line_col.line), u64::from(line_col.col_utf16));
268 Some(TextDocumentPositionParams {
269 text_document: TextDocumentIdentifier::new(pos.file_id.try_conv_with(world)?),
274 let mut document_changes: Vec<DocumentChangeOperation> = Vec::new();
275 for resource_op in self.file_system_edits.try_conv_with(world)? {
276 document_changes.push(DocumentChangeOperation::Op(resource_op));
278 for text_document_edit in self.source_file_edits.try_conv_with(world)? {
279 document_changes.push(DocumentChangeOperation::Edit(text_document_edit));
281 let workspace_edit = WorkspaceEdit {
283 document_changes: Some(DocumentChanges::Operations(document_changes)),
285 Ok(req::SourceChange {
293 impl TryConvWith for SourceFileEdit {
294 type Ctx = ServerWorld;
295 type Output = TextDocumentEdit;
296 fn try_conv_with(self, world: &ServerWorld) -> Result<TextDocumentEdit> {
297 let text_document = VersionedTextDocumentIdentifier {
298 uri: self.file_id.try_conv_with(world)?,
301 let line_index = world.analysis().file_line_index(self.file_id);
306 .map_conv_with(&line_index)
308 Ok(TextDocumentEdit {
315 impl TryConvWith for FileSystemEdit {
316 type Ctx = ServerWorld;
317 type Output = ResourceOp;
318 fn try_conv_with(self, world: &ServerWorld) -> Result<ResourceOp> {
319 let res = match self {
320 FileSystemEdit::CreateFile { source_root, path } => {
321 let uri = world.path_to_uri(source_root, &path)?.to_string();
322 ResourceOp::Create(CreateFile { uri, options: None })
324 FileSystemEdit::MoveFile {
329 let old_uri = world.file_id_to_uri(src)?.to_string();
330 let new_uri = world.path_to_uri(dst_source_root, &dst_path)?.to_string();
331 ResourceOp::Rename(RenameFile {
342 impl TryConvWith for &NavigationTarget {
343 type Ctx = ServerWorld;
344 type Output = Location;
345 fn try_conv_with(self, world: &ServerWorld) -> Result<Location> {
346 let line_index = world.analysis().file_line_index(self.file_id());
347 to_location(self.file_id(), self.range(), &world, &line_index)
355 line_index: &LineIndex,
356 ) -> Result<Location> {
357 let url = file_id.try_conv_with(world)?;
358 let loc = Location::new(url, range.conv_with(line_index));
362 pub trait MapConvWith<'a>: Sized + 'a {
366 fn map_conv_with(self, ctx: &'a Self::Ctx) -> ConvWithIter<'a, Self, Self::Ctx> {
367 ConvWithIter { iter: self, ctx }
371 impl<'a, I> MapConvWith<'a> for I
376 type Ctx = <I::Item as ConvWith>::Ctx;
377 type Output = <I::Item as ConvWith>::Output;
380 pub struct ConvWithIter<'a, I, Ctx: 'a> {
385 impl<'a, I, Ctx> Iterator for ConvWithIter<'a, I, Ctx>
388 I::Item: ConvWith<Ctx = Ctx>,
390 type Item = <I::Item as ConvWith>::Output;
392 fn next(&mut self) -> Option<Self::Item> {
393 self.iter.next().map(|item| item.conv_with(self.ctx))