diff options
| author | DanielGavin <danielgavin5@hotmail.com> | 2021-03-12 16:31:09 +0100 |
|---|---|---|
| committer | DanielGavin <danielgavin5@hotmail.com> | 2021-03-12 16:31:09 +0100 |
| commit | 00ccd7e03e17dac40efb9b34a048d968dd77c218 (patch) | |
| tree | 24e9e8d9743bc3e98b63183d1f976b11ab6d93d2 /src/server/analysis.odin | |
| parent | baf86e02a2c45170d58ab828a13f52361129b255 (diff) | |
ran odinfmt on project
Diffstat (limited to 'src/server/analysis.odin')
| -rw-r--r-- | src/server/analysis.odin | 4238 |
1 files changed, 2008 insertions, 2230 deletions
diff --git a/src/server/analysis.odin b/src/server/analysis.odin index d9b9dc2..8dc4509 100644 --- a/src/server/analysis.odin +++ b/src/server/analysis.odin @@ -18,2701 +18,2479 @@ import "shared:common" import "shared:index" /* - TODO(replace all of the possible ast walking with the new odin visitor function) - TODO(improve the current_package logic, kinda confusing switching between different packages with selectors) + TODO(replace all of the possible ast walking with the new odin visitor function) + TODO(improve the current_package logic, kinda confusing switching between different packages with selectors) */ -bool_lit := "bool"; -int_lit := "int"; +bool_lit := "bool"; +int_lit := "int"; string_lit := "string"; DocumentPositionContextHint :: enum { - Completion, - SignatureHelp, - Definition, - Hover, -}; + Completion, + SignatureHelp, + Definition, + Hover, +} DocumentPositionContext :: struct { - file: ast.File, - position: common.AbsolutePosition, - line: int, - function: ^ast.Proc_Lit, //used to help with type resolving in function scope - selector: ^ast.Expr, //used for completion - identifier: ^ast.Node, - field: ^ast.Expr, //used for completion - call: ^ast.Expr, //used for signature help - returns: ^ast.Return_Stmt, //used for completion - comp_lit: ^ast.Comp_Lit, //used for completion - parent_comp_lit: ^ast.Comp_Lit, //used for completion - implicit: bool, //used for completion - arrow: bool, - binary: ^ast.Binary_Expr, //used for completion - assign: ^ast.Assign_Stmt, //used for completion - switch_stmt: ^ast.Switch_Stmt, //used for completion - switch_type_stmt: ^ast.Type_Switch_Stmt, //used for completion - case_clause: ^ast.Case_Clause, //used for completion - value_decl: ^ast.Value_Decl, //used for completion - hint: DocumentPositionContextHint, -}; + file: ast.File, + position: common.AbsolutePosition, + line: int, + function: ^ast.Proc_Lit, //used to help with type resolving in function scope + selector: ^ast.Expr, //used for completion + identifier: ^ast.Node, + field: ^ast.Expr, //used for completion + call: ^ast.Expr, //used for signature help + returns: ^ast.Return_Stmt, //used for completion + comp_lit: ^ast.Comp_Lit, //used for completion + parent_comp_lit: ^ast.Comp_Lit, //used for completion + implicit: bool, //used for completion + arrow: bool, + binary: ^ast.Binary_Expr, //used for completion + assign: ^ast.Assign_Stmt, //used for completion + switch_stmt: ^ast.Switch_Stmt, //used for completion + switch_type_stmt: ^ast.Type_Switch_Stmt, //used for completion + case_clause: ^ast.Case_Clause, //used for completion + value_decl: ^ast.Value_Decl, //used for completion + hint: DocumentPositionContextHint, +} DocumentLocal :: struct { - expr: ^ast.Expr, - offset: int, + expr: ^ast.Expr, + offset: int, } AstContext :: struct { - locals: map [string] [dynamic] DocumentLocal, //locals all the way to the document position - globals: map [string] ^ast.Expr, - variables: map [string] bool, - parameters: map [string] bool, - in_package: map[string] string, //sometimes you have to extract types from arrays/maps and you lose package information - usings: [dynamic] string, - file: ast.File, - allocator: mem.Allocator, - imports: [] Package, //imports for the current document - current_package: string, - document_package: string, - use_globals: bool, - use_locals: bool, - call: ^ast.Call_Expr, //used to determene the types for generics and the correct function for overloaded functions - position: common.AbsolutePosition, - value_decl: ^ast.Value_Decl, - field_name: string, -}; - -make_ast_context :: proc(file: ast.File, imports: [] Package, package_name: string, allocator := context.temp_allocator) -> AstContext { - - ast_context := AstContext { - locals = make(map [string] [dynamic] DocumentLocal, 0, allocator), - globals = make(map [string] ^ast.Expr, 0, allocator), - variables = make(map [string] bool, 0, allocator), - usings = make([dynamic] string, allocator), - parameters = make(map [string] bool, 0, allocator), - in_package = make(map[string] string, 0, allocator), - file = file, - imports = imports, - use_locals = true, - use_globals = true, - document_package = package_name, - current_package = package_name, - }; - return ast_context; + locals: map[string][dynamic]DocumentLocal, //locals all the way to the document position + globals: map[string]^ast.Expr, + variables: map[string]bool, + parameters: map[string]bool, + in_package: map[string]string, //sometimes you have to extract types from arrays/maps and you lose package information + usings: [dynamic]string, + file: ast.File, + allocator: mem.Allocator, + imports: []Package, //imports for the current document + current_package: string, + document_package: string, + use_globals: bool, + use_locals: bool, + call: ^ast.Call_Expr, //used to determene the types for generics and the correct function for overloaded functions + position: common.AbsolutePosition, + value_decl: ^ast.Value_Decl, + field_name: string, } -tokenizer_error_handler :: proc(pos: tokenizer.Pos, msg: string, args: ..any) { +make_ast_context :: proc (file: ast.File, imports: []Package, package_name: string, allocator := context.temp_allocator) -> AstContext { + + ast_context := AstContext { + locals = make(map[string][dynamic]DocumentLocal, 0, allocator), + globals = make(map[string]^ast.Expr, 0, allocator), + variables = make(map[string]bool, 0, allocator), + usings = make([dynamic]string, allocator), + parameters = make(map[string]bool, 0, allocator), + in_package = make(map[string]string, 0, allocator), + file = file, + imports = imports, + use_locals = true, + use_globals = true, + document_package = package_name, + current_package = package_name, + }; + return ast_context; +} +tokenizer_error_handler :: proc (pos: tokenizer.Pos, msg: string, args: ..any) { } /* - Walk through the type expression while both the call expression and specialization type are the same - */ + Walk through the type expression while both the call expression and specialization type are the same +*/ resolve_poly_spec :: proc { - resolve_poly_spec_node, - resolve_poly_spec_array, - resolve_poly_spec_dynamic_array, -}; +resolve_poly_spec_node, +resolve_poly_spec_array, +resolve_poly_spec_dynamic_array}; -resolve_poly_spec_array :: proc(ast_context: ^AstContext, call_array: $A/[]^$T, spec_array: $D/[]^$K, poly_map: ^map[string]^ast.Expr) { +resolve_poly_spec_array :: proc (ast_context: ^AstContext, call_array: $A/[]^$T, spec_array: $D/[]^$K, poly_map: ^map[string]^ast.Expr) { - if len(call_array) != len(spec_array) { - return; - } - - for elem, i in call_array { - resolve_poly_spec(ast_context, elem, spec_array[i], poly_map); - } + if len(call_array) != len(spec_array) { + return; + } + for elem, i in call_array { + resolve_poly_spec(ast_context, elem, spec_array[i], poly_map); + } } -resolve_poly_spec_dynamic_array :: proc(ast_context: ^AstContext, call_array: $A/[dynamic]^$T, spec_array: $D/[dynamic]^$K, poly_map: ^map[string]^ast.Expr) { +resolve_poly_spec_dynamic_array :: proc (ast_context: ^AstContext, call_array: $A/[dynamic]^$T, spec_array: $D/[dynamic]^$K, poly_map: ^map[string]^ast.Expr) { - if len(call_array) != len(spec_array) { - return; - } - - for elem, i in call_array { - resolve_poly_spec(ast_context, elem, spec_array[i], poly_map); - } + if len(call_array) != len(spec_array) { + return; + } + for elem, i in call_array { + resolve_poly_spec(ast_context, elem, spec_array[i], poly_map); + } } -get_poly_node_to_expr :: proc(node: ^ast.Node) -> ^ast.Expr { - - using ast; +get_poly_node_to_expr :: proc (node: ^ast.Node) -> ^ast.Expr { - switch v in node.derived { - case Ident: - return cast(^Expr)node; - case: - log.warnf("Unhandled poly to node kind %v", v); - } + using ast; - return nil; -} - -resolve_poly_spec_node :: proc(ast_context: ^AstContext, call_node: ^ast.Node, spec_node: ^ast.Node, poly_map: ^map[string]^ast.Expr) { - - /* - Note(Daniel, uncertain about the switch cases being enough or too little) - */ - - using ast; - - if call_node == nil || spec_node == nil { - return; - } - - switch m in spec_node.derived { - case Bad_Expr: - case Ident: - case Implicit: - case Undef: - case Basic_Lit: - case Poly_Type: - if expr := get_poly_node_to_expr(call_node); expr != nil { - poly_map[m.type.name] = expr; - } - case Ellipsis: - if n, ok := call_node.derived.(Ellipsis); ok { - resolve_poly_spec(ast_context, n.expr, m.expr, poly_map); - } - case Tag_Expr: - if n, ok := call_node.derived.(Tag_Expr); ok { - resolve_poly_spec(ast_context, n.expr, m.expr, poly_map); - } - case Unary_Expr: - if n, ok := call_node.derived.(Unary_Expr); ok { - resolve_poly_spec(ast_context, n.expr, m.expr, poly_map); - } - case Binary_Expr: - if n, ok := call_node.derived.(Binary_Expr); ok { - resolve_poly_spec(ast_context, n.left, m.left, poly_map); - resolve_poly_spec(ast_context, n.right, m.right, poly_map); - } - case Paren_Expr: - if n, ok := call_node.derived.(Paren_Expr); ok { - resolve_poly_spec(ast_context, n.expr, m.expr, poly_map); - } - case Selector_Expr: - if n, ok := call_node.derived.(Selector_Expr); ok { - resolve_poly_spec(ast_context, n.expr, m.expr, poly_map); - resolve_poly_spec(ast_context, n.field, m.field, poly_map); - } - case Slice_Expr: - if n, ok := call_node.derived.(Slice_Expr); ok { - resolve_poly_spec(ast_context, n.expr, m.expr, poly_map); - resolve_poly_spec(ast_context, n.low, m.low, poly_map); - resolve_poly_spec(ast_context, n.high, m.high, poly_map); - } - case Distinct_Type: - if n, ok := call_node.derived.(Distinct_Type); ok { - resolve_poly_spec(ast_context, n.type, m.type, poly_map); - } - case Proc_Type: - if n, ok := call_node.derived.(Proc_Type); ok { - resolve_poly_spec(ast_context, n.params, m.params, poly_map); - resolve_poly_spec(ast_context, n.results, m.results, poly_map); - } - case Pointer_Type: - if n, ok := call_node.derived.(Pointer_Type); ok { - resolve_poly_spec(ast_context, n.elem, m.elem, poly_map); - } - case Array_Type: - if n, ok := call_node.derived.(Array_Type); ok { - resolve_poly_spec(ast_context, n.len, m.len, poly_map); - resolve_poly_spec(ast_context, n.elem, m.elem, poly_map); - } - case Dynamic_Array_Type: - if n, ok := call_node.derived.(Dynamic_Array_Type); ok { - resolve_poly_spec(ast_context, n.elem, m.elem, poly_map); - } - case Struct_Type: - if n, ok := call_node.derived.(Struct_Type); ok { - resolve_poly_spec(ast_context, n.poly_params, m.poly_params, poly_map); - resolve_poly_spec(ast_context, n.align, m.align, poly_map); - resolve_poly_spec(ast_context, n.fields, m.fields, poly_map); - } - case Field: - if n, ok := call_node.derived.(Field); ok { - resolve_poly_spec(ast_context, n.names, m.names, poly_map); - resolve_poly_spec(ast_context, n.type, m.type, poly_map); - resolve_poly_spec(ast_context, n.default_value, m.default_value, poly_map); - } - case Field_List: - if n, ok := call_node.derived.(Field_List); ok { - resolve_poly_spec(ast_context, n.list, m.list, poly_map); - } - case Field_Value: - if n, ok := call_node.derived.(Field_Value); ok { - resolve_poly_spec(ast_context, n.field, m.field, poly_map); - resolve_poly_spec(ast_context, n.value, m.value, poly_map); - } - case Union_Type: - if n, ok := call_node.derived.(Union_Type); ok { - resolve_poly_spec(ast_context, n.poly_params, m.poly_params, poly_map); - resolve_poly_spec(ast_context, n.align, m.align, poly_map); - resolve_poly_spec(ast_context, n.variants, m.variants, poly_map); - } - case Enum_Type: - if n, ok := call_node.derived.(Enum_Type); ok { - resolve_poly_spec(ast_context, n.base_type, m.base_type, poly_map); - resolve_poly_spec(ast_context, n.fields, m.fields, poly_map); - } - case Bit_Set_Type: - if n, ok := call_node.derived.(Bit_Set_Type); ok { - resolve_poly_spec(ast_context, n.elem, m.elem, poly_map); - resolve_poly_spec(ast_context, n.underlying, m.underlying, poly_map); - } - case Map_Type: - if n, ok := call_node.derived.(Map_Type); ok { - resolve_poly_spec(ast_context, n.key, m.key, poly_map); - resolve_poly_spec(ast_context, n.value, m.value, poly_map); - } - case Call_Expr: - if n, ok := call_node.derived.(Call_Expr); ok { - resolve_poly_spec(ast_context, n.expr, m.expr, poly_map); - resolve_poly_spec(ast_context, n.args, m.args, poly_map); - } - case Typeid_Type: - if n, ok := call_node.derived.(Typeid_Type); ok { - resolve_poly_spec(ast_context, n.specialization, m.specialization, poly_map); - } - case: - log.error("Unhandled poly node kind: %T", m); - } + switch v in node.derived { + case Ident: + return cast(^Expr)node; + case: + log.warnf("Unhandled poly to node kind %v", v); + } + return nil; } -resolve_type_comp_literal :: proc(ast_context: ^AstContext, position_context: ^DocumentPositionContext, current_symbol: index.Symbol, current_comp_lit: ^ast.Comp_Lit) -> (index.Symbol, bool) { - - if position_context.comp_lit == current_comp_lit { - return current_symbol, true; - } +resolve_poly_spec_node :: proc (ast_context: ^AstContext, call_node: ^ast.Node, spec_node: ^ast.Node, poly_map: ^map[string]^ast.Expr) { - for elem in current_comp_lit.elems { + /* + Note(Daniel, uncertain about the switch cases being enough or too little) + */ - if !position_in_node(elem, position_context.position) { - continue; - } + using ast; - if field_value, ok := elem.derived.(ast.Field_Value); ok { + if call_node == nil || spec_node == nil { + return; + } - if comp_lit, ok := field_value.value.derived.(ast.Comp_Lit); ok { + switch m in spec_node.derived { + case Bad_Expr: + case Ident: + case Implicit: + case Undef: + case Basic_Lit: + case Poly_Type: + if expr := get_poly_node_to_expr(call_node); expr != nil { + poly_map[m.type.name] = expr; + } + case Ellipsis: + if n, ok := call_node.derived.(Ellipsis); ok { + resolve_poly_spec(ast_context, n.expr, m.expr, poly_map); + } + case Tag_Expr: + if n, ok := call_node.derived.(Tag_Expr); ok { + resolve_poly_spec(ast_context, n.expr, m.expr, poly_map); + } + case Unary_Expr: + if n, ok := call_node.derived.(Unary_Expr); ok { + resolve_poly_spec(ast_context, n.expr, m.expr, poly_map); + } + case Binary_Expr: + if n, ok := call_node.derived.(Binary_Expr); ok { + resolve_poly_spec(ast_context, n.left, m.left, poly_map); + resolve_poly_spec(ast_context, n.right, m.right, poly_map); + } + case Paren_Expr: + if n, ok := call_node.derived.(Paren_Expr); ok { + resolve_poly_spec(ast_context, n.expr, m.expr, poly_map); + } + case Selector_Expr: + if n, ok := call_node.derived.(Selector_Expr); ok { + resolve_poly_spec(ast_context, n.expr, m.expr, poly_map); + resolve_poly_spec(ast_context, n.field, m.field, poly_map); + } + case Slice_Expr: + if n, ok := call_node.derived.(Slice_Expr); ok { + resolve_poly_spec(ast_context, n.expr, m.expr, poly_map); + resolve_poly_spec(ast_context, n.low, m.low, poly_map); + resolve_poly_spec(ast_context, n.high, m.high, poly_map); + } + case Distinct_Type: + if n, ok := call_node.derived.(Distinct_Type); ok { + resolve_poly_spec(ast_context, n.type, m.type, poly_map); + } + case Proc_Type: + if n, ok := call_node.derived.(Proc_Type); ok { + resolve_poly_spec(ast_context, n.params, m.params, poly_map); + resolve_poly_spec(ast_context, n.results, m.results, poly_map); + } + case Pointer_Type: + if n, ok := call_node.derived.(Pointer_Type); ok { + resolve_poly_spec(ast_context, n.elem, m.elem, poly_map); + } + case Array_Type: + if n, ok := call_node.derived.(Array_Type); ok { + resolve_poly_spec(ast_context, n.len, m.len, poly_map); + resolve_poly_spec(ast_context, n.elem, m.elem, poly_map); + } + case Dynamic_Array_Type: + if n, ok := call_node.derived.(Dynamic_Array_Type); ok { + resolve_poly_spec(ast_context, n.elem, m.elem, poly_map); + } + case Struct_Type: + if n, ok := call_node.derived.(Struct_Type); ok { + resolve_poly_spec(ast_context, n.poly_params, m.poly_params, poly_map); + resolve_poly_spec(ast_context, n.align, m.align, poly_map); + resolve_poly_spec(ast_context, n.fields, m.fields, poly_map); + } + case Field: + if n, ok := call_node.derived.(Field); ok { + resolve_poly_spec(ast_context, n.names, m.names, poly_map); + resolve_poly_spec(ast_context, n.type, m.type, poly_map); + resolve_poly_spec(ast_context, n.default_value, m.default_value, poly_map); + } + case Field_List: + if n, ok := call_node.derived.(Field_List); ok { + resolve_poly_spec(ast_context, n.list, m.list, poly_map); + } + case Field_Value: + if n, ok := call_node.derived.(Field_Value); ok { + resolve_poly_spec(ast_context, n.field, m.field, poly_map); + resolve_poly_spec(ast_context, n.value, m.value, poly_map); + } + case Union_Type: + if n, ok := call_node.derived.(Union_Type); ok { + resolve_poly_spec(ast_context, n.poly_params, m.poly_params, poly_map); + resolve_poly_spec(ast_context, n.align, m.align, poly_map); + resolve_poly_spec(ast_context, n.variants, m.variants, poly_map); + } + case Enum_Type: + if n, ok := call_node.derived.(Enum_Type); ok { + resolve_poly_spec(ast_context, n.base_type, m.base_type, poly_map); + resolve_poly_spec(ast_context, n.fields, m.fields, poly_map); + } + case Bit_Set_Type: + if n, ok := call_node.derived.(Bit_Set_Type); ok { + resolve_poly_spec(ast_context, n.elem, m.elem, poly_map); + resolve_poly_spec(ast_context, n.underlying, m.underlying, poly_map); + } + case Map_Type: + if n, ok := call_node.derived.(Map_Type); ok { + resolve_poly_spec(ast_context, n.key, m.key, poly_map); + resolve_poly_spec(ast_context, n.value, m.value, poly_map); + } + case Call_Expr: + if n, ok := call_node.derived.(Call_Expr); ok { + resolve_poly_spec(ast_context, n.expr, m.expr, poly_map); + resolve_poly_spec(ast_context, n.args, m.args, poly_map); + } + case Typeid_Type: + if n, ok := call_node.derived.(Typeid_Type); ok { + resolve_poly_spec(ast_context, n.specialization, m.specialization, poly_map); + } + case: + log.error("Unhandled poly node kind: %T", m); + } +} - if s, ok := current_symbol.value.(index.SymbolStructValue); ok { +resolve_type_comp_literal :: proc (ast_context: ^AstContext, position_context: ^DocumentPositionContext, current_symbol: index.Symbol, current_comp_lit: ^ast.Comp_Lit) -> (index.Symbol, bool) { - for name, i in s.names { + if position_context.comp_lit == current_comp_lit { + return current_symbol, true; + } - if name == field_value.field.derived.(ast.Ident).name { + for elem in current_comp_lit.elems { - if symbol, ok := resolve_type_expression(ast_context, s.types[i]); ok { - return resolve_type_comp_literal(ast_context, position_context, symbol, cast(^ast.Comp_Lit)field_value.value); - } + if !position_in_node(elem, position_context.position) { + continue; + } - } + if field_value, ok := elem.derived.(ast.Field_Value); ok { - } + if comp_lit, ok := field_value.value.derived.(ast.Comp_Lit); ok { - } + if s, ok := current_symbol.value.(index.SymbolStructValue); ok { - } + for name, i in s.names { - } + if name == field_value.field.derived.(ast.Ident).name { - } + if symbol, ok := resolve_type_expression(ast_context, s.types[i]); ok { + return resolve_type_comp_literal(ast_context, position_context, symbol, cast(^ast.Comp_Lit)field_value.value); + } + } + } + } + } + } + } - return current_symbol, true; + return current_symbol, true; } resolve_generic_function :: proc { - resolve_generic_function_ast, - resolve_generic_function_symbol, -}; - -resolve_generic_function_symbol :: proc(ast_context: ^AstContext, params: []^ast.Field, results: []^ast.Field) -> (index.Symbol, bool) { - using ast; - - if params == nil { - return index.Symbol {}, false; - } - - if results == nil { - return index.Symbol {}, false; - } - - if ast_context.call == nil { - return index.Symbol {}, false; - } - - call_expr := ast_context.call; - poly_map := make(map[string]^Expr, 0, context.temp_allocator); - i := 0; - +resolve_generic_function_ast, +resolve_generic_function_symbol}; - for param in params { +resolve_generic_function_symbol :: proc (ast_context: ^AstContext, params: []^ast.Field, results: []^ast.Field) -> (index.Symbol, bool) { + using ast; - for name in param.names { - - if len(call_expr.args) <= i { - break; - } - - if poly, ok := name.derived.(Poly_Type); ok { - poly_map[poly.type.name] = call_expr.args[i]; - } - - if param.type == nil { - continue; - } - - if poly, ok := param.type.derived.(Poly_Type); ok { - - if arg_eval, ok := resolve_type_expression(ast_context, call_expr.args[i]); ok { + if params == nil { + return index.Symbol {}, false; + } - if value, ok := arg_eval.value.(index.SymbolGenericValue); ok { - resolve_poly_spec_node(ast_context, value.expr, poly.specialization, &poly_map); - } + if results == nil { + return index.Symbol {}, false; + } - } - } + if ast_context.call == nil { + return index.Symbol {}, false; + } - i += 1; - } + call_expr := ast_context.call; + poly_map := make(map[string]^Expr, 0, context.temp_allocator); + i := 0; - } + for param in params { - function_name := ""; - function_range: common.Range; + for name in param.names { - if ident, ok := call_expr.expr.derived.(Ident); ok { - function_name = ident.name; - function_range = common.get_token_range(ident, ast_context.file.src); - } + if len(call_expr.args) <= i { + break; + } - else if selector, ok := call_expr.expr.derived.(Selector_Expr); ok { - function_name = selector.field.name; - function_range = common.get_token_range(selector, ast_context.file.src); - } + if poly, ok := name.derived.(Poly_Type); ok { + poly_map[poly.type.name] = call_expr.args[i]; + } - else { - log.debug("call expr expr could not be derived correctly"); - return index.Symbol {}, false; - } + if param.type == nil { + continue; + } - symbol := index.Symbol { - range = function_range, - type = .Function, - name = function_name, - }; + if poly, ok := param.type.derived.(Poly_Type); ok { - return_types := make([dynamic] ^ast.Field, context.temp_allocator); + if arg_eval, ok := resolve_type_expression(ast_context, call_expr.args[i]); ok { - for result in results { + if value, ok := arg_eval.value.(index.SymbolGenericValue); ok { + resolve_poly_spec_node(ast_context, value.expr, poly.specialization, &poly_map); + } + } + } - if result.type == nil { - continue; - } + i += 1; + } + } - if ident, ok := result.type.derived.(Ident); ok { - field := cast(^Field)index.clone_node(result, context.temp_allocator, nil); + function_name := ""; + function_range: common.Range; + + if ident, ok := call_expr.expr.derived.(Ident); ok { + function_name = ident.name; + function_range = common.get_token_range(ident, ast_context.file.src); + } else if selector, ok := call_expr.expr.derived.(Selector_Expr); ok { + function_name = selector.field.name; + function_range = common.get_token_range(selector, ast_context.file.src); + } else { + log.debug("call expr expr could not be derived correctly"); + return index.Symbol {}, false; + } - if m := &poly_map[ident.name]; m != nil { - field.type = poly_map[ident.name]; - append(&return_types, field); - } + symbol := index.Symbol { + range = function_range, + type = .Function, + name = function_name, + }; - else{ - return index.Symbol {}, false; - } + return_types := make([dynamic]^ast.Field, context.temp_allocator); - } + for result in results { - } + if result.type == nil { + continue; + } + if ident, ok := result.type.derived.(Ident); ok { + field := cast(^Field)index.clone_node(result, context.temp_allocator, nil); - symbol.value = index.SymbolProcedureValue { - return_types = return_types[:], - arg_types = params, - }; + if m := &poly_map[ident.name]; m != nil { + field.type = poly_map[ident.name]; + append(&return_types, field); + } else { + return index.Symbol {}, false; + } + } + } + symbol.value = index.SymbolProcedureValue { + return_types = return_types[:], + arg_types = params, + }; - //log.infof("return %v", poly_map); + //log.infof("return %v", poly_map); - return symbol, true; + return symbol, true; } -resolve_generic_function_ast :: proc(ast_context: ^AstContext, proc_lit: ast.Proc_Lit) -> (index.Symbol, bool) { +resolve_generic_function_ast :: proc (ast_context: ^AstContext, proc_lit: ast.Proc_Lit) -> (index.Symbol, bool) { - using ast; + using ast; - if proc_lit.type.params == nil { - return index.Symbol {}, false; - } + if proc_lit.type.params == nil { + return index.Symbol {}, false; + } - if proc_lit.type.results == nil { - return index.Symbol {}, false; - } + if proc_lit.type.results == nil { + return index.Symbol {}, false; + } - if ast_context.call == nil { - return index.Symbol {}, false; - } + if ast_context.call == nil { + return index.Symbol {}, false; + } - return resolve_generic_function_symbol(ast_context, proc_lit.type.params.list, proc_lit.type.results.list); + return resolve_generic_function_symbol(ast_context, proc_lit.type.params.list, proc_lit.type.results.list); } - /* - Figure out which function the call expression is using out of the list from proc group - */ -resolve_function_overload :: proc(ast_context: ^AstContext, group: ast.Proc_Group) -> (index.Symbol, bool) { - - using ast; - - //log.info("overload"); - - if ast_context.call == nil { - //log.info("no call"); - return index.Symbol {}, false; - } - - call_expr := ast_context.call; - - for arg_expr in group.args { + Figure out which function the call expression is using out of the list from proc group +*/ +resolve_function_overload :: proc (ast_context: ^AstContext, group: ast.Proc_Group) -> (index.Symbol, bool) { - next_fn: if f, ok := resolve_type_expression(ast_context, arg_expr); ok { + using ast; - if procedure, ok := f.value.(index.SymbolProcedureValue); ok { + //log.info("overload"); - if len(procedure.arg_types) < len(call_expr.args) { - continue; - } + if ast_context.call == nil { + //log.info("no call"); + return index.Symbol {}, false; + } - for arg, i in call_expr.args { + call_expr := ast_context.call; - if eval_call_expr, ok := resolve_type_expression(ast_context, arg); ok { + for arg_expr in group.args { - #partial switch v in eval_call_expr.value { - case index.SymbolProcedureValue: - case index.SymbolGenericValue: - if !common.node_equal(v.expr, procedure.arg_types[i].type) { - break next_fn; - } - case index.SymbolStructValue: - } + next_fn: if f, ok := resolve_type_expression(ast_context, arg_expr); ok { - } + if procedure, ok := f.value.(index.SymbolProcedureValue); ok { - else { - //log.debug("Failed to resolve call expr"); - return index.Symbol {}, false; - } - } + if len(procedure.arg_types) < len(call_expr.args) { + continue; + } - //log.debugf("return overload %v", f); - return f, true; - } + for arg, i in call_expr.args { - } + if eval_call_expr, ok := resolve_type_expression(ast_context, arg); ok { - } + #partial switch v in eval_call_expr.value { + case index.SymbolProcedureValue: + case index.SymbolGenericValue: + if !common.node_equal(v.expr, procedure.arg_types[i].type) { + break next_fn; + } + case index.SymbolStructValue: + } + } else { + //log.debug("Failed to resolve call expr"); + return index.Symbol {}, false; + } + } + //log.debugf("return overload %v", f); + return f, true; + } + } + } - return index.Symbol {}, false; + return index.Symbol {}, false; } -resolve_basic_lit :: proc(ast_context: ^AstContext, basic_lit: ast.Basic_Lit) -> (index.Symbol, bool) { +resolve_basic_lit :: proc (ast_context: ^AstContext, basic_lit: ast.Basic_Lit) -> (index.Symbol, bool) { - /* - This is temporary, since basic lit is untyped, but either way it's going to be an ident representing a keyword. + /* + This is temporary, since basic lit is untyped, but either way it's going to be an ident representing a keyword. - Could perhaps name them "$integer", "$float", etc. - */ + Could perhaps name them "$integer", "$float", etc. + */ - ident := index.new_type(ast.Ident, basic_lit.pos, basic_lit.end, context.temp_allocator); + ident := index.new_type(ast.Ident, basic_lit.pos, basic_lit.end, context.temp_allocator); - symbol := index.Symbol { - type = .Keyword, - }; - - if v, ok := strconv.parse_bool(basic_lit.tok.text); ok { - ident.name = bool_lit; - } - - else if v, ok := strconv.parse_int(basic_lit.tok.text); ok { - ident.name = int_lit; - } - - else { - ident.name = string_lit; - } + symbol := index.Symbol { + type = .Keyword + }; - symbol.value = index.SymbolGenericValue { - expr = ident, - }; + if v, ok := strconv.parse_bool(basic_lit.tok.text); ok { + ident.name = bool_lit; + } else if v, ok := strconv.parse_int(basic_lit.tok.text); ok { + ident.name = int_lit; + } else { + ident.name = string_lit; + } - return symbol, true; -} + symbol.value = index.SymbolGenericValue { + expr = ident + }; -resolve_type_expression :: proc(ast_context: ^AstContext, node: ^ast.Expr) -> (index.Symbol, bool) { - - if node == nil { - return {}, false; - } - - using ast; - - switch v in node.derived { - case Proc_Type: - return make_symbol_procedure_from_ast(ast_context, node, v, ast_context.field_name), true; - case Ident: - return resolve_type_identifier(ast_context, v); - case Basic_Lit: - return resolve_basic_lit(ast_context, v); - case Type_Cast: - return resolve_type_expression(ast_context, v.type); - case Auto_Cast: - return resolve_type_expression(ast_context, v.expr); - case Unary_Expr: - if v.op.kind == .And { - return resolve_type_expression(ast_context, make_pointer_ast(v.expr)); - } - - else { - return resolve_type_expression(ast_context, v.expr); - } - case Deref_Expr: - return resolve_type_expression(ast_context, v.expr); - case Paren_Expr: - return resolve_type_expression(ast_context, v.expr); - case Slice_Expr: - return resolve_type_expression(ast_context, v.expr); - case Tag_Expr: - return resolve_type_expression(ast_context, v.expr); - case Helper_Type: - return resolve_type_expression(ast_context, v.type); - case Ellipsis: - return resolve_type_expression(ast_context, v.expr); - case Implicit: - ident := index.new_type(Ident, v.node.pos, v.node.end, context.temp_allocator); - ident.name = v.tok.text; - return resolve_type_identifier(ast_context, ident^); - case Type_Assertion: - return resolve_type_expression(ast_context, v.type); - case Proc_Lit: - if v.type.results != nil { - if len(v.type.results.list) == 1 { - return resolve_type_expression(ast_context, v.type.results.list[0].type); - } - } - case Pointer_Type: - - /* - Add flag to not pull out a type from a pointer for function overloading. - */ - - if v2, ok := v.elem.derived.(ast.Pointer_Type); !ok { - return resolve_type_expression(ast_context, v.elem); - } - - else if v2, ok := v.elem.derived.(ast.Type_Assertion); !ok { - return resolve_type_expression(ast_context, v.elem); - } - - else { - return make_symbol_generic_from_ast(ast_context, node), true; - } - - case Index_Expr: - indexed, ok := resolve_type_expression(ast_context, v.expr); - - if generic, ok := indexed.value.(index.SymbolGenericValue); ok { - - switch c in generic.expr.derived { - case Array_Type: - return resolve_type_expression(ast_context, c.elem); - case Dynamic_Array_Type: - return resolve_type_expression(ast_context, c.elem); - case Map_Type: - return resolve_type_expression(ast_context, c.value); - } - - } - - return index.Symbol {}, false; - case Call_Expr: - ast_context.call = cast(^Call_Expr)node; - return resolve_type_expression(ast_context, v.expr); - case Implicit_Selector_Expr: - return index.Symbol {}, false; - case Selector_Call_Expr: - return resolve_type_expression(ast_context, v.expr); - case Selector_Expr: - - if selector, ok := resolve_type_expression(ast_context, v.expr); ok { - - ast_context.use_locals = false; - - #partial switch s in selector.value { - case index.SymbolProcedureValue: - - if len(s.return_types) == 1 { - selector_expr := index.new_type(ast.Selector_Expr, s.return_types[0].node.pos, s.return_types[0].node.end, context.temp_allocator); - selector_expr.expr = s.return_types[0].type; - selector_expr.field = v.field; - return resolve_type_expression(ast_context, selector_expr); - } - case index.SymbolStructValue: - if selector.pkg != "" { - ast_context.current_package = selector.pkg; - } - - else { - ast_context.current_package = ast_context.document_package; - } - - for name, i in s.names { - if v.field != nil && name == v.field.name { - ast_context.field_name = v.field.name; - return resolve_type_expression(ast_context, s.types[i]); - } - } - case index.SymbolPackageValue: - - ast_context.current_package = selector.pkg; - - if v.field != nil { - return resolve_symbol_return(ast_context, index.lookup(v.field.name, selector.pkg)); - } - - else { - return index.Symbol {}, false; - } - } - } - - else { - return index.Symbol {}, false; - } - case: - log.warnf("default node kind, resolve_type_expression: %T", v); - - if v == nil { - return {}, false; - } - - return make_symbol_generic_from_ast(ast_context, node), true; - } - - return index.Symbol {}, false; + return symbol, true; } -store_local :: proc(ast_context: ^AstContext, expr: ^ast.Expr, offset: int, name: string) { +resolve_type_expression :: proc (ast_context: ^AstContext, node: ^ast.Expr) -> (index.Symbol, bool) { - local_stack := &ast_context.locals[name]; + if node == nil { + return {}, false; + } - if local_stack == nil { - ast_context.locals[name] = make([dynamic] DocumentLocal, context.temp_allocator); - local_stack = &ast_context.locals[name]; - } + using ast; + + switch v in node.derived { + case Proc_Type: + return make_symbol_procedure_from_ast(ast_context, node, v, ast_context.field_name), true; + case Ident: + return resolve_type_identifier(ast_context, v); + case Basic_Lit: + return resolve_basic_lit(ast_context, v); + case Type_Cast: + return resolve_type_expression(ast_context, v.type); + case Auto_Cast: + return resolve_type_expression(ast_context, v.expr); + case Unary_Expr: + if v.op.kind == .And { + return resolve_type_expression(ast_context, make_pointer_ast(v.expr)); + } else { + return resolve_type_expression(ast_context, v.expr); + } + case Deref_Expr: + return resolve_type_expression(ast_context, v.expr); + case Paren_Expr: + return resolve_type_expression(ast_context, v.expr); + case Slice_Expr: + return resolve_type_expression(ast_context, v.expr); + case Tag_Expr: + return resolve_type_expression(ast_context, v.expr); + case Helper_Type: + return resolve_type_expression(ast_context, v.type); + case Ellipsis: + return resolve_type_expression(ast_context, v.expr); + case Implicit: + ident := index.new_type(Ident, v.node.pos, v.node.end, context.temp_allocator); + ident.name = v.tok.text; + return resolve_type_identifier(ast_context, ident^); + case Type_Assertion: + return resolve_type_expression(ast_context, v.type); + case Proc_Lit: + if v.type.results != nil { + if len(v.type.results.list) == 1 { + return resolve_type_expression(ast_context, v.type.results.list[0].type); + } + } + case Pointer_Type: + + /* + Add flag to not pull out a type from a pointer for function overloading. + */ + + if v2, ok := v.elem.derived.(ast.Pointer_Type); !ok { + return resolve_type_expression(ast_context, v.elem); + } else if v2, ok := v.elem.derived.(ast.Type_Assertion); !ok { + return resolve_type_expression(ast_context, v.elem); + } else { + return make_symbol_generic_from_ast(ast_context, node), true; + } + + case Index_Expr: + indexed, ok := resolve_type_expression(ast_context, v.expr); + + if generic, ok := indexed.value.(index.SymbolGenericValue); ok { + + switch c in generic.expr.derived { + case Array_Type: + return resolve_type_expression(ast_context, c.elem); + case Dynamic_Array_Type: + return resolve_type_expression(ast_context, c.elem); + case Map_Type: + return resolve_type_expression(ast_context, c.value); + } + } + + return index.Symbol {}, false; + case Call_Expr: + ast_context.call = cast(^Call_Expr)node; + return resolve_type_expression(ast_context, v.expr); + case Implicit_Selector_Expr: + return index.Symbol {}, false; + case Selector_Call_Expr: + return resolve_type_expression(ast_context, v.expr); + case Selector_Expr: + + if selector, ok := resolve_type_expression(ast_context, v.expr); ok { + + ast_context.use_locals = false; + + #partial switch s in selector.value { + case index.SymbolProcedureValue: + + if len(s.return_types) == 1 { + selector_expr := index.new_type(ast.Selector_Expr, s.return_types[0].node.pos, s.return_types[0].node.end, context.temp_allocator); + selector_expr.expr = s.return_types[0].type; + selector_expr.field = v.field; + return resolve_type_expression(ast_context, selector_expr); + } + case index.SymbolStructValue: + if selector.pkg != "" { + ast_context.current_package = selector.pkg; + } else { + ast_context.current_package = ast_context.document_package; + } + + for name, i in s.names { + if v.field != nil && name == v.field.name { + ast_context.field_name = v.field.name; + return resolve_type_expression(ast_context, s.types[i]); + } + } + case index.SymbolPackageValue: + + ast_context.current_package = selector.pkg; + + if v.field != nil { + return resolve_symbol_return(ast_context, index.lookup(v.field.name, selector.pkg)); + } else { + return index.Symbol {}, false; + } + } + } else { + return index.Symbol {}, false; + } + case: + log.warnf("default node kind, resolve_type_expression: %T", v); + + if v == nil { + return {}, false; + } + + return make_symbol_generic_from_ast(ast_context, node), true; + } - append(local_stack, DocumentLocal { expr = expr, offset = offset }); + return index.Symbol {}, false; } -get_local :: proc(ast_context: ^AstContext, offset: int, name: string) -> ^ast.Expr { - - previous := 0; +store_local :: proc (ast_context: ^AstContext, expr: ^ast.Expr, offset: int, name: string) { - //is the local we are getting being declared? - if ast_context.value_decl != nil { + local_stack := &ast_context.locals[name]; - for value_decl_name in ast_context.value_decl.names { + if local_stack == nil { + ast_context.locals[name] = make([dynamic]DocumentLocal, context.temp_allocator); + local_stack = &ast_context.locals[name]; + } - if ident, ok := value_decl_name.derived.(ast.Ident); ok { + append(local_stack, DocumentLocal {expr = expr, offset = offset}); +} - if ident.name == name { - previous = 1; - break; - } - } +get_local :: proc (ast_context: ^AstContext, offset: int, name: string) -> ^ast.Expr { - } + previous := 0; + //is the local we are getting being declared? + if ast_context.value_decl != nil { - } + for value_decl_name in ast_context.value_decl.names { - if local_stack, ok := ast_context.locals[name]; ok { + if ident, ok := value_decl_name.derived.(ast.Ident); ok { - for i := len(local_stack)-1; i >= 0; i -= 1 { + if ident.name == name { + previous = 1; + break; + } + } + } + } - if local_stack[i].offset <= offset { - return local_stack[max(0, i - previous)].expr; - } + if local_stack, ok := ast_context.locals[name]; ok { - } + for i := len(local_stack) - 1; i >= 0; i -= 1 { - } + if local_stack[i].offset <= offset { + return local_stack[max(0, i - previous)].expr; + } + } + } - return nil; + return nil; } /* - Function recusively goes through the identifier until it hits a struct, enum, procedure literals, since you can - have chained variable declarations. ie. a := foo { test = 2}; b := a; c := b; - */ -resolve_type_identifier :: proc(ast_context: ^AstContext, node: ast.Ident) -> (index.Symbol, bool) { - - using ast; - - if pkg, ok := ast_context.in_package[node.name]; ok { - ast_context.current_package = pkg; - } - - if _, ok := ast_context.parameters[node.name]; ok { - for imp in ast_context.imports { - - if strings.compare(imp.base, node.name) == 0 { - - symbol := index.Symbol { - type = .Package, - pkg = imp.name, - value = index.SymbolPackageValue { - } - }; - - return symbol, true; - } - - } - } - - //note(Daniel, if global and local ends up being 100% same just make a function that takes the map) - if local := get_local(ast_context, node.pos.offset, node.name); local != nil && ast_context.use_locals { - - switch v in local.derived { - case Ident: - - if node.name == v.name { - break; - } - - return resolve_type_identifier(ast_context, v); - case Union_Type: - return make_symbol_union_from_ast(ast_context, v, node), true; - case Enum_Type: - return make_symbol_enum_from_ast(ast_context, v, node), true; - case Struct_Type: - return make_symbol_struct_from_ast(ast_context, v, node), true; - case Bit_Set_Type: - return make_symbol_bitset_from_ast(ast_context, v, node), true; - case Proc_Lit: - if !v.type.generic { - return make_symbol_procedure_from_ast(ast_context, local, v.type^, node.name), true; - } - else { - return resolve_generic_function(ast_context, v); - } - case Proc_Group: - return resolve_function_overload(ast_context, v); - case Array_Type: - return make_symbol_generic_from_ast(ast_context, local), true; - case Dynamic_Array_Type: - return make_symbol_generic_from_ast(ast_context, local), true; - case Call_Expr: - return resolve_type_expression(ast_context, local); - case: - log.warnf("default type node kind: %T", v); - return resolve_type_expression(ast_context, local); - //return make_symbol_generic_from_ast(ast_context, local), true; - } - } - - else if global, ok := ast_context.globals[node.name]; ast_context.use_globals && ok { - - switch v in global.derived { - case Ident: - - if node.name == v.name { - break; - } - - return resolve_type_identifier(ast_context, v); - case Struct_Type: - return make_symbol_struct_from_ast(ast_context, v, node), true; - case Bit_Set_Type: - return make_symbol_bitset_from_ast(ast_context, v, node), true; - case Union_Type: - return make_symbol_union_from_ast(ast_context, v, node), true; - case Enum_Type: - return make_symbol_enum_from_ast(ast_context, v, node), true; - case Proc_Lit: - if !v.type.generic { - return make_symbol_procedure_from_ast(ast_context, global, v.type^, node.name), true; - } - else { - return resolve_generic_function(ast_context, v); - } - case Proc_Group: - return resolve_function_overload(ast_context, v); - case Array_Type: - return make_symbol_generic_from_ast(ast_context, global), true; - case Dynamic_Array_Type: - return make_symbol_generic_from_ast(ast_context, global), true; - case Call_Expr: - return resolve_type_expression(ast_context, global); - case: - log.warnf("default type node kind: %T", v); - return resolve_type_expression(ast_context, global); - } - - } - - //if there are more of these variables that hard builtin, move them to the indexer - else if node.name == "context" { - return index.lookup("Context", ast_context.current_package); - } - //keywords - else if v, ok := common.keyword_map[node.name]; ok { - - ident := index.new_type(Ident, node.pos, node.end, context.temp_allocator); - ident.name = node.name; - - symbol := index.Symbol { - type = .Keyword, - signature = node.name, - pkg = ast_context.current_package, - value = index.SymbolGenericValue { - expr = ident, - }, - }; - - return symbol, true; - } - - else { - - //right now we replace the package ident with the absolute directory name, so it should have '/' which is not a valid ident character - if strings.contains(node.name, "/") { - - symbol := index.Symbol { - type = .Package, - pkg = node.name, - value = index.SymbolPackageValue { - } - }; - - return symbol, true; - - } - - //part of the ast so we check the imports of the document - else { - - for imp in ast_context.imports { - - if strings.compare(imp.base, node.name) == 0 { - - symbol := index.Symbol { - type = .Package, - pkg = imp.name, - value = index.SymbolPackageValue { - } - }; - - return symbol, true; - } - - } - - } - - //last option is to check the index - - if symbol, ok := index.lookup(node.name, ast_context.current_package); ok { - return resolve_symbol_return(ast_context, symbol); - } - - for u in ast_context.usings { - - //TODO(Daniel, make into a map, not really required for performance but looks nicer) - for imp in ast_context.imports { - - if strings.compare(imp.base, u) == 0 { - - if symbol, ok := index.lookup(node.name, imp.name); ok { - return resolve_symbol_return(ast_context, symbol); - } - } - - } - } - - //TODO(daniel, index can be used on identifiers if using is in the function scope) - } - - return index.Symbol {}, false; -} + Function recusively goes through the identifier until it hits a struct, enum, procedure literals, since you can + have chained variable declarations. ie. a := foo { test = 2}; b := a; c := b; +*/ +resolve_type_identifier :: proc (ast_context: ^AstContext, node: ast.Ident) -> (index.Symbol, bool) { -resolve_ident_is_variable :: proc(ast_context: ^AstContext, node: ast.Ident) -> bool { + using ast; - if v, ok := ast_context.variables[node.name]; ok && v { - return true; - } + if pkg, ok := ast_context.in_package[node.name]; ok { + ast_context.current_package = pkg; + } - if symbol, ok := index.lookup(node.name, ast_context.current_package); ok { - return symbol.type == .Variable; - } + if _, ok := ast_context.parameters[node.name]; ok { + for imp in ast_context.imports { - return false; -} + if strings.compare(imp.base, node.name) == 0 { -resolve_ident_is_package :: proc(ast_context: ^AstContext, node: ast.Ident) -> bool { + symbol := index.Symbol { + type = .Package, + pkg = imp.name, + value = index.SymbolPackageValue {}, + }; - if strings.contains(node.name, "/") { - return true; - } + return symbol, true; + } + } + } - else { + //note(Daniel, if global and local ends up being 100% same just make a function that takes the map) + if local := get_local(ast_context, node.pos.offset, node.name); local != nil && ast_context.use_locals { + + switch v in local.derived { + case Ident: + + if node.name == v.name { + break; + } + + return resolve_type_identifier(ast_context, v); + case Union_Type: + return make_symbol_union_from_ast(ast_context, v, node), true; + case Enum_Type: + return make_symbol_enum_from_ast(ast_context, v, node), true; + case Struct_Type: + return make_symbol_struct_from_ast(ast_context, v, node), true; + case Bit_Set_Type: + return make_symbol_bitset_from_ast(ast_context, v, node), true; + case Proc_Lit: + if !v.type.generic { + return make_symbol_procedure_from_ast(ast_context, local, v.type^, node.name), true; + } else { + return resolve_generic_function(ast_context, v); + } + case Proc_Group: + return resolve_function_overload(ast_context, v); + case Array_Type: + return make_symbol_generic_from_ast(ast_context, local), true; + case Dynamic_Array_Type: + return make_symbol_generic_from_ast(ast_context, local), true; + case Call_Expr: + return resolve_type_expression(ast_context, local); + case: + log.warnf("default type node kind: %T", v); + return resolve_type_expression(ast_context, local); + //return make_symbol_generic_from_ast(ast_context, local), true; + } + } else if global, ok := ast_context.globals[node.name]; ast_context.use_globals && ok { + + switch v in global.derived { + case Ident: + + if node.name == v.name { + break; + } + + return resolve_type_identifier(ast_context, v); + case Struct_Type: + return make_symbol_struct_from_ast(ast_context, v, node), true; + case Bit_Set_Type: + return make_symbol_bitset_from_ast(ast_context, v, node), true; + case Union_Type: + return make_symbol_union_from_ast(ast_context, v, node), true; + case Enum_Type: + return make_symbol_enum_from_ast(ast_context, v, node), true; + case Proc_Lit: + if !v.type.generic { + return make_symbol_procedure_from_ast(ast_context, global, v.type^, node.name), true; + } else { + return resolve_generic_function(ast_context, v); + } + case Proc_Group: + return resolve_function_overload(ast_context, v); + case Array_Type: + return make_symbol_generic_from_ast(ast_context, global), true; + case Dynamic_Array_Type: + return make_symbol_generic_from_ast(ast_context, global), true; + case Call_Expr: + return resolve_type_expression(ast_context, global); + case: + log.warnf("default type node kind: %T", v); + return resolve_type_expression(ast_context, global); + } + } else + + //if there are more of these variables that hard builtin, move them to the indexer + if node.name == "context" { + return index.lookup("Context", ast_context.current_package); + } else + //keywords + if v, ok := common.keyword_map[node.name]; ok { + + ident := index.new_type(Ident, node.pos, node.end, context.temp_allocator); + ident.name = node.name; + + symbol := index.Symbol { + type = .Keyword, + signature = node.name, + pkg = ast_context.current_package, + value = index.SymbolGenericValue { + expr = ident + }, + }; + + return symbol, true; + } else { + + //right now we replace the package ident with the absolute directory name, so it should have '/' which is not a valid ident character + if strings.contains(node.name, "/") { + + symbol := index.Symbol { + type = .Package, + pkg = node.name, + value = index.SymbolPackageValue {}, + }; + + return symbol, true; + } else + + //part of the ast so we check the imports of the document + { + + for imp in ast_context.imports { + + if strings.compare(imp.base, node.name) == 0 { + + symbol := index.Symbol { + type = .Package, + pkg = imp.name, + value = index.SymbolPackageValue {}, + }; + + return symbol, true; + } + } + } + + //last option is to check the index + + if symbol, ok := index.lookup(node.name, ast_context.current_package); ok { + return resolve_symbol_return(ast_context, symbol); + } + + for u in ast_context.usings { + + //TODO(Daniel, make into a map, not really required for performance but looks nicer) + for imp in ast_context.imports { + + if strings.compare(imp.base, u) == 0 { + + if symbol, ok := index.lookup(node.name, imp.name); ok { + return resolve_symbol_return(ast_context, symbol); + } + } + } + } + + //TODO(daniel, index can be used on identifiers if using is in the function scope) + } - for imp in ast_context.imports { + return index.Symbol {}, false; +} - if imp.base == node.name { - return true; - } +resolve_ident_is_variable :: proc (ast_context: ^AstContext, node: ast.Ident) -> bool { - } + if v, ok := ast_context.variables[node.name]; ok && v { + return true; + } - } + if symbol, ok := index.lookup(node.name, ast_context.current_package); ok { + return symbol.type == .Variable; + } - return false; + return false; } -expand_struct_usings :: proc(ast_context: ^AstContext, symbol: index.Symbol, value: index.SymbolStructValue) -> index.SymbolStructValue { +resolve_ident_is_package :: proc (ast_context: ^AstContext, node: ast.Ident) -> bool { - //ERROR no completion or over on names and types - generic resolve error - names := slice.to_dynamic(value.names, context.temp_allocator); - types := slice.to_dynamic(value.types, context.temp_allocator); + if strings.contains(node.name, "/") { + return true; + } else { - //ERROR no hover on k and v(completion works) - for k, v in value.usings { + for imp in ast_context.imports { - ast_context.current_package = symbol.pkg; + if imp.base == node.name { + return true; + } + } + } - field_expr: ^ast.Expr; + return false; +} - for name, i in value.names { +expand_struct_usings :: proc (ast_context: ^AstContext, symbol: index.Symbol, value: index.SymbolStructValue) -> index.SymbolStructValue { - if name == k && v { - field_expr = value.types[i]; - } + //ERROR no completion or over on names and types - generic resolve error + names := slice.to_dynamic(value.names, context.temp_allocator); + types := slice.to_dynamic(value.types, context.temp_allocator); - } + //ERROR no hover on k and v(completion works) + for k, v in value.usings { - if field_expr == nil { - continue; - } + ast_context.current_package = symbol.pkg; - if s, ok := resolve_type_expression(ast_context, field_expr); ok { + field_expr: ^ast.Expr; - if struct_value, ok := s.value.(index.SymbolStructValue); ok { + for name, i in value.names { - for name in struct_value.names { - append(&names, name); - } + if name == k && v { + field_expr = value.types[i]; + } + } - for type in struct_value.types { - append(&types, type); - } + if field_expr == nil { + continue; + } - } + if s, ok := resolve_type_expression(ast_context, field_expr); ok { - } + if struct_value, ok := s.value.(index.SymbolStructValue); ok { - } + for name in struct_value.names { + append(&names, name); + } - return { - names = names[:], - types = types[:], - }; + for type in struct_value.types { + append(&types, type); + } + } + } + } + return { + names = names[:], + types = types[:], + }; } -resolve_symbol_return :: proc(ast_context: ^AstContext, symbol: index.Symbol, ok := true) -> (index.Symbol, bool) { - - if !ok { - return symbol, ok; - } - - #partial switch v in symbol.value { - case index.SymbolProcedureGroupValue: - if symbol, ok := resolve_function_overload(ast_context, v.group.derived.(ast.Proc_Group)); ok { - return symbol, true; - } - else { - return symbol, false; - } - case index.SymbolProcedureValue: - if v.generic { - return resolve_generic_function_symbol(ast_context, v.arg_types, v.return_types); - } - else { - return symbol, true; - } - case index.SymbolStructValue: - - //expand the types and names from the using - can't be done while indexing without complicating everything(this also saves memory) - if len(v.usings) > 0 { - expanded := symbol; - expanded.value = expand_struct_usings(ast_context, symbol, v); - return expanded, true; - } - else { - return symbol, true; - } - - case index.SymbolGenericValue: - return resolve_type_expression(ast_context, v.expr); - } - - return symbol, true; -} +resolve_symbol_return :: proc (ast_context: ^AstContext, symbol: index.Symbol, ok := true) -> (index.Symbol, bool) { -resolve_location_identifier :: proc(ast_context: ^AstContext, node: ast.Ident) -> (index.Symbol, bool) { + if !ok { + return symbol, ok; + } - symbol: index.Symbol; + #partial switch v in symbol.value { + case index.SymbolProcedureGroupValue: + if symbol, ok := resolve_function_overload(ast_context, v.group.derived.(ast.Proc_Group)); ok { + return symbol, true; + } else { + return symbol, false; + } + case index.SymbolProcedureValue: + if v.generic { + return resolve_generic_function_symbol(ast_context, v.arg_types, v.return_types); + } else { + return symbol, true; + } + case index.SymbolStructValue: + + //expand the types and names from the using - can't be done while indexing without complicating everything(this also saves memory) + if len(v.usings) > 0 { + expanded := symbol; + expanded.value = expand_struct_usings(ast_context, symbol, v); + return expanded, true; + } else { + return symbol, true; + } + + case index.SymbolGenericValue: + return resolve_type_expression(ast_context, v.expr); + } - if local := get_local(ast_context, node.pos.offset, node.name); local != nil { - symbol.range = common.get_token_range(get_local(ast_context, node.pos.offset, node.name), ast_context.file.src); - return symbol, true; - } + return symbol, true; +} - else if global, ok := ast_context.globals[node.name]; ok { - symbol.range = common.get_token_range(global, ast_context.file.src); - return symbol, true; - } +resolve_location_identifier :: proc (ast_context: ^AstContext, node: ast.Ident) -> (index.Symbol, bool) { + symbol: index.Symbol; - return index.lookup(node.name, ast_context.document_package); -} + if local := get_local(ast_context, node.pos.offset, node.name); local != nil { + symbol.range = common.get_token_range(get_local(ast_context, node.pos.offset, node.name), ast_context.file.src); + return symbol, true; + } else if global, ok := ast_context.globals[node.name]; ok { + symbol.range = common.get_token_range(global, ast_context.file.src); + return symbol, true; + } -make_pointer_ast :: proc(elem: ^ast.Expr) -> ^ast.Pointer_Type { - pointer := index.new_type(ast.Pointer_Type, elem.pos, elem.end, context.temp_allocator); - pointer.elem = elem; - return pointer; + return index.lookup(node.name, ast_context.document_package); } -make_bool_ast :: proc() -> ^ast.Ident { - ident := index.new_type(ast.Ident, {}, {}, context.temp_allocator); - ident.name = bool_lit; - return ident; +make_pointer_ast :: proc (elem: ^ast.Expr) -> ^ast.Pointer_Type { + pointer := index.new_type(ast.Pointer_Type, elem.pos, elem.end, context.temp_allocator); + pointer.elem = elem; + return pointer; } -make_int_ast :: proc() -> ^ast.Ident { - ident := index.new_type(ast.Ident, {}, {}, context.temp_allocator); - ident.name = int_lit; - return ident; +make_bool_ast :: proc () -> ^ast.Ident { + ident := index.new_type(ast.Ident, {}, {}, context.temp_allocator); + ident.name = bool_lit; + return ident; } -get_package_from_node :: proc(node: ast.Node) -> string { - slashed, _ := filepath.to_slash(node.pos.file, context.temp_allocator); - ret := strings.to_lower(path.dir(slashed, context.temp_allocator), context.temp_allocator); - return ret; +make_int_ast :: proc () -> ^ast.Ident { + ident := index.new_type(ast.Ident, {}, {}, context.temp_allocator); + ident.name = int_lit; + return ident; } -get_using_packages :: proc(ast_context: ^AstContext) -> [] string { - - usings := make([] string, len(ast_context.usings), context.temp_allocator); +get_package_from_node :: proc (node: ast.Node) -> string { + slashed, _ := filepath.to_slash(node.pos.file, context.temp_allocator); + ret := strings.to_lower(path.dir(slashed, context.temp_allocator), context.temp_allocator); + return ret; +} - if len(ast_context.usings) == 0 { - return usings; - } +get_using_packages :: proc (ast_context: ^AstContext) -> []string { - //probably map instead - for u, i in ast_context.usings { + usings := make([]string, len(ast_context.usings), context.temp_allocator); - for imp in ast_context.imports { + if len(ast_context.usings) == 0 { + return usings; + } - if strings.compare(imp.base, u) == 0 { - usings[i] = imp.name; - } + //probably map instead + for u, i in ast_context.usings { - } + for imp in ast_context.imports { - } + if strings.compare(imp.base, u) == 0 { + usings[i] = imp.name; + } + } + } - return usings; + return usings; } -make_symbol_procedure_from_ast :: proc(ast_context: ^AstContext, n: ^ast.Node, v: ast.Proc_Type, name: string) -> index.Symbol { +make_symbol_procedure_from_ast :: proc (ast_context: ^AstContext, n: ^ast.Node, v: ast.Proc_Type, name: string) -> index.Symbol { - symbol := index.Symbol { - range = common.get_token_range(n^, ast_context.file.src), - type = .Function, - pkg = get_package_from_node(n^), - }; - - symbol.name = name; + symbol := index.Symbol { + range = common.get_token_range(n^, ast_context.file.src), + type = .Function, + pkg = get_package_from_node(n^), + }; - return_types := make([dynamic] ^ast.Field, context.temp_allocator); - arg_types := make([dynamic] ^ast.Field, context.temp_allocator); + symbol.name = name; - if v.results != nil { + return_types := make([dynamic]^ast.Field, context.temp_allocator); + arg_types := make([dynamic]^ast.Field, context.temp_allocator); - for ret in v.results.list { - append(&return_types, ret); - } + if v.results != nil { - symbol.returns = strings.concatenate( {"(", string(ast_context.file.src[v.results.pos.offset:v.results.end.offset]), ")"}, context.temp_allocator); + for ret in v.results.list { + append(&return_types, ret); + } - } + symbol.returns = strings.concatenate({"(", string(ast_context.file.src[v.results.pos.offset:v.results.end.offset]), ")"}, context.temp_allocator); + } - if v.params != nil { + if v.params != nil { - for param in v.params.list { - append(&arg_types, param); - } + for param in v.params.list { + append(&arg_types, param); + } - symbol.signature = strings.concatenate( {"(", string(ast_context.file.src[v.params.pos.offset:v.params.end.offset]), ")"}, context.temp_allocator); - } + symbol.signature = strings.concatenate({"(", string(ast_context.file.src[v.params.pos.offset:v.params.end.offset]), ")"}, context.temp_allocator); + } - symbol.value = index.SymbolProcedureValue { - return_types = return_types[:], - arg_types = arg_types[:], - }; + symbol.value = index.SymbolProcedureValue { + return_types = return_types[:], + arg_types = arg_types[:], + }; - return symbol; + return symbol; } -make_symbol_generic_from_ast :: proc(ast_context: ^AstContext, expr: ^ast.Expr) -> index.Symbol { +make_symbol_generic_from_ast :: proc (ast_context: ^AstContext, expr: ^ast.Expr) -> index.Symbol { - symbol := index.Symbol { - range = common.get_token_range(expr, ast_context.file.src), - type = .Variable, - signature = index.node_to_string(expr), - pkg = get_package_from_node(expr^), - }; + symbol := index.Symbol { + range = common.get_token_range(expr, ast_context.file.src), + type = .Variable, + signature = index.node_to_string(expr), + pkg = get_package_from_node(expr^), + }; - symbol.value = index.SymbolGenericValue { - expr = expr, - }; + symbol.value = index.SymbolGenericValue { + expr = expr + }; - return symbol; + return symbol; } -make_symbol_union_from_ast :: proc(ast_context: ^AstContext, v: ast.Union_Type, ident: ast.Ident) -> index.Symbol { - - symbol := index.Symbol { - range = common.get_token_range(v, ast_context.file.src), - type = .Enum, - name = ident.name, - pkg = get_package_from_node(v.node), - }; +make_symbol_union_from_ast :: proc (ast_context: ^AstContext, v: ast.Union_Type, ident: ast.Ident) -> index.Symbol { - names := make([dynamic] string, context.temp_allocator); + symbol := index.Symbol { + range = common.get_token_range(v, ast_context.file.src), + type = .Enum, + name = ident.name, + pkg = get_package_from_node(v.node), + }; - for variant in v.variants { + names := make([dynamic]string, context.temp_allocator); - if ident, ok := variant.derived.(ast.Ident); ok { - append(&names, ident.name); - } + for variant in v.variants { - } + if ident, ok := variant.derived.(ast.Ident); ok { + append(&names, ident.name); + } + } - symbol.value = index.SymbolUnionValue { - names = names[:], - }; + symbol.value = index.SymbolUnionValue { + names = names[:] + }; - return symbol; + return symbol; } -make_symbol_enum_from_ast :: proc(ast_context: ^AstContext, v: ast.Enum_Type, ident: ast.Ident) -> index.Symbol { - - symbol := index.Symbol { - range = common.get_token_range(v, ast_context.file.src), - type = .Enum, - name = ident.name, - pkg = get_package_from_node(v.node), - }; +make_symbol_enum_from_ast :: proc (ast_context: ^AstContext, v: ast.Enum_Type, ident: ast.Ident) -> index.Symbol { - names := make([dynamic] string, context.temp_allocator); - - for n in v.fields { + symbol := index.Symbol { + range = common.get_token_range(v, ast_context.file.src), + type = .Enum, + name = ident.name, + pkg = get_package_from_node(v.node), + }; - if ident, ok := n.derived.(ast.Ident); ok { - append(&names, ident.name); - } + names := make([dynamic]string, context.temp_allocator); - else if field, ok := n.derived.(ast.Field_Value); ok { - append(&names, field.field.derived.(ast.Ident).name); - } + for n in v.fields { - } + if ident, ok := n.derived.(ast.Ident); ok { + append(&names, ident.name); + } else if field, ok := n.derived.(ast.Field_Value); ok { + append(&names, field.field.derived.(ast.Ident).name); + } + } - symbol.value = index.SymbolEnumValue { - names = names[:], - }; + symbol.value = index.SymbolEnumValue { + names = names[:] + }; - return symbol; + return symbol; } -make_symbol_bitset_from_ast :: proc(ast_context: ^AstContext, v: ast.Bit_Set_Type, ident: ast.Ident) -> index.Symbol { +make_symbol_bitset_from_ast :: proc (ast_context: ^AstContext, v: ast.Bit_Set_Type, ident: ast.Ident) -> index.Symbol { - symbol := index.Symbol { - range = common.get_token_range(v, ast_context.file.src), - type = .Enum, - name = ident.name, - pkg = get_package_from_node(v.node), - }; + symbol := index.Symbol { + range = common.get_token_range(v, ast_context.file.src), + type = .Enum, + name = ident.name, + pkg = get_package_from_node(v.node), + }; - symbol.value = index.SymbolBitSetValue { - expr = v.elem, - }; + symbol.value = index.SymbolBitSetValue { + expr = v.elem + }; - return symbol; + return symbol; } -make_symbol_struct_from_ast :: proc(ast_context: ^AstContext, v: ast.Struct_Type, ident: ast.Ident) -> index.Symbol { - - symbol := index.Symbol { - range = common.get_token_range(v, ast_context.file.src), - type = .Struct, - name = ident.name, - pkg = get_package_from_node(v.node), - }; +make_symbol_struct_from_ast :: proc (ast_context: ^AstContext, v: ast.Struct_Type, ident: ast.Ident) -> index.Symbol { - names := make([dynamic] string, context.temp_allocator); - types := make([dynamic] ^ast.Expr, context.temp_allocator); - usings := make(map [string] bool, 0, context.temp_allocator); - - for field in v.fields.list { + symbol := index.Symbol { + range = common.get_token_range(v, ast_context.file.src), + type = .Struct, + name = ident.name, + pkg = get_package_from_node(v.node), + }; - for n in field.names { - if identifier, ok := n.derived.(ast.Ident); ok { - append(&names, identifier.name); - append(&types, index.clone_type(field.type, context.temp_allocator, nil)); + names := make([dynamic]string, context.temp_allocator); + types := make([dynamic]^ast.Expr, context.temp_allocator); + usings := make(map[string]bool, 0, context.temp_allocator); - if .Using in field.flags { - usings[identifier.name] = true; - } + for field in v.fields.list { - } - } + for n in field.names { + if identifier, ok := n.derived.(ast.Ident); ok { + append(&names, identifier.name); + append(&types, index.clone_type(field.type, context.temp_allocator, nil)); - } + if .Using in field.flags { + usings[identifier.name] = true; + } + } + } + } - symbol.value = index.SymbolStructValue { - names = names[:], - types = types[:], - usings = usings, - }; + symbol.value = index.SymbolStructValue { + names = names[:], + types = types[:], + usings = usings, + }; - if v.poly_params != nil { - resolve_poly_struct(ast_context, v, &symbol); - } + if v.poly_params != nil { + resolve_poly_struct(ast_context, v, &symbol); + } - //TODO change the expand to not double copy the array, but just pass the dynamic arrays - if len(usings) > 0 { - symbol.value = expand_struct_usings(ast_context, symbol, symbol.value.(index.SymbolStructValue)); - } + //TODO change the expand to not double copy the array, but just pass the dynamic arrays + if len(usings) > 0 { + symbol.value = expand_struct_usings(ast_context, symbol, symbol.value.(index.SymbolStructValue)); + } - return symbol; + return symbol; } -resolve_poly_struct :: proc(ast_context: ^AstContext, v: ast.Struct_Type, symbol: ^index.Symbol) { - - if ast_context.call == nil { - log.infof("no call"); - return; - } - - symbol_value := &symbol.value.(index.SymbolStructValue); - - if symbol_value == nil { - log.infof("no value"); - return; - } - - i := 0; - - poly_map := make(map [string] ^ast.Expr, 0, context.temp_allocator); - - for param in v.poly_params.list { - - for name in param.names { - - if len(ast_context.call.args) <= i { - break; - } - - if param.type == nil { - continue; - } - - if poly, ok := param.type.derived.(ast.Typeid_Type); ok { - - if ident, ok := name.derived.(ast.Ident); ok { - poly_map[ident.name] = ast_context.call.args[i]; - } - - } +resolve_poly_struct :: proc (ast_context: ^AstContext, v: ast.Struct_Type, symbol: ^index.Symbol) { + if ast_context.call == nil { + log.infof("no call"); + return; + } - i += 1; - } + symbol_value := &symbol.value.(index.SymbolStructValue); + if symbol_value == nil { + log.infof("no value"); + return; + } - } + i := 0; - for type, i in symbol_value.types { + poly_map := make(map[string]^ast.Expr, 0, context.temp_allocator); - if ident, ok := type.derived.(ast.Ident); ok { + for param in v.poly_params.list { - if expr, ok := poly_map[ident.name]; ok { - symbol_value.types[i] = expr; - } + for name in param.names { - } + if len(ast_context.call.args) <= i { + break; + } - else if call_expr, ok := type.derived.(ast.Call_Expr); ok { + if param.type == nil { + continue; + } - if call_expr.args == nil { - continue; - } + if poly, ok := param.type.derived.(ast.Typeid_Type); ok { - for arg, i in call_expr.args { + if ident, ok := name.derived.(ast.Ident); ok { + poly_map[ident.name] = ast_context.call.args[i]; + } + } - if ident, ok := arg.derived.(ast.Ident); ok { + i += 1; + } + } - if expr, ok := poly_map[ident.name]; ok { - call_expr.args[i] = expr; - } + for type, i in symbol_value.types { - } + if ident, ok := type.derived.(ast.Ident); ok { + if expr, ok := poly_map[ident.name]; ok { + symbol_value.types[i] = expr; + } + } else if call_expr, ok := type.derived.(ast.Call_Expr); ok { - } + if call_expr.args == nil { + continue; + } - } + for arg, i in call_expr.args { - } + if ident, ok := arg.derived.(ast.Ident); ok { + if expr, ok := poly_map[ident.name]; ok { + call_expr.args[i] = expr; + } + } + } + } + } } -get_globals :: proc(file: ast.File, ast_context: ^AstContext) { +get_globals :: proc (file: ast.File, ast_context: ^AstContext) { - ast_context.variables["context"] = true; + ast_context.variables["context"] = true; - exprs := common.collect_globals(file); + exprs := common.collect_globals(file); - for expr in exprs { - ast_context.globals[expr.name] = expr.expr; - ast_context.variables[expr.name] = expr.mutable; - } + for expr in exprs { + ast_context.globals[expr.name] = expr.expr; + ast_context.variables[expr.name] = expr.mutable; + } } -get_generic_assignment :: proc(file: ast.File, value: ^ast.Expr, ast_context: ^AstContext, results: ^[dynamic]^ast.Expr) { - - using ast; - - ast_context.use_locals = true; - ast_context.use_globals = true; - - switch v in value.derived { - case Call_Expr: - - ast_context.call = cast(^ast.Call_Expr)value; - - if symbol, ok := resolve_type_expression(ast_context, v.expr); ok { - - if procedure, ok := symbol.value.(index.SymbolProcedureValue); ok { - - for ret in procedure.return_types { - append(results, ret.type); - } - - } - - } - - case Comp_Lit: - if v.type != nil { - append(results, v.type); - } - case Array_Type: - if v.elem != nil { - append(results, v.elem); - } - case Dynamic_Array_Type: - if v.elem != nil { - append(results, v.elem); - } - case Selector_Expr: - if v.expr != nil { - append(results, value); - } - case Type_Assertion: - if v.type != nil { - append(results, v.type); - append(results, make_bool_ast()); - } - case: - //log.debugf("default node get_generic_assignment %v", v); - append(results, value); - } - +get_generic_assignment :: proc (file: ast.File, value: ^ast.Expr, ast_context: ^AstContext, results: ^[dynamic]^ast.Expr) { + + using ast; + + ast_context.use_locals = true; + ast_context.use_globals = true; + + switch v in value.derived { + case Call_Expr: + + ast_context.call = cast(^ast.Call_Expr)value; + + if symbol, ok := resolve_type_expression(ast_context, v.expr); ok { + + if procedure, ok := symbol.value.(index.SymbolProcedureValue); ok { + + for ret in procedure.return_types { + append(results, ret.type); + } + } + } + + case Comp_Lit: + if v.type != nil { + append(results, v.type); + } + case Array_Type: + if v.elem != nil { + append(results, v.elem); + } + case Dynamic_Array_Type: + if v.elem != nil { + append(results, v.elem); + } + case Selector_Expr: + if v.expr != nil { + append(results, value); + } + case Type_Assertion: + if v.type != nil { + append(results, v.type); + append(results, make_bool_ast()); + } + case: + //log.debugf("default node get_generic_assignment %v", v); + append(results, value); + } } -get_locals_value_decl :: proc(file: ast.File, value_decl: ast.Value_Decl, ast_context: ^AstContext) { - - using ast; - - if len(value_decl.names) <= 0 { - return; - } - - - if value_decl.type != nil { - str := common.get_ast_node_string(value_decl.names[0], file.src); - ast_context.variables[str] = value_decl.is_mutable; - store_local(ast_context, value_decl.type, value_decl.pos.offset, str); - return; - } +get_locals_value_decl :: proc (file: ast.File, value_decl: ast.Value_Decl, ast_context: ^AstContext) { - results := make([dynamic]^Expr, context.temp_allocator); + using ast; - for value in value_decl.values { - get_generic_assignment(file, value, ast_context, &results); - } - - for name, i in value_decl.names { - if i < len(results) { - str := common.get_ast_node_string(name, file.src); - ast_context.in_package[str] = get_package_from_node(results[i]); - store_local(ast_context, results[i], name.pos.offset, str); - ast_context.variables[str] = value_decl.is_mutable; - } - } - -} + if len(value_decl.names) <= 0 { + return; + } -get_locals_stmt :: proc(file: ast.File, stmt: ^ast.Stmt, ast_context: ^AstContext, document_position: ^DocumentPositionContext, save_assign := false) { - - ast_context.use_locals = true; - ast_context.use_globals = true; - ast_context.current_package = ast_context.document_package; - - using ast; - - if stmt == nil { - return; - } - - if stmt.pos.offset > document_position.position { - return; - } - - switch v in stmt.derived { - case Value_Decl: - get_locals_value_decl(file, v, ast_context); - case Type_Switch_Stmt: - get_locals_type_switch_stmt(file, v, ast_context, document_position); - case Switch_Stmt: - get_locals_switch_stmt(file, v, ast_context, document_position); - case For_Stmt: - get_locals_for_stmt(file, v, ast_context, document_position); - case Inline_Range_Stmt: - get_locals_stmt(file, v.body, ast_context, document_position); - case Range_Stmt: - get_locals_for_range_stmt(file, v, ast_context, document_position); - case If_Stmt: - get_locals_if_stmt(file, v, ast_context, document_position); - case Block_Stmt: - for stmt in v.stmts { - get_locals_stmt(file, stmt, ast_context, document_position); - } - case Proc_Lit: - get_locals_stmt(file, v.body, ast_context, document_position); - case Assign_Stmt: - if save_assign { - get_locals_assign_stmt(file, v, ast_context); - } - case Using_Stmt: - get_locals_using_stmt(file, v, ast_context); - case When_Stmt: - get_locals_stmt(file, v.else_stmt, ast_context, document_position); - get_locals_stmt(file, v.body, ast_context, document_position); - case: - //log.debugf("default node local stmt %v", v); - } + if value_decl.type != nil { + str := common.get_ast_node_string(value_decl.names[0], file.src); + ast_context.variables[str] = value_decl.is_mutable; + store_local(ast_context, value_decl.type, value_decl.pos.offset, str); + return; + } + results := make([dynamic]^Expr, context.temp_allocator); + for value in value_decl.values { + get_generic_assignment(file, value, ast_context, &results); + } + for name, i in value_decl.names { + if i < len(results) { + str := common.get_ast_node_string(name, file.src); + ast_context.in_package[str] = get_package_from_node(results[i]); + store_local(ast_context, results[i], name.pos.offset, str); + ast_context.variables[str] = value_decl.is_mutable; + } + } } -get_locals_using_stmt :: proc(file: ast.File, stmt: ast.Using_Stmt, ast_context: ^AstContext) { - - for u in stmt.list { - - if symbol, ok := resolve_type_expression(ast_context, u); ok { +get_locals_stmt :: proc (file: ast.File, stmt: ^ast.Stmt, ast_context: ^AstContext, document_position: ^DocumentPositionContext, save_assign := false) { - #partial switch v in symbol.value { - case index.SymbolPackageValue: - if ident, ok := u.derived.(ast.Ident); ok { - append(&ast_context.usings, ident.name); - } - case index.SymbolStructValue: - for name, i in v.names { - selector := index.new_type(ast.Selector_Expr, v.types[i].pos, v.types[i].end, context.temp_allocator); - selector.expr = u; - selector.field = index.new_type(ast.Ident, v.types[i].pos, v.types[i].end, context.temp_allocator); - selector.field.name = name; - store_local(ast_context, selector, 0, name); - ast_context.variables[name] = true; - } + ast_context.use_locals = true; + ast_context.use_globals = true; + ast_context.current_package = ast_context.document_package; - } - - - } + using ast; + if stmt == nil { + return; + } - } + if stmt.pos.offset > document_position.position { + return; + } + switch v in stmt.derived { + case Value_Decl: + get_locals_value_decl(file, v, ast_context); + case Type_Switch_Stmt: + get_locals_type_switch_stmt(file, v, ast_context, document_position); + case Switch_Stmt: + get_locals_switch_stmt(file, v, ast_context, document_position); + case For_Stmt: + get_locals_for_stmt(file, v, ast_context, document_position); + case Inline_Range_Stmt: + get_locals_stmt(file, v.body, ast_context, document_position); + case Range_Stmt: + get_locals_for_range_stmt(file, v, ast_context, document_position); + case If_Stmt: + get_locals_if_stmt(file, v, ast_context, document_position); + case Block_Stmt: + for stmt in v.stmts { + get_locals_stmt(file, stmt, ast_context, document_position); + } + case Proc_Lit: + get_locals_stmt(file, v.body, ast_context, document_position); + case Assign_Stmt: + if save_assign { + get_locals_assign_stmt(file, v, ast_context); + } + case Using_Stmt: + get_locals_using_stmt(file, v, ast_context); + case When_Stmt: + get_locals_stmt(file, v.else_stmt, ast_context, document_position); + get_locals_stmt(file, v.body, ast_context, document_position); + case: + //log.debugf("default node local stmt %v", v); + } } -get_locals_assign_stmt :: proc(file: ast.File, stmt: ast.Assign_Stmt, ast_context: ^AstContext) { - - using ast; +get_locals_using_stmt :: proc (file: ast.File, stmt: ast.Using_Stmt, ast_context: ^AstContext) { + + for u in stmt.list { + + if symbol, ok := resolve_type_expression(ast_context, u); ok { + + #partial switch v in symbol.value { + case index.SymbolPackageValue: + if ident, ok := u.derived.(ast.Ident); ok { + append(&ast_context.usings, ident.name); + } + case index.SymbolStructValue: + for name, i in v.names { + selector := index.new_type(ast.Selector_Expr, v.types[i].pos, v.types[i].end, context.temp_allocator); + selector.expr = u; + selector.field = index.new_type(ast.Ident, v.types[i].pos, v.types[i].end, context.temp_allocator); + selector.field.name = name; + store_local(ast_context, selector, 0, name); + ast_context.variables[name] = true; + } + } + } + } +} - if stmt.lhs == nil || stmt.rhs == nil { - return; - } +get_locals_assign_stmt :: proc (file: ast.File, stmt: ast.Assign_Stmt, ast_context: ^AstContext) { - results := make([dynamic]^Expr, context.temp_allocator); + using ast; - for rhs in stmt.rhs { - get_generic_assignment(file, rhs, ast_context, &results); - } + if stmt.lhs == nil || stmt.rhs == nil { + return; + } + results := make([dynamic]^Expr, context.temp_allocator); - if len(stmt.lhs) != len(results) { - return; - } + for rhs in stmt.rhs { + get_generic_assignment(file, rhs, ast_context, &results); + } - for lhs, i in stmt.lhs { - if ident, ok := lhs.derived.(ast.Ident); ok { - store_local(ast_context, results[i], ident.pos.offset, ident.name); - ast_context.variables[ident.name] = true; - } - } + if len(stmt.lhs) != len(results) { + return; + } + for lhs, i in stmt.lhs { + if ident, ok := lhs.derived.(ast.Ident); ok { + store_local(ast_context, results[i], ident.pos.offset, ident.name); + ast_context.variables[ident.name] = true; + } + } } -get_locals_if_stmt :: proc(file: ast.File, stmt: ast.If_Stmt, ast_context: ^AstContext, document_position: ^DocumentPositionContext) { +get_locals_if_stmt :: proc (file: ast.File, stmt: ast.If_Stmt, ast_context: ^AstContext, document_position: ^DocumentPositionContext) { - if !(stmt.pos.offset <= document_position.position && document_position.position <= stmt.end.offset) { - return; - } + if !(stmt.pos.offset <= document_position.position && document_position.position <= stmt.end.offset) { + return; + } - get_locals_stmt(file, stmt.init, ast_context, document_position, true); - get_locals_stmt(file, stmt.body, ast_context, document_position); - get_locals_stmt(file, stmt.else_stmt, ast_context, document_position); + get_locals_stmt(file, stmt.init, ast_context, document_position, true); + get_locals_stmt(file, stmt.body, ast_context, document_position); + get_locals_stmt(file, stmt.else_stmt, ast_context, document_position); } -get_locals_for_range_stmt :: proc(file: ast.File, stmt: ast.Range_Stmt, ast_context: ^AstContext, document_position: ^DocumentPositionContext) { - - using ast; - - if !(stmt.body.pos.offset <= document_position.position && document_position.position <= stmt.body.end.offset) { - return; - } - - results := make([dynamic]^Expr, context.temp_allocator); - - if stmt.expr == nil { - return; - } - - if symbol, ok := resolve_type_expression(ast_context, stmt.expr); ok { - - if generic, ok := symbol.value.(index.SymbolGenericValue); ok { +get_locals_for_range_stmt :: proc (file: ast.File, stmt: ast.Range_Stmt, ast_context: ^AstContext, document_position: ^DocumentPositionContext) { - switch v in generic.expr.derived { - case Map_Type: - if stmt.val0 != nil { + using ast; - if ident, ok := stmt.val0.derived.(Ident); ok { - store_local(ast_context, v.key, ident.pos.offset, ident.name); - ast_context.variables[ident.name] = true; - ast_context.in_package[ident.name] = symbol.pkg; - } - - } - - if stmt.val1 != nil { - - if ident, ok := stmt.val1.derived.(Ident); ok { - store_local(ast_context, v.value, ident.pos.offset, ident.name); - ast_context.variables[ident.name] = true; - ast_context.in_package[ident.name] = symbol.pkg; - } - - } - case Dynamic_Array_Type: - if stmt.val0 != nil { - - if ident, ok := stmt.val0.derived.(Ident); ok { - store_local(ast_context, v.elem, ident.pos.offset, ident.name); - ast_context.variables[ident.name] = true; - ast_context.in_package[ident.name] = symbol.pkg; - } - - } - - if stmt.val1 != nil { - - if ident, ok := stmt.val1.derived.(Ident); ok { - store_local(ast_context, make_int_ast(), ident.pos.offset, ident.name); - ast_context.variables[ident.name] = true; - ast_context.in_package[ident.name] = symbol.pkg; - } - - } - case Array_Type: - if stmt.val0 != nil { - - if ident, ok := stmt.val0.derived.(Ident); ok { - store_local(ast_context, v.elem, ident.pos.offset, ident.name); - ast_context.variables[ident.name] = true; - ast_context.in_package[ident.name] = symbol.pkg; - } - - } - - if stmt.val1 != nil { - - if ident, ok := stmt.val1.derived.(Ident); ok { - store_local(ast_context, make_int_ast(), ident.pos.offset, ident.name); - ast_context.variables[ident.name] = true; - ast_context.in_package[ident.name] = symbol.pkg; - } + if !(stmt.body.pos.offset <= document_position.position && document_position.position <= stmt.body.end.offset) { + return; + } - } - } + results := make([dynamic]^Expr, context.temp_allocator); - } + if stmt.expr == nil { + return; + } - } + if symbol, ok := resolve_type_expression(ast_context, stmt.expr); ok { + + if generic, ok := symbol.value.(index.SymbolGenericValue); ok { + + switch v in generic.expr.derived { + case Map_Type: + if stmt.val0 != nil { + + if ident, ok := stmt.val0.derived.(Ident); ok { + store_local(ast_context, v.key, ident.pos.offset, ident.name); + ast_context.variables[ident.name] = true; + ast_context.in_package[ident.name] = symbol.pkg; + } + } + + if stmt.val1 != nil { + + if ident, ok := stmt.val1.derived.(Ident); ok { + store_local(ast_context, v.value, ident.pos.offset, ident.name); + ast_context.variables[ident.name] = true; + ast_context.in_package[ident.name] = symbol.pkg; + } + } + case Dynamic_Array_Type: + if stmt.val0 != nil { + + if ident, ok := stmt.val0.derived.(Ident); ok { + store_local(ast_context, v.elem, ident.pos.offset, ident.name); + ast_context.variables[ident.name] = true; + ast_context.in_package[ident.name] = symbol.pkg; + } + } + + if stmt.val1 != nil { + + if ident, ok := stmt.val1.derived.(Ident); ok { + store_local(ast_context, make_int_ast(), ident.pos.offset, ident.name); + ast_context.variables[ident.name] = true; + ast_context.in_package[ident.name] = symbol.pkg; + } + } + case Array_Type: + if stmt.val0 != nil { + + if ident, ok := stmt.val0.derived.(Ident); ok { + store_local(ast_context, v.elem, ident.pos.offset, ident.name); + ast_context.variables[ident.name] = true; + ast_context.in_package[ident.name] = symbol.pkg; + } + } + + if stmt.val1 != nil { + + if ident, ok := stmt.val1.derived.(Ident); ok { + store_local(ast_context, make_int_ast(), ident.pos.offset, ident.name); + ast_context.variables[ident.name] = true; + ast_context.in_package[ident.name] = symbol.pkg; + } + } + } + } + } - get_locals_stmt(file, stmt.body, ast_context, document_position); + get_locals_stmt(file, stmt.body, ast_context, document_position); } -get_locals_for_stmt :: proc(file: ast.File, stmt: ast.For_Stmt, ast_context: ^AstContext, document_position: ^DocumentPositionContext) { +get_locals_for_stmt :: proc (file: ast.File, stmt: ast.For_Stmt, ast_context: ^AstContext, document_position: ^DocumentPositionContext) { - if !(stmt.pos.offset <= document_position.position && document_position.position <= stmt.end.offset) { - return; - } + if !(stmt.pos.offset <= document_position.position && document_position.position <= stmt.end.offset) { + return; + } - get_locals_stmt(file, stmt.init, ast_context, document_position, true); - get_locals_stmt(file, stmt.body, ast_context, document_position); + get_locals_stmt(file, stmt.init, ast_context, document_position, true); + get_locals_stmt(file, stmt.body, ast_context, document_position); } -get_locals_switch_stmt :: proc(file: ast.File, stmt: ast.Switch_Stmt, ast_context: ^AstContext, document_position: ^DocumentPositionContext) { +get_locals_switch_stmt :: proc (file: ast.File, stmt: ast.Switch_Stmt, ast_context: ^AstContext, document_position: ^DocumentPositionContext) { - if !(stmt.pos.offset <= document_position.position && document_position.position <= stmt.end.offset) { - return; - } + if !(stmt.pos.offset <= document_position.position && document_position.position <= stmt.end.offset) { + return; + } - get_locals_stmt(file, stmt.body, ast_context, document_position); + get_locals_stmt(file, stmt.body, ast_context, document_position); } -get_locals_type_switch_stmt :: proc(file: ast.File, stmt: ast.Type_Switch_Stmt, ast_context: ^AstContext, document_position: ^DocumentPositionContext) { - - using ast; +get_locals_type_switch_stmt :: proc (file: ast.File, stmt: ast.Type_Switch_Stmt, ast_context: ^AstContext, document_position: ^DocumentPositionContext) { - if !(stmt.pos.offset <= document_position.position && document_position.position <= stmt.end.offset) { - return; - } + using ast; - if stmt.body == nil { - return; - } - - if block, ok := stmt.body.derived.(Block_Stmt); ok { - - for block_stmt in block.stmts { - - if cause, ok := block_stmt.derived.(Case_Clause); ok && cause.pos.offset <= document_position.position && document_position.position <= cause.end.offset { + if !(stmt.pos.offset <= document_position.position && document_position.position <= stmt.end.offset) { + return; + } - for b in cause.body { - get_locals_stmt(file, b, ast_context, document_position); - } + if stmt.body == nil { + return; + } - tag := stmt.tag.derived.(Assign_Stmt); + if block, ok := stmt.body.derived.(Block_Stmt); ok { - if len(tag.lhs) == 1 && len(cause.list) == 1 { - ident := tag.lhs[0].derived.(Ident); - store_local(ast_context, cause.list[0], ident.pos.offset, ident.name); - ast_context.variables[ident.name] = true; - } + for block_stmt in block.stmts { - } + if cause, ok := block_stmt.derived.(Case_Clause); ok && cause.pos.offset <= document_position.position && document_position.position <= cause.end.offset { - } + for b in cause.body { + get_locals_stmt(file, b, ast_context, document_position); + } - } + tag := stmt.tag.derived.(Assign_Stmt); + if len(tag.lhs) == 1 && len(cause.list) == 1 { + ident := tag.lhs[0].derived.(Ident); + store_local(ast_context, cause.list[0], ident.pos.offset, ident.name); + ast_context.variables[ident.name] = true; + } + } + } + } } -get_locals :: proc(file: ast.File, function: ^ast.Node, ast_context: ^AstContext, document_position: ^DocumentPositionContext) { - - proc_lit, ok := function.derived.(ast.Proc_Lit); - - if !ok || proc_lit.body == nil { - return; - } - - - if proc_lit.type != nil && proc_lit.type.params != nil { +get_locals :: proc (file: ast.File, function: ^ast.Node, ast_context: ^AstContext, document_position: ^DocumentPositionContext) { - for arg in proc_lit.type.params.list { + proc_lit, ok := function.derived.(ast.Proc_Lit); - for name in arg.names { - if arg.type != nil { - str := common.get_ast_node_string(name, file.src); - store_local(ast_context, arg.type, name.pos.offset, str); - ast_context.variables[str] = true; - ast_context.parameters[str] = true; - - log.info(arg.flags); + if !ok || proc_lit.body == nil { + return; + } - if .Using in arg.flags { - using_stmt: ast.Using_Stmt; - using_stmt.list = make([] ^ ast.Expr, 1, context.temp_allocator); - using_stmt.list[0] = arg.type; - get_locals_using_stmt(file, using_stmt, ast_context); - } - } - } + if proc_lit.type != nil && proc_lit.type.params != nil { - } + for arg in proc_lit.type.params.list { + for name in arg.names { + if arg.type != nil { + str := common.get_ast_node_string(name, file.src); + store_local(ast_context, arg.type, name.pos.offset, str); + ast_context.variables[str] = true; + ast_context.parameters[str] = true; - } + log.info(arg.flags); - block: ast.Block_Stmt; - block, ok = proc_lit.body.derived.(ast.Block_Stmt); + if .Using in arg.flags { + using_stmt: ast.Using_Stmt; + using_stmt.list = make([]^ast.Expr, 1, context.temp_allocator); + using_stmt.list[0] = arg.type; + get_locals_using_stmt(file, using_stmt, ast_context); + } + } + } + } + } - if !ok { - log.error("Proc_List body not block"); - return; - } + block: ast.Block_Stmt; + block, ok = proc_lit.body.derived.(ast.Block_Stmt); - for stmt in block.stmts { - get_locals_stmt(file, stmt, ast_context, document_position); - } + if !ok { + log.error("Proc_List body not block"); + return; + } + for stmt in block.stmts { + get_locals_stmt(file, stmt, ast_context, document_position); + } } -clear_locals :: proc(ast_context: ^AstContext) { - clear(&ast_context.locals); - clear(&ast_context.parameters); - clear(&ast_context.variables); - clear(&ast_context.usings); +clear_locals :: proc (ast_context: ^AstContext) { + clear(&ast_context.locals); + clear(&ast_context.parameters); + clear(&ast_context.variables); + clear(&ast_context.usings); } -concatenate_symbols_information :: proc(ast_context: ^AstContext, symbol: index.Symbol) -> string { - - pkg := path.base(symbol.pkg, false, context.temp_allocator); - - if symbol.type == .Function { - - if symbol.returns != "" { - return fmt.tprintf("%v.%v: proc %v -> %v", pkg, symbol.name, symbol.signature, symbol.returns); - } - - else { - return fmt.tprintf("%v.%v: proc%v", pkg, symbol.name, symbol.signature); - } +concatenate_symbols_information :: proc (ast_context: ^AstContext, symbol: index.Symbol) -> string { - } + pkg := path.base(symbol.pkg, false, context.temp_allocator); - else if symbol.type == .Package { - return symbol.name; - } + if symbol.type == .Function { - else { + if symbol.returns != "" { + return fmt.tprintf("%v.%v: proc %v -> %v", pkg, symbol.name, symbol.signature, symbol.returns); + } else { + return fmt.tprintf("%v.%v: proc%v", pkg, symbol.name, symbol.signature); + } + } else if symbol.type == .Package { + return symbol.name; + } else { - if symbol.signature != "" { - return fmt.tprintf("%v.%v: %v", pkg, symbol.name, symbol.signature); - } - - else { - return fmt.tprintf("%v.%v", pkg, symbol.name); - } - - } + if symbol.signature != "" { + return fmt.tprintf("%v.%v: %v", pkg, symbol.name, symbol.signature); + } else { + return fmt.tprintf("%v.%v", pkg, symbol.name); + } + } - return ""; //weird bug requires this + return ""; //weird bug requires this } -get_definition_location :: proc(document: ^Document, position: common.Position) -> (common.Location, bool) { - - location: common.Location; - - ast_context := make_ast_context(document.ast, document.imports, document.package_name); - - uri: string; - - position_context, ok := get_document_position_context(document, position, .Definition); - - if !ok { - log.warn("Failed to get position context"); - return location, false; - } - - get_globals(document.ast, &ast_context); - - if position_context.function != nil { - get_locals(document.ast, position_context.function, &ast_context, &position_context); - } - - if position_context.selector != nil { - - //if the base selector is the client wants to go to. - if base, ok := position_context.selector.derived.(ast.Ident); ok && position_context.identifier != nil { - - ident := position_context.identifier.derived.(ast.Ident); - - if ident.name == base.name { - - if resolved, ok := resolve_location_identifier(&ast_context, ident); ok { - location.range = resolved.range; - - if resolved.uri == "" { - location.uri = document.uri.uri; - } - - else { - location.uri = resolved.uri; - } - - return location, true; - } - - else { - return location, false; - } +get_definition_location :: proc (document: ^Document, position: common.Position) -> (common.Location, bool) { - } + location: common.Location; - } + ast_context := make_ast_context(document.ast, document.imports, document.package_name); - //otherwise it's the field the client wants to go to. + uri: string; - selector: index.Symbol; + position_context, ok := get_document_position_context(document, position, .Definition); - ast_context.use_locals = true; - ast_context.use_globals = true; - ast_context.current_package = ast_context.document_package; - - selector, ok = resolve_type_expression(&ast_context, position_context.selector); - - if !ok { - return location, false; - } - - field: string; - - if position_context.field != nil { - - switch v in position_context.field.derived { - case ast.Ident: - field = v.name; - } - - } - - uri = selector.uri; - - #partial switch v in selector.value { - case index.SymbolEnumValue: - location.range = selector.range; - case index.SymbolStructValue: - for name, i in v.names { - if strings.compare(name, field) == 0 { - location.range = common.get_token_range(v.types[i]^, document.ast.src); - } - } - case index.SymbolPackageValue: - if symbol, ok := index.lookup(field, selector.pkg); ok { - location.range = symbol.range; - uri = symbol.uri; - } - else { - return location, false; - } - } - - if !ok { - return location, false; - } - - } - - else if position_context.identifier != nil { - - if resolved, ok := resolve_location_identifier(&ast_context, position_context.identifier.derived.(ast.Ident)); ok { - location.range = resolved.range; - uri = resolved.uri; - } - - else { - return location, false; - } + if !ok { + log.warn("Failed to get position context"); + return location, false; + } - } + get_globals(document.ast, &ast_context); - else { - return location, false; - } + if position_context.function != nil { + get_locals(document.ast, position_context.function, &ast_context, &position_context); + } - //if the symbol is generated by the ast we don't set the uri. - if uri == "" { - location.uri = document.uri.uri; - } + if position_context.selector != nil { + + //if the base selector is the client wants to go to. + if base, ok := position_context.selector.derived.(ast.Ident); ok && position_context.identifier != nil { + + ident := position_context.identifier.derived.(ast.Ident); + + if ident.name == base.name { + + if resolved, ok := resolve_location_identifier(&ast_context, ident); ok { + location.range = resolved.range; + + if resolved.uri == "" { + location.uri = document.uri.uri; + } else { + location.uri = resolved.uri; + } + + return location, true; + } else { + return location, false; + } + } + } + + //otherwise it's the field the client wants to go to. + + selector: index.Symbol; + + ast_context.use_locals = true; + ast_context.use_globals = true; + ast_context.current_package = ast_context.document_package; + + selector, ok = resolve_type_expression(&ast_context, position_context.selector); + + if !ok { + return location, false; + } + + field: string; + + if position_context.field != nil { + + switch v in position_context.field.derived { + case ast.Ident: + field = v.name; + } + } + + uri = selector.uri; + + #partial switch v in selector.value { + case index.SymbolEnumValue: + location.range = selector.range; + case index.SymbolStructValue: + for name, i in v.names { + if strings.compare(name, field) == 0 { + location.range = common.get_token_range(v.types[i]^, document.ast.src); + } + } + case index.SymbolPackageValue: + if symbol, ok := index.lookup(field, selector.pkg); ok { + location.range = symbol.range; + uri = symbol.uri; + } else { + return location, false; + } + } + + if !ok { + return location, false; + } + } else if position_context.identifier != nil { + + if resolved, ok := resolve_location_identifier(&ast_context, position_context.identifier.derived.(ast.Ident)); ok { + location.range = resolved.range; + uri = resolved.uri; + } else { + return location, false; + } + } else { + return location, false; + } - else { - location.uri = uri; - } + //if the symbol is generated by the ast we don't set the uri. + if uri == "" { + location.uri = document.uri.uri; + } else { + location.uri = uri; + } - return location, true; + return location, true; } -write_hover_content :: proc(ast_context: ^AstContext, symbol: index.Symbol) -> MarkupContent { - content: MarkupContent; +write_hover_content :: proc (ast_context: ^AstContext, symbol: index.Symbol) -> MarkupContent { + content: MarkupContent; - cat := concatenate_symbols_information(ast_context, symbol); + cat := concatenate_symbols_information(ast_context, symbol); - if cat != "" { - content.kind = "markdown"; - content.value = fmt.tprintf("```odin\n %v\n```\n%v", cat, symbol.doc); - } - - else { - content.kind = "plaintext"; - } + if cat != "" { + content.kind = "markdown"; + content.value = fmt.tprintf("```odin\n %v\n```\n%v", cat, symbol.doc); + } else { + content.kind = "plaintext"; + } - return content; + return content; } -get_signature :: proc(ast_context: ^AstContext, ident: ast.Ident, symbol: index.Symbol, was_variable := false) -> string { +get_signature :: proc (ast_context: ^AstContext, ident: ast.Ident, symbol: index.Symbol, was_variable := false) -> string { - if symbol.type == .Function { - return symbol.signature; - } - - if is_variable, ok := ast_context.variables[ident.name]; ok && is_variable { - - if local := get_local(ast_context, ident.pos.offset, ident.name); local != nil { - - if i, ok := local.derived.(ast.Ident); ok { - return get_signature(ast_context, i, symbol, true); - } + if symbol.type == .Function { + return symbol.signature; + } - else { - return index.node_to_string(local); - } - } + if is_variable, ok := ast_context.variables[ident.name]; ok && is_variable { - if global, ok := ast_context.globals[ident.name]; ok { - if i, ok := global.derived.(ast.Ident); ok { - return get_signature(ast_context, i, symbol, true); - } + if local := get_local(ast_context, ident.pos.offset, ident.name); local != nil { - else { - return index.node_to_string(global); - } - } + if i, ok := local.derived.(ast.Ident); ok { + return get_signature(ast_context, i, symbol, true); + } else { + return index.node_to_string(local); + } + } - } + if global, ok := ast_context.globals[ident.name]; ok { + if i, ok := global.derived.(ast.Ident); ok { + return get_signature(ast_context, i, symbol, true); + } else { + return index.node_to_string(global); + } + } + } - if !was_variable { - #partial switch v in symbol.value { - case index.SymbolStructValue: - return "struct"; - case index.SymbolUnionValue: - return "union"; - case index.SymbolEnumValue: - return "enum"; - } - } + if !was_variable { + #partial switch v in symbol.value { + case index.SymbolStructValue: + return "struct"; + case index.SymbolUnionValue: + return "union"; + case index.SymbolEnumValue: + return "enum"; + } + } - return ident.name; + return ident.name; } -get_signature_information :: proc(document: ^Document, position: common.Position) -> (SignatureHelp, bool) { +get_signature_information :: proc (document: ^Document, position: common.Position) -> (SignatureHelp, bool) { - signature_help: SignatureHelp; + signature_help: SignatureHelp; - ast_context := make_ast_context(document.ast, document.imports, document.package_name); + ast_context := make_ast_context(document.ast, document.imports, document.package_name); - position_context, ok := get_document_position_context(document, position, .SignatureHelp); + position_context, ok := get_document_position_context(document, position, .SignatureHelp); - if !ok { - return signature_help, true; - } + if !ok { + return signature_help, true; + } - if position_context.call == nil { - return signature_help, true; - } + if position_context.call == nil { + return signature_help, true; + } - get_globals(document.ast, &ast_context); + get_globals(document.ast, &ast_context); - if position_context.function != nil { - get_locals(document.ast, position_context.function, &ast_context, &position_context); - } + if position_context.function != nil { + get_locals(document.ast, position_context.function, &ast_context, &position_context); + } - call: index.Symbol; - call, ok = resolve_type_expression(&ast_context, position_context.call); + call: index.Symbol; + call, ok = resolve_type_expression(&ast_context, position_context.call); - if symbol, ok := call.value.(index.SymbolProcedureValue); !ok { - return signature_help, true; - } + if symbol, ok := call.value.(index.SymbolProcedureValue); !ok { + return signature_help, true; + } - signature_information := make([] SignatureInformation, 1, context.temp_allocator); + signature_information := make([]SignatureInformation, 1, context.temp_allocator); - signature_information[0].label = concatenate_symbols_information(&ast_context, call); - signature_information[0].documentation = call.doc; + signature_information[0].label = concatenate_symbols_information(&ast_context, call); + signature_information[0].documentation = call.doc; - signature_help.signatures = signature_information; + signature_help.signatures = signature_information; - return signature_help, true; + return signature_help, true; } -get_document_symbols :: proc(document: ^Document) -> [] DocumentSymbol { +get_document_symbols :: proc (document: ^Document) -> []DocumentSymbol { - ast_context := make_ast_context(document.ast, document.imports, document.package_name); + ast_context := make_ast_context(document.ast, document.imports, document.package_name); - get_globals(document.ast, &ast_context); + get_globals(document.ast, &ast_context); - symbols := make([dynamic] DocumentSymbol, context.temp_allocator); + symbols := make([dynamic]DocumentSymbol, context.temp_allocator); - package_symbol: DocumentSymbol; + package_symbol: DocumentSymbol; - if len(document.ast.decls) == 0 { - return {}; - } - - package_symbol.kind = .Package; - package_symbol.name = path.base(document.package_name, false, context.temp_allocator); - package_symbol.range = { - start = { - line = document.ast.decls[0].pos.line, - }, - end = { - line = document.ast.decls[len(document.ast.decls)-1].end.line, - }, - }; - package_symbol.selectionRange = package_symbol.range; - - children_symbols := make([dynamic] DocumentSymbol, context.temp_allocator); - - for k, expr in ast_context.globals { - - symbol: DocumentSymbol; - - symbol.range = common.get_token_range(expr, ast_context.file.src); - symbol.selectionRange = symbol.range; - symbol.name = k; - - switch v in expr.derived { - case ast.Struct_Type: - symbol.kind = .Struct; - case ast.Proc_Lit, ast.Proc_Group: - symbol.kind = .Function; - case ast.Enum_Type, ast.Union_Type: - symbol.kind = .Enum; - case: - symbol.kind = .Variable; - } + if len(document.ast.decls) == 0 { + return {}; + } - append(&children_symbols, symbol); - } + package_symbol.kind = .Package; + package_symbol.name = path.base(document.package_name, false, context.temp_allocator); + package_symbol.range = { + start = { + line = document.ast.decls[0].pos.line + }, + end = { + line = document.ast.decls[len(document.ast.decls) - 1].end.line + }, + }; + package_symbol.selectionRange = package_symbol.range; + + children_symbols := make([dynamic]DocumentSymbol, context.temp_allocator); + + for k, expr in ast_context.globals { + + symbol: DocumentSymbol; + + symbol.range = common.get_token_range(expr, ast_context.file.src); + symbol.selectionRange = symbol.range; + symbol.name = k; + + switch v in expr.derived { + case ast.Struct_Type: + symbol.kind = .Struct; + case ast.Proc_Lit,ast.Proc_Group: + symbol.kind = .Function; + case ast.Enum_Type,ast.Union_Type: + symbol.kind = .Enum; + case: + symbol.kind = .Variable; + } + + append(&children_symbols, symbol); + } - package_symbol.children = children_symbols[:]; + package_symbol.children = children_symbols[:]; - append(&symbols, package_symbol); + append(&symbols, package_symbol); - return symbols[:]; + return symbols[:]; } /* - All these fallback functions are not perfect and should be fixed. A lot of weird use of the odin tokenizer and parser. + All these fallback functions are not perfect and should be fixed. A lot of weird use of the odin tokenizer and parser. */ /* - Figure out what exactly is at the given position and whether it is in a function, struct, etc. + Figure out what exactly is at the given position and whether it is in a function, struct, etc. */ -get_document_position_context :: proc(document: ^Document, position: common.Position, hint: DocumentPositionContextHint) -> (DocumentPositionContext, bool) { - - position_context: DocumentPositionContext; +get_document_position_context :: proc (document: ^Document, position: common.Position, hint: DocumentPositionContextHint) -> (DocumentPositionContext, bool) { - position_context.hint = hint; - position_context.file = document.ast; - position_context.line = position.line; + position_context: DocumentPositionContext; - absolute_position, ok := common.get_absolute_position(position, document.text); + position_context.hint = hint; + position_context.file = document.ast; + position_context.line = position.line; - if !ok { - log.error("failed to get absolute position"); - return position_context, false; - } + absolute_position, ok := common.get_absolute_position(position, document.text); - position_context.position = absolute_position; + if !ok { + log.error("failed to get absolute position"); + return position_context, false; + } - for decl in document.ast.decls { - get_document_position(decl, &position_context); - } + position_context.position = absolute_position; - if hint == .Completion && position_context.selector == nil && position_context.field == nil { - fallback_position_context_completion(document, position, &position_context); - } + for decl in document.ast.decls { + get_document_position(decl, &position_context); + } - else if hint == .SignatureHelp && position_context.call == nil { - fallback_position_context_signature(document, position, &position_context); - } + if hint == .Completion && position_context.selector == nil && position_context.field == nil { + fallback_position_context_completion(document, position, &position_context); + } else if hint == .SignatureHelp && position_context.call == nil { + fallback_position_context_signature(document, position, &position_context); + } - return position_context, true; + return position_context, true; } +fallback_position_context_completion :: proc (document: ^Document, position: common.Position, position_context: ^DocumentPositionContext) { + + paren_count: int; + bracket_count: int; + end: int; + start: int; + empty_dot: bool; + empty_arrow: bool; + last_dot: bool; + last_arrow: bool; + + i := position_context.position - 1; + + end = i; + + for i > 0 { + + c := position_context.file.src[i]; + + if c == '(' && paren_count == 0 { + start = i + 1; + break; + } else if c == '[' && bracket_count == 0 { + start = i + 1; + break; + } else if c == ']' && !last_dot { + start = i + 1; + break; + } else if c == ')' && !last_dot { + start = i + 1; + break; + } else if c == ')' { + paren_count -= 1; + } else if c == '(' { + paren_count += 1; + } else if c == '[' { + bracket_count += 1; + } else if c == ']' { + bracket_count -= 1; + } else if c == '.' { + last_dot = true; + i -= 1; + continue; + } else if position_context.file.src[max(0, i - 1)] == '-' && c == '>' { + last_arrow = true; + i -= 2; + continue; + } + + //yeah.. + if c == ' ' || c == '{' || c == ',' || + c == '}' || c == '^' || c == ':' || + c == '\n' || c == '\r' || c == '=' || + c == '<' || c == '>' || c == '-' || + c == '+' || c == '&' { + start = i + 1; + break; + } + + last_dot = false; + last_arrow = false; + + i -= 1; + } -fallback_position_context_completion :: proc(document: ^Document, position: common.Position, position_context: ^DocumentPositionContext) { - - paren_count: int; - bracket_count: int; - end: int; - start: int; - empty_dot: bool; - empty_arrow: bool; - last_dot: bool; - last_arrow: bool; - - i := position_context.position-1; - - end = i; - - for i > 0 { - - c := position_context.file.src[i]; - - if c == '(' && paren_count == 0 { - start = i+1; - break; - } - - else if c == '[' && bracket_count == 0 { - start = i+1; - break; - } - - else if c == ']' && !last_dot { - start = i+1; - break; - } - - else if c == ')' && !last_dot { - start = i+1; - break; - } - - else if c == ')' { - paren_count -= 1; - } - - else if c == '(' { - paren_count += 1; - } - - else if c == '[' { - bracket_count += 1; - } - - else if c == ']' { - bracket_count -= 1; - } - - else if c == '.' { - last_dot = true; - i -= 1; - continue; - } - - else if position_context.file.src[max(0, i-1)] == '-' && c == '>' { - last_arrow = true; - i -= 2; - continue; - } - - //yeah.. - if c == ' ' || c == '{' || c == ',' || - c == '}' || c == '^' || c == ':' || - c == '\n' || c == '\r' || c == '=' || - c == '<' || c == '>' || c == '-' || - c == '+' || c == '&' { - start = i+1; - break; - } - - last_dot = false; - last_arrow = false; - - i -= 1; - } - - if i >= 0 && position_context.file.src[end] == '.' { - empty_dot = true; - end -= 1; - } - - else if i >= 0 && position_context.file.src[max(0, end-1)] == '-' && position_context.file.src[end] == '>' { - empty_arrow = true; - end -= 2; - position_context.arrow = true; - } - - begin_offset := max(0, start); - end_offset := max(start, end+1); - - str := position_context.file.src[0:end_offset]; - - if empty_dot && end_offset - begin_offset == 0 { - position_context.implicit = true; - return; - } - - p := parser.Parser { - err = parser_warning_handler, //empty - warn = parser_warning_handler, //empty - file = &position_context.file, - }; - - tokenizer.init(&p.tok, str, position_context.file.fullpath, parser_warning_handler); + if i >= 0 && position_context.file.src[end] == '.' { + empty_dot = true; + end -= 1; + } else if i >= 0 && position_context.file.src[max(0, end - 1)] == '-' && position_context.file.src[end] == '>' { + empty_arrow = true; + end -= 2; + position_context.arrow = true; + } - p.tok.ch = ' '; - p.tok.line_count = position.line; - p.tok.offset = begin_offset; - p.tok.read_offset = begin_offset; + begin_offset := max(0, start); + end_offset := max(start, end + 1); - tokenizer.advance_rune(&p.tok); + str := position_context.file.src[0:end_offset]; - if p.tok.ch == utf8.RUNE_BOM { - tokenizer.advance_rune(&p.tok); + if empty_dot && end_offset - begin_offset == 0 { + position_context.implicit = true; + return; } - parser.advance_token(&p); - - context.allocator = context.temp_allocator; + p := parser.Parser { + err = parser_warning_handler, //empty + warn = parser_warning_handler, //empty + file = &position_context.file, + }; - e := parser.parse_expr(&p, true); + tokenizer.init(&p.tok, str, position_context.file.fullpath, parser_warning_handler); - if empty_dot || empty_arrow { - position_context.selector = e; - } + p.tok.ch = ' '; + p.tok.line_count = position.line; + p.tok.offset = begin_offset; + p.tok.read_offset = begin_offset; - else if s, ok := e.derived.(ast.Selector_Expr); ok { - position_context.selector = s.expr; - position_context.field = s.field; - } + tokenizer.advance_rune(&p.tok); - else if s, ok := e.derived.(ast.Implicit_Selector_Expr); ok { - position_context.implicit = true; - } + if p.tok.ch == utf8.RUNE_BOM { + tokenizer.advance_rune(&p.tok); + } - else if bad_expr, ok := e.derived.(ast.Bad_Expr); ok { - //this is most likely because of use of 'in', 'context', etc. - //try to go back one dot. + parser.advance_token(&p); - src_with_dot := string(position_context.file.src[0:end_offset+1]); - last_dot := strings.last_index(src_with_dot, "."); + context.allocator = context.temp_allocator; - if last_dot == -1 { - return; - } + e := parser.parse_expr(&p, true); - tokenizer.init(&p.tok, position_context.file.src[0:last_dot], position_context.file.fullpath, parser_warning_handler); + if empty_dot || empty_arrow { + position_context.selector = e; + } else if s, ok := e.derived.(ast.Selector_Expr); ok { + position_context.selector = s.expr; + position_context.field = s.field; + } else if s, ok := e.derived.(ast.Implicit_Selector_Expr); ok { + position_context.implicit = true; + } else if bad_expr, ok := e.derived.(ast.Bad_Expr); ok { + //this is most likely because of use of 'in', 'context', etc. + //try to go back one dot. - p.tok.ch = ' '; - p.tok.line_count = position.line; - p.tok.offset = begin_offset; - p.tok.read_offset = begin_offset; + src_with_dot := string(position_context.file.src[0:end_offset + 1]); + last_dot := strings.last_index(src_with_dot, "."); - tokenizer.advance_rune(&p.tok); + if last_dot == -1 { + return; + } - if p.tok.ch == utf8.RUNE_BOM { - tokenizer.advance_rune(&p.tok); - } + tokenizer.init(&p.tok, position_context.file.src[0:last_dot], position_context.file.fullpath, parser_warning_handler); - parser.advance_token(&p); + p.tok.ch = ' '; + p.tok.line_count = position.line; + p.tok.offset = begin_offset; + p.tok.read_offset = begin_offset; - e := parser.parse_expr(&p, true); + tokenizer.advance_rune(&p.tok); - if e == nil { - return; - } + if p.tok.ch == utf8.RUNE_BOM { + tokenizer.advance_rune(&p.tok); + } - position_context.selector = e; + parser.advance_token(&p); - ident := index.new_type(ast.Ident, e.pos, e.end, context.temp_allocator); - ident.name = string(position_context.file.src[last_dot+1:end_offset]); + e := parser.parse_expr(&p, true); - if ident.name != "" { - position_context.field = ident; - } + if e == nil { + return; + } - } + position_context.selector = e; - else { - position_context.identifier = e; - } + ident := index.new_type(ast.Ident, e.pos, e.end, context.temp_allocator); + ident.name = string(position_context.file.src[last_dot + 1:end_offset]); + if ident.name != "" { + position_context.field = ident; + } + } else { + position_context.identifier = e; + } } -fallback_position_context_signature :: proc(document: ^Document, position: common.Position, position_context: ^DocumentPositionContext) { - - paren_count: int; - end: int; - start: int; - first_paren: bool; - i := position_context.position-1; - - for i > 0 { - - c := position_context.file.src[i]; - - if c == '(' && (paren_count == 0 || paren_count == -1) { - end = i; - first_paren = true; - } - - else if c == ')' { - paren_count -= 1; - } - - else if c == '(' { - paren_count += 1; - } - - else if c == ' ' && end != 0 { - start = i+1; - break; - } - //not good enough if you want multi function signature help - else if c == '\n' || c == '\r' { - start = i+1; - break; - } - - i -= 1; - } +fallback_position_context_signature :: proc (document: ^Document, position: common.Position, position_context: ^DocumentPositionContext) { + + paren_count: int; + end: int; + start: int; + first_paren: bool; + i := position_context.position - 1; + + for i > 0 { + + c := position_context.file.src[i]; + + if c == '(' && (paren_count == 0 || paren_count == -1) { + end = i; + first_paren = true; + } else if c == ')' { + paren_count -= 1; + } else if c == '(' { + paren_count += 1; + } else if c == ' ' && end != 0 { + start = i + 1; + break; + } else + //not good enough if you want multi function signature help + if c == '\n' || c == '\r' { + start = i + 1; + break; + } + + i -= 1; + } - if !first_paren { - return; - } + if !first_paren { + return; + } - begin_offset := max(0, start); - end_offset := max(start, end+1); + begin_offset := max(0, start); + end_offset := max(start, end + 1); - str := position_context.file.src[0:end_offset]; + str := position_context.file.src[0:end_offset]; - p := parser.Parser { - err = parser_warning_handler, //empty + p := parser.Parser { + err = parser_warning_handler, //empty warn = parser_warning_handler, //empty - file = &position_context.file, + file = &position_context.file, }; - tokenizer.init(&p.tok, str, position_context.file.fullpath, parser_warning_handler); + tokenizer.init(&p.tok, str, position_context.file.fullpath, parser_warning_handler); - p.tok.ch = ' '; - p.tok.line_count = position.line; - p.tok.offset = begin_offset; - p.tok.read_offset = begin_offset; + p.tok.ch = ' '; + p.tok.line_count = position.line; + p.tok.offset = begin_offset; + p.tok.read_offset = begin_offset; - tokenizer.advance_rune(&p.tok); + tokenizer.advance_rune(&p.tok); if p.tok.ch == utf8.RUNE_BOM { tokenizer.advance_rune(&p.tok); } - parser.advance_token(&p); + parser.advance_token(&p); - context.allocator = context.temp_allocator; + context.allocator = context.temp_allocator; - e := parser.parse_expr(&p, true); + e := parser.parse_expr(&p, true); - position_context.call = e; + position_context.call = e; } +get_document_position :: proc { +get_document_position_array, +get_document_position_dynamic_array, +get_document_position_node}; -get_document_position :: proc{ - get_document_position_array, - get_document_position_dynamic_array, - get_document_position_node, -}; +get_document_position_array :: proc (array: $A/[]^$T, position_context: ^DocumentPositionContext) { -get_document_position_array :: proc(array: $A/[]^$T, position_context: ^DocumentPositionContext) { + for elem, i in array { + get_document_position(elem, position_context); + } +} - for elem, i in array { - get_document_position(elem, position_context); - } +get_document_position_dynamic_array :: proc (array: $A/[dynamic]^$T, position_context: ^DocumentPositionContext) { + for elem, i in array { + get_document_position(elem, position_context); + } } -get_document_position_dynamic_array :: proc(array: $A/[dynamic]^$T, position_context: ^DocumentPositionContext) { +position_in_node :: proc (node: ^ast.Node, position: common.AbsolutePosition) -> bool { + return node != nil && node.pos.offset <= position && position <= node.end.offset; +} - for elem, i in array { - get_document_position(elem, position_context); - } +get_document_position_node :: proc (node: ^ast.Node, position_context: ^DocumentPositionContext) { -} + using ast; -position_in_node :: proc(node: ^ast.Node, position: common.AbsolutePosition) -> bool { - return node != nil && node.pos.offset <= position && position <= node.end.offset; -} + if node == nil { + return; + } -get_document_position_node :: proc(node: ^ast.Node, position_context: ^DocumentPositionContext) { - - using ast; - - if node == nil { - return; - } - - if !position_in_node(node, position_context.position) { - return; - } - - switch n in node.derived { - case Bad_Expr: - case Ident: - position_context.identifier = node; - case Implicit: - case Undef: - case Basic_Lit: - case Ellipsis: - get_document_position(n.expr, position_context); - case Proc_Lit: - get_document_position(n.type, position_context); - - if position_in_node(n.body, position_context.position) { - position_context.function = cast(^Proc_Lit)node; - get_document_position(n.body, position_context); - } - case Comp_Lit: - //only set this for the parent comp literal, since we will need to walk through it to infer types. - if position_context.parent_comp_lit == nil { - position_context.parent_comp_lit = cast(^Comp_Lit)node; - } - - position_context.comp_lit = cast(^Comp_Lit)node; - - get_document_position(n.type, position_context); - get_document_position(n.elems, position_context); - case Tag_Expr: - get_document_position(n.expr, position_context); - case Unary_Expr: - get_document_position(n.expr, position_context); - case Binary_Expr: - position_context.binary = cast(^Binary_Expr)node; - get_document_position(n.left, position_context); - get_document_position(n.right, position_context); - case Paren_Expr: - get_document_position(n.expr, position_context); - case Call_Expr: - if position_context.hint == .SignatureHelp { - position_context.call = cast(^Expr)node; - } - get_document_position(n.expr, position_context); - get_document_position(n.args, position_context); - case Selector_Expr: - if position_context.hint == .Completion { - if n.field != nil && n.field.pos.line == position_context.line { - position_context.selector = n.expr; - position_context.field = n.field; - } - } - - else if (position_context.hint == .Definition || position_context.hint == .Hover) && n.field != nil { - position_context.selector = n.expr; - position_context.field = n.field; - get_document_position(n.expr, position_context); - get_document_position(n.field, position_context); - } - - else { - get_document_position(n.expr, position_context); - get_document_position(n.field, position_context); - } - case Index_Expr: - get_document_position(n.expr, position_context); - get_document_position(n.index, position_context); - case Deref_Expr: - get_document_position(n.expr, position_context); - case Slice_Expr: - get_document_position(n.expr, position_context); - get_document_position(n.low, position_context); - get_document_position(n.high, position_context); - case Field_Value: - get_document_position(n.field, position_context); - get_document_position(n.value, position_context); - case Ternary_Expr: - get_document_position(n.cond, position_context); - get_document_position(n.x, position_context); - get_document_position(n.y, position_context); - case Ternary_If_Expr: - get_document_position(n.x, position_context); - get_document_position(n.cond, position_context); - get_document_position(n.y, position_context); - case Ternary_When_Expr: - get_document_position(n.x, position_context); - get_document_position(n.cond, position_context); - get_document_position(n.y, position_context); - case Type_Assertion: - get_document_position(n.expr, position_context); - get_document_position(n.type, position_context); - case Type_Cast: - get_document_position(n.type, position_context); - get_document_position(n.expr, position_context); - case Auto_Cast: - get_document_position(n.expr, position_context); - case Bad_Stmt: - case Empty_Stmt: - case Expr_Stmt: - get_document_position(n.expr, position_context); - case Tag_Stmt: - r := cast(^Tag_Stmt)node; - get_document_position(r.stmt, position_context); - case Assign_Stmt: - position_context.assign = cast(^Assign_Stmt)node; - get_document_position(n.lhs, position_context); - get_document_position(n.rhs, position_context); - case Block_Stmt: - get_document_position(n.label, position_context); - get_document_position(n.stmts, position_context); - case If_Stmt: - get_document_position(n.label, position_context); - get_document_position(n.init, position_context); - get_document_position(n.cond, position_context); - get_document_position(n.body, position_context); - get_document_position(n.else_stmt, position_context); - case When_Stmt: - get_document_position(n.cond, position_context); - get_document_position(n.body, position_context); - get_document_position(n.else_stmt, position_context); - case Return_Stmt: - position_context.returns = cast(^Return_Stmt)node; - get_document_position(n.results, position_context); - case Defer_Stmt: - get_document_position(n.stmt, position_context); - case For_Stmt: - get_document_position(n.label, position_context); - get_document_position(n.init, position_context); - get_document_position(n.cond, position_context); - get_document_position(n.post, position_context); - get_document_position(n.body, position_context); - case Range_Stmt: - get_document_position(n.label, position_context); - get_document_position(n.val0, position_context); - get_document_position(n.val1, position_context); - get_document_position(n.expr, position_context); - get_document_position(n.body, position_context); - case Case_Clause: - position_context.case_clause = cast(^Case_Clause)node; - get_document_position(n.list, position_context); - get_document_position(n.body, position_context); - case Switch_Stmt: - position_context.switch_stmt = cast(^Switch_Stmt)node; - get_document_position(n.label, position_context); - get_document_position(n.init, position_context); - get_document_position(n.cond, position_context); - get_document_position(n.body, position_context); - case Type_Switch_Stmt: - position_context.switch_type_stmt = cast(^Type_Switch_Stmt)node; - get_document_position(n.label, position_context); - get_document_position(n.tag, position_context); - get_document_position(n.expr, position_context); - get_document_position(n.body, position_context); - case Branch_Stmt: - get_document_position(n.label, position_context); - case Using_Stmt: - get_document_position(n.list, position_context); - case Bad_Decl: - case Value_Decl: - position_context.value_decl = cast(^Value_Decl)node; - get_document_position(n.attributes, position_context); - get_document_position(n.names, position_context); - get_document_position(n.type, position_context); - get_document_position(n.values, position_context); - case Package_Decl: - case Import_Decl: - case Foreign_Block_Decl: - get_document_position(n.attributes, position_context); - get_document_position(n.foreign_library, position_context); - get_document_position(n.body, position_context); - case Foreign_Import_Decl: - get_document_position(n.name, position_context); - case Proc_Group: - get_document_position(n.args, position_context); - case Attribute: - get_document_position(n.elems, position_context); - case Field: - get_document_position(n.names, position_context); - get_document_position(n.type, position_context); - get_document_position(n.default_value, position_context); - case Field_List: - get_document_position(n.list, position_context); - case Typeid_Type: - get_document_position(n.specialization, position_context); - case Helper_Type: - get_document_position(n.type, position_context); - case Distinct_Type: - get_document_position(n.type, position_context); - case Poly_Type: - get_document_position(n.type, position_context); - get_document_position(n.specialization, position_context); - case Proc_Type: - get_document_position(n.params, position_context); - get_document_position(n.results, position_context); - case Pointer_Type: - get_document_position(n.elem, position_context); - case Array_Type: - get_document_position(n.len, position_context); - get_document_position(n.elem, position_context); - case Dynamic_Array_Type: - get_document_position(n.elem, position_context); - case Struct_Type: - get_document_position(n.poly_params, position_context); - get_document_position(n.align, position_context); - get_document_position(n.fields, position_context); - case Union_Type: - get_document_position(n.poly_params, position_context); - get_document_position(n.align, position_context); - get_document_position(n.variants, position_context); - case Enum_Type: - get_document_position(n.base_type, position_context); - get_document_position(n.fields, position_context); - case Bit_Set_Type: - get_document_position(n.elem, position_context); - get_document_position(n.underlying, position_context); - case Map_Type: - get_document_position(n.key, position_context); - get_document_position(n.value, position_context); - case Implicit_Selector_Expr: - position_context.implicit = true; - get_document_position(n.field, position_context); - case: - log.errorf("Unhandled node kind: %T", n); - } + if !position_in_node(node, position_context.position) { + return; + } + switch n in node.derived { + case Bad_Expr: + case Ident: + position_context.identifier = node; + case Implicit: + case Undef: + case Basic_Lit: + case Ellipsis: + get_document_position(n.expr, position_context); + case Proc_Lit: + get_document_position(n.type, position_context); + + if position_in_node(n.body, position_context.position) { + position_context.function = cast(^Proc_Lit)node; + get_document_position(n.body, position_context); + } + case Comp_Lit: + //only set this for the parent comp literal, since we will need to walk through it to infer types. + if position_context.parent_comp_lit == nil { + position_context.parent_comp_lit = cast(^Comp_Lit)node; + } + + position_context.comp_lit = cast(^Comp_Lit)node; + + get_document_position(n.type, position_context); + get_document_position(n.elems, position_context); + case Tag_Expr: + get_document_position(n.expr, position_context); + case Unary_Expr: + get_document_position(n.expr, position_context); + case Binary_Expr: + position_context.binary = cast(^Binary_Expr)node; + get_document_position(n.left, position_context); + get_document_position(n.right, position_context); + case Paren_Expr: + get_document_position(n.expr, position_context); + case Call_Expr: + if position_context.hint == .SignatureHelp { + position_context.call = cast(^Expr)node; + } + get_document_position(n.expr, position_context); + get_document_position(n.args, position_context); + case Selector_Expr: + if position_context.hint == .Completion { + if n.field != nil && n.field.pos.line == position_context.line { + position_context.selector = n.expr; + position_context.field = n.field; + } + } else if (position_context.hint == .Definition || position_context.hint == .Hover) && n.field != nil { + position_context.selector = n.expr; + position_context.field = n.field; + get_document_position(n.expr, position_context); + get_document_position(n.field, position_context); + } else { + get_document_position(n.expr, position_context); + get_document_position(n.field, position_context); + } + case Index_Expr: + get_document_position(n.expr, position_context); + get_document_position(n.index, position_context); + case Deref_Expr: + get_document_position(n.expr, position_context); + case Slice_Expr: + get_document_position(n.expr, position_context); + get_document_position(n.low, position_context); + get_document_position(n.high, position_context); + case Field_Value: + get_document_position(n.field, position_context); + get_document_position(n.value, position_context); + case Ternary_Expr: + get_document_position(n.cond, position_context); + get_document_position(n.x, position_context); + get_document_position(n.y, position_context); + case Ternary_If_Expr: + get_document_position(n.x, position_context); + get_document_position(n.cond, position_context); + get_document_position(n.y, position_context); + case Ternary_When_Expr: + get_document_position(n.x, position_context); + get_document_position(n.cond, position_context); + get_document_position(n.y, position_context); + case Type_Assertion: + get_document_position(n.expr, position_context); + get_document_position(n.type, position_context); + case Type_Cast: + get_document_position(n.type, position_context); + get_document_position(n.expr, position_context); + case Auto_Cast: + get_document_position(n.expr, position_context); + case Bad_Stmt: + case Empty_Stmt: + case Expr_Stmt: + get_document_position(n.expr, position_context); + case Tag_Stmt: + r := cast(^Tag_Stmt)node; + get_document_position(r.stmt, position_context); + case Assign_Stmt: + position_context.assign = cast(^Assign_Stmt)node; + get_document_position(n.lhs, position_context); + get_document_position(n.rhs, position_context); + case Block_Stmt: + get_document_position(n.label, position_context); + get_document_position(n.stmts, position_context); + case If_Stmt: + get_document_position(n.label, position_context); + get_document_position(n.init, position_context); + get_document_position(n.cond, position_context); + get_document_position(n.body, position_context); + get_document_position(n.else_stmt, position_context); + case When_Stmt: + get_document_position(n.cond, position_context); + get_document_position(n.body, position_context); + get_document_position(n.else_stmt, position_context); + case Return_Stmt: + position_context.returns = cast(^Return_Stmt)node; + get_document_position(n.results, position_context); + case Defer_Stmt: + get_document_position(n.stmt, position_context); + case For_Stmt: + get_document_position(n.label, position_context); + get_document_position(n.init, position_context); + get_document_position(n.cond, position_context); + get_document_position(n.post, position_context); + get_document_position(n.body, position_context); + case Range_Stmt: + get_document_position(n.label, position_context); + get_document_position(n.val0, position_context); + get_document_position(n.val1, position_context); + get_document_position(n.expr, position_context); + get_document_position(n.body, position_context); + case Case_Clause: + position_context.case_clause = cast(^Case_Clause)node; + get_document_position(n.list, position_context); + get_document_position(n.body, position_context); + case Switch_Stmt: + position_context.switch_stmt = cast(^Switch_Stmt)node; + get_document_position(n.label, position_context); + get_document_position(n.init, position_context); + get_document_position(n.cond, position_context); + get_document_position(n.body, position_context); + case Type_Switch_Stmt: + position_context.switch_type_stmt = cast(^Type_Switch_Stmt)node; + get_document_position(n.label, position_context); + get_document_position(n.tag, position_context); + get_document_position(n.expr, position_context); + get_document_position(n.body, position_context); + case Branch_Stmt: + get_document_position(n.label, position_context); + case Using_Stmt: + get_document_position(n.list, position_context); + case Bad_Decl: + case Value_Decl: + position_context.value_decl = cast(^Value_Decl)node; + get_document_position(n.attributes, position_context); + get_document_position(n.names, position_context); + get_document_position(n.type, position_context); + get_document_position(n.values, position_context); + case Package_Decl: + case Import_Decl: + case Foreign_Block_Decl: + get_document_position(n.attributes, position_context); + get_document_position(n.foreign_library, position_context); + get_document_position(n.body, position_context); + case Foreign_Import_Decl: + get_document_position(n.name, position_context); + case Proc_Group: + get_document_position(n.args, position_context); + case Attribute: + get_document_position(n.elems, position_context); + case Field: + get_document_position(n.names, position_context); + get_document_position(n.type, position_context); + get_document_position(n.default_value, position_context); + case Field_List: + get_document_position(n.list, position_context); + case Typeid_Type: + get_document_position(n.specialization, position_context); + case Helper_Type: + get_document_position(n.type, position_context); + case Distinct_Type: + get_document_position(n.type, position_context); + case Poly_Type: + get_document_position(n.type, position_context); + get_document_position(n.specialization, position_context); + case Proc_Type: + get_document_position(n.params, position_context); + get_document_position(n.results, position_context); + case Pointer_Type: + get_document_position(n.elem, position_context); + case Array_Type: + get_document_position(n.len, position_context); + get_document_position(n.elem, position_context); + case Dynamic_Array_Type: + get_document_position(n.elem, position_context); + case Struct_Type: + get_document_position(n.poly_params, position_context); + get_document_position(n.align, position_context); + get_document_position(n.fields, position_context); + case Union_Type: + get_document_position(n.poly_params, position_context); + get_document_position(n.align, position_context); + get_document_position(n.variants, position_context); + case Enum_Type: + get_document_position(n.base_type, position_context); + get_document_position(n.fields, position_context); + case Bit_Set_Type: + get_document_position(n.elem, position_context); + get_document_position(n.underlying, position_context); + case Map_Type: + get_document_position(n.key, position_context); + get_document_position(n.value, position_context); + case Implicit_Selector_Expr: + position_context.implicit = true; + get_document_position(n.field, position_context); + case: + log.errorf("Unhandled node kind: %T", n); + } }
\ No newline at end of file |