2018-08-15 09:24:20 -05:00
|
|
|
use languageserver_types::{
|
2018-12-11 12:07:17 -06:00
|
|
|
self, Location, Position, Range, SymbolKind, TextDocumentEdit, TextDocumentIdentifier,
|
|
|
|
TextDocumentItem, TextDocumentPositionParams, Url, VersionedTextDocumentIdentifier,
|
2018-08-17 11:54:08 -05:00
|
|
|
};
|
2018-12-21 02:15:23 -06:00
|
|
|
use ra_analysis::{FileId, FileSystemEdit, SourceChange, SourceFileEdit, FilePosition};
|
2018-12-10 15:09:12 -06:00
|
|
|
use ra_editor::{LineCol, LineIndex};
|
2018-12-11 12:07:17 -06:00
|
|
|
use ra_text_edit::{AtomTextEdit, TextEdit};
|
2018-10-15 16:44:23 -05:00
|
|
|
use ra_syntax::{SyntaxKind, TextRange, TextUnit};
|
|
|
|
|
|
|
|
use crate::{req, server_world::ServerWorld, Result};
|
2018-08-13 08:35:17 -05:00
|
|
|
|
2018-08-12 13:02:56 -05:00
|
|
|
pub trait Conv {
|
|
|
|
type Output;
|
2018-08-12 18:38:34 -05:00
|
|
|
fn conv(self) -> Self::Output;
|
2018-08-12 13:02:56 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
pub trait ConvWith {
|
|
|
|
type Ctx;
|
|
|
|
type Output;
|
2018-08-12 18:38:34 -05:00
|
|
|
fn conv_with(self, ctx: &Self::Ctx) -> Self::Output;
|
2018-08-12 13:02:56 -05:00
|
|
|
}
|
|
|
|
|
2018-08-13 08:35:17 -05:00
|
|
|
pub trait TryConvWith {
|
|
|
|
type Ctx;
|
|
|
|
type Output;
|
|
|
|
fn try_conv_with(self, ctx: &Self::Ctx) -> Result<Self::Output>;
|
|
|
|
}
|
|
|
|
|
2018-08-12 13:02:56 -05:00
|
|
|
impl Conv for SyntaxKind {
|
|
|
|
type Output = SymbolKind;
|
|
|
|
|
2018-08-12 18:38:34 -05:00
|
|
|
fn conv(self) -> <Self as Conv>::Output {
|
|
|
|
match self {
|
2018-08-13 10:27:26 -05:00
|
|
|
SyntaxKind::FN_DEF => SymbolKind::Function,
|
|
|
|
SyntaxKind::STRUCT_DEF => SymbolKind::Struct,
|
|
|
|
SyntaxKind::ENUM_DEF => SymbolKind::Enum,
|
|
|
|
SyntaxKind::TRAIT_DEF => SymbolKind::Interface,
|
2018-08-12 13:02:56 -05:00
|
|
|
SyntaxKind::MODULE => SymbolKind::Module,
|
2018-08-13 10:27:26 -05:00
|
|
|
SyntaxKind::TYPE_DEF => SymbolKind::TypeParameter,
|
|
|
|
SyntaxKind::STATIC_DEF => SymbolKind::Constant,
|
|
|
|
SyntaxKind::CONST_DEF => SymbolKind::Constant,
|
2018-08-14 03:20:09 -05:00
|
|
|
SyntaxKind::IMPL_ITEM => SymbolKind::Object,
|
2018-08-12 13:02:56 -05:00
|
|
|
_ => SymbolKind::Variable,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl ConvWith for Position {
|
|
|
|
type Ctx = LineIndex;
|
|
|
|
type Output = TextUnit;
|
|
|
|
|
2018-08-12 18:38:34 -05:00
|
|
|
fn conv_with(self, line_index: &LineIndex) -> TextUnit {
|
2018-08-12 13:02:56 -05:00
|
|
|
let line_col = LineCol {
|
|
|
|
line: self.line as u32,
|
2018-11-16 05:02:45 -06:00
|
|
|
col_utf16: self.character as u32,
|
2018-08-12 13:02:56 -05:00
|
|
|
};
|
|
|
|
line_index.offset(line_col)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl ConvWith for TextUnit {
|
|
|
|
type Ctx = LineIndex;
|
|
|
|
type Output = Position;
|
|
|
|
|
2018-08-12 18:38:34 -05:00
|
|
|
fn conv_with(self, line_index: &LineIndex) -> Position {
|
|
|
|
let line_col = line_index.line_col(self);
|
2018-11-29 14:30:49 -06:00
|
|
|
Position::new(u64::from(line_col.line), u64::from(line_col.col_utf16))
|
2018-08-12 13:02:56 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl ConvWith for TextRange {
|
|
|
|
type Ctx = LineIndex;
|
|
|
|
type Output = Range;
|
|
|
|
|
2018-08-12 18:38:34 -05:00
|
|
|
fn conv_with(self, line_index: &LineIndex) -> Range {
|
2018-08-12 13:02:56 -05:00
|
|
|
Range::new(
|
|
|
|
self.start().conv_with(line_index),
|
|
|
|
self.end().conv_with(line_index),
|
|
|
|
)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl ConvWith for Range {
|
|
|
|
type Ctx = LineIndex;
|
|
|
|
type Output = TextRange;
|
|
|
|
|
2018-08-12 18:38:34 -05:00
|
|
|
fn conv_with(self, line_index: &LineIndex) -> TextRange {
|
2018-08-12 13:02:56 -05:00
|
|
|
TextRange::from_to(
|
|
|
|
self.start.conv_with(line_index),
|
|
|
|
self.end.conv_with(line_index),
|
|
|
|
)
|
|
|
|
}
|
|
|
|
}
|
2018-08-12 18:38:34 -05:00
|
|
|
|
2018-12-11 12:07:17 -06:00
|
|
|
impl ConvWith for TextEdit {
|
2018-08-12 18:38:34 -05:00
|
|
|
type Ctx = LineIndex;
|
2018-12-11 12:07:17 -06:00
|
|
|
type Output = Vec<languageserver_types::TextEdit>;
|
2018-08-12 18:38:34 -05:00
|
|
|
|
2018-12-11 12:07:17 -06:00
|
|
|
fn conv_with(self, line_index: &LineIndex) -> Vec<languageserver_types::TextEdit> {
|
2018-12-21 02:24:16 -06:00
|
|
|
self.as_atoms()
|
2018-08-12 18:38:34 -05:00
|
|
|
.into_iter()
|
|
|
|
.map_conv_with(line_index)
|
|
|
|
.collect()
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-12-21 02:24:16 -06:00
|
|
|
impl<'a> ConvWith for &'a AtomTextEdit {
|
2018-08-12 18:38:34 -05:00
|
|
|
type Ctx = LineIndex;
|
2018-12-11 12:07:17 -06:00
|
|
|
type Output = languageserver_types::TextEdit;
|
2018-08-12 18:38:34 -05:00
|
|
|
|
2018-12-11 12:07:17 -06:00
|
|
|
fn conv_with(self, line_index: &LineIndex) -> languageserver_types::TextEdit {
|
|
|
|
languageserver_types::TextEdit {
|
2018-08-12 18:38:34 -05:00
|
|
|
range: self.delete.conv_with(line_index),
|
2018-12-21 02:24:16 -06:00
|
|
|
new_text: self.insert.clone(),
|
2018-08-12 18:38:34 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-15 16:23:22 -05:00
|
|
|
impl<T: ConvWith> ConvWith for Option<T> {
|
|
|
|
type Ctx = <T as ConvWith>::Ctx;
|
|
|
|
type Output = Option<<T as ConvWith>::Output>;
|
|
|
|
fn conv_with(self, ctx: &Self::Ctx) -> Self::Output {
|
|
|
|
self.map(|x| ConvWith::conv_with(x, ctx))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-15 09:24:20 -05:00
|
|
|
impl<'a> TryConvWith for &'a Url {
|
2018-08-17 11:54:08 -05:00
|
|
|
type Ctx = ServerWorld;
|
2018-08-15 09:24:20 -05:00
|
|
|
type Output = FileId;
|
2018-08-17 11:54:08 -05:00
|
|
|
fn try_conv_with(self, world: &ServerWorld) -> Result<FileId> {
|
|
|
|
world.uri_to_file_id(self)
|
2018-08-15 09:24:20 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl TryConvWith for FileId {
|
2018-08-17 11:54:08 -05:00
|
|
|
type Ctx = ServerWorld;
|
2018-08-15 09:24:20 -05:00
|
|
|
type Output = Url;
|
2018-08-17 11:54:08 -05:00
|
|
|
fn try_conv_with(self, world: &ServerWorld) -> Result<Url> {
|
|
|
|
world.file_id_to_uri(self)
|
2018-08-15 09:24:20 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl<'a> TryConvWith for &'a TextDocumentItem {
|
2018-08-17 11:54:08 -05:00
|
|
|
type Ctx = ServerWorld;
|
2018-08-15 09:24:20 -05:00
|
|
|
type Output = FileId;
|
2018-08-17 11:54:08 -05:00
|
|
|
fn try_conv_with(self, world: &ServerWorld) -> Result<FileId> {
|
|
|
|
self.uri.try_conv_with(world)
|
2018-08-15 09:24:20 -05:00
|
|
|
}
|
|
|
|
}
|
2018-08-13 08:35:17 -05:00
|
|
|
|
2018-08-15 09:24:20 -05:00
|
|
|
impl<'a> TryConvWith for &'a VersionedTextDocumentIdentifier {
|
2018-08-17 11:54:08 -05:00
|
|
|
type Ctx = ServerWorld;
|
2018-08-15 09:24:20 -05:00
|
|
|
type Output = FileId;
|
2018-08-17 11:54:08 -05:00
|
|
|
fn try_conv_with(self, world: &ServerWorld) -> Result<FileId> {
|
|
|
|
self.uri.try_conv_with(world)
|
2018-08-15 09:24:20 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl<'a> TryConvWith for &'a TextDocumentIdentifier {
|
2018-08-17 11:54:08 -05:00
|
|
|
type Ctx = ServerWorld;
|
2018-08-15 09:24:20 -05:00
|
|
|
type Output = FileId;
|
2018-08-17 11:54:08 -05:00
|
|
|
fn try_conv_with(self, world: &ServerWorld) -> Result<FileId> {
|
|
|
|
world.uri_to_file_id(&self.uri)
|
2018-08-15 09:24:20 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-11-05 05:57:41 -06:00
|
|
|
impl<'a> TryConvWith for &'a TextDocumentPositionParams {
|
|
|
|
type Ctx = ServerWorld;
|
|
|
|
type Output = FilePosition;
|
|
|
|
fn try_conv_with(self, world: &ServerWorld) -> Result<FilePosition> {
|
|
|
|
let file_id = self.text_document.try_conv_with(world)?;
|
|
|
|
let line_index = world.analysis().file_line_index(file_id);
|
|
|
|
let offset = self.position.conv_with(&line_index);
|
|
|
|
Ok(FilePosition { file_id, offset })
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-29 10:03:14 -05:00
|
|
|
impl<T: TryConvWith> TryConvWith for Vec<T> {
|
|
|
|
type Ctx = <T as TryConvWith>::Ctx;
|
|
|
|
type Output = Vec<<T as TryConvWith>::Output>;
|
|
|
|
fn try_conv_with(self, ctx: &Self::Ctx) -> Result<Self::Output> {
|
|
|
|
let mut res = Vec::with_capacity(self.len());
|
|
|
|
for item in self {
|
|
|
|
res.push(item.try_conv_with(ctx)?);
|
|
|
|
}
|
|
|
|
Ok(res)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl TryConvWith for SourceChange {
|
|
|
|
type Ctx = ServerWorld;
|
|
|
|
type Output = req::SourceChange;
|
|
|
|
fn try_conv_with(self, world: &ServerWorld) -> Result<req::SourceChange> {
|
|
|
|
let cursor_position = match self.cursor_position {
|
|
|
|
None => None,
|
|
|
|
Some(pos) => {
|
|
|
|
let line_index = world.analysis().file_line_index(pos.file_id);
|
2018-10-15 16:44:23 -05:00
|
|
|
let edits = self
|
|
|
|
.source_file_edits
|
|
|
|
.iter()
|
|
|
|
.find(|it| it.file_id == pos.file_id)
|
2018-12-21 02:24:16 -06:00
|
|
|
.map(|it| it.edit.as_atoms())
|
2018-10-15 16:44:23 -05:00
|
|
|
.unwrap_or(&[]);
|
2018-10-09 08:00:20 -05:00
|
|
|
let line_col = translate_offset_with_edit(&*line_index, pos.offset, edits);
|
2018-11-29 14:30:49 -06:00
|
|
|
let position =
|
|
|
|
Position::new(u64::from(line_col.line), u64::from(line_col.col_utf16));
|
2018-08-29 10:03:14 -05:00
|
|
|
Some(TextDocumentPositionParams {
|
|
|
|
text_document: TextDocumentIdentifier::new(pos.file_id.try_conv_with(world)?),
|
2018-10-09 08:00:20 -05:00
|
|
|
position,
|
2018-08-29 10:03:14 -05:00
|
|
|
})
|
|
|
|
}
|
|
|
|
};
|
|
|
|
let source_file_edits = self.source_file_edits.try_conv_with(world)?;
|
|
|
|
let file_system_edits = self.file_system_edits.try_conv_with(world)?;
|
|
|
|
Ok(req::SourceChange {
|
|
|
|
label: self.label,
|
|
|
|
source_file_edits,
|
|
|
|
file_system_edits,
|
|
|
|
cursor_position,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-10-09 08:00:20 -05:00
|
|
|
// HACK: we should translate offset to line/column using linde_index *with edits applied*.
|
|
|
|
// A naive version of this function would be to apply `edits` to the original text,
|
|
|
|
// construct a new line index and use that, but it would be slow.
|
|
|
|
//
|
|
|
|
// Writing fast & correct version is issue #105, let's use a quick hack in the meantime
|
|
|
|
fn translate_offset_with_edit(
|
|
|
|
pre_edit_index: &LineIndex,
|
|
|
|
offset: TextUnit,
|
2018-12-11 12:07:17 -06:00
|
|
|
edits: &[AtomTextEdit],
|
2018-10-09 08:00:20 -05:00
|
|
|
) -> LineCol {
|
|
|
|
let fallback = pre_edit_index.line_col(offset);
|
|
|
|
let edit = match edits.first() {
|
|
|
|
None => return fallback,
|
2018-10-15 16:44:23 -05:00
|
|
|
Some(edit) => edit,
|
2018-10-09 08:00:20 -05:00
|
|
|
};
|
|
|
|
let end_offset = edit.delete.start() + TextUnit::of_str(&edit.insert);
|
|
|
|
if !(edit.delete.start() <= offset && offset <= end_offset) {
|
2018-10-15 16:44:23 -05:00
|
|
|
return fallback;
|
2018-10-09 08:00:20 -05:00
|
|
|
}
|
|
|
|
let rel_offset = offset - edit.delete.start();
|
|
|
|
let in_edit_line_col = LineIndex::new(&edit.insert).line_col(rel_offset);
|
|
|
|
let edit_line_col = pre_edit_index.line_col(edit.delete.start());
|
|
|
|
if in_edit_line_col.line == 0 {
|
|
|
|
LineCol {
|
|
|
|
line: edit_line_col.line,
|
2018-11-16 05:02:45 -06:00
|
|
|
col_utf16: edit_line_col.col_utf16 + in_edit_line_col.col_utf16,
|
2018-10-09 08:00:20 -05:00
|
|
|
}
|
|
|
|
} else {
|
|
|
|
LineCol {
|
|
|
|
line: edit_line_col.line + in_edit_line_col.line,
|
2018-11-16 05:02:45 -06:00
|
|
|
col_utf16: in_edit_line_col.col_utf16,
|
2018-10-09 08:00:20 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-12-21 02:15:23 -06:00
|
|
|
impl TryConvWith for SourceFileEdit {
|
2018-08-29 10:03:14 -05:00
|
|
|
type Ctx = ServerWorld;
|
|
|
|
type Output = TextDocumentEdit;
|
|
|
|
fn try_conv_with(self, world: &ServerWorld) -> Result<TextDocumentEdit> {
|
|
|
|
let text_document = VersionedTextDocumentIdentifier {
|
|
|
|
uri: self.file_id.try_conv_with(world)?,
|
|
|
|
version: None,
|
|
|
|
};
|
|
|
|
let line_index = world.analysis().file_line_index(self.file_id);
|
2018-12-21 02:24:16 -06:00
|
|
|
let edits = self
|
|
|
|
.edit
|
|
|
|
.as_atoms()
|
|
|
|
.iter()
|
|
|
|
.map_conv_with(&line_index)
|
|
|
|
.collect();
|
2018-10-15 16:44:23 -05:00
|
|
|
Ok(TextDocumentEdit {
|
|
|
|
text_document,
|
|
|
|
edits,
|
|
|
|
})
|
2018-08-29 10:03:14 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl TryConvWith for FileSystemEdit {
|
|
|
|
type Ctx = ServerWorld;
|
|
|
|
type Output = req::FileSystemEdit;
|
|
|
|
fn try_conv_with(self, world: &ServerWorld) -> Result<req::FileSystemEdit> {
|
|
|
|
let res = match self {
|
|
|
|
FileSystemEdit::CreateFile { anchor, path } => {
|
|
|
|
let uri = world.file_id_to_uri(anchor)?;
|
|
|
|
let path = &path.as_str()[3..]; // strip `../` b/c url is weird
|
|
|
|
let uri = uri.join(path)?;
|
|
|
|
req::FileSystemEdit::CreateFile { uri }
|
2018-10-15 16:44:23 -05:00
|
|
|
}
|
2018-08-29 10:03:14 -05:00
|
|
|
FileSystemEdit::MoveFile { file, path } => {
|
|
|
|
let src = world.file_id_to_uri(file)?;
|
|
|
|
let path = &path.as_str()[3..]; // strip `../` b/c url is weird
|
|
|
|
let dst = src.join(path)?;
|
|
|
|
req::FileSystemEdit::MoveFile { src, dst }
|
2018-10-15 16:44:23 -05:00
|
|
|
}
|
2018-08-29 10:03:14 -05:00
|
|
|
};
|
|
|
|
Ok(res)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2018-08-15 09:24:20 -05:00
|
|
|
pub fn to_location(
|
|
|
|
file_id: FileId,
|
|
|
|
range: TextRange,
|
2018-08-17 11:54:08 -05:00
|
|
|
world: &ServerWorld,
|
2018-08-15 09:24:20 -05:00
|
|
|
line_index: &LineIndex,
|
|
|
|
) -> Result<Location> {
|
2018-10-15 16:44:23 -05:00
|
|
|
let url = file_id.try_conv_with(world)?;
|
|
|
|
let loc = Location::new(url, range.conv_with(line_index));
|
|
|
|
Ok(loc)
|
2018-08-13 08:35:17 -05:00
|
|
|
}
|
|
|
|
|
2018-10-15 12:15:53 -05:00
|
|
|
pub trait MapConvWith<'a>: Sized + 'a {
|
2018-08-12 18:38:34 -05:00
|
|
|
type Ctx;
|
|
|
|
type Output;
|
|
|
|
|
|
|
|
fn map_conv_with(self, ctx: &'a Self::Ctx) -> ConvWithIter<'a, Self, Self::Ctx> {
|
|
|
|
ConvWithIter { iter: self, ctx }
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
impl<'a, I> MapConvWith<'a> for I
|
2018-10-15 16:44:23 -05:00
|
|
|
where
|
|
|
|
I: Iterator + 'a,
|
|
|
|
I::Item: ConvWith,
|
2018-08-12 18:38:34 -05:00
|
|
|
{
|
|
|
|
type Ctx = <I::Item as ConvWith>::Ctx;
|
|
|
|
type Output = <I::Item as ConvWith>::Output;
|
|
|
|
}
|
|
|
|
|
|
|
|
pub struct ConvWithIter<'a, I, Ctx: 'a> {
|
|
|
|
iter: I,
|
|
|
|
ctx: &'a Ctx,
|
|
|
|
}
|
|
|
|
|
|
|
|
impl<'a, I, Ctx> Iterator for ConvWithIter<'a, I, Ctx>
|
2018-10-15 16:44:23 -05:00
|
|
|
where
|
|
|
|
I: Iterator,
|
|
|
|
I::Item: ConvWith<Ctx = Ctx>,
|
2018-08-12 18:38:34 -05:00
|
|
|
{
|
|
|
|
type Item = <I::Item as ConvWith>::Output;
|
|
|
|
|
|
|
|
fn next(&mut self) -> Option<Self::Item> {
|
|
|
|
self.iter.next().map(|item| item.conv_with(self.ctx))
|
|
|
|
}
|
|
|
|
}
|