diff options
Diffstat (limited to 'src/server')
| -rw-r--r-- | src/server/analysis.odin | 430 | ||||
| -rw-r--r-- | src/server/completion.odin | 447 |
2 files changed, 447 insertions, 430 deletions
diff --git a/src/server/analysis.odin b/src/server/analysis.odin index 5e929e5..69fd218 100644 --- a/src/server/analysis.odin +++ b/src/server/analysis.odin @@ -1941,436 +1941,6 @@ get_hover_information :: proc(document: ^Document, position: common.Position) -> return hover, true; } -get_completion_list :: proc(document: ^Document, position: common.Position) -> (CompletionList, bool) { - - list: CompletionList; - - ast_context := make_ast_context(document.ast, document.imports, document.package_name); - - position_context, ok := get_document_position_context(document, position, .Completion); - - get_globals(document.ast, &ast_context); - - if position_context.function != nil { - get_locals(document.ast, position_context.function, &ast_context, &position_context); - } - - items := make([dynamic] CompletionItem, context.temp_allocator); - - if position_context.selector != nil { - - ast_context.current_package = ast_context.document_package; - - if ident, ok := position_context.selector.derived.(ast.Ident); ok { - - if !resolve_ident_is_variable(&ast_context, ident) && !resolve_ident_is_package(&ast_context, ident) { - return list, true; - } - - } - - symbols := make([dynamic] index.Symbol, context.temp_allocator); - - selector: index.Symbol; - - ast_context.use_locals = true; - ast_context.use_globals = true; - - selector, ok = resolve_type_expression(&ast_context, position_context.selector); - - if !ok { - log.info(position_context.selector.derived); - log.error("Failed to resolve type selector in completion list"); - return list, true; - } - - if selector.pkg != "" { - ast_context.current_package = selector.pkg; - } - - else { - ast_context.current_package = ast_context.document_package; - } - - field: string; - - if position_context.field != nil { - - switch v in position_context.field.derived { - case ast.Ident: - field = v.name; - } - - } - - - if s, ok := selector.value.(index.SymbolProcedureValue); ok { - if len(s.return_types) == 1 { - if selector, ok = resolve_type_expression(&ast_context, s.return_types[0].type); !ok { - return list, true; - } - } - } - - - #partial switch v in selector.value { - case index.SymbolEnumValue: - list.isIncomplete = false; - - for name in v.names { - symbol: index.Symbol; - symbol.name = name; - symbol.type = .EnumMember; - append(&symbols, symbol); - } - - case index.SymbolStructValue: - list.isIncomplete = false; - - for name, i in v.names { - - if selector.pkg != "" { - ast_context.current_package = selector.pkg; - } - - else { - ast_context.current_package = ast_context.document_package; - } - - if symbol, ok := resolve_type_expression(&ast_context, v.types[i]); ok { - symbol.name = name; - symbol.type = .Field; - symbol.pkg = selector.name; - symbol.signature = index.node_to_string(v.types[i]); - append(&symbols, symbol); - } - - else { - //just give some generic symbol with name. - symbol: index.Symbol; - symbol.name = name; - symbol.type = .Field; - append(&symbols, symbol); - } - - } - - - case index.SymbolPackageValue: - - list.isIncomplete = true; - - log.infof("search field %v, pkg %v", field, selector.pkg); - - if searched, ok := index.fuzzy_search(field, {selector.pkg}); ok { - - for search in searched { - append(&symbols, search.symbol); - } - - } - - else { - log.errorf("Failed to fuzzy search, field: %v, package: %v", field, selector.pkg); - return list, true; - } - - - case index.SymbolGenericValue: - - list.isIncomplete = false; - - if ptr, ok := v.expr.derived.(ast.Pointer_Type); ok { - - if symbol, ok := resolve_type_expression(&ast_context, ptr.elem); ok { - - #partial switch s in symbol.value { - case index.SymbolStructValue: - for name, i in s.names { - //ERROR no completion on name - - if selector.pkg != "" { - ast_context.current_package = selector.pkg; - } - - else { - ast_context.current_package = ast_context.document_package; - } - - if symbol, ok := resolve_type_expression(&ast_context, s.types[i]); ok { - symbol.name = name; - symbol.type = .Field; - symbol.pkg = symbol.name; - symbol.signature = index.node_to_string(s.types[i]); - append(&symbols, symbol); - } - - else { - symbol: index.Symbol; - symbol.name = name; - symbol.type = .Field; - append(&symbols, symbol); - } - } - } - } - - } - - } - - for symbol, i in symbols { - - item := CompletionItem { - label = symbol.name, - kind = cast(CompletionItemKind) symbol.type, - detail = concatenate_symbols_information(&ast_context, symbol), - documentation = symbol.doc, - }; - - append(&items, item); - } - - list.items = items[:]; - } - - else if position_context.implicit { - - list.isIncomplete = false; - - selector: index.Symbol; - - ast_context.use_locals = true; - ast_context.use_globals = true; - - if selector.pkg != "" { - ast_context.current_package = selector.pkg; - } - - else { - ast_context.current_package = ast_context.document_package; - } - - if position_context.binary != nil && (position_context.binary.op.text == "==" || position_context.binary.op.text == "!=") { - - context_node: ^ast.Expr; - enum_node: ^ast.Expr; - - if position_in_node(position_context.binary.right, position_context.position) { - context_node = position_context.binary.right; - enum_node = position_context.binary.left; - } - - else if position_in_node(position_context.binary.left, position_context.position) { - context_node = position_context.binary.left; - enum_node = position_context.binary.right; - } - - if context_node != nil && enum_node != nil { - - if lhs, ok := resolve_type_expression(&ast_context, enum_node); ok { - - #partial switch v in lhs.value { - case index.SymbolEnumValue: - for name in v.names { - - item := CompletionItem { - label = name, - kind = .EnumMember, - detail = name, - }; - - append(&items, item); - - } - - - } - - - - } - - - - } - - - - } - - else if position_context.returns != nil && position_context.function != nil { - - //function := position_context.function.derived.(ast.Call_Expr); - - - - - - - //if len(position_context.returns.results) == position_context.function. - - - } - - list.items = items[:]; - - - - - } - - else { - - list.isIncomplete = true; - - CombinedResult :: struct { - score: f32, - symbol: index.Symbol, - variable: ^ast.Ident, - }; - - combined_sort_interface :: proc(s: ^[dynamic] CombinedResult) -> sort.Interface { - return sort.Interface{ - collection = rawptr(s), - len = proc(it: sort.Interface) -> int { - s := (^[dynamic] CombinedResult)(it.collection); - return len(s^); - }, - less = proc(it: sort.Interface, i, j: int) -> bool { - s := (^[dynamic] CombinedResult)(it.collection); - return s[i].score > s[j].score; - }, - swap = proc(it: sort.Interface, i, j: int) { - s := (^[dynamic] CombinedResult)(it.collection); - s[i], s[j] = s[j], s[i]; - }, - }; - } - - combined := make([dynamic] CombinedResult); - - lookup := ""; - - if ident, ok := position_context.identifier.derived.(ast.Ident); ok { - lookup = ident.name; - } - - pkgs := make([dynamic] string, context.temp_allocator); - - usings := get_using_packages(&ast_context); - - for u in usings { - append(&pkgs, u); - } - - append(&pkgs, ast_context.document_package); - - if results, ok := index.fuzzy_search(lookup, pkgs[:]); ok { - - for r in results { - append(&combined, CombinedResult { score = r.score, symbol = r.symbol}); - } - - } - - matcher := common.make_fuzzy_matcher(lookup); - - global: for k, v in ast_context.globals { - - //combined is sorted and should do binary search instead. - for result in combined { - if result.symbol.name == k { - continue global; - } - } - - ast_context.use_locals = true; - ast_context.use_globals = true; - ast_context.current_package = ast_context.document_package; - - ident := index.new_type(ast.Ident, tokenizer.Pos {}, tokenizer.Pos {}, context.temp_allocator); - ident.name = k; - - if symbol, ok := resolve_type_identifier(&ast_context, ident^); ok { - symbol.name = ident.name; - symbol.signature = get_signature(&ast_context, ident^, symbol); - - if score, ok := common.fuzzy_match(matcher, symbol.name); ok { - append(&combined, CombinedResult { score = score * 1.1, symbol = symbol, variable = ident }); - } - - } - } - - for k, v in ast_context.locals { - - ast_context.use_locals = true; - ast_context.use_globals = true; - ast_context.current_package = ast_context.document_package; - - ident := index.new_type(ast.Ident, tokenizer.Pos {}, tokenizer.Pos {}, context.temp_allocator); - ident.name = k; - - if symbol, ok := resolve_type_identifier(&ast_context, ident^); ok { - symbol.name = ident.name; - symbol.signature = get_signature(&ast_context, ident^, symbol); - - if score, ok := common.fuzzy_match(matcher, symbol.name); ok { - append(&combined, CombinedResult { score = score * 1.1, symbol = symbol, variable = ident }); - } - - } - } - - for pkg in ast_context.imports { - - symbol: index.Symbol; - - symbol.name = pkg.base; - symbol.type = .Package; - - if score, ok := common.fuzzy_match(matcher, symbol.name); ok { - append(&combined, CombinedResult { score = score * 1.1, symbol = symbol }); - } - } - - sort.sort(combined_sort_interface(&combined)); - - //hard code for now - top_results := combined[0:(min(20, len(combined)))]; - - for result in top_results { - - item := CompletionItem { - label = result.symbol.name, - detail = concatenate_symbols_information(&ast_context, result.symbol), - }; - - if result.variable != nil { - if ok := resolve_ident_is_variable(&ast_context, result.variable^); ok { - item.kind = .Variable; - } - - else { - item.kind = cast(CompletionItemKind)result.symbol.type; - } - } - - else { - item.kind = cast(CompletionItemKind)result.symbol.type; - } - - append(&items, item); - } - - - list.items = items[:]; - } - - return list, true; -} - get_signature_information :: proc(document: ^Document, position: common.Position) -> (SignatureHelp, bool) { signature_help: SignatureHelp; diff --git a/src/server/completion.odin b/src/server/completion.odin new file mode 100644 index 0000000..025ff9f --- /dev/null +++ b/src/server/completion.odin @@ -0,0 +1,447 @@ +package server + +import "core:odin/parser" +import "core:odin/ast" +import "core:odin/tokenizer" +import "core:fmt" +import "core:log" +import "core:strings" +import "core:path" +import "core:mem" +import "core:strconv" +import "core:path/filepath" +import "core:sort" +import "core:slice" + +import "shared:common" +import "shared:index" + +get_completion_list :: proc(document: ^Document, position: common.Position) -> (CompletionList, bool) { + + list: CompletionList; + + ast_context := make_ast_context(document.ast, document.imports, document.package_name); + + position_context, ok := get_document_position_context(document, position, .Completion); + + get_globals(document.ast, &ast_context); + + if position_context.function != nil { + get_locals(document.ast, position_context.function, &ast_context, &position_context); + } + + items := make([dynamic] CompletionItem, context.temp_allocator); + + if position_context.selector != nil { + + ast_context.current_package = ast_context.document_package; + + if ident, ok := position_context.selector.derived.(ast.Ident); ok { + + if !resolve_ident_is_variable(&ast_context, ident) && !resolve_ident_is_package(&ast_context, ident) { + return list, true; + } + + } + + symbols := make([dynamic] index.Symbol, context.temp_allocator); + + selector: index.Symbol; + + ast_context.use_locals = true; + ast_context.use_globals = true; + + selector, ok = resolve_type_expression(&ast_context, position_context.selector); + + if !ok { + log.info(position_context.selector.derived); + log.error("Failed to resolve type selector in completion list"); + return list, true; + } + + if selector.pkg != "" { + ast_context.current_package = selector.pkg; + } + + else { + ast_context.current_package = ast_context.document_package; + } + + field: string; + + if position_context.field != nil { + + switch v in position_context.field.derived { + case ast.Ident: + field = v.name; + } + + } + + + if s, ok := selector.value.(index.SymbolProcedureValue); ok { + if len(s.return_types) == 1 { + if selector, ok = resolve_type_expression(&ast_context, s.return_types[0].type); !ok { + return list, true; + } + } + } + + + #partial switch v in selector.value { + case index.SymbolEnumValue: + list.isIncomplete = false; + + for name in v.names { + symbol: index.Symbol; + symbol.name = name; + symbol.type = .EnumMember; + append(&symbols, symbol); + } + + case index.SymbolStructValue: + list.isIncomplete = false; + + for name, i in v.names { + + if selector.pkg != "" { + ast_context.current_package = selector.pkg; + } + + else { + ast_context.current_package = ast_context.document_package; + } + + if symbol, ok := resolve_type_expression(&ast_context, v.types[i]); ok { + symbol.name = name; + symbol.type = .Field; + symbol.pkg = selector.name; + symbol.signature = index.node_to_string(v.types[i]); + append(&symbols, symbol); + } + + else { + //just give some generic symbol with name. + symbol: index.Symbol; + symbol.name = name; + symbol.type = .Field; + append(&symbols, symbol); + } + + } + + + case index.SymbolPackageValue: + + list.isIncomplete = true; + + log.infof("search field %v, pkg %v", field, selector.pkg); + + if searched, ok := index.fuzzy_search(field, {selector.pkg}); ok { + + for search in searched { + append(&symbols, search.symbol); + } + + } + + else { + log.errorf("Failed to fuzzy search, field: %v, package: %v", field, selector.pkg); + return list, true; + } + + + case index.SymbolGenericValue: + + list.isIncomplete = false; + + if ptr, ok := v.expr.derived.(ast.Pointer_Type); ok { + + if symbol, ok := resolve_type_expression(&ast_context, ptr.elem); ok { + + #partial switch s in symbol.value { + case index.SymbolStructValue: + for name, i in s.names { + //ERROR no completion on name + + if selector.pkg != "" { + ast_context.current_package = selector.pkg; + } + + else { + ast_context.current_package = ast_context.document_package; + } + + if symbol, ok := resolve_type_expression(&ast_context, s.types[i]); ok { + symbol.name = name; + symbol.type = .Field; + symbol.pkg = symbol.name; + symbol.signature = index.node_to_string(s.types[i]); + append(&symbols, symbol); + } + + else { + symbol: index.Symbol; + symbol.name = name; + symbol.type = .Field; + append(&symbols, symbol); + } + } + } + } + + } + + } + + for symbol, i in symbols { + + item := CompletionItem { + label = symbol.name, + kind = cast(CompletionItemKind) symbol.type, + detail = concatenate_symbols_information(&ast_context, symbol), + documentation = symbol.doc, + }; + + append(&items, item); + } + + list.items = items[:]; + } + + else if position_context.implicit { + + list.isIncomplete = false; + + selector: index.Symbol; + + ast_context.use_locals = true; + ast_context.use_globals = true; + + if selector.pkg != "" { + ast_context.current_package = selector.pkg; + } + + else { + ast_context.current_package = ast_context.document_package; + } + + if position_context.binary != nil && (position_context.binary.op.text == "==" || position_context.binary.op.text == "!=") { + + context_node: ^ast.Expr; + enum_node: ^ast.Expr; + + if position_in_node(position_context.binary.right, position_context.position) { + context_node = position_context.binary.right; + enum_node = position_context.binary.left; + } + + else if position_in_node(position_context.binary.left, position_context.position) { + context_node = position_context.binary.left; + enum_node = position_context.binary.right; + } + + if context_node != nil && enum_node != nil { + + if lhs, ok := resolve_type_expression(&ast_context, enum_node); ok { + + #partial switch v in lhs.value { + case index.SymbolEnumValue: + for name in v.names { + + item := CompletionItem { + label = name, + kind = .EnumMember, + detail = name, + }; + + append(&items, item); + + } + + + } + + + + } + + + + } + + + + } + + else if position_context.returns != nil && position_context.function != nil { + + //function := position_context.function.derived.(ast.Call_Expr); + + + + + + + //if len(position_context.returns.results) == position_context.function. + + + } + + list.items = items[:]; + + + + + } + + else { + + list.isIncomplete = true; + + CombinedResult :: struct { + score: f32, + symbol: index.Symbol, + variable: ^ast.Ident, + }; + + combined_sort_interface :: proc(s: ^[dynamic] CombinedResult) -> sort.Interface { + return sort.Interface{ + collection = rawptr(s), + len = proc(it: sort.Interface) -> int { + s := (^[dynamic] CombinedResult)(it.collection); + return len(s^); + }, + less = proc(it: sort.Interface, i, j: int) -> bool { + s := (^[dynamic] CombinedResult)(it.collection); + return s[i].score > s[j].score; + }, + swap = proc(it: sort.Interface, i, j: int) { + s := (^[dynamic] CombinedResult)(it.collection); + s[i], s[j] = s[j], s[i]; + }, + }; + } + + combined := make([dynamic] CombinedResult); + + lookup := ""; + + if ident, ok := position_context.identifier.derived.(ast.Ident); ok { + lookup = ident.name; + } + + pkgs := make([dynamic] string, context.temp_allocator); + + usings := get_using_packages(&ast_context); + + for u in usings { + append(&pkgs, u); + } + + append(&pkgs, ast_context.document_package); + + if results, ok := index.fuzzy_search(lookup, pkgs[:]); ok { + + for r in results { + append(&combined, CombinedResult { score = r.score, symbol = r.symbol}); + } + + } + + matcher := common.make_fuzzy_matcher(lookup); + + global: for k, v in ast_context.globals { + + //combined is sorted and should do binary search instead. + for result in combined { + if result.symbol.name == k { + continue global; + } + } + + ast_context.use_locals = true; + ast_context.use_globals = true; + ast_context.current_package = ast_context.document_package; + + ident := index.new_type(ast.Ident, tokenizer.Pos {}, tokenizer.Pos {}, context.temp_allocator); + ident.name = k; + + if symbol, ok := resolve_type_identifier(&ast_context, ident^); ok { + symbol.name = ident.name; + symbol.signature = get_signature(&ast_context, ident^, symbol); + + if score, ok := common.fuzzy_match(matcher, symbol.name); ok { + append(&combined, CombinedResult { score = score * 1.1, symbol = symbol, variable = ident }); + } + + } + } + + for k, v in ast_context.locals { + + ast_context.use_locals = true; + ast_context.use_globals = true; + ast_context.current_package = ast_context.document_package; + + ident := index.new_type(ast.Ident, tokenizer.Pos {}, tokenizer.Pos {}, context.temp_allocator); + ident.name = k; + + if symbol, ok := resolve_type_identifier(&ast_context, ident^); ok { + symbol.name = ident.name; + symbol.signature = get_signature(&ast_context, ident^, symbol); + + if score, ok := common.fuzzy_match(matcher, symbol.name); ok { + append(&combined, CombinedResult { score = score * 1.1, symbol = symbol, variable = ident }); + } + + } + } + + for pkg in ast_context.imports { + + symbol: index.Symbol; + + symbol.name = pkg.base; + symbol.type = .Package; + + if score, ok := common.fuzzy_match(matcher, symbol.name); ok { + append(&combined, CombinedResult { score = score * 1.1, symbol = symbol }); + } + } + + sort.sort(combined_sort_interface(&combined)); + + //hard code for now + top_results := combined[0:(min(20, len(combined)))]; + + for result in top_results { + + item := CompletionItem { + label = result.symbol.name, + detail = concatenate_symbols_information(&ast_context, result.symbol), + }; + + if result.variable != nil { + if ok := resolve_ident_is_variable(&ast_context, result.variable^); ok { + item.kind = .Variable; + } + + else { + item.kind = cast(CompletionItemKind)result.symbol.type; + } + } + + else { + item.kind = cast(CompletionItemKind)result.symbol.type; + } + + append(&items, item); + } + + + list.items = items[:]; + } + + return list, true; +} |