diff options
Diffstat (limited to 'typecheck.c')
| -rw-r--r-- | typecheck.c | 42 |
1 files changed, 35 insertions, 7 deletions
diff --git a/typecheck.c b/typecheck.c index ac6ed811..47fe8c64 100644 --- a/typecheck.c +++ b/typecheck.c @@ -403,13 +403,41 @@ type_t *get_type(env_t *env, ast_t *ast) } case MethodCall: { auto call = Match(ast, MethodCall); - type_t *fn_type_t = get_method_type(env, call->self, call->name); - if (!fn_type_t) - code_err(ast, "No such method!"); - if (fn_type_t->tag != FunctionType) - code_err(ast, "This isn't a method, it's a %T", fn_type_t); - auto fn_type = Match(fn_type_t, FunctionType); - return fn_type->ret; + type_t *self_value_t = value_type(get_type(env, call->self)); + switch (self_value_t->tag) { + case ArrayType: { + if (streq(call->name, "insert")) return Type(VoidType); + else if (streq(call->name, "insert_all")) return Type(VoidType); + else if (streq(call->name, "remove")) return Type(VoidType); + else if (streq(call->name, "sort")) return Type(VoidType); + else if (streq(call->name, "shuffle")) return Type(VoidType); + else if (streq(call->name, "random")) + return Type(PointerType, .pointed=Match(self_value_t, ArrayType)->item_type, .is_optional=true, .is_readonly=true); + else if (streq(call->name, "clear")) return Type(VoidType); + else if (streq(call->name, "compact")) return Type(VoidType); + else if (streq(call->name, "contains")) return Type(VoidType); + else if (streq(call->name, "slice")) return self_value_t; + else if (streq(call->name, "concat")) return self_value_t; + else code_err(ast, "There is no '%s' method for arrays", call->name); + } + case TableType: { + auto table = Match(self_value_t, TableType); + if (streq(call->name, "get")) return Type(PointerType, .pointed=table->value_type, .is_readonly=true, .is_optional=true); + else if (streq(call->name, "set")) return Type(VoidType); + else if (streq(call->name, "remove")) return Type(VoidType); + else if (streq(call->name, "clear")) return Type(VoidType); + else code_err(ast, "There is no '%s' method for tables", call->name); + } + default: { + type_t *fn_type_t = get_method_type(env, call->self, call->name); + if (!fn_type_t) + code_err(ast, "No such method!"); + if (fn_type_t->tag != FunctionType) + code_err(ast, "This isn't a method, it's a %T", fn_type_t); + auto fn_type = Match(fn_type_t, FunctionType); + return fn_type->ret; + } + } } case Block: { auto block = Match(ast, Block); |