New poly resolve system

5 files changed, 731 insertions, 565 deletions

diff --git a/src/common/ast.odin b/src/common/ast.odin
index 9adc57c..a2e5d42 100644
--- a/src/common/ast.odin
+++ b/src/common/ast.odin
@@ -163,7 +163,7 @@ get_attribute_objc_is_class_method :: proc(
 	return false
 }
 
-unwrap_pointer :: proc(expr: ^ast.Expr) -> (ast.Ident, int, bool) {
+unwrap_pointer_ident :: proc(expr: ^ast.Expr) -> (ast.Ident, int, bool) {
 	n := 0
 	expr := expr
 	for expr != nil {
@@ -186,6 +186,25 @@ unwrap_pointer :: proc(expr: ^ast.Expr) -> (ast.Ident, int, bool) {
 	return {}, n, false
 }
 
+unwrap_pointer_expr :: proc(expr: ^ast.Expr) -> (^ast.Expr, int, bool) {
+	n := 0
+	expr := expr
+	for expr != nil {
+		if pointer, ok := expr.derived.(^ast.Pointer_Type); ok {
+			expr = pointer.elem
+			n += 1
+		} else {
+			break
+		}
+	}
+
+	if expr == nil {
+		return {}, n, false
+	}
+
+	return expr, n, true
+}
+
 is_expr_basic_lit :: proc(expr: ^ast.Expr) -> bool {
 	_, ok := expr.derived.(^ast.Basic_Lit)
 	return ok
@@ -198,10 +217,10 @@ collect_value_decl :: proc(
 	skip_private: bool,
 ) {
 	if value_decl, ok := stmt.derived.(^ast.Value_Decl); ok {
-		is_deprecated   := false
+		is_deprecated := false
 		is_private_file := false
-		is_private_pkg  := false
-		is_builtin      := false
+		is_private_pkg := false
+		is_builtin := false
 
 		for attribute in value_decl.attributes {
 			for elem in attribute.elems {
@@ -238,7 +257,7 @@ collect_value_decl :: proc(
 		if is_private_file && skip_private {
 			return
 		}
-		
+
 		// If a private status is not explicitly set with an attribute above the declaration
 		// check the file comment.
 		if !is_private_file && !is_private_pkg && file.docs != nil {
@@ -247,7 +266,7 @@ collect_value_decl :: proc(
 				if strings.has_prefix(txt, "//+private") {
 					txt = strings.trim_prefix(txt, "//+private")
 					is_private_pkg = true
-						
+
 					if strings.has_prefix(txt, " ") {
 						txt = strings.trim_space(txt)
 						if txt == "file" {
@@ -264,7 +283,7 @@ collect_value_decl :: proc(
 			if value_decl.type != nil {
 				append(
 					exprs,
-					GlobalExpr{
+					GlobalExpr {
 						name = str,
 						name_expr = name,
 						expr = value_decl.type,
@@ -280,7 +299,7 @@ collect_value_decl :: proc(
 				if len(value_decl.values) > i {
 					append(
 						exprs,
-						GlobalExpr{
+						GlobalExpr {
 							name = str,
 							name_expr = name,
 							expr = value_decl.values[i],
diff --git a/src/server/analysis.odin b/src/server/analysis.odin
index c56f63c..d7c6263 100644
--- a/src/server/analysis.odin
+++ b/src/server/analysis.odin
@@ -135,217 +135,6 @@ reset_ast_context :: proc(ast_context: ^AstContext) {
 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
-*/
-
-resolve_poly_spec :: proc {
-	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,
-) {
-	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,
-) {
-	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
-
-	#partial switch v in node.derived {
-	case ^Ident:
-		return cast(^Expr)node
-	case ^Comp_Lit:
-		return v.type
-	case:
-		log.warnf("Unhandled poly to node kind %v", v)
-	}
-
-	return nil
-}
-
-resolve_poly_spec_node :: proc(
-	ast_context: ^AstContext,
-	call_node: ^ast.Node,
-	spec_node: ^ast.Node,
-	poly_map: ^map[string]^ast.Expr,
-) {
-	using ast
-
-	if call_node == nil || spec_node == nil {
-		return
-	}
-
-	#partial 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)
-	}
-}
 
 resolve_type_comp_literal :: proc(
 	ast_context: ^AstContext,
@@ -463,185 +252,6 @@ resolve_type_comp_literal :: proc(
 	return current_symbol, current_comp_lit, 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,
-) -> (
-	Symbol,
-	bool,
-) {
-	if params == nil {
-		return {}, false
-	}
-
-	if results == nil {
-		return {}, false
-	}
-
-	if ast_context.call == nil {
-		return {}, false
-	}
-
-	call_expr := ast_context.call
-	poly_map := make(map[string]^ast.Expr, 0, context.temp_allocator)
-	i := 0
-	count_required_params := 0
-
-	for param in params {
-		if param.default_value == nil {
-			count_required_params += 1
-		}
-
-		for name in param.names {
-			if len(call_expr.args) <= i {
-				break
-			}
-
-			if poly, ok := name.derived.(^ast.Poly_Type); ok {
-				poly_map[poly.type.name] = call_expr.args[i]
-			}
-
-			if param.type == nil {
-				continue
-			}
-
-			if type_id, ok := param.type.derived.(^ast.Typeid_Type); ok {
-				if type_id.specialization != nil &&
-				   !common.node_equal(
-						   call_expr.args[i],
-						   type_id.specialization,
-					   ) {
-					return {}, false
-				}
-			}
-
-			resolve_poly_spec_node(
-				ast_context,
-				call_expr.args[i],
-				param.type,
-				&poly_map,
-			)
-
-			i += 1
-		}
-	}
-
-	if count_required_params > len(call_expr.args) ||
-	   count_required_params == 0 ||
-	   len(call_expr.args) == 0 {
-		return {}, false
-	}
-
-	function_name := ""
-	function_range: common.Range
-
-	if ident, ok := call_expr.expr.derived.(^ast.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.(^ast.Selector_Expr); ok {
-		function_name = selector.field.name
-		function_range = common.get_token_range(selector, ast_context.file.src)
-	} else {
-		return {}, false
-	}
-
-	symbol := Symbol {
-		range = function_range,
-		type  = .Function,
-		name  = function_name,
-	}
-
-	return_types := make([dynamic]^ast.Field, ast_context.allocator)
-	argument_types := make([dynamic]^ast.Field, ast_context.allocator)
-
-	for result in results {
-		if result.type == nil {
-			continue
-		}
-
-		ident, wrapped, ok := common.unwrap_pointer(result.type)
-
-		if ok {
-			if m, ok := poly_map[ident.name]; ok {
-				field := cast(^ast.Field)clone_node(
-					result,
-					ast_context.allocator,
-					nil,
-				)
-				field.type = wrap_pointer(m, wrapped)
-				append(&return_types, field)
-			} else {
-				append(&return_types, result)
-			}
-		} else {
-			append(&return_types, result)
-		}
-	}
-
-	for param in params {
-		if len(param.names) == 0 {
-			continue
-		}
-
-		//check the name for poly
-		if poly_type, ok := param.names[0].derived.(^ast.Poly_Type);
-		   ok && param.type != nil {
-			if m, ok := poly_map[poly_type.type.name]; ok {
-				field := cast(^ast.Field)clone_node(
-					param,
-					ast_context.allocator,
-					nil,
-				)
-				field.type = m
-				append(&argument_types, field)
-			}
-		} else {
-			append(&argument_types, param)
-		}
-	}
-
-	symbol.value = SymbolProcedureValue {
-		return_types = return_types[:],
-		arg_types    = argument_types[:],
-	}
-
-	return symbol, true
-}
-
-resolve_generic_function_ast :: proc(
-	ast_context: ^AstContext,
-	proc_lit: ast.Proc_Lit,
-) -> (
-	Symbol,
-	bool,
-) {
-
-	using ast
-
-	if proc_lit.type.params == nil {
-		return Symbol{}, false
-	}
-
-	if proc_lit.type.results == nil {
-		return Symbol{}, false
-	}
-
-	if ast_context.call == nil {
-		return Symbol{}, false
-	}
-
-	return resolve_generic_function_symbol(
-		ast_context,
-		proc_lit.type.params.list,
-		proc_lit.type.results.list,
-	)
-}
 
 is_symbol_same_typed :: proc(
 	ast_context: ^AstContext,
@@ -1024,6 +634,8 @@ resolve_basic_lit :: proc(
 
 	value: SymbolUntypedValue
 
+	value.tok = basic_lit.tok
+
 	if len(basic_lit.tok.text) == 0 {
 		return {}, false
 	}
@@ -1688,7 +1300,7 @@ internal_resolve_type_identifier :: proc(
 				make_symbol_bitset_from_ast(ast_context, v^, node), true
 			return_symbol.name = node.name
 		case ^Proc_Lit:
-			if !v.type.generic {
+			if !is_procedure_generic(v.type) {
 				return_symbol, ok =
 					make_symbol_procedure_from_ast(
 						ast_context,
@@ -1811,7 +1423,7 @@ internal_resolve_type_identifier :: proc(
 				make_symbol_enum_from_ast(ast_context, v^, node), true
 			return_symbol.name = node.name
 		case ^Proc_Lit:
-			if !v.type.generic {
+			if !is_procedure_generic(v.type) {
 				return_symbol, ok =
 					make_symbol_procedure_from_ast(
 						ast_context,
@@ -2708,6 +2320,17 @@ make_int_ast :: proc(
 	return ident
 }
 
+make_ident_ast :: proc(
+	ast_context: ^AstContext,
+	pos: tokenizer.Pos,
+	end: tokenizer.Pos,
+	name: string,
+) -> ^ast.Ident {
+	ident := new_type(ast.Ident, pos, end, ast_context.allocator)
+	ident.name = name
+	return ident
+}
+
 make_int_basic_value :: proc(
 	ast_context: ^AstContext,
 	n: int,
@@ -3121,6 +2744,7 @@ make_symbol_struct_from_ast :: proc(
 		types  = types[:],
 		ranges = ranges[:],
 		usings = usings,
+		poly   = v.poly_params,
 	}
 
 	if _, ok := common.get_attribute_objc_class_name(attributes); ok {
@@ -3146,166 +2770,6 @@ make_symbol_struct_from_ast :: proc(
 	return symbol
 }
 
-resolve_poly_union :: proc(
-	ast_context: ^AstContext,
-	poly_params: ^ast.Field_List,
-	symbol: ^Symbol,
-) {
-	if ast_context.call == nil {
-		return
-	}
-
-	symbol_value := &symbol.value.(SymbolUnionValue)
-
-	if symbol_value == nil {
-		return
-	}
-
-	i := 0
-
-	poly_map := make(map[string]^ast.Expr, 0, context.temp_allocator)
-
-	for param in 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]
-				} else if poly, ok := name.derived.(^ast.Poly_Type); ok {
-					if poly.type != nil {
-						poly_map[poly.type.name] = ast_context.call.args[i]
-					}
-				}
-			}
-
-			i += 1
-		}
-	}
-
-	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 {
-						symbol_value.types[i] = expr
-					}
-				}
-			}
-		}
-	}
-}
-
-resolve_poly_struct :: proc(
-	ast_context: ^AstContext,
-	poly_params: ^ast.Field_List,
-	symbol: ^Symbol,
-) {
-	if ast_context.call == nil {
-		return
-	}
-
-	symbol_value := &symbol.value.(SymbolStructValue)
-
-	if symbol_value == nil {
-		return
-	}
-
-	i := 0
-
-	poly_map := make(map[string]^ast.Expr, 0, context.temp_allocator)
-
-	for param in 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]
-				} else if poly, ok := name.derived.(^ast.Poly_Type); ok {
-					if poly.type != nil {
-						poly_map[poly.type.name] = ast_context.call.args[i]
-					}
-				}
-			}
-
-			i += 1
-		}
-	}
-
-	Visit_Data :: struct {
-		poly_map:     map[string]^ast.Expr,
-		symbol_value: ^SymbolStructValue,
-		parent:       ^ast.Node,
-		i:            int,
-	}
-
-	visit :: proc(visitor: ^ast.Visitor, node: ^ast.Node) -> ^ast.Visitor {
-		if node == nil || visitor == nil {
-			return nil
-		}
-
-		data := cast(^Visit_Data)visitor.data
-
-		if ident, ok := node.derived.(^ast.Ident); ok {
-			if expr, ok := data.poly_map[ident.name]; ok {
-				if data.parent != nil {
-					#partial switch &v in data.parent.derived {
-					case ^ast.Array_Type:
-						v.elem = expr
-					case ^ast.Dynamic_Array_Type:
-						v.elem = expr
-					}
-				} else {
-					data.symbol_value.types[data.i] = expr
-				}
-			}
-		}
-
-		#partial switch v in node.derived {
-		case ^ast.Array_Type, ^ast.Dynamic_Array_Type, ^ast.Selector_Expr:
-			data.parent = node
-		}
-
-		return visitor
-	}
-
-	for type, i in symbol_value.types {
-		data := Visit_Data {
-			poly_map     = poly_map,
-			symbol_value = symbol_value,
-			i            = i,
-		}
-
-		visitor := ast.Visitor {
-			data  = &data,
-			visit = visit,
-		}
-
-		ast.walk(&visitor, type)
-	}
-}
-
 get_globals :: proc(file: ast.File, ast_context: ^AstContext) {
 	exprs := common.collect_globals(file)
 
diff --git a/src/server/collector.odin b/src/server/collector.odin
index 917dea6..5d91039 100644
--- a/src/server/collector.odin
+++ b/src/server/collector.odin
@@ -75,7 +75,7 @@ make_symbol_collection :: proc(
 	config: ^common.Config,
 ) -> SymbolCollection {
 	return(
-		SymbolCollection{
+		SymbolCollection {
 			allocator = allocator,
 			config = config,
 			packages = make(map[string]SymbolPackage, 16, allocator),
@@ -141,7 +141,7 @@ collect_procedure_fields :: proc(
 	value := SymbolProcedureValue {
 		return_types = returns[:],
 		arg_types    = args[:],
-		generic      = proc_type.generic,
+		generic      = is_procedure_generic(proc_type),
 	}
 
 	return value
@@ -435,7 +435,7 @@ collect_method :: proc(collection: ^SymbolCollection, symbol: Symbol) {
 			return
 		}
 
-		expr, _, ok := common.unwrap_pointer(value.arg_types[0].type)
+		expr, _, ok := common.unwrap_pointer_ident(value.arg_types[0].type)
 
 		if !ok {
 			return
@@ -514,7 +514,7 @@ collect_objc :: proc(
 
 				append(
 					&objc_struct.functions,
-					ObjcFunction{
+					ObjcFunction {
 						logical_name = get_index_unique_string_collection(
 							collection,
 							objc_name,
diff --git a/src/server/generics.odin b/src/server/generics.odin
new file mode 100644
index 0000000..5d07181
--- /dev/null
+++ b/src/server/generics.odin
@@ -0,0 +1,629 @@
+package server
+
+import "core:fmt"
+import "core:log"
+import "core:mem"
+import "core:odin/ast"
+import "core:odin/parser"
+import "core:odin/tokenizer"
+import "core:path/filepath"
+import path "core:path/slashpath"
+import "core:reflect"
+import "core:slice"
+import "core:sort"
+import "core:strconv"
+import "core:strings"
+import "core:unicode/utf8"
+
+import "shared:common"
+
+resolve_poly :: proc(
+	ast_context: ^AstContext,
+	call_node: ^ast.Expr,
+	call_symbol: Symbol,
+	poly_node: ^ast.Expr,
+	poly_map: ^map[string]^ast.Expr,
+) -> bool {
+	if poly_node == nil || call_node == nil {
+		return false
+	}
+
+	specialization: ^ast.Expr
+	type: ^ast.Expr
+
+	poly_node := poly_node
+	poly_node, _, _ = common.unwrap_pointer_expr(poly_node)
+
+	#partial switch v in poly_node.derived {
+	case ^ast.Typeid_Type:
+		specialization = v.specialization
+	case ^ast.Poly_Type:
+		specialization = v.specialization
+		type = v.type
+	}
+
+	if specialization == nil {
+		if type != nil {
+			if ident, ok := unwrap_ident(type); ok {
+				poly_map[ident.name] = make_ident_ast(
+					ast_context,
+					call_node.pos,
+					call_node.end,
+					call_symbol.name,
+				)
+			}
+		}
+		return true
+	}
+
+	#partial switch p in specialization.derived {
+	case ^ast.Call_Expr:
+		if call_struct, ok := call_node.derived.(^ast.Struct_Type); ok {
+			arg_index := 0
+			struct_value := call_symbol.value.(SymbolStructValue)
+
+			for arg in p.args {
+				if poly_type, ok := arg.derived.(^ast.Poly_Type); ok {
+					if poly_type.type == nil ||
+					   struct_value.poly == nil ||
+					   len(struct_value.args) <= arg_index {
+						return false
+					}
+
+					poly_map[poly_type.type.name] =
+						struct_value.args[arg_index]
+
+					arg_index += 1
+				}
+			}
+		}
+	case ^ast.Struct_Type:
+
+	case ^ast.Dynamic_Array_Type:
+		if call_array, ok := call_node.derived.(^ast.Dynamic_Array_Type); ok {
+			if poly_type, ok := p.elem.derived.(^ast.Poly_Type); ok {
+				if ident, ok := unwrap_ident(poly_type.type); ok {
+					poly_map[ident.name] = call_array.elem
+				}
+
+				if poly_type.specialization != nil {
+					return resolve_poly(
+						ast_context,
+						call_array.elem,
+						call_symbol,
+						p.elem,
+						poly_map,
+					)
+				}
+				return true
+			}
+		}
+	case ^ast.Array_Type:
+		if call_array, ok := call_node.derived.(^ast.Array_Type); ok {
+			if poly_type, ok := p.elem.derived.(^ast.Poly_Type); ok {
+				if ident, ok := unwrap_ident(poly_type.type); ok {
+					poly_map[ident.name] = call_array.elem
+				}
+
+				if poly_type.specialization != nil {
+					return resolve_poly(
+						ast_context,
+						call_array.elem,
+						call_symbol,
+						p.elem,
+						poly_map,
+					)
+				}
+				return true
+			}
+		}
+	case ^ast.Map_Type:
+		if call_map, ok := call_node.derived.(^ast.Map_Type); ok {
+			if poly_type, ok := p.key.derived.(^ast.Poly_Type); ok {
+				if ident, ok := unwrap_ident(poly_type.type); ok {
+					poly_map[ident.name] = call_map.key
+				}
+
+				if poly_type.specialization != nil {
+					return resolve_poly(
+						ast_context,
+						call_map.key,
+						call_symbol,
+						p.key,
+						poly_map,
+					)
+				}
+				return true
+			} else if poly_type, ok := p.value.derived.(^ast.Poly_Type); ok {
+				if ident, ok := unwrap_ident(poly_type.type); ok {
+					poly_map[ident.name] = call_map.value
+				}
+
+				if poly_type.specialization != nil {
+					return resolve_poly(
+						ast_context,
+						call_map.value,
+						call_symbol,
+						p.value,
+						poly_map,
+					)
+				}
+				return true
+			}
+		}
+	case ^ast.Multi_Pointer_Type:
+		if call_pointer, ok := call_node.derived.(^ast.Multi_Pointer_Type);
+		   ok {
+			if poly_type, ok := p.elem.derived.(^ast.Poly_Type); ok {
+				if ident, ok := unwrap_ident(poly_type.type); ok {
+					poly_map[ident.name] = call_pointer.elem
+				}
+
+				if poly_type.specialization != nil {
+					return resolve_poly(
+						ast_context,
+						call_pointer.elem,
+						call_symbol,
+						p.elem,
+						poly_map,
+					)
+				}
+				return true
+			}
+		}
+	case ^ast.Ident:
+		if n, ok := call_node.derived.(^ast.Ident); ok {
+			return true
+		}
+	case:
+		log.panicf("Unhandled specialization %v", specialization.derived)
+	}
+
+	return false
+}
+
+find_and_replace_poly_type :: proc(
+	expr: ^ast.Expr,
+	poly_map: ^map[string]^ast.Expr,
+) {
+	is_in_poly_map :: proc(
+		node: ^ast.Node,
+		poly_map: ^map[string]^ast.Expr,
+	) -> (
+		^ast.Expr,
+		bool,
+	) {
+		if node == nil {
+			return {}, false
+		}
+
+		if ident, ok := node.derived.(^ast.Ident); ok {
+			if v, ok := poly_map[ident.name]; ok {
+				return v, ok
+			}
+		}
+
+		return nil, false
+	}
+
+	visit_function :: proc(
+		visitor: ^ast.Visitor,
+		node: ^ast.Node,
+	) -> ^ast.Visitor {
+		if node == nil {
+			return nil
+		}
+
+		poly_map := cast(^map[string]^ast.Expr)visitor.data
+
+		#partial switch v in node.derived {
+		case ^ast.Dynamic_Array_Type:
+			if expr, ok := is_in_poly_map(v.elem, poly_map); ok {
+				v.elem = expr
+			}
+		case ^ast.Array_Type:
+			if expr, ok := is_in_poly_map(v.elem, poly_map); ok {
+				v.elem = expr
+			} else if expr, ok := is_in_poly_map(v.len, poly_map); ok {
+				v.len = expr
+			}
+		case ^ast.Multi_Pointer_Type:
+			if expr, ok := is_in_poly_map(v.elem, poly_map); ok {
+				v.elem = expr
+			}
+		case ^ast.Pointer_Type:
+			if expr, ok := is_in_poly_map(v.elem, poly_map); ok {
+				v.elem = expr
+			}
+		}
+
+		return visitor
+	}
+
+	visitor := ast.Visitor {
+		data  = poly_map,
+		visit = visit_function,
+	}
+
+	ast.walk(&visitor, expr)
+}
+
+resolve_generic_function :: proc {
+	resolve_generic_function_ast,
+	resolve_generic_function_symbol,
+}
+
+resolve_generic_function_ast :: proc(
+	ast_context: ^AstContext,
+	proc_lit: ast.Proc_Lit,
+) -> (
+	Symbol,
+	bool,
+) {
+
+	using ast
+
+	if proc_lit.type.params == nil {
+		return Symbol{}, false
+	}
+
+	if proc_lit.type.results == nil {
+		return Symbol{}, false
+	}
+
+	if ast_context.call == nil {
+		return Symbol{}, false
+	}
+
+	return resolve_generic_function_symbol(
+		ast_context,
+		proc_lit.type.params.list,
+		proc_lit.type.results.list,
+	)
+}
+
+
+resolve_generic_function_symbol :: proc(
+	ast_context: ^AstContext,
+	params: []^ast.Field,
+	results: []^ast.Field,
+) -> (
+	Symbol,
+	bool,
+) {
+	if params == nil {
+		return {}, false
+	}
+
+	if results == nil {
+		return {}, false
+	}
+
+	if ast_context.call == nil {
+		return {}, false
+	}
+
+	call_expr := ast_context.call
+
+	poly_map := make(map[string]^ast.Expr, 0, context.temp_allocator)
+
+	i := 0
+	count_required_params := 0
+
+	for param in params {
+		if param.default_value == nil {
+			count_required_params += 1
+		}
+
+		for name in param.names {
+			if len(call_expr.args) <= i {
+				break
+			}
+
+			if param.type == nil {
+				continue
+			}
+
+			reset_ast_context(ast_context)
+
+			if symbol, ok := resolve_type_expression(
+				ast_context,
+				call_expr.args[i],
+			); ok {
+				symbol_expr := symbol_to_expr(
+					symbol,
+					call_expr.args[i].pos.file,
+					context.temp_allocator,
+				)
+
+				if symbol_expr == nil {
+					continue
+				}
+
+				if resolve_poly(
+					ast_context,
+					symbol_expr,
+					symbol,
+					param.type,
+					&poly_map,
+				) {
+					if poly, ok := name.derived.(^ast.Poly_Type); ok {
+						poly_map[poly.type.name] = call_expr.args[i]
+					}
+				}
+			}
+
+			i += 1
+		}
+	}
+
+	for k, v in poly_map {
+		//fmt.println(k, v.derived)
+	}
+
+	if count_required_params > len(call_expr.args) ||
+	   count_required_params == 0 ||
+	   len(call_expr.args) == 0 {
+		return {}, false
+	}
+
+	function_name := ""
+	function_range: common.Range
+
+	if ident, ok := call_expr.expr.derived.(^ast.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.(^ast.Selector_Expr); ok {
+		function_name = selector.field.name
+		function_range = common.get_token_range(selector, ast_context.file.src)
+	} else {
+		return {}, false
+	}
+
+	symbol := Symbol {
+		range = function_range,
+		type  = .Function,
+		name  = function_name,
+		pkg   = ast_context.current_package,
+	}
+
+	return_types := make([dynamic]^ast.Field, ast_context.allocator)
+	argument_types := make([dynamic]^ast.Field, ast_context.allocator)
+
+	for result in results {
+		if result.type == nil {
+			continue
+		}
+
+		field := cast(^ast.Field)clone_node(result, ast_context.allocator, nil)
+
+		if ident, ok := unwrap_ident(field.type); ok {
+			if expr, ok := poly_map[ident.name]; ok {
+				field.type = expr
+			}
+		}
+
+		find_and_replace_poly_type(field.type, &poly_map)
+
+		append(&return_types, field)
+	}
+
+	for param in params {
+		if len(param.names) == 0 {
+			continue
+		}
+
+		//check the name for poly
+		if poly_type, ok := param.names[0].derived.(^ast.Poly_Type);
+		   ok && param.type != nil {
+			if m, ok := poly_map[poly_type.type.name]; ok {
+				field := cast(^ast.Field)clone_node(
+					param,
+					ast_context.allocator,
+					nil,
+				)
+				field.type = m
+				append(&argument_types, field)
+			}
+		} else {
+			append(&argument_types, param)
+		}
+	}
+
+	symbol.value = SymbolProcedureValue {
+		return_types = return_types[:],
+		arg_types    = argument_types[:],
+	}
+
+	return symbol, true
+}
+
+is_procedure_generic :: proc(proc_type: ^ast.Proc_Type) -> bool {
+	if proc_type.generic {
+		return true
+	}
+
+	for param in proc_type.params.list {
+		if param.type == nil {
+			continue
+		}
+
+		if expr, _, ok := common.unwrap_pointer_expr(param.type); ok {
+			if _, ok := expr.derived.(^ast.Poly_Type); ok {
+				return true
+			}
+		}
+
+	}
+
+	return false
+}
+
+
+resolve_poly_struct :: proc(
+	ast_context: ^AstContext,
+	poly_params: ^ast.Field_List,
+	symbol: ^Symbol,
+) {
+	if ast_context.call == nil {
+		return
+	}
+
+	symbol_value := &symbol.value.(SymbolStructValue)
+
+	if symbol_value == nil {
+		return
+	}
+
+	i := 0
+
+	poly_map := make(map[string]^ast.Expr, 0, context.temp_allocator)
+	args := make([dynamic]^ast.Expr, 0, context.temp_allocator)
+
+	for param in 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]
+				} else if poly, ok := name.derived.(^ast.Poly_Type); ok {
+					if poly.type != nil {
+						poly_map[poly.type.name] = ast_context.call.args[i]
+					}
+				}
+			}
+
+			append(&args, ast_context.call.args[i])
+
+			i += 1
+		}
+	}
+
+	Visit_Data :: struct {
+		poly_map:     map[string]^ast.Expr,
+		symbol_value: ^SymbolStructValue,
+		parent:       ^ast.Node,
+		i:            int,
+		poly_index:   int,
+	}
+
+	visit :: proc(visitor: ^ast.Visitor, node: ^ast.Node) -> ^ast.Visitor {
+		if node == nil || visitor == nil {
+			return nil
+		}
+
+		data := cast(^Visit_Data)visitor.data
+
+		if ident, ok := node.derived.(^ast.Ident); ok {
+			if expr, ok := data.poly_map[ident.name]; ok {
+				if data.parent != nil {
+					#partial switch &v in data.parent.derived {
+					case ^ast.Array_Type:
+						v.elem = expr
+					case ^ast.Dynamic_Array_Type:
+						v.elem = expr
+					}
+				} else {
+					data.symbol_value.types[data.i] = expr
+					data.poly_index += 1
+				}
+			}
+		}
+
+		#partial switch v in node.derived {
+		case ^ast.Array_Type, ^ast.Dynamic_Array_Type, ^ast.Selector_Expr:
+			data.parent = node
+		}
+
+		return visitor
+	}
+
+	for type, i in symbol_value.types {
+		data := Visit_Data {
+			poly_map     = poly_map,
+			symbol_value = symbol_value,
+			i            = i,
+		}
+
+		visitor := ast.Visitor {
+			data  = &data,
+			visit = visit,
+		}
+
+		ast.walk(&visitor, type)
+	}
+
+	symbol_value.args = args[:]
+}
+
+
+resolve_poly_union :: proc(
+	ast_context: ^AstContext,
+	poly_params: ^ast.Field_List,
+	symbol: ^Symbol,
+) {
+	if ast_context.call == nil {
+		return
+	}
+
+	symbol_value := &symbol.value.(SymbolUnionValue)
+
+	if symbol_value == nil {
+		return
+	}
+
+	i := 0
+
+	poly_map := make(map[string]^ast.Expr, 0, context.temp_allocator)
+
+	for param in 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]
+				} else if poly, ok := name.derived.(^ast.Poly_Type); ok {
+					if poly.type != nil {
+						poly_map[poly.type.name] = ast_context.call.args[i]
+					}
+				}
+			}
+
+			i += 1
+		}
+	}
+
+	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 {
+						symbol_value.types[i] = expr
+					}
+				}
+			}
+		}
+	}
+}
diff --git a/src/server/symbol.odin b/src/server/symbol.odin
index d6a6aa5..a62c8cf 100644
--- a/src/server/symbol.odin
+++ b/src/server/symbol.odin
@@ -1,8 +1,11 @@
 package server
 
+import "core:fmt"
 import "core:hash"
+import "core:log"
 import "core:mem"
 import "core:odin/ast"
+import "core:odin/tokenizer"
 import "core:path/filepath"
 import path "core:path/slashpath"
 import "core:slice"
@@ -22,6 +25,7 @@ SymbolStructValue :: struct {
 	types:  []^ast.Expr,
 	usings: map[int]bool,
 	poly:   ^ast.Field_List,
+	args:   []^ast.Expr, //The arguments in the call expression for poly
 }
 
 SymbolPackageValue :: struct {}
@@ -83,6 +87,7 @@ SymbolUntypedValue :: struct {
 		String,
 		Bool,
 	},
+	tok:  tokenizer.Token,
 }
 
 SymbolMapValue :: struct {
@@ -227,7 +232,9 @@ free_symbol :: proc(symbol: Symbol, allocator: mem.Allocator) {
 	case SymbolMapValue:
 		common.free_ast(v.key, allocator)
 		common.free_ast(v.value, allocator)
-	case SymbolUntypedValue, SymbolPackageValue:
+	case SymbolUntypedValue:
+		delete(v.tok.text)
+	case SymbolPackageValue:
 	}
 }
 
@@ -284,3 +291,50 @@ symbol_kind_to_type :: proc(type: SymbolType) -> SymbolKind {
 		return .Null
 	}
 }
+
+symbol_to_expr :: proc(
+	symbol: Symbol,
+	file: string,
+	allocator := context.temp_allocator,
+) -> ^ast.Expr {
+
+	pos := tokenizer.Pos {
+		file = file,
+	}
+
+	end := tokenizer.Pos {
+		file = file,
+	}
+
+	#partial switch v in symbol.value {
+	case SymbolDynamicArrayValue:
+		type := new_type(ast.Dynamic_Array_Type, pos, end, allocator)
+		type.elem = v.expr
+		return type
+	case SymbolFixedArrayValue:
+		type := new_type(ast.Array_Type, pos, end, allocator)
+		type.elem = v.expr
+		return type
+	case SymbolMapValue:
+		type := new_type(ast.Map_Type, pos, end, allocator)
+		type.key = v.key
+		type.value = v.value
+		return type
+	case SymbolBasicValue:
+		return v.ident
+	case SymbolSliceValue:
+		type := new_type(ast.Array_Type, pos, end, allocator)
+		type.elem = v.expr
+		return type
+	case SymbolStructValue:
+		type := new_type(ast.Struct_Type, pos, end, allocator)
+		return type
+	case SymbolUntypedValue:
+		type := new_type(ast.Basic_Lit, pos, end, allocator)
+		return type
+	case:
+		log.errorf("Unhandled symbol %v", symbol)
+	}
+
+	return nil
+}