traxys/tree-sitter
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Implement class and call tagging.

Browse source
This commit is contained in:
Patrick Thomson 2020-03-06 16:48:15 -05:00
parent 680a9e0531
commit 8546a71c31
1 changed files with 24 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

30
tags/src/lib.rs
View file

@ -153,17 +153,26 @@ impl TagsContext {
}
}
if let (Some(function), Some(name)) = (function_node, name_node) {
let tag_from_node = |kind, name: Node, node| {
if let Ok(name) = str::from_utf8(&source[name.byte_range()]) {
return Some(Tag {
name,
line: "",
loc: loc_for_node(function),
kind: TagKind::Function,
line: "TODO",
loc: loc_for_node(node),
kind: kind,
docs: doc_node
.and_then(|n| str::from_utf8(&source[n.byte_range()]).ok()),
});
}
};
return None;
};
if let (Some(function), Some(name)) = (function_node, name_node) {
return tag_from_node(TagKind::Function, name, function);
} else if let (Some(call), Some(name)) = (call_node, name_node) {
return tag_from_node(TagKind::Call, name, call);
} else if let (Some(class), Some(name)) = (class_node, name_node) {
return tag_from_node(TagKind::Class, name, class);
}
None
@ -191,6 +200,15 @@ impl Serialize for TagKind {
fn loc_for_node(node: Node) -> Loc {
Loc {
byte_range: node.byte_range(),
span: node.start_position()..node.start_position(),
span: node.start_position().into()..node.start_position().into(),
}
}
impl From<tree_sitter::Point> for Pos {
fn from(point: tree_sitter::Point) -> Self {
return Pos {
line: point.row as i64,
column: point.column as i64,
};
}
}