Implemented go to definition
This commit is contained in:
parent
acd85c0f39
commit
b1ce06a80d
@ -180,26 +180,42 @@ pub fn isPascalCase(name: []const u8) bool {
|
|||||||
|
|
||||||
// ANALYSIS ENGINE
|
// ANALYSIS ENGINE
|
||||||
|
|
||||||
/// Gets the child of node
|
pub fn getDeclNameToken(tree: *ast.Tree, node: *ast.Node) ?ast.TokenIndex {
|
||||||
pub fn getChild(tree: *ast.Tree, node: *ast.Node, name: []const u8) ?*ast.Node {
|
switch (node.id) {
|
||||||
var index: usize = 0;
|
|
||||||
while (node.iterate(index)) |child| {
|
|
||||||
switch (child.id) {
|
|
||||||
.VarDecl => {
|
.VarDecl => {
|
||||||
const vari = child.cast(ast.Node.VarDecl).?;
|
const vari = node.cast(ast.Node.VarDecl).?;
|
||||||
if (std.mem.eql(u8, tree.tokenSlice(vari.name_token), name)) return child;
|
return vari.name_token;
|
||||||
|
},
|
||||||
|
.ParamDecl => {
|
||||||
|
const decl = node.cast(ast.Node.ParamDecl).?;
|
||||||
|
if (decl.name_token == null) return null;
|
||||||
|
return decl.name_token.?;
|
||||||
},
|
},
|
||||||
.FnProto => {
|
.FnProto => {
|
||||||
const func = child.cast(ast.Node.FnProto).?;
|
const func = node.cast(ast.Node.FnProto).?;
|
||||||
if (func.name_token != null and std.mem.eql(u8, tree.tokenSlice(func.name_token.?), name)) return child;
|
if (func.name_token == null) return null;
|
||||||
|
return func.name_token.?;
|
||||||
},
|
},
|
||||||
.ContainerField => {
|
.ContainerField => {
|
||||||
const field = child.cast(ast.Node.ContainerField).?;
|
const field = node.cast(ast.Node.ContainerField).?;
|
||||||
if (std.mem.eql(u8, tree.tokenSlice(field.name_token), name)) return child;
|
return field.name_token;
|
||||||
},
|
},
|
||||||
else => {},
|
else => {},
|
||||||
}
|
}
|
||||||
index += 1;
|
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
|
fn getDeclName(tree: *ast.Tree, node: *ast.Node) ?[]const u8 {
|
||||||
|
return tree.tokenSlice(getDeclNameToken(tree, node) orelse return null);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Gets the child of node
|
||||||
|
pub fn getChild(tree: *ast.Tree, node: *ast.Node, name: []const u8) ?*ast.Node {
|
||||||
|
var index: usize = 0;
|
||||||
|
while (node.iterate(index)) |child| : (index += 1) {
|
||||||
|
const child_name = getDeclName(tree, child) orelse continue;
|
||||||
|
if (std.mem.eql(u8, child_name, name)) return child;
|
||||||
}
|
}
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
@ -207,25 +223,8 @@ pub fn getChild(tree: *ast.Tree, node: *ast.Node, name: []const u8) ?*ast.Node {
|
|||||||
/// Gets the child of slice
|
/// Gets the child of slice
|
||||||
pub fn getChildOfSlice(tree: *ast.Tree, nodes: []*ast.Node, name: []const u8) ?*ast.Node {
|
pub fn getChildOfSlice(tree: *ast.Tree, nodes: []*ast.Node, name: []const u8) ?*ast.Node {
|
||||||
for (nodes) |child| {
|
for (nodes) |child| {
|
||||||
switch (child.id) {
|
const child_name = getDeclName(tree, child) orelse continue;
|
||||||
.VarDecl => {
|
if (std.mem.eql(u8, child_name, name)) return child;
|
||||||
const vari = child.cast(ast.Node.VarDecl).?;
|
|
||||||
if (std.mem.eql(u8, tree.tokenSlice(vari.name_token), name)) return child;
|
|
||||||
},
|
|
||||||
.ParamDecl => {
|
|
||||||
const decl = child.cast(ast.Node.ParamDecl).?;
|
|
||||||
if (decl.name_token != null and std.mem.eql(u8, tree.tokenSlice(decl.name_token.?), name)) return child;
|
|
||||||
},
|
|
||||||
.FnProto => {
|
|
||||||
const func = child.cast(ast.Node.FnProto).?;
|
|
||||||
if (func.name_token != null and std.mem.eql(u8, tree.tokenSlice(func.name_token.?), name)) return child;
|
|
||||||
},
|
|
||||||
.ContainerField => {
|
|
||||||
const field = child.cast(ast.Node.ContainerField).?;
|
|
||||||
if (std.mem.eql(u8, tree.tokenSlice(field.name_token), name)) return child;
|
|
||||||
},
|
|
||||||
else => {},
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
|
114
src/main.zig
114
src/main.zig
@ -17,7 +17,7 @@ var allocator: *std.mem.Allocator = undefined;
|
|||||||
var document_store: DocumentStore = undefined;
|
var document_store: DocumentStore = undefined;
|
||||||
|
|
||||||
const initialize_response =
|
const initialize_response =
|
||||||
\\,"result":{"capabilities":{"signatureHelpProvider":{"triggerCharacters":["(",","]},"textDocumentSync":1,"completionProvider":{"resolveProvider":false,"triggerCharacters":[".",":","@"]},"documentHighlightProvider":false,"codeActionProvider":false,"workspace":{"workspaceFolders":{"supported":true}}}}}
|
\\,"result":{"capabilities":{"signatureHelpProvider":{"triggerCharacters":["(",","]},"textDocumentSync":1,"completionProvider":{"resolveProvider":false,"triggerCharacters":[".",":","@"]},"documentHighlightProvider":false,"codeActionProvider":false,"declarationProvider":true,"definitionProvider":true,"typeDefinitionProvider":true,"workspace":{"workspaceFolders":{"supported":true}}}}}
|
||||||
;
|
;
|
||||||
|
|
||||||
const not_implemented_response =
|
const not_implemented_response =
|
||||||
@ -172,7 +172,7 @@ fn publishDiagnostics(handle: DocumentStore.Handle, config: Config) !void {
|
|||||||
|
|
||||||
fn containerToCompletion(list: *std.ArrayList(types.CompletionItem), tree: *std.zig.ast.Tree, container: *std.zig.ast.Node, config: Config) !void {
|
fn containerToCompletion(list: *std.ArrayList(types.CompletionItem), tree: *std.zig.ast.Tree, container: *std.zig.ast.Node, config: Config) !void {
|
||||||
var index: usize = 0;
|
var index: usize = 0;
|
||||||
while (container.iterate(index)) |child_node| : (index+=1) {
|
while (container.iterate(index)) |child_node| : (index += 1) {
|
||||||
if (analysis.isNodePublic(tree, child_node)) {
|
if (analysis.isNodePublic(tree, child_node)) {
|
||||||
try nodeToCompletion(list, tree, child_node, config);
|
try nodeToCompletion(list, tree, child_node, config);
|
||||||
}
|
}
|
||||||
@ -258,6 +258,78 @@ fn nodeToCompletion(list: *std.ArrayList(types.CompletionItem), tree: *std.zig.a
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn identifierFromPosition(pos_index: usize, handle: DocumentStore.Handle) []const u8 {
|
||||||
|
var start_idx = pos_index;
|
||||||
|
while (start_idx > 0 and
|
||||||
|
(std.ascii.isAlNum(handle.document.text[start_idx]) or handle.document.text[start_idx] == '_')) : (start_idx -= 1)
|
||||||
|
{}
|
||||||
|
|
||||||
|
var end_idx = pos_index;
|
||||||
|
while (end_idx < handle.document.text.len and
|
||||||
|
(std.ascii.isAlNum(handle.document.text[end_idx]) or handle.document.text[end_idx] == '_')) : (end_idx += 1)
|
||||||
|
{}
|
||||||
|
|
||||||
|
return handle.document.text[start_idx + 1 .. end_idx];
|
||||||
|
}
|
||||||
|
|
||||||
|
fn gotoDefinitionGlobal(id: i64, pos_index: usize, handle: DocumentStore.Handle) !void {
|
||||||
|
var tree = try handle.tree(allocator);
|
||||||
|
defer tree.deinit();
|
||||||
|
|
||||||
|
const name = identifierFromPosition(pos_index, handle);
|
||||||
|
var arena = std.heap.ArenaAllocator.init(allocator);
|
||||||
|
defer arena.deinit();
|
||||||
|
|
||||||
|
var decl_nodes = std.ArrayList(*std.zig.ast.Node).init(&arena.allocator);
|
||||||
|
try analysis.declsFromIndex(&decl_nodes, tree, pos_index);
|
||||||
|
|
||||||
|
const decl = analysis.getChildOfSlice(tree, decl_nodes.items, name) orelse return try respondGeneric(id, null_result_response);
|
||||||
|
const name_token = analysis.getDeclNameToken(tree, decl) orelse unreachable;
|
||||||
|
|
||||||
|
try send(types.Response{
|
||||||
|
.id = .{ .Integer = id },
|
||||||
|
.result = .{
|
||||||
|
.Location = .{
|
||||||
|
.uri = handle.document.uri,
|
||||||
|
.range = astLocationToRange(tree.tokenLocation(0, name_token)),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
fn gotoDefinitionFieldAccess(id: i64, handle: *DocumentStore.Handle, position: types.Position, line_start_idx: usize) !void {
|
||||||
|
var arena = std.heap.ArenaAllocator.init(allocator);
|
||||||
|
defer arena.deinit();
|
||||||
|
|
||||||
|
var analysis_ctx = try document_store.analysisContext(handle, &arena, position);
|
||||||
|
defer analysis_ctx.deinit();
|
||||||
|
|
||||||
|
const pos_index = try handle.document.positionToIndex(position);
|
||||||
|
var name = identifierFromPosition(pos_index, handle.*);
|
||||||
|
|
||||||
|
const line = try handle.document.getLine(@intCast(usize, position.line));
|
||||||
|
var tokenizer = std.zig.Tokenizer.init(line[line_start_idx..]);
|
||||||
|
|
||||||
|
const line_length = @ptrToInt(name.ptr) - @ptrToInt(line.ptr) + name.len - line_start_idx;
|
||||||
|
name = try std.mem.dupe(&arena.allocator, u8, name);
|
||||||
|
|
||||||
|
if (analysis.getFieldAccessTypeNode(&analysis_ctx, &tokenizer, line_length)) |container| {
|
||||||
|
const decl = analysis.getChild(analysis_ctx.tree, container, name) orelse return try respondGeneric(id, null_result_response);
|
||||||
|
const name_token = analysis.getDeclNameToken(analysis_ctx.tree, decl) orelse unreachable;
|
||||||
|
return try send(types.Response{
|
||||||
|
.id = .{ .Integer = id },
|
||||||
|
.result = .{
|
||||||
|
.Location = .{
|
||||||
|
.uri = analysis_ctx.handle.document.uri,
|
||||||
|
.range = astLocationToRange(analysis_ctx.tree.tokenLocation(0, name_token)),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
try respondGeneric(id, null_result_response);
|
||||||
|
}
|
||||||
|
|
||||||
fn completeGlobal(id: i64, pos_index: usize, handle: DocumentStore.Handle, config: Config) !void {
|
fn completeGlobal(id: i64, pos_index: usize, handle: DocumentStore.Handle, config: Config) !void {
|
||||||
var tree = try handle.tree(allocator);
|
var tree = try handle.tree(allocator);
|
||||||
defer tree.deinit();
|
defer tree.deinit();
|
||||||
@ -531,7 +603,7 @@ fn processJsonRpc(parser: *std.json.Parser, json: []const u8, config: Config) !v
|
|||||||
|
|
||||||
const handle = document_store.getHandle(uri) orelse {
|
const handle = document_store.getHandle(uri) orelse {
|
||||||
std.debug.warn("Trying to complete in non existent document {}", .{uri});
|
std.debug.warn("Trying to complete in non existent document {}", .{uri});
|
||||||
return;
|
return try respondGeneric(id, no_completions_response);
|
||||||
};
|
};
|
||||||
|
|
||||||
const pos = types.Position{
|
const pos = types.Position{
|
||||||
@ -560,18 +632,36 @@ fn processJsonRpc(parser: *std.json.Parser, json: []const u8, config: Config) !v
|
|||||||
try respondGeneric(id, no_completions_response);
|
try respondGeneric(id, no_completions_response);
|
||||||
}
|
}
|
||||||
} else if (std.mem.eql(u8, method, "textDocument/signatureHelp")) {
|
} else if (std.mem.eql(u8, method, "textDocument/signatureHelp")) {
|
||||||
// try respondGeneric(id,
|
|
||||||
// \\,"result":{"signatures":[{
|
|
||||||
// \\"label": "nameOfFunction(aNumber: u8)",
|
|
||||||
// \\"documentation": {"kind": "markdown", "value": "Description of the function in **Markdown**!"},
|
|
||||||
// \\"parameters": [
|
|
||||||
// \\{"label": [15, 27], "documentation": {"kind": "markdown", "value": "An argument"}}
|
|
||||||
// \\]
|
|
||||||
// \\}]}}
|
|
||||||
// );
|
|
||||||
try respondGeneric(id,
|
try respondGeneric(id,
|
||||||
\\,"result":{"signatures":[]}}
|
\\,"result":{"signatures":[]}}
|
||||||
);
|
);
|
||||||
|
} else if (std.mem.eql(u8, method, "textDocument/definition") or
|
||||||
|
std.mem.eql(u8, method, "textDocument/declaration") or
|
||||||
|
std.mem.eql(u8, method, "textDocument/typeDefinition"))
|
||||||
|
{
|
||||||
|
const document = params.getValue("textDocument").?.Object;
|
||||||
|
const uri = document.getValue("uri").?.String;
|
||||||
|
const position = params.getValue("position").?.Object;
|
||||||
|
|
||||||
|
const handle = document_store.getHandle(uri) orelse {
|
||||||
|
std.debug.warn("Trying to got to definition in non existent document {}", .{uri});
|
||||||
|
return try respondGeneric(id, null_result_response);
|
||||||
|
};
|
||||||
|
|
||||||
|
const pos = types.Position{
|
||||||
|
.line = position.getValue("line").?.Integer,
|
||||||
|
.character = position.getValue("character").?.Integer - 1,
|
||||||
|
};
|
||||||
|
if (pos.character >= 0) {
|
||||||
|
const pos_index = try handle.document.positionToIndex(pos);
|
||||||
|
const pos_context = documentPositionContext(handle.document, pos_index);
|
||||||
|
|
||||||
|
switch (pos_context) {
|
||||||
|
.var_access => try gotoDefinitionGlobal(id, pos_index, handle.*),
|
||||||
|
.field_access => |start_idx| try gotoDefinitionFieldAccess(id, handle, pos, start_idx),
|
||||||
|
else => try respondGeneric(id, null_result_response),
|
||||||
|
}
|
||||||
|
}
|
||||||
} else if (root.Object.getValue("id")) |_| {
|
} else if (root.Object.getValue("id")) |_| {
|
||||||
std.debug.warn("Method with return value not implemented: {}", .{method});
|
std.debug.warn("Method with return value not implemented: {}", .{method});
|
||||||
try respondGeneric(id, not_implemented_response);
|
try respondGeneric(id, not_implemented_response);
|
||||||
|
@ -51,7 +51,8 @@ pub const NotificationParams = union(enum) {
|
|||||||
|
|
||||||
/// Params of a response (result)
|
/// Params of a response (result)
|
||||||
pub const ResponseParams = union(enum) {
|
pub const ResponseParams = union(enum) {
|
||||||
CompletionList: CompletionList
|
CompletionList: CompletionList,
|
||||||
|
Location: Location,
|
||||||
};
|
};
|
||||||
|
|
||||||
/// JSONRPC error
|
/// JSONRPC error
|
||||||
@ -282,4 +283,3 @@ pub const CompletionItem = struct {
|
|||||||
documentation: ?MarkupContent = null
|
documentation: ?MarkupContent = null
|
||||||
// filterText: String = .NotDefined,
|
// filterText: String = .NotDefined,
|
||||||
};
|
};
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user