Forgot file...

1 files changed, 141 insertions, 0 deletions

diff --git a/src/server/methods.odin b/src/server/methods.odin
new file mode 100644
index 0000000..60e7cd8
--- /dev/null
+++ b/src/server/methods.odin
@@ -0,0 +1,141 @@
+package server
+
+import "core:odin/parser"
+import "core:odin/ast"
+import "core:odin/tokenizer"
+import "core:fmt"
+import "core:log"
+import "core:strings"
+import path "core:path/slashpath"
+import "core:mem"
+import "core:strconv"
+import "core:path/filepath"
+import "core:sort"
+import "core:slice"
+import "core:os"
+
+
+import "shared:common"
+
+
+@(private)
+create_remove_edit :: proc(
+	position_context: ^DocumentPositionContext,
+) -> (
+	[]TextEdit,
+	bool,
+) {
+	range, ok := get_range_from_selection_start_to_dot(position_context)
+
+	if !ok {
+		return {}, false
+	}
+
+	remove_range := common.Range {
+		start = range.start,
+		end   = range.end,
+	}
+
+	remove_edit := TextEdit {
+		range   = remove_range,
+		newText = "",
+	}
+
+	additionalTextEdits := make([]TextEdit, 1, context.temp_allocator)
+	additionalTextEdits[0] = remove_edit
+
+	return additionalTextEdits, true
+}
+
+append_method_completion :: proc(
+	ast_context: ^AstContext,
+	symbol: Symbol,
+	position_context: ^DocumentPositionContext,
+	items: ^[dynamic]CompletionItem,
+) {
+	if symbol.type != .Variable {
+		return
+	}
+
+	remove_edit, ok := create_remove_edit(position_context)
+
+	if !ok {
+		return
+	}
+
+	for k, v in indexer.index.collection.packages {
+		method := Method {
+			name = symbol.name,
+			pkg  = symbol.pkg,
+		}
+		if symbols, ok := &v.methods[method]; ok {
+			for symbol in symbols {
+				resolve_unresolved_symbol(ast_context, &symbol)
+				build_procedure_symbol_signature(&symbol)
+
+				range, ok := get_range_from_selection_start_to_dot(
+					position_context,
+				)
+
+				if !ok {
+					return
+				}
+
+				value: SymbolProcedureValue
+				value, ok = symbol.value.(SymbolProcedureValue)
+
+				if !ok {
+					continue
+				}
+
+				first_arg: Symbol
+				first_arg, ok = resolve_type_expression(
+					ast_context,
+					value.arg_types[0].type,
+				)
+
+				if !ok {
+					continue
+				}
+
+				pointers_to_add := first_arg.pointers - symbol.pointers
+
+				if pointers_to_add != 1 {
+					pointers_to_add = 0
+				}
+
+				new_text := ""
+
+				if symbol.pkg != ast_context.document_package {
+					new_text = fmt.tprintf(
+						"%v.%v($0)",
+						path.base(symbol.pkg, false, ast_context.allocator),
+						symbol.name,
+					)
+				} else {
+					new_text = fmt.tprintf("%v($0)", symbol.name)
+				}
+
+				item := CompletionItem {
+					label = symbol.name,
+					kind = cast(CompletionItemKind)symbol.type,
+					detail = concatenate_symbol_information(
+						ast_context,
+						symbol,
+						true,
+					),
+					additionalTextEdits = remove_edit,
+					textEdit = TextEdit{
+						newText = new_text,
+						range = {start = range.end, end = range.end},
+					},
+					insertTextFormat = .Snippet,
+					InsertTextMode = .adjustIndentation,
+					documentation = symbol.doc,
+				}
+
+				append(items, item)
+			}
+		}
+	}
+}