v/vlib/v/checker/checker.v

5900 lines
187 KiB
V
Raw Normal View History

// Copyright (c) 2019-2021 Alexander Medvednikov. All rights reserved.
2020-08-27 06:46:18 +02:00
// Use of this source code is governed by an MIT license that can be found in the LICENSE file.
module checker
import os
import strings
2020-04-25 17:49:16 +02:00
import v.ast
import v.vmod
2020-04-25 17:49:16 +02:00
import v.table
import v.token
import v.pref
import v.util
import v.errors
import v.pkgconfig
2020-02-19 19:54:36 +01:00
const (
max_nr_errors = 300
match_exhaustive_cutoff_limit = 10
int_min = int(0x80000000)
int_max = 0x7FFFFFFF
2020-02-19 19:54:36 +01:00
)
const (
valid_comp_if_os = ['windows', 'ios', 'macos', 'mach', 'darwin', 'hpux', 'gnu', 'qnx',
'linux', 'freebsd', 'openbsd', 'netbsd', 'bsd', 'dragonfly', 'android', 'solaris', 'haiku',
'linux_or_macos',
]
valid_comp_if_compilers = ['gcc', 'tinyc', 'clang', 'mingw', 'msvc', 'cplusplus']
valid_comp_if_platforms = ['amd64', 'aarch64', 'x64', 'x32', 'little_endian', 'big_endian']
valid_comp_if_other = ['js', 'debug', 'test', 'glibc', 'prealloc', 'no_bounds_checking']
array_builtin_methods = ['filter', 'clone', 'repeat', 'reverse', 'map', 'slice', 'sort',
'contains', 'index']
)
pub struct Checker {
pref &pref.Preferences // Preferences shared from V struct
2020-05-09 15:16:48 +02:00
pub mut:
table &table.Table
file &ast.File = 0
nr_errors int
nr_warnings int
errors []errors.Error
warnings []errors.Warning
error_lines []int // to avoid printing multiple errors for the same line
expected_type table.Type
expected_or_type table.Type // fn() or { 'this type' } eg. string. expected or block type
cur_fn &ast.FnDecl // current function
const_decl string
const_deps []string
const_names []string
global_names []string
locked_names []string // vars that are currently locked
rlocked_names []string // vars that are currently read-locked
in_for_count int // if checker is currently in a for loop
2020-10-14 12:49:55 +02:00
// checked_ident string // to avoid infinite checker loops
returns bool
scope_returns bool
mod string // current module name
is_builtin_mod bool // are we in `builtin`?
inside_unsafe bool
inside_const bool
inside_anon_fn bool
skip_flags bool // should `#flag` and `#include` be skipped
cur_generic_types []table.Type
mut:
2021-02-05 08:05:13 +01:00
files []ast.File
expr_level int // to avoid infinite recursion segfaults due to compiler bugs
inside_sql bool // to handle sql table fields pseudo variables
cur_orm_ts table.TypeSymbol
error_details []string
for_in_mut_val_name string
fn_mut_arg_names []string
vmod_file_content string // needed for @VMOD_FILE, contents of the file, *NOT its path**
vweb_gen_types []table.Type // vweb route checks
prevent_sum_type_unwrapping_once bool // needed for assign new values to sum type, stopping unwrapping then
loop_label string // set when inside a labelled for loop
timers &util.Timers = util.new_timers(false)
comptime_fields_type map[string]table.Type
fn_scope &ast.Scope = voidptr(0)
2021-02-05 08:05:13 +01:00
used_fns map[string]bool // used_fns['println'] == true
main_fn_decl_node ast.FnDecl
}
2021-02-05 08:27:14 +01:00
pub fn new_checker(table &table.Table, pref &pref.Preferences) Checker {
2020-12-19 12:16:18 +01:00
mut timers_should_print := false
$if time_checking ? {
timers_should_print = true
}
2020-04-09 15:33:46 +02:00
return Checker{
table: table
pref: pref
2020-05-23 08:51:15 +02:00
cur_fn: 0
timers: util.new_timers(timers_should_print)
}
}
pub fn (mut c Checker) check(ast_file &ast.File) {
c.file = ast_file
2020-04-30 09:33:12 +02:00
for i, ast_import in ast_file.imports {
2020-04-30 18:06:14 +02:00
for j in 0 .. i {
2020-04-30 09:33:12 +02:00
if ast_import.mod == ast_file.imports[j].mod {
c.error('module name `$ast_import.mod` duplicate', ast_import.mod_pos)
2020-04-30 09:33:12 +02:00
}
}
}
for stmt in ast_file.stmts {
c.expr_level = 0
c.stmt(stmt)
}
c.check_scope_vars(c.file.scope)
}
pub fn (mut c Checker) check_scope_vars(sc &ast.Scope) {
for _, obj in sc.objects {
match obj {
ast.Var {
if !c.pref.is_repl {
if !obj.is_used && obj.name[0] != `_` {
2020-10-24 19:29:24 +02:00
c.warn('unused variable: `$obj.name`', obj.pos)
}
}
if obj.is_mut && !obj.is_changed && !c.is_builtin_mod && obj.name != 'it' {
// if obj.is_mut && !obj.is_changed && !c.is_builtin { //TODO C error bad field not checked
// c.warn('`$obj.name` is declared as mutable, but it was never changed',
// obj.pos)
}
}
else {}
}
}
for _, child in sc.children {
c.check_scope_vars(child)
}
}
// not used right now
pub fn (mut c Checker) check2(ast_file &ast.File) []errors.Error {
2020-02-20 17:05:16 +01:00
c.file = ast_file
for stmt in ast_file.stmts {
c.stmt(stmt)
}
return c.errors
}
2020-04-25 17:49:16 +02:00
pub fn (mut c Checker) check_files(ast_files []ast.File) {
2021-02-05 08:05:13 +01:00
// c.files = ast_files
mut has_main_mod_file := false
mut has_main_fn := false
mut files_from_main_module := []&ast.File{}
for i in 0 .. ast_files.len {
file := unsafe { &ast_files[i] }
c.timers.start('checker_check $file.path')
c.check(file)
if file.mod.name == 'main' {
files_from_main_module << file
has_main_mod_file = true
if c.file_has_main_fn(file) {
has_main_fn = true
}
}
c.timers.show('checker_check $file.path')
}
if has_main_mod_file && !has_main_fn && files_from_main_module.len > 0 {
if c.pref.is_script && !c.pref.is_test {
// files_from_main_module contain preludes at the start
mut the_main_file := files_from_main_module.last()
the_main_file.stmts << ast.FnDecl{
name: 'main.main'
mod: 'main'
file: the_main_file.path
return_type: table.void_type
scope: &ast.Scope{
parent: 0
}
}
has_main_fn = true
}
}
c.timers.start('checker_post_process_generic_fns')
last_file := c.file
last_mod := c.mod
// post process generic functions. must be done after all files have been
// checked, to eunsure all generic calls are processed as this information
// is needed when the generic type is auto inferred from the call argument
for i in 0 .. ast_files.len {
file := unsafe { &ast_files[i] }
if file.generic_fns.len > 0 {
c.file = file
c.mod = file.mod.name
c.post_process_generic_fns()
}
}
// restore the original c.file && c.mod after post processing
c.file = last_file
c.mod = last_mod
c.timers.show('checker_post_process_generic_fns')
//
c.timers.start('checker_verify_all_vweb_routes')
c.verify_all_vweb_routes()
c.timers.show('checker_verify_all_vweb_routes')
//
if c.pref.is_test {
mut n_test_fns := 0
for _, f in c.table.fns {
if f.name.contains('.test_') {
n_test_fns++
}
}
if n_test_fns == 0 {
c.add_error_detail('The name of a test function in V, should start with `test_`.')
c.add_error_detail('The test function should take 0 parameters, and no return type. Example:')
c.add_error_detail('fn test_xyz(){ assert 2 + 2 == 4 }')
c.error('a _test.v file should have *at least* one `test_` function', token.Position{})
}
}
if c.pref.skip_unused {
c.mark_used(ast_files)
}
// Make sure fn main is defined in non lib builds
if c.pref.build_mode == .build_module || c.pref.is_test {
return
}
2020-04-18 17:46:23 +02:00
if c.pref.is_shared {
// shared libs do not need to have a main
return
}
if !has_main_mod_file {
2020-05-06 12:57:40 +02:00
c.error('project must include a `main` module or be a shared library (compile with `v -shared`)',
token.Position{})
} else if !has_main_fn {
c.error('function `main` must be declared in the main module', token.Position{})
}
}
2020-04-19 00:07:57 +02:00
// do checks specific to files in main module
// returns `true` if a main function is in the file
fn (mut c Checker) file_has_main_fn(file ast.File) bool {
2020-04-19 00:07:57 +02:00
mut has_main_fn := false
for stmt in file.stmts {
if stmt is ast.FnDecl {
if stmt.name == 'main.main' {
if has_main_fn {
c.error('function `main` is already defined', stmt.pos)
}
has_main_fn = true
if stmt.params.len > 0 {
c.error('function `main` cannot have arguments', stmt.pos)
2020-04-19 00:07:57 +02:00
}
if stmt.return_type != table.void_type {
c.error('function `main` cannot return values', stmt.pos)
}
if stmt.no_body {
c.error('function `main` must declare a body', stmt.pos)
}
} else if stmt.attrs.contains('console') {
c.error('only `main` can have the `[console]` attribute', stmt.pos)
2020-04-19 00:07:57 +02:00
}
}
}
2020-04-19 00:07:57 +02:00
return has_main_fn
}
fn (mut c Checker) check_valid_snake_case(name string, identifier string, pos token.Position) {
if !c.pref.is_vweb && name.len > 0 && (name[0] == `_` || name.contains('._')) {
2020-05-16 16:12:23 +02:00
c.error('$identifier `$name` cannot start with `_`', pos)
}
2020-08-09 03:57:54 +02:00
if !c.pref.experimental && !c.pref.translated && util.contains_capital(name) {
2020-05-16 16:12:23 +02:00
c.error('$identifier `$name` cannot contain uppercase letters, use snake_case instead',
pos)
}
}
fn stripped_name(name string) string {
idx := name.last_index('.') or { -1 }
2020-05-16 16:12:23 +02:00
return name[(idx + 1)..]
}
fn (mut c Checker) check_valid_pascal_case(name string, identifier string, pos token.Position) {
sname := stripped_name(name)
if sname.len > 0 && !sname[0].is_capital() && !c.pref.translated {
2020-05-16 16:12:23 +02:00
c.error('$identifier `$name` must begin with capital letter', pos)
}
}
pub fn (mut c Checker) type_decl(node ast.TypeDecl) {
match node {
ast.AliasTypeDecl {
// TODO Replace `c.file.mod.name != 'time'` by `it.language != .v` once available
2020-08-27 06:46:18 +02:00
if c.file.mod.name != 'time' && c.file.mod.name != 'builtin' {
2020-06-18 20:38:59 +02:00
c.check_valid_pascal_case(node.name, 'type alias', node.pos)
2020-05-16 16:12:23 +02:00
}
2020-06-18 20:38:59 +02:00
typ_sym := c.table.get_type_symbol(node.parent_type)
if typ_sym.kind in [.placeholder, .int_literal, .float_literal] {
c.error("type `$typ_sym.name` doesn't exist", node.pos)
} else if typ_sym.kind == .alias {
orig_sym := c.table.get_type_symbol((typ_sym.info as table.Alias).parent_type)
c.error('type `$typ_sym.str()` is an alias, use the original alias type `$orig_sym.name` instead',
node.pos)
} else if typ_sym.kind == .chan {
c.error('aliases of `chan` types are not allowed.', node.pos)
}
}
ast.FnTypeDecl {
2020-06-18 20:38:59 +02:00
c.check_valid_pascal_case(node.name, 'fn type', node.pos)
typ_sym := c.table.get_type_symbol(node.typ)
fn_typ_info := typ_sym.info as table.FnType
fn_info := fn_typ_info.func
ret_sym := c.table.get_type_symbol(fn_info.return_type)
if ret_sym.kind == .placeholder {
c.error("type `$ret_sym.name` doesn't exist", node.pos)
}
2020-09-27 03:32:56 +02:00
for arg in fn_info.params {
arg_sym := c.table.get_type_symbol(arg.typ)
if arg_sym.kind == .placeholder {
c.error("type `$arg_sym.name` doesn't exist", node.pos)
}
}
}
ast.SumTypeDecl {
c.check_valid_pascal_case(node.name, 'sum type', node.pos)
mut names_used := []string{}
for variant in node.variants {
if variant.typ.is_ptr() {
c.error('sum type cannot hold a reference type', variant.pos)
}
mut sym := c.table.get_type_symbol(variant.typ)
if sym.name in names_used {
c.error('sum type $node.name cannot hold the type `$sym.name` more than once',
variant.pos)
} else if sym.kind in [.placeholder, .int_literal, .float_literal] {
c.error("type `$sym.name` doesn't exist", variant.pos)
} else if sym.kind == .interface_ {
c.error('sum type cannot hold an interface', variant.pos)
}
names_used << sym.name
}
}
}
}
2020-05-16 16:12:23 +02:00
pub fn (mut c Checker) interface_decl(decl ast.InterfaceDecl) {
c.check_valid_pascal_case(decl.name, 'interface name', decl.pos)
for method in decl.methods {
c.check_valid_snake_case(method.name, 'method name', method.pos)
}
// TODO: copy pasta from StructDecl
for i, field in decl.fields {
c.check_valid_snake_case(field.name, 'field name', field.pos)
sym := c.table.get_type_symbol(field.typ)
for j in 0 .. i {
if field.name == decl.fields[j].name {
c.error('field name `$field.name` duplicate', field.pos)
}
}
if sym.kind == .placeholder && !sym.name.starts_with('C.') {
c.error(util.new_suggestion(sym.name, c.table.known_type_names()).say('unknown type `$sym.name`'),
field.type_pos)
}
// Separate error condition for `int_literal` and `float_literal` because `util.suggestion` may give different
// suggestions due to f32 comparision issue.
if sym.kind in [.int_literal, .float_literal] {
msg := if sym.kind == .int_literal {
'unknown type `$sym.name`.\nDid you mean `int`?'
} else {
'unknown type `$sym.name`.\nDid you mean `f64`?'
}
c.error(msg, field.type_pos)
}
if sym.kind == .array {
array_info := sym.array_info()
elem_sym := c.table.get_type_symbol(array_info.elem_type)
if elem_sym.kind == .placeholder {
c.error(util.new_suggestion(elem_sym.name, c.table.known_type_names()).say('unknown type `$elem_sym.name`'),
field.type_pos)
}
}
if sym.kind == .struct_ {
info := sym.info as table.Struct
if info.is_ref_only && !field.typ.is_ptr() {
c.error('`$sym.name` type can only be used as a reference: `&$sym.name`',
field.type_pos)
}
}
if sym.kind == .map {
info := sym.map_info()
key_sym := c.table.get_type_symbol(info.key_type)
value_sym := c.table.get_type_symbol(info.value_type)
if key_sym.kind == .placeholder {
c.error('unknown type `$key_sym.name`', field.type_pos)
}
if value_sym.kind == .placeholder {
c.error('unknown type `$value_sym.name`', field.type_pos)
}
}
}
2020-05-16 16:12:23 +02:00
}
pub fn (mut c Checker) struct_decl(mut decl ast.StructDecl) {
if decl.language == .v && !c.is_builtin_mod {
2020-05-16 16:12:23 +02:00
c.check_valid_pascal_case(decl.name, 'struct name', decl.pos)
}
mut struct_sym := c.table.find_type(decl.name) or { table.TypeSymbol{} }
if mut struct_sym.info is table.Struct {
for embed in decl.embeds {
embed_sym := c.table.get_type_symbol(embed.typ)
if embed_sym.kind != .struct_ {
c.error('`$embed_sym.name` is not a struct', embed.pos)
}
}
for attr in decl.attrs {
if attr.name == 'typedef' && decl.language != .c {
c.error('`typedef` attribute can only be used with C structs', decl.pos)
}
}
for i, field in decl.fields {
if decl.language == .v {
c.check_valid_snake_case(field.name, 'field name', field.pos)
}
sym := c.table.get_type_symbol(field.typ)
for j in 0 .. i {
if field.name == decl.fields[j].name {
c.error('field name `$field.name` duplicate', field.pos)
}
}
if sym.kind == .placeholder && decl.language != .c && !sym.name.starts_with('C.') {
c.error(util.new_suggestion(sym.name, c.table.known_type_names()).say('unknown type `$sym.name`'),
field.type_pos)
2020-05-09 23:45:41 +02:00
}
// Separate error condition for `int_literal` and `float_literal` because `util.suggestion` may give different
// suggestions due to f32 comparision issue.
if sym.kind in [.int_literal, .float_literal] {
msg := if sym.kind == .int_literal {
'unknown type `$sym.name`.\nDid you mean `int`?'
} else {
'unknown type `$sym.name`.\nDid you mean `f64`?'
}
c.error(msg, field.type_pos)
}
if sym.kind == .array {
array_info := sym.array_info()
elem_sym := c.table.get_type_symbol(array_info.elem_type)
if elem_sym.kind == .placeholder {
c.error(util.new_suggestion(elem_sym.name, c.table.known_type_names()).say('unknown type `$elem_sym.name`'),
field.type_pos)
}
}
if sym.kind == .struct_ {
info := sym.info as table.Struct
if info.is_ref_only && !field.typ.is_ptr() {
c.error('`$sym.name` type can only be used as a reference: `&$sym.name`',
field.type_pos)
}
}
if sym.kind == .map {
info := sym.map_info()
key_sym := c.table.get_type_symbol(info.key_type)
value_sym := c.table.get_type_symbol(info.value_type)
if key_sym.kind == .placeholder {
c.error('unknown type `$key_sym.name`', field.type_pos)
}
if value_sym.kind == .placeholder {
c.error('unknown type `$value_sym.name`', field.type_pos)
}
2020-09-09 14:05:56 +02:00
}
if field.has_default_expr {
c.expected_type = field.typ
field_expr_type := c.expr(field.default_expr)
c.check_expected(field_expr_type, field.typ) or {
c.error('incompatible initializer for field `$field.name`: $err',
field.default_expr.position())
}
// Check for unnecessary inits like ` = 0` and ` = ''`
if field.typ.is_ptr() {
continue
}
if field.default_expr is ast.IntegerLiteral {
if field.default_expr.val == '0' {
c.warn('unnecessary default value of `0`: struct fields are zeroed by default',
field.default_expr.pos)
}
} else if field.default_expr is ast.StringLiteral {
if field.default_expr.val == '' {
c.warn("unnecessary default value of '': struct fields are zeroed by default",
field.default_expr.pos)
}
} else if field.default_expr is ast.BoolLiteral {
if field.default_expr.val == false {
c.warn('unnecessary default value `false`: struct fields are zeroed by default',
field.default_expr.pos)
}
}
}
}
}
}
2020-05-11 14:38:25 +02:00
pub fn (mut c Checker) struct_init(mut struct_init ast.StructInit) table.Type {
// typ := c.table.find_type(struct_init.typ.typ.name) or {
2020-02-10 13:58:24 +01:00
// c.error('unknown struct: $struct_init.typ.typ.name', struct_init.pos)
// panic('')
// }
2020-04-01 23:23:20 +02:00
if struct_init.typ == table.void_type {
// Short syntax `({foo: bar})`
if c.expected_type == table.void_type {
c.error('unexpected short struct syntax', struct_init.pos)
return table.void_type
}
sym := c.table.get_type_symbol(c.expected_type)
if sym.kind == .array {
struct_init.typ = c.table.value_type(c.expected_type)
} else {
struct_init.typ = c.expected_type
}
2020-04-01 23:23:20 +02:00
}
2020-08-06 17:23:54 +02:00
if struct_init.typ == 0 {
c.error('unknown type', struct_init.pos)
}
utyp := c.unwrap_generic(struct_init.typ)
type_sym := c.table.get_type_symbol(utyp)
if type_sym.kind == .sum_type && struct_init.fields.len == 1 {
sexpr := struct_init.fields[0].expr.str()
c.error('cast to sum type using `${type_sym.name}($sexpr)` not `$type_sym.name{$sexpr}`',
struct_init.pos)
}
if type_sym.kind == .interface_ {
c.error('cannot instantiate interface `$type_sym.name`', struct_init.pos)
}
if type_sym.info is table.Alias {
if type_sym.info.parent_type.is_number() {
c.error('cannot instantiate number type alias `$type_sym.name`', struct_init.pos)
2020-06-24 16:52:23 +02:00
return table.void_type
}
}
// allow init structs from generic if they're private except the type is from builtin module
if !type_sym.is_public && type_sym.kind != .placeholder && type_sym.language != .c
&& (type_sym.mod != c.mod && !(struct_init.typ.has_flag(.generic)
&& type_sym.mod != 'builtin')) {
c.error('type `$type_sym.name` is private', struct_init.pos)
}
2021-01-08 04:49:13 +01:00
if type_sym.kind == .struct_ {
info := type_sym.info as table.Struct
if info.attrs.len > 0 && info.attrs[0].name == 'noinit' && type_sym.mod != c.mod {
c.error('struct `$type_sym.name` is declared with a `[noinit]` attribute, so ' +
'it cannot be initialized with `$type_sym.name{}`', struct_init.pos)
2021-01-08 04:49:13 +01:00
}
}
if type_sym.name.len == 1 && c.cur_fn.generic_params.len == 0 {
c.error('unknown struct `$type_sym.name`', struct_init.pos)
return 0
}
match type_sym.kind {
.placeholder {
c.error('unknown struct: $type_sym.name', struct_init.pos)
return table.void_type
}
// string & array are also structs but .kind of string/array
.struct_, .string, .array, .alias {
mut info := table.Struct{}
if type_sym.kind == .alias {
info_t := type_sym.info as table.Alias
sym := c.table.get_type_symbol(info_t.parent_type)
if sym.kind == .placeholder { // pending import symbol did not resolve
c.error('unknown struct: $type_sym.name', struct_init.pos)
return table.void_type
}
if sym.kind != .struct_ {
c.error('alias type name: $sym.name is not struct type', struct_init.pos)
}
info = sym.info as table.Struct
} else {
info = type_sym.info as table.Struct
}
if struct_init.is_short {
exp_len := info.fields.len
got_len := struct_init.fields.len
if exp_len != got_len {
amount := if exp_len < got_len { 'many' } else { 'few' }
c.error('too $amount fields in `$type_sym.name` literal (expecting $exp_len, got $got_len)',
struct_init.pos)
}
}
mut inited_fields := []string{}
for i, field in struct_init.fields {
mut info_field := table.Field{}
mut embed_type := table.Type(0)
mut is_embed := false
mut field_name := ''
if struct_init.is_short {
if i >= info.fields.len {
// It doesn't make sense to check for fields that don't exist.
// We should just stop here.
break
}
info_field = info.fields[i]
field_name = info_field.name
struct_init.fields[i].name = field_name
} else {
field_name = field.name
mut exists := false
for f in info.fields {
if f.name == field_name {
info_field = f
exists = true
break
}
}
if !exists {
for embed in info.embeds {
embed_sym := c.table.get_type_symbol(embed)
if embed_sym.embed_name() == field_name {
exists = true
embed_type = embed
is_embed = true
2020-10-30 07:09:26 +01:00
break
}
}
}
if !exists {
c.error('unknown field `$field.name` in struct literal of type `$type_sym.name`',
2020-04-25 17:49:16 +02:00
field.pos)
continue
}
if field_name in inited_fields {
2020-06-24 14:44:06 +02:00
c.error('duplicate field name in struct literal: `$field_name`',
field.pos)
continue
}
}
if is_embed {
c.expected_type = embed_type
expr_type := c.expr(field.expr)
expr_type_sym := c.table.get_type_symbol(expr_type)
if expr_type != table.void_type && expr_type_sym.kind != .placeholder {
c.check_expected(expr_type, embed_type) or {
c.error('cannot assign to field `$info_field.name`: $err',
field.pos)
}
}
struct_init.fields[i].typ = expr_type
struct_init.fields[i].expected_type = embed_type
} else {
inited_fields << field_name
field_type_sym := c.table.get_type_symbol(info_field.typ)
c.expected_type = info_field.typ
expr_type := c.expr(field.expr)
expr_type_sym := c.table.get_type_symbol(expr_type)
if field_type_sym.kind == .interface_ {
c.type_implements(expr_type, info_field.typ, field.pos)
} else if expr_type != table.void_type && expr_type_sym.kind != .placeholder {
c.check_expected(expr_type, info_field.typ) or {
c.error('cannot assign to field `$info_field.name`: $err',
field.pos)
}
}
if info_field.typ.has_flag(.shared_f) {
if !expr_type.has_flag(.shared_f) && expr_type.is_ptr() {
c.error('`shared` field must be initialized with `shared` or value',
field.pos)
}
} else {
if info_field.typ.is_ptr() && !expr_type.is_ptr() && !expr_type.is_pointer()
&& !expr_type.is_number() {
c.error('reference field must be initialized with reference',
field.pos)
}
}
struct_init.fields[i].typ = expr_type
struct_init.fields[i].expected_type = info_field.typ
}
}
// Check uninitialized refs
for field in info.fields {
if field.has_default_expr || field.name in inited_fields {
continue
}
if field.typ.is_ptr() && !struct_init.has_update_expr && !c.pref.translated {
c.error('reference field `${type_sym.name}.$field.name` must be initialized',
2020-04-07 18:51:39 +02:00
struct_init.pos)
}
// Check for `[required]` struct attr
if field.attrs.contains('required') && !struct_init.is_short {
mut found := false
for init_field in struct_init.fields {
if field.name == init_field.name {
found = true
break
}
}
if !found {
c.error('field `${type_sym.name}.$field.name` must be initialized',
struct_init.pos)
}
}
}
}
else {}
}
if struct_init.has_update_expr {
update_type := c.expr(struct_init.update_expr)
struct_init.update_expr_type = update_type
if c.table.type_kind(update_type) != .struct_ {
s := c.table.type_to_str(update_type)
c.error('expected struct, found `$s`', struct_init.update_expr.position())
} else if update_type != struct_init.typ {
from_sym := c.table.get_type_symbol(update_type)
to_sym := c.table.get_type_symbol(struct_init.typ)
from_info := from_sym.info as table.Struct
to_info := to_sym.info as table.Struct
// TODO this check is too strict
if !c.check_struct_signature(from_info, to_info) {
c.error('struct `$from_sym.name` is not compatible with struct `$to_sym.name`',
struct_init.update_expr.position())
}
}
if !struct_init.update_expr.is_lvalue() {
// cgen will repeat `update_expr` for each field
// so enforce an lvalue for efficiency
c.error('expression is not an lvalue', struct_init.update_expr.position())
}
}
return struct_init.typ
}
2020-12-11 04:44:07 +01:00
fn (mut c Checker) check_div_mod_by_zero(expr ast.Expr, op_kind token.Kind) {
match mut expr {
ast.FloatLiteral {
if expr.val.f64() == 0.0 {
oper := if op_kind == .div { 'division' } else { 'modulo' }
c.error('$oper by zero', expr.pos)
}
}
ast.IntegerLiteral {
if expr.val.int() == 0 {
oper := if op_kind == .div { 'division' } else { 'modulo' }
c.error('$oper by zero', expr.pos)
}
}
ast.CastExpr {
c.check_div_mod_by_zero(expr.expr, op_kind)
}
else {}
}
}
2020-05-11 14:38:25 +02:00
pub fn (mut c Checker) infix_expr(mut infix_expr ast.InfixExpr) table.Type {
2020-02-20 12:16:18 +01:00
// println('checker: infix expr(op $infix_expr.op.str())')
former_expected_type := c.expected_type
defer {
c.expected_type = former_expected_type
}
left_type := c.expr(infix_expr.left)
// left_type = c.unwrap_genric(c.expr(infix_expr.left))
2020-03-07 00:34:14 +01:00
infix_expr.left_type = left_type
c.expected_type = left_type
right_type := c.expr(infix_expr.right)
// right_type = c.unwrap_genric(c.expr(infix_expr.right))
2020-03-15 00:46:08 +01:00
infix_expr.right_type = right_type
mut right := c.table.get_type_symbol(right_type)
mut left := c.table.get_type_symbol(left_type)
left_pos := infix_expr.left.position()
right_pos := infix_expr.right.position()
if (left_type.is_ptr() || left.is_pointer()) && infix_expr.op in [.plus, .minus] {
if !c.inside_unsafe {
c.warn('pointer arithmetic is only allowed in `unsafe` blocks', left_pos)
}
if left_type == table.voidptr_type {
c.error('`$infix_expr.op` cannot be used with `voidptr`', left_pos)
}
}
mut return_type := left_type
if infix_expr.op != .key_is {
match mut infix_expr.left {
ast.Ident, ast.SelectorExpr {
if infix_expr.left.is_mut {
c.error('remove unnecessary `mut`', infix_expr.left.mut_pos)
}
}
else {}
}
}
2020-04-25 20:28:49 +02:00
// Single side check
// Place these branches according to ops' usage frequency to accelerate.
// TODO: First branch includes ops where single side check is not needed, or needed but hasn't been implemented.
// TODO: Some of the checks are not single side. Should find a better way to organize them.
2020-04-25 20:28:49 +02:00
match infix_expr.op {
2020-05-12 19:38:43 +02:00
// .eq, .ne, .gt, .lt, .ge, .le, .and, .logical_or, .dot, .key_as, .right_shift {}
2020-11-28 23:34:26 +01:00
.eq, .ne {
is_mismatch := (left.kind == .alias && right.kind in [.struct_, .array])
|| (right.kind == .alias && left.kind in [.struct_, .array])
if is_mismatch {
2020-11-28 23:34:26 +01:00
c.error('possible type mismatch of compared values of `$infix_expr.op` operation',
infix_expr.pos)
}
}
2020-04-25 20:28:49 +02:00
.key_in, .not_in {
match right.kind {
.array {
2020-08-27 06:46:18 +02:00
elem_type := right.array_info().elem_type
// if left_default.kind != right_sym.kind {
c.check_expected(left_type, elem_type) or {
c.error('left operand to `$infix_expr.op` does not match the array element type: $err',
2020-05-18 15:43:46 +02:00
infix_expr.pos)
2020-04-25 20:28:49 +02:00
}
}
.map {
elem_type := right.map_info().key_type
c.check_expected(left_type, elem_type) or {
c.error('left operand to `$infix_expr.op` does not match the map key type: $err',
2020-05-18 15:43:46 +02:00
infix_expr.pos)
2020-04-25 21:51:44 +02:00
}
}
2020-04-25 20:28:49 +02:00
.string {
c.check_expected(left_type, right_type) or {
c.error('left operand to `$infix_expr.op` does not match: $err',
2020-05-18 15:43:46 +02:00
infix_expr.pos)
2020-04-25 20:28:49 +02:00
}
2020-04-25 21:51:44 +02:00
}
2020-04-25 20:28:49 +02:00
else {
2020-05-18 15:43:46 +02:00
c.error('`$infix_expr.op.str()` can only be used with an array/map/string',
infix_expr.pos)
2020-04-25 20:28:49 +02:00
}
}
2020-04-25 20:28:49 +02:00
return table.bool_type
2020-02-19 19:54:36 +01:00
}
.plus, .minus, .mul, .div, .mod, .xor, .amp, .pipe { // binary operators that expect matching types
if right.info is table.Alias
&& (right.info as table.Alias).language != .c && c.mod == c.table.type_to_str(right_type).split('.')[0] {
right = c.table.get_type_symbol((right.info as table.Alias).parent_type)
}
if left.info is table.Alias
&& (left.info as table.Alias).language != .c && c.mod == c.table.type_to_str(left_type).split('.')[0] {
left = c.table.get_type_symbol((left.info as table.Alias).parent_type)
}
if left.kind in [.array, .array_fixed, .map, .struct_] {
if left.has_method(infix_expr.op.str()) {
if method := left.find_method(infix_expr.op.str()) {
return_type = method.return_type
} else {
return_type = left_type
}
} else {
left_name := c.table.type_to_str(left_type)
right_name := c.table.type_to_str(right_type)
if left_name == right_name {
c.error('undefined operation `$left_name` $infix_expr.op.str() `$right_name`',
left_pos)
} else {
c.error('mismatched types `$left_name` and `$right_name`', left_pos)
}
}
} else if right.kind in [.array, .array_fixed, .map, .struct_] {
if right.has_method(infix_expr.op.str()) {
if method := right.find_method(infix_expr.op.str()) {
return_type = method.return_type
} else {
return_type = right_type
}
} else {
left_name := c.table.type_to_str(left_type)
right_name := c.table.type_to_str(right_type)
if left_name == right_name {
c.error('undefined operation `$left_name` $infix_expr.op.str() `$right_name`',
right_pos)
} else {
c.error('mismatched types `$left_name` and `$right_name`', right_pos)
}
}
} else {
promoted_type := c.promote(c.table.unalias_num_type(left_type), c.table.unalias_num_type(right_type))
if promoted_type.idx() == table.void_type_idx {
left_name := c.table.type_to_str(left_type)
right_name := c.table.type_to_str(right_type)
c.error('mismatched types `$left_name` and `$right_name`', infix_expr.pos)
} else if promoted_type.has_flag(.optional) {
s := c.table.type_to_str(promoted_type)
c.error('`$infix_expr.op` cannot be used with `$s`', infix_expr.pos)
} else if promoted_type.is_float() {
if infix_expr.op in [.mod, .xor, .amp, .pipe] {
side := if left_type == promoted_type { 'left' } else { 'right' }
pos := if left_type == promoted_type { left_pos } else { right_pos }
2020-05-28 05:50:57 +02:00
name := if left_type == promoted_type { left.name } else { right.name }
if infix_expr.op == .mod {
2020-06-24 14:44:06 +02:00
c.error('float modulo not allowed, use math.fmod() instead',
pos)
} else {
c.error('$side type of `$infix_expr.op.str()` cannot be non-integer type `$name`',
2020-05-28 05:50:57 +02:00
pos)
}
}
}
if infix_expr.op in [.div, .mod] {
2020-12-11 04:44:07 +01:00
c.check_div_mod_by_zero(infix_expr.right, infix_expr.op)
}
return_type = promoted_type
}
2020-04-25 20:28:49 +02:00
}
.gt, .lt, .ge, .le {
if left.kind in [.array, .array_fixed] && right.kind in [.array, .array_fixed] {
c.error('only `==` and `!=` are defined on arrays', infix_expr.pos)
} else if left.kind == .struct_ && right.kind == .struct_ && infix_expr.op in [.eq, .lt] {
2021-01-10 11:24:46 +01:00
if !(left.has_method(infix_expr.op.str()) && right.has_method(infix_expr.op.str())) {
left_name := c.table.type_to_str(left_type)
right_name := c.table.type_to_str(right_type)
if left_name == right_name {
c.error('undefined operation `$left_name` $infix_expr.op.str() `$right_name`',
2021-01-10 11:24:46 +01:00
infix_expr.pos)
} else {
c.error('mismatched types `$left_name` and `$right_name`', infix_expr.pos)
}
}
}
if left.kind == .struct_ && right.kind == .struct_ {
if !left.has_method('<') && infix_expr.op in [.ge, .le] {
c.error('cannot use `$infix_expr.op` as `<` operator method is not defined',
infix_expr.pos)
} else if !left.has_method('<') && infix_expr.op == .gt {
c.error('cannot use `>` as `<=` operator method is not defined', infix_expr.pos)
}
}
}
2020-04-25 20:28:49 +02:00
.left_shift {
if left.kind == .array {
// `array << elm`
infix_expr.auto_locked, _ = c.fail_if_immutable(infix_expr.left)
left_value_type := c.table.value_type(left_type)
left_value_sym := c.table.get_type_symbol(left_value_type)
if left_value_sym.kind == .interface_ {
if right.kind != .array {
// []Animal << Cat
c.type_implements(right_type, left_value_type, right_pos)
} else {
// []Animal << Cat
2020-05-28 05:50:57 +02:00
c.type_implements(c.table.value_type(right_type), left_value_type,
right_pos)
}
return table.void_type
}
2020-04-25 20:28:49 +02:00
// the expressions have different types (array_x and x)
2020-05-24 04:43:00 +02:00
if c.check_types(right_type, left_value_type) { // , right_type) {
2020-04-25 20:28:49 +02:00
// []T << T
return table.void_type
}
if right.kind == .array
&& c.check_types(left_value_type, c.table.value_type(right_type)) {
2020-04-25 20:28:49 +02:00
// []T << []T
return table.void_type
}
c.error('cannot append `$right.name` to `$left.name`', right_pos)
2020-04-25 20:28:49 +02:00
return table.void_type
} else {
return c.check_shift(left_type, right_type, left_pos, right_pos)
2020-04-25 06:14:17 +02:00
}
}
.right_shift {
return c.check_shift(left_type, right_type, left_pos, right_pos)
}
2020-06-02 16:18:12 +02:00
.key_is, .not_is {
2020-04-25 20:28:49 +02:00
type_expr := infix_expr.right as ast.Type
typ_sym := c.table.get_type_symbol(type_expr.typ)
if typ_sym.kind == .placeholder {
c.error('$infix_expr.op.str(): type `$typ_sym.name` does not exist', type_expr.pos)
2020-04-25 20:28:49 +02:00
}
if left.kind !in [.interface_, .sum_type] {
2020-06-17 00:59:33 +02:00
c.error('`$infix_expr.op.str()` can only be used with interfaces and sum types',
infix_expr.pos)
} else if mut left.info is table.SumType {
2020-11-24 17:55:24 +01:00
if type_expr.typ !in left.info.variants {
c.error('`$left.name` has no variant `$right.name`', infix_expr.pos)
}
}
2020-04-25 20:28:49 +02:00
return table.bool_type
2020-04-20 14:49:26 +02:00
}
.arrow { // `chan <- elem`
if left.kind == .chan {
chan_info := left.chan_info()
elem_type := chan_info.elem_type
if !c.check_types(right_type, elem_type) {
c.error('cannot push `$right.name` on `$left.name`', right_pos)
}
if chan_info.is_mut {
// TODO: The error message of the following could be more specific...
c.fail_if_immutable(infix_expr.right)
}
if elem_type.is_ptr() && !right_type.is_ptr() {
c.error('cannot push non-reference `$right.name` on `$left.name`',
right_pos)
}
c.stmts(infix_expr.or_block.stmts)
} else {
c.error('cannot push on non-channel `$left.name`', left_pos)
}
return table.void_type
}
2020-11-28 23:34:26 +01:00
.and, .logical_or {
if infix_expr.left_type != table.bool_type_idx {
c.error('left operand for `$infix_expr.op` is not a boolean', infix_expr.left.position())
}
if infix_expr.right_type != table.bool_type_idx {
c.error('right operand for `$infix_expr.op` is not a boolean', infix_expr.right.position())
}
2020-06-02 22:21:40 +02:00
// use `()` to make the boolean expression clear error
// for example: `(a && b) || c` instead of `a && b || c`
2020-11-28 23:34:26 +01:00
if mut infix_expr.left is ast.InfixExpr {
if infix_expr.left.op != infix_expr.op && infix_expr.left.op in [.logical_or, .and] {
c.error('use `()` to make the boolean expression clear', infix_expr.pos)
2020-06-02 22:21:40 +02:00
}
}
}
2020-11-28 23:34:26 +01:00
else {}
2020-04-19 22:29:45 +02:00
}
2020-04-25 20:28:49 +02:00
// TODO: Absorb this block into the above single side check block to accelerate.
if left_type == table.bool_type && infix_expr.op !in [.eq, .ne, .logical_or, .and] {
2020-04-25 21:51:44 +02:00
c.error('bool types only have the following operators defined: `==`, `!=`, `||`, and `&&`',
2020-04-14 04:12:24 +02:00
infix_expr.pos)
} else if left_type == table.string_type
&& infix_expr.op !in [.plus, .eq, .ne, .lt, .gt, .le, .ge] {
2020-04-14 04:12:24 +02:00
// TODO broken !in
c.error('string types only have the following operators defined: `==`, `!=`, `<`, `>`, `<=`, `>=`, and `+`',
2020-04-14 04:12:24 +02:00
infix_expr.pos)
} else if left.kind == .enum_ && right.kind == .enum_ && infix_expr.op !in [.ne, .eq] {
c.error('only `==` and `!=` are defined on `enum`, use an explicit cast to `int` if needed',
infix_expr.pos)
2020-04-14 04:12:24 +02:00
}
// sum types can't have any infix operation except of "is", is is checked before and doesn't reach this
if c.table.type_kind(left_type) == .sum_type {
c.error('cannot use operator `$infix_expr.op` with `$left.name`', infix_expr.pos)
} else if c.table.type_kind(right_type) == .sum_type {
c.error('cannot use operator `$infix_expr.op` with `$right.name`', infix_expr.pos)
}
// TODO move this to symmetric_check? Right now it would break `return 0` for `fn()?int `
left_is_optional := left_type.has_flag(.optional)
right_is_optional := right_type.has_flag(.optional)
if (left_is_optional && !right_is_optional) || (!left_is_optional && right_is_optional) {
c.error('unwrapped optional cannot be used in an infix expression', infix_expr.pos)
}
2020-04-25 20:28:49 +02:00
// Dual sides check (compatibility check)
2020-08-11 00:51:15 +02:00
if !c.symmetric_check(right_type, left_type) && !c.pref.translated {
2020-04-20 14:49:26 +02:00
// for type-unresolved consts
if left_type == table.void_type || right_type == table.void_type {
return table.void_type
}
2020-04-25 17:49:16 +02:00
c.error('infix expr: cannot use `$right.name` (right expression) as `$left.name`',
infix_expr.pos)
2020-04-20 14:49:26 +02:00
}
2020-07-16 19:40:14 +02:00
/*
if (infix_expr.left is ast.InfixExpr &&
(infix_expr.left as ast.InfixExpr).op == .inc) ||
(infix_expr.right is ast.InfixExpr && (infix_expr.right as ast.InfixExpr).op == .inc) {
c.warn('`++` and `--` are statements, not expressions', infix_expr.pos)
}
*/
return if infix_expr.op.is_relational() {
table.bool_type
} else {
return_type
}
}
// returns name and position of variable that needs write lock
fn (mut c Checker) needs_rlock(expr ast.Expr) (string, token.Position) {
mut to_lock := '' // name of variable that needs lock
mut pos := token.Position{} // and its position
match mut expr {
ast.Ident {
if expr.obj is ast.Var {
mut v := expr.obj as ast.Var
if v.typ.share() == .shared_t {
if expr.name !in c.rlocked_names && expr.name !in c.locked_names {
to_lock = expr.name
pos = expr.pos
}
}
}
}
else {
to_lock = ''
pos = token.Position{}
}
}
return to_lock, pos
}
// returns name and position of variable that needs write lock
// also sets `is_changed` to true (TODO update the name to reflect this?)
fn (mut c Checker) fail_if_immutable(expr ast.Expr) (string, token.Position) {
mut to_lock := '' // name of variable that needs lock
mut pos := token.Position{} // and its position
mut explicit_lock_needed := false
match mut expr {
ast.CastExpr {
// TODO
return '', pos
}
ast.ComptimeSelector {
return '', pos
}
ast.Ident {
if expr.obj is ast.Var {
mut v := expr.obj as ast.Var
2020-09-27 03:40:59 +02:00
if !v.is_mut && !c.pref.translated && !c.inside_unsafe {
c.error('`$expr.name` is immutable, declare it with `mut` to make it mutable',
expr.pos)
}
v.is_changed = true
if v.typ.share() == .shared_t {
if expr.name !in c.locked_names {
if c.locked_names.len > 0 || c.rlocked_names.len > 0 {
if expr.name in c.rlocked_names {
c.error('$expr.name has an `rlock` but needs a `lock`',
expr.pos)
} else {
c.error('$expr.name must be added to the `lock` list above',
expr.pos)
}
}
to_lock = expr.name
pos = expr.pos
}
}
} else if expr.obj is ast.ConstField && expr.name in c.const_names {
2020-06-18 20:38:59 +02:00
c.error('cannot modify constant `$expr.name`', expr.pos)
}
}
ast.IndexExpr {
to_lock, pos = c.fail_if_immutable(expr.left)
}
ast.ParExpr {
to_lock, pos = c.fail_if_immutable(expr.expr)
}
ast.PrefixExpr {
to_lock, pos = c.fail_if_immutable(expr.right)
}
ast.SelectorExpr {
// retrieve table.Field
2020-06-18 20:38:59 +02:00
if expr.expr_type == 0 {
c.error('0 type in SelectorExpr', expr.pos)
return '', pos
}
mut typ_sym := c.table.get_type_symbol(c.unwrap_generic(expr.expr_type))
if mut typ_sym.info is table.Alias {
typ_sym = c.table.get_type_symbol(typ_sym.info.parent_type)
}
match typ_sym.kind {
.struct_ {
struct_info := typ_sym.info as table.Struct
mut has_field := true
mut field_info := struct_info.find_field(expr.field_name) or {
has_field = false
table.Field{}
}
if !has_field {
for embed in struct_info.embeds {
embed_sym := c.table.get_type_symbol(embed)
embed_struct_info := embed_sym.info as table.Struct
if embed_field_info := embed_struct_info.find_field(expr.field_name) {
has_field = true
field_info = embed_field_info
break
}
}
}
if !has_field {
2020-06-18 20:38:59 +02:00
type_str := c.table.type_to_str(expr.expr_type)
c.error('unknown field `${type_str}.$expr.field_name`', expr.pos)
return '', pos
}
2020-08-09 03:57:54 +02:00
if !field_info.is_mut && !c.pref.translated {
2020-06-18 20:38:59 +02:00
type_str := c.table.type_to_str(expr.expr_type)
c.error('field `$expr.field_name` of struct `$type_str` is immutable',
expr.pos)
}
to_lock, pos = c.fail_if_immutable(expr.expr)
if to_lock != '' {
// No automatic lock for struct access
explicit_lock_needed = true
}
}
.interface_ {
interface_info := typ_sym.info as table.Interface
mut field_info := interface_info.find_field(expr.field_name) or {
type_str := c.table.type_to_str(expr.expr_type)
c.error('unknown field `${type_str}.$expr.field_name`', expr.pos)
return '', pos
}
if !field_info.is_mut {
type_str := c.table.type_to_str(expr.expr_type)
c.error('field `$expr.field_name` of interface `$type_str` is immutable',
expr.pos)
}
c.fail_if_immutable(expr.expr)
}
.array, .string {
// This should only happen in `builtin`
// TODO Remove `crypto.rand` when possible (see vlib/crypto/rand/rand.v,
// if `c_array_to_bytes_tmp` doesn't exist, then it's safe to remove it)
if c.file.mod.name !in ['builtin', 'crypto.rand'] {
2020-06-18 20:38:59 +02:00
c.error('`$typ_sym.kind` can not be modified', expr.pos)
}
}
else {
2020-06-18 20:38:59 +02:00
c.error('unexpected symbol `$typ_sym.kind`', expr.pos)
}
}
}
ast.CallExpr {
// TODO: should only work for builtin method
if expr.name == 'slice' {
to_lock, pos = c.fail_if_immutable(expr.left)
if to_lock != '' {
// No automatic lock for array slicing (yet(?))
explicit_lock_needed = true
}
}
}
ast.ArrayInit {
return '', pos
}
ast.StructInit {
return '', pos
}
else {
c.error('unexpected expression `$expr.type_name()`', expr.position())
}
}
if explicit_lock_needed {
c.error('`$to_lock` is `shared` and needs explicit lock for `$expr.type_name()`',
pos)
to_lock = ''
}
return to_lock, pos
}
2020-05-11 14:38:25 +02:00
pub fn (mut c Checker) call_expr(mut call_expr ast.CallExpr) table.Type {
// First check everything that applies to both fns and methods
// TODO merge logic from call_method and call_fn
/*
for i, call_arg in call_expr.args {
if call_arg.is_mut {
c.fail_if_immutable(call_arg.expr)
if !arg.is_mut {
tok := call_arg.share.str()
c.error('`$call_expr.name` parameter `$arg.name` is not `$tok`, `$tok` is not needed`',
call_arg.expr.position())
} else if arg.typ.share() != call_arg.share {
c.error('wrong shared type', call_arg.expr.position())
}
} else {
if arg.is_mut && (!call_arg.is_mut || arg.typ.share() != call_arg.share) {
tok := call_arg.share.str()
c.error('`$call_expr.name` parameter `$arg.name` is `$tok`, you need to provide `$tok` e.g. `$tok arg${i+1}`',
call_arg.expr.position())
}
}
}
*/
// Now call `call_method` or `call_fn` for specific checks.
typ := if call_expr.is_method { c.call_method(mut call_expr) } else { c.call_fn(mut call_expr) }
2020-10-22 03:51:25 +02:00
// autofree: mark args that have to be freed (after saving them in tmp exprs)
free_tmp_arg_vars := c.pref.autofree && !c.is_builtin_mod && call_expr.args.len > 0
&& !call_expr.args[0].typ.has_flag(.optional)
2020-11-21 19:07:47 +01:00
if free_tmp_arg_vars && !c.inside_const {
2020-09-05 12:00:35 +02:00
for i, arg in call_expr.args {
if arg.typ != table.string_type {
continue
}
if arg.expr is ast.Ident || arg.expr is ast.StringLiteral
|| arg.expr is ast.SelectorExpr {
// Simple expressions like variables, string literals, selector expressions
// (`x.field`) can't result in allocations and don't need to be assigned to
// temporary vars.
// Only expressions like `str + 'b'` need to be freed.
2020-09-05 12:00:35 +02:00
continue
}
call_expr.args[i].is_tmp_autofree = true
}
// TODO copy pasta from above
if call_expr.receiver_type == table.string_type && !(call_expr.left is ast.Ident
|| call_expr.left is ast.StringLiteral
|| call_expr.left is ast.SelectorExpr) {
2020-10-22 03:51:25 +02:00
call_expr.free_receiver = true
}
}
c.expected_or_type = call_expr.return_type.clear_flag(.optional)
c.stmts(call_expr.or_block.stmts)
c.expected_or_type = table.void_type
if call_expr.or_block.kind == .propagate && !c.cur_fn.return_type.has_flag(.optional)
&& !c.inside_const {
cur_names := c.cur_fn.name.split('.')
if cur_names[cur_names.len - 1] != 'main' {
c.error('to propagate the optional call, `$c.cur_fn.name` must return an optional',
call_expr.or_block.pos)
}
}
2020-09-05 12:00:35 +02:00
return typ
}
fn (mut c Checker) check_map_and_filter(is_map bool, elem_typ table.Type, call_expr ast.CallExpr) {
if call_expr.args.len != 1 {
c.error('expected 1 arguments, but got $call_expr.args.len', call_expr.pos)
// Finish early so that it doesn't fail later
return
}
elem_sym := c.table.get_type_symbol(elem_typ)
arg_expr := call_expr.args[0].expr
match arg_expr {
ast.AnonFn {
2020-09-27 03:32:56 +02:00
if arg_expr.decl.params.len > 1 {
c.error('function needs exactly 1 argument', arg_expr.decl.pos)
} else if is_map && (arg_expr.decl.return_type == table.void_type
|| arg_expr.decl.params[0].typ != elem_typ) {
c.error('type mismatch, should use `fn(a $elem_sym.name) T {...}`', arg_expr.decl.pos)
} else if !is_map && (arg_expr.decl.return_type != table.bool_type
|| arg_expr.decl.params[0].typ != elem_typ) {
c.error('type mismatch, should use `fn(a $elem_sym.name) bool {...}`',
arg_expr.decl.pos)
}
}
ast.Ident {
if arg_expr.kind == .function {
func := c.table.find_fn(arg_expr.name) or {
2021-02-05 08:05:13 +01:00
c.error('$arg_expr.name does not exist', arg_expr.pos)
return
}
2020-09-27 03:32:56 +02:00
if func.params.len > 1 {
c.error('function needs exactly 1 argument', call_expr.pos)
} else if is_map
&& (func.return_type == table.void_type || func.params[0].typ != elem_typ) {
c.error('type mismatch, should use `fn(a $elem_sym.name) T {...}`',
arg_expr.pos)
} else if !is_map
&& (func.return_type != table.bool_type || func.params[0].typ != elem_typ) {
c.error('type mismatch, should use `fn(a $elem_sym.name) bool {...}`',
arg_expr.pos)
}
}
}
else {}
}
}
2020-05-11 14:38:25 +02:00
pub fn (mut c Checker) call_method(mut call_expr ast.CallExpr) table.Type {
left_type := c.expr(call_expr.left)
c.expected_type = left_type
is_generic := left_type.has_flag(.generic)
call_expr.left_type = left_type
// Set default values for .return_type & .receiver_type too,
// or there will be hard to diagnose 0 type panics in cgen.
call_expr.return_type = left_type
call_expr.receiver_type = left_type
2020-05-28 01:19:03 +02:00
left_type_sym := c.table.get_type_symbol(c.unwrap_generic(left_type))
method_name := call_expr.name
mut unknown_method_msg := 'unknown method: `${left_type_sym.name}.$method_name`'
2020-06-23 11:39:58 +02:00
if left_type.has_flag(.optional) {
c.error('optional type cannot be called directly', call_expr.left.position())
return table.void_type
}
if left_type_sym.kind == .sum_type && method_name == 'type_name' {
return table.string_type
}
mut has_generic_generic := false // x.foo<T>() instead of x.foo<int>()
mut generic_types := []table.Type{}
for generic_type in call_expr.generic_types {
if generic_type.has_flag(.generic) {
has_generic_generic = true
generic_types << c.unwrap_generic(generic_type)
} else {
generic_types << generic_type
}
}
if has_generic_generic {
if c.mod != '' {
// Need to prepend the module when adding a generic type to a function
c.table.register_fn_gen_type(c.mod + '.' + call_expr.name, generic_types)
} else {
c.table.register_fn_gen_type(call_expr.name, generic_types)
}
}
// TODO: remove this for actual methods, use only for compiler magic
2020-05-01 00:29:54 +02:00
// FIXME: Argument count != 1 will break these
2021-01-25 10:26:20 +01:00
if left_type_sym.kind == .array && method_name in checker.array_builtin_methods {
mut elem_typ := table.void_type
2020-08-12 05:54:51 +02:00
is_filter_map := method_name in ['filter', 'map']
is_sort := method_name == 'sort'
is_slice := method_name == 'slice'
if is_slice && !c.is_builtin_mod {
c.error('.slice() is a private method, use `x[start..end]` instead', call_expr.pos)
}
2020-08-12 05:54:51 +02:00
if is_filter_map || is_sort {
array_info := left_type_sym.info as table.Array
2020-08-12 05:54:51 +02:00
if is_filter_map {
// position of `it` doesn't matter
scope_register_it(mut call_expr.scope, call_expr.pos, array_info.elem_type)
2020-08-12 05:54:51 +02:00
} else if is_sort {
2020-08-21 20:17:55 +02:00
c.fail_if_immutable(call_expr.left)
// position of `a` and `b` doesn't matter, they're the same
scope_register_ab(mut call_expr.scope, call_expr.pos, array_info.elem_type)
// Verify `.sort(a < b)`
if call_expr.args.len > 0 {
if call_expr.args[0].expr !is ast.InfixExpr {
c.error(
'`.sort()` requires a `<` or `>` comparison as the first and only argument' +
'\ne.g. `users.sort(a.id < b.id)`', call_expr.pos)
}
}
2020-08-12 05:54:51 +02:00
}
elem_typ = array_info.elem_type
}
2020-05-02 15:26:58 +02:00
// map/filter are supposed to have 1 arg only
2020-05-01 00:29:54 +02:00
mut arg_type := left_type
for arg in call_expr.args {
2020-05-01 00:29:54 +02:00
arg_type = c.expr(arg.expr)
}
2020-05-02 15:26:58 +02:00
if method_name == 'map' {
// check fn
c.check_map_and_filter(true, elem_typ, call_expr)
arg_sym := c.table.get_type_symbol(arg_type)
// FIXME: match expr failed for now
mut ret_type := 0
match mut arg_sym.info {
table.FnType { ret_type = arg_sym.info.func.return_type }
else { ret_type = arg_type }
}
call_expr.return_type = c.table.find_or_register_array(ret_type)
} else if method_name == 'filter' {
// check fn
c.check_map_and_filter(false, elem_typ, call_expr)
2020-05-01 00:29:54 +02:00
} else if method_name == 'clone' {
// need to return `array_xxx` instead of `array`
// in ['clone', 'str'] {
2020-04-25 09:08:53 +02:00
call_expr.receiver_type = left_type.to_ptr()
call_expr.return_type = call_expr.receiver_type.set_nr_muls(0)
2020-08-20 16:29:40 +02:00
} else if method_name == 'sort' {
call_expr.return_type = table.void_type
} else if method_name == 'contains' {
call_expr.return_type = table.bool_type
} else if method_name == 'index' {
call_expr.return_type = table.int_type
}
2020-05-01 00:29:54 +02:00
return call_expr.return_type
} else if left_type_sym.kind == .map && method_name in ['clone', 'keys'] {
mut ret_type := table.void_type
match method_name {
'clone' {
ret_type = left_type
}
'keys' {
info := left_type_sym.info as table.Map
typ := c.table.find_or_register_array(info.key_type)
ret_type = table.Type(typ)
}
else {}
}
2020-05-26 02:09:26 +02:00
call_expr.receiver_type = left_type.to_ptr()
call_expr.return_type = ret_type
2020-05-26 02:09:26 +02:00
return call_expr.return_type
2020-07-14 18:55:44 +02:00
} else if left_type_sym.kind == .array && method_name in ['first', 'last', 'pop'] {
info := left_type_sym.info as table.Array
call_expr.return_type = info.elem_type
2020-07-14 18:55:44 +02:00
if method_name == 'pop' {
call_expr.receiver_type = left_type.to_ptr()
} else {
call_expr.receiver_type = left_type
}
2020-05-01 00:29:54 +02:00
return call_expr.return_type
} else if left_type_sym.kind == .array && method_name in ['insert', 'prepend'] {
info := left_type_sym.info as table.Array
arg_expr := if method_name == 'insert' {
call_expr.args[1].expr
} else {
call_expr.args[0].expr
}
arg_type := c.expr(arg_expr)
arg_sym := c.table.get_type_symbol(arg_type)
if !c.check_types(arg_type, info.elem_type) && !c.check_types(left_type, arg_type) {
c.error('cannot $method_name `$arg_sym.name` to `$left_type_sym.name`', arg_expr.position())
}
} else if left_type_sym.kind == .gohandle && method_name == 'wait' {
info := left_type_sym.info as table.GoHandle
if call_expr.args.len > 0 {
c.error('wait() does not have any arguments', call_expr.args[0].pos)
}
return info.return_type
}
mut method := table.Fn{}
mut has_method := false
mut is_method_from_embed := false
if m := c.table.type_find_method(left_type_sym, method_name) {
method = m
has_method = true
} else {
// can this logic be moved to table.type_find_method() so it can be used from anywhere
if left_type_sym.info is table.Struct {
mut found_methods := []table.Fn{}
mut embed_of_found_methods := []table.Type{}
for embed in left_type_sym.info.embeds {
embed_sym := c.table.get_type_symbol(embed)
if m := c.table.type_find_method(embed_sym, method_name) {
found_methods << m
embed_of_found_methods << embed
}
}
if found_methods.len == 1 {
method = found_methods[0]
has_method = true
is_method_from_embed = true
call_expr.from_embed_type = embed_of_found_methods[0]
} else if found_methods.len > 1 {
c.error('ambiguous method `$method_name`', call_expr.pos)
}
}
if left_type_sym.kind == .aggregate {
// the error message contains the problematic type
unknown_method_msg = err
}
}
if has_method {
if !method.is_pub && !c.is_builtin_mod && !c.pref.is_test && left_type_sym.mod != c.mod
&& left_type_sym.mod != '' { // method.mod != c.mod {
// If a private method is called outside of the module
// its receiver type is defined in, show an error.
2020-04-25 17:49:16 +02:00
// println('warn $method_name lef.mod=$left_type_sym.mod c.mod=$c.mod')
c.error('method `${left_type_sym.name}.$method_name` is private', call_expr.pos)
}
rec_share := method.params[0].typ.share()
if rec_share == .shared_t && (c.locked_names.len > 0 || c.rlocked_names.len > 0) {
c.error('method with `shared` receiver cannot be called inside `lock`/`rlock` block',
call_expr.pos)
}
2020-09-27 03:32:56 +02:00
if method.params[0].is_mut {
to_lock, pos := c.fail_if_immutable(call_expr.left)
2020-06-06 16:05:16 +02:00
// call_expr.is_mut = true
if to_lock != '' && rec_share != .shared_t {
c.error('$to_lock is `shared` and must be `lock`ed to be passed as `mut`',
pos)
}
} else {
if left_type.has_flag(.shared_f) {
to_lock, pos := c.needs_rlock(call_expr.left)
if to_lock != '' {
c.error('$to_lock is `shared` and must be `rlock`ed or `lock`ed to be used as non-mut receiver',
pos)
}
}
}
if (!left_type_sym.is_builtin() && method.mod != 'builtin') && method.language == .v
&& method.no_body {
c.error('cannot call a method that does not have a body', call_expr.pos)
}
if method.return_type == table.void_type && method.ctdefine.len > 0
&& method.ctdefine !in c.pref.compile_defines {
call_expr.should_be_skipped = true
}
2020-09-27 03:32:56 +02:00
nr_args := if method.params.len == 0 { 0 } else { method.params.len - 1 }
min_required_args := method.params.len - if method.is_variadic && method.params.len > 1 {
2
} else {
1
}
if call_expr.args.len < min_required_args {
c.error('expected $min_required_args arguments, but got $call_expr.args.len',
call_expr.pos)
2020-04-22 20:20:49 +02:00
} else if !method.is_variadic && call_expr.args.len > nr_args {
unexpected_arguments := call_expr.args[min_required_args..]
unexpected_arguments_pos := unexpected_arguments[0].pos.extend(unexpected_arguments.last().pos)
c.error('expected $nr_args arguments, but got $call_expr.args.len', unexpected_arguments_pos)
return method.return_type
}
// if method_name == 'clone' {
// println('CLONE nr args=$method.args.len')
// }
// call_expr.args << method.args[0].typ
// call_expr.exp_arg_types << method.args[0].typ
for i, arg in call_expr.args {
exp_arg_typ := if method.is_variadic && i >= method.params.len - 1 {
method.params[method.params.len - 1].typ
} else {
method.params[i + 1].typ
}
exp_arg_sym := c.table.get_type_symbol(exp_arg_typ)
2020-05-04 17:32:40 +02:00
c.expected_type = exp_arg_typ
got_arg_typ := c.expr(arg.expr)
call_expr.args[i].typ = got_arg_typ
2020-06-24 14:44:06 +02:00
if method.is_variadic && got_arg_typ.has_flag(.variadic) && call_expr.args.len - 1 > i {
c.error('when forwarding a varg variable, it must be the final argument',
call_expr.pos)
}
if exp_arg_sym.kind == .interface_ {
c.type_implements(got_arg_typ, exp_arg_typ, arg.expr.position())
continue
}
if method.generic_names.len > 0 {
continue
}
c.check_expected_call_arg(got_arg_typ, exp_arg_typ) or {
2020-05-04 17:32:40 +02:00
// str method, allow type with str method if fn arg is string
// Passing an int or a string array produces a c error here
// Deleting this condition results in propper V error messages
// if arg_typ_sym.kind == .string && typ_sym.has_method('str') {
2020-06-24 14:44:06 +02:00
// continue
// }
if got_arg_typ != table.void_type {
c.error('$err in argument ${i + 1} to `${left_type_sym.name}.$method_name`',
call_expr.pos)
}
2020-05-04 17:32:40 +02:00
}
param := if method.is_variadic && i >= method.params.len - 1 {
method.params[method.params.len - 1]
} else {
method.params[i + 1]
}
param_share := param.typ.share()
if param_share == .shared_t && (c.locked_names.len > 0 || c.rlocked_names.len > 0) {
c.error('method with `shared` arguments cannot be called inside `lock`/`rlock` block',
call_expr.pos)
}
if arg.is_mut {
to_lock, pos := c.fail_if_immutable(arg.expr)
if !param.is_mut {
tok := arg.share.str()
c.error('`$call_expr.name` parameter `$param.name` is not `$tok`, `$tok` is not needed`',
arg.expr.position())
} else {
if param.typ.share() != arg.share {
c.error('wrong shared type', arg.expr.position())
}
if to_lock != '' && param_share != .shared_t {
c.error('$to_lock is `shared` and must be `lock`ed to be passed as `mut`',
pos)
}
}
} else {
if param.is_mut {
tok := arg.share.str()
c.error('`$call_expr.name` parameter `$param.name` is `$tok`, you need to provide `$tok` e.g. `$tok arg${
i + 1}`', arg.expr.position())
} else {
to_lock, pos := c.needs_rlock(arg.expr)
if to_lock != '' {
c.error('$to_lock is `shared` and must be `rlock`ed or `locked` to be passed as non-mut argument',
pos)
}
}
}
}
if method.is_unsafe && !c.inside_unsafe {
c.warn('method `${left_type_sym.name}.$method_name` must be called from an `unsafe` block',
call_expr.pos)
}
// TODO: typ optimize.. this node can get processed more than once
2020-04-03 15:18:17 +02:00
if call_expr.expected_arg_types.len == 0 {
2020-09-27 03:32:56 +02:00
for i in 1 .. method.params.len {
call_expr.expected_arg_types << method.params[i].typ
}
}
if is_method_from_embed {
call_expr.receiver_type = call_expr.from_embed_type.derive(method.params[0].typ)
} else if is_generic {
2020-05-28 05:50:57 +02:00
// We need the receiver to be T in cgen.
// TODO: cant we just set all these to the concrete type in checker? then no need in gen
2020-09-27 03:32:56 +02:00
call_expr.receiver_type = left_type.derive(method.params[0].typ).set_flag(.generic)
2020-05-28 05:50:57 +02:00
} else {
2020-09-27 03:32:56 +02:00
call_expr.receiver_type = method.params[0].typ
2020-05-28 05:50:57 +02:00
}
call_expr.return_type = method.return_type
if method.generic_names.len != call_expr.generic_types.len {
// no type arguments given in call, attempt implicit instantiation
c.infer_fn_types(method, mut call_expr)
}
if call_expr.generic_types.len > 0 && method.return_type != 0 {
2021-01-23 22:41:32 +01:00
if typ := c.resolve_generic_type(method.return_type, method.generic_names,
call_expr.generic_types)
{
call_expr.return_type = typ
return typ
}
}
if call_expr.generic_types.len > 0 && method.generic_names.len == 0 {
c.error('a non generic function called like a generic one', call_expr.generic_list_pos)
}
if method.generic_names.len > 0 {
return call_expr.return_type
}
return method.return_type
}
// TODO: str methods
if method_name == 'str' {
if left_type_sym.kind == .interface_ {
iname := left_type_sym.name
c.error('interface `$iname` does not have a .str() method. Use typeof() instead',
call_expr.pos)
}
call_expr.receiver_type = left_type
call_expr.return_type = table.string_type
2020-07-04 13:45:11 +02:00
if call_expr.args.len > 0 {
c.error('.str() method calls should have no arguments', call_expr.pos)
}
return table.string_type
}
// call struct field fn type
// TODO: can we use SelectorExpr for all? this dosent really belong here
if field := c.table.find_field(left_type_sym, method_name) {
field_type_sym := c.table.get_type_symbol(field.typ)
if field_type_sym.kind == .function {
// call_expr.is_method = false
call_expr.is_field = true
info := field_type_sym.info as table.FnType
call_expr.return_type = info.func.return_type
mut earg_types := []table.Type{}
for mut arg in call_expr.args {
targ := c.expr(arg.expr)
arg.typ = targ
earg_types << targ
}
call_expr.expected_arg_types = earg_types
return info.func.return_type
}
}
if left_type != table.void_type {
suggestion := util.new_suggestion(method_name, left_type_sym.methods.map(it.name))
c.error(suggestion.say(unknown_method_msg), call_expr.pos)
}
return table.void_type
}
2020-05-11 14:38:25 +02:00
pub fn (mut c Checker) call_fn(mut call_expr ast.CallExpr) table.Type {
fn_name := call_expr.name
2020-04-28 07:21:50 +02:00
if fn_name == 'main' {
c.error('the `main` function cannot be called in the program', call_expr.pos)
}
if fn_name == 'typeof' {
// TODO: impl typeof properly (probably not going to be a fn call)
return table.string_type
}
mut has_generic_generic := false // foo<T>() instead of foo<int>()
mut generic_types := []table.Type{}
for generic_type in call_expr.generic_types {
if generic_type.has_flag(.generic) {
has_generic_generic = true
generic_types << c.unwrap_generic(generic_type)
} else {
generic_types << generic_type
}
}
if has_generic_generic {
if c.mod != '' {
// Need to prepend the module when adding a generic type to a function
c.table.register_fn_gen_type(c.mod + '.' + fn_name, generic_types)
} else {
c.table.register_fn_gen_type(fn_name, generic_types)
}
}
if fn_name == 'json.encode' {
} else if fn_name == 'json.decode' && call_expr.args.len > 0 {
2020-05-04 16:46:36 +02:00
expr := call_expr.args[0].expr
2020-07-08 15:46:58 +02:00
if expr !is ast.Type {
typ := expr.type_name()
2020-05-05 14:19:31 +02:00
c.error('json.decode: first argument needs to be a type, got `$typ`', call_expr.pos)
2020-05-04 16:46:36 +02:00
return table.void_type
}
c.expected_type = table.string_type
call_expr.args[1].typ = c.expr(call_expr.args[1].expr)
2020-05-13 21:59:05 +02:00
if call_expr.args[1].typ != table.string_type {
c.error('json.decode: second argument needs to be a string', call_expr.pos)
}
2020-05-04 16:46:36 +02:00
typ := expr as ast.Type
2020-05-21 22:35:43 +02:00
ret_type := typ.typ.set_flag(.optional)
call_expr.return_type = ret_type
return ret_type
2020-05-01 12:02:31 +02:00
}
// look for function in format `mod.fn` or `fn` (builtin)
mut f := table.Fn{}
mut found := false
mut found_in_args := false
// anon fn direct call
if mut call_expr.left is ast.AnonFn {
// it was set to anon for checker errors, clear for gen
call_expr.name = ''
c.expr(call_expr.left)
anon_fn_sym := c.table.get_type_symbol(call_expr.left.typ)
f = (anon_fn_sym.info as table.FnType).func
found = true
}
// try prefix with current module as it would have never gotten prefixed
if !found && !fn_name.contains('.') && call_expr.mod != 'builtin' {
name_prefixed := '${call_expr.mod}.$fn_name'
if f1 := c.table.find_fn(name_prefixed) {
call_expr.name = name_prefixed
found = true
f = f1
c.table.fns[name_prefixed].usages++
}
}
if !found && call_expr.left is ast.IndexExpr {
c.expr(call_expr.left)
expr := call_expr.left as ast.IndexExpr
sym := c.table.get_type_symbol(expr.left_type)
if sym.kind == .array {
info := sym.info as table.Array
elem_typ := c.table.get_type_symbol(info.elem_type)
if elem_typ.info is table.FnType {
return elem_typ.info.func.return_type
}
} else if sym.kind == .map {
info := sym.info as table.Map
value_typ := c.table.get_type_symbol(info.value_type)
if value_typ.info is table.FnType {
return value_typ.info.func.return_type
}
} else if sym.kind == .array_fixed {
info := sym.info as table.ArrayFixed
elem_typ := c.table.get_type_symbol(info.elem_type)
if elem_typ.info is table.FnType {
return elem_typ.info.func.return_type
}
}
found = true
return table.string_type
}
// already prefixed (mod.fn) or C/builtin/main
if !found {
if f1 := c.table.find_fn(fn_name) {
found = true
f = f1
c.table.fns[fn_name].usages++
}
}
if c.pref.is_script && !found {
os_name := 'os.$fn_name'
if f1 := c.table.find_fn(os_name) {
if f1.generic_names.len == call_expr.generic_types.len {
c.table.fn_gen_types[os_name] = c.table.fn_gen_types['${call_expr.mod}.$call_expr.name']
}
call_expr.name = os_name
found = true
f = f1
c.table.fns[os_name].usages++
}
}
// check for arg (var) of fn type
if !found {
if v := call_expr.scope.find_var(fn_name) {
2020-04-15 01:45:27 +02:00
if v.typ != 0 {
vts := c.table.get_type_symbol(v.typ)
if vts.kind == .function {
info := vts.info as table.FnType
f = info.func
found = true
found_in_args = true
}
}
}
}
if !found {
c.error('unknown function: $fn_name', call_expr.pos)
return table.void_type
}
if !found_in_args {
if _ := call_expr.scope.find_var(fn_name) {
2020-04-29 12:20:22 +02:00
c.error('ambiguous call to: `$fn_name`, may refer to fn `$fn_name` or variable `$fn_name`',
call_expr.pos)
}
}
if !f.is_pub && f.language == .v && f.name.len > 0 && f.mod.len > 0 && f.mod != c.mod {
c.error('function `$f.name` is private, so you can not import it in module `$c.mod`',
call_expr.pos)
2020-06-23 16:25:24 +02:00
}
if f.is_deprecated {
c.warn('function `$f.name` has been deprecated', call_expr.pos)
}
if f.is_unsafe && !c.inside_unsafe && f.language == .c && f.name[2] in [`m`, `s`]
&& f.mod == 'builtin' {
// builtin C.m*, C.s* only - temp
2020-07-20 22:30:09 +02:00
c.warn('function `$f.name` must be called from an `unsafe` block', call_expr.pos)
}
if f.mod != 'builtin' && f.language == .v && f.no_body {
c.error('cannot call a function that does not have a body', call_expr.pos)
}
for generic_type in call_expr.generic_types {
sym := c.table.get_type_symbol(generic_type)
if sym.kind == .placeholder {
c.error('unknown type `$sym.name`', call_expr.generic_list_pos)
}
}
if f.generic_names.len > 0 && f.return_type.has_flag(.generic) {
rts := c.table.get_type_symbol(f.return_type)
if rts.info is table.Struct {
if rts.info.generic_types.len > 0 {
gts := c.table.get_type_symbol(call_expr.generic_types[0])
nrt := '$rts.name<$gts.name>'
idx := c.table.type_idxs[nrt]
if idx == 0 {
c.error('unknown type: $nrt', call_expr.pos)
}
call_expr.return_type = table.new_type(idx).derive(f.return_type)
}
}
} else {
call_expr.return_type = f.return_type
}
if f.return_type == table.void_type && f.ctdefine.len > 0
&& f.ctdefine !in c.pref.compile_defines {
call_expr.should_be_skipped = true
}
if f.language != .v || call_expr.language != .v {
// ignore C function of type `fn()`, assume untyped
// For now don't check C functions that are variadic, underscored, capitalized
// or have no params and return int
if f.language == .c && f.params.len != call_expr.args.len && !f.is_variadic
&& f.name[2] != `_` && !f.name[2].is_capital()
&& (f.params.len != 0 || f.return_type !in [table.void_type, table.int_type]) {
// change to error later
c.warn('expected $f.params.len arguments, but got $call_expr.args.len', call_expr.pos)
}
for arg in call_expr.args {
c.expr(arg.expr)
}
return f.return_type
}
2020-09-27 03:32:56 +02:00
min_required_args := if f.is_variadic { f.params.len - 1 } else { f.params.len }
if call_expr.args.len < min_required_args {
c.error('expected $min_required_args arguments, but got $call_expr.args.len',
call_expr.pos)
2020-09-27 03:32:56 +02:00
} else if !f.is_variadic && call_expr.args.len > f.params.len {
unexpected_arguments := call_expr.args[min_required_args..]
unexpected_arguments_pos := unexpected_arguments[0].pos.extend(unexpected_arguments.last().pos)
c.error('expected $min_required_args arguments, but got $call_expr.args.len',
unexpected_arguments_pos)
return f.return_type
}
// println / eprintln can print anything
if fn_name in ['println', 'print', 'eprintln', 'eprint'] && call_expr.args.len > 0 {
c.expected_type = table.string_type
call_expr.args[0].typ = c.expr(call_expr.args[0].expr)
/*
// TODO: optimize `struct T{} fn (t &T) str() string {return 'abc'} mut a := []&T{} a << &T{} println(a[0])`
// It currently generates:
// `println(T_str_no_ptr(*(*(T**)array_get(a, 0))));`
// ... which works, but could be just:
// `println(T_str(*(T**)array_get(a, 0)));`
prexpr := call_expr.args[0].expr
prtyp := call_expr.args[0].typ
prtyp_sym := c.table.get_type_symbol(prtyp)
prtyp_is_ptr := prtyp.is_ptr()
prhas_str, prexpects_ptr, prnr_args := prtyp_sym.str_method_info()
eprintln('>>> println hack typ: ${prtyp} | sym.name: ${prtyp_sym.name} | is_ptr: $prtyp_is_ptr | has_str: $prhas_str | expects_ptr: $prexpects_ptr | nr_args: $prnr_args | expr: ${prexpr.str()} ')
*/
return f.return_type
2020-03-26 11:32:29 +01:00
}
// TODO: typ optimize.. this node can get processed more than once
if call_expr.expected_arg_types.len == 0 {
2020-09-27 03:32:56 +02:00
for param in f.params {
call_expr.expected_arg_types << param.typ
}
}
for i, call_arg in call_expr.args {
arg := if f.is_variadic && i >= f.params.len - 1 {
f.params[f.params.len - 1]
} else {
f.params[i]
}
c.expected_type = arg.typ
typ := c.expr(call_arg.expr)
call_expr.args[i].typ = typ
typ_sym := c.table.get_type_symbol(typ)
arg_typ_sym := c.table.get_type_symbol(arg.typ)
2020-06-04 14:38:54 +02:00
if f.is_variadic && typ.has_flag(.variadic) && call_expr.args.len - 1 > i {
2020-06-24 14:44:06 +02:00
c.error('when forwarding a varg variable, it must be the final argument',
call_expr.pos)
}
arg_share := arg.typ.share()
if arg_share == .shared_t && (c.locked_names.len > 0 || c.rlocked_names.len > 0) {
c.error('function with `shared` arguments cannot be called inside `lock`/`rlock` block',
call_expr.pos)
}
if call_arg.is_mut {
to_lock, pos := c.fail_if_immutable(call_arg.expr)
if !arg.is_mut {
tok := call_arg.share.str()
c.error('`$call_expr.name` parameter `$arg.name` is not `$tok`, `$tok` is not needed`',
call_arg.expr.position())
} else {
if arg.typ.share() != call_arg.share {
c.error('wrong shared type', call_arg.expr.position())
}
if to_lock != '' && !arg.typ.has_flag(.shared_f) {
c.error('$to_lock is `shared` and must be `lock`ed to be passed as `mut`',
pos)
}
}
} else {
if arg.is_mut {
tok := call_arg.share.str()
c.error('`$call_expr.name` parameter `$arg.name` is `$tok`, you need to provide `$tok` e.g. `$tok arg${
i + 1}`', call_arg.expr.position())
} else {
to_lock, pos := c.needs_rlock(call_arg.expr)
if to_lock != '' {
c.error('$to_lock is `shared` and must be `rlock`ed or `lock`ed to be passed as non-mut argument',
pos)
}
}
}
// Handle expected interface
if arg_typ_sym.kind == .interface_ {
c.type_implements(typ, arg.typ, call_arg.expr.position())
continue
}
c.check_expected_call_arg(typ, arg.typ) or {
// str method, allow type with str method if fn arg is string
// Passing an int or a string array produces a c error here
// Deleting this condition results in propper V error messages
// if arg_typ_sym.kind == .string && typ_sym.has_method('str') {
2020-09-13 16:21:40 +02:00
// continue
// }
if typ_sym.kind == .void && arg_typ_sym.kind == .string {
continue
}
if f.generic_names.len > 0 {
continue
}
c.error('$err in argument ${i + 1} to `$fn_name`', call_arg.pos)
}
}
if f.generic_names.len != call_expr.generic_types.len {
// no type arguments given in call, attempt implicit instantiation
c.infer_fn_types(f, mut call_expr)
}
if call_expr.generic_types.len > 0 && f.return_type != 0 {
2021-01-23 22:41:32 +01:00
if typ := c.resolve_generic_type(f.return_type, f.generic_names, call_expr.generic_types) {
call_expr.return_type = typ
return typ
2020-05-21 18:15:04 +02:00
}
}
if call_expr.generic_types.len > 0 && f.generic_names.len == 0 {
c.error('a non generic function called like a generic one', call_expr.generic_list_pos)
}
if f.generic_names.len > 0 {
return call_expr.return_type
}
return f.return_type
}
fn (mut c Checker) type_implements(typ table.Type, inter_typ table.Type, pos token.Position) bool {
$if debug_interface_type_implements ? {
eprintln('> type_implements typ: $typ.debug() | inter_typ: $inter_typ.debug()')
}
typ_sym := c.table.get_type_symbol(typ)
mut inter_sym := c.table.get_type_symbol(inter_typ)
styp := c.table.type_to_str(typ)
same_base_type := typ.idx() == inter_typ.idx()
if typ_sym.kind == .interface_ && inter_sym.kind == .interface_ && !same_base_type {
c.error('cannot implement interface `$inter_sym.name` with a different interface `$styp`',
pos)
}
imethods := if inter_sym.kind == .interface_ {
(inter_sym.info as table.Interface).methods
} else {
inter_sym.methods
}
for imethod in imethods {
if method := typ_sym.find_method(imethod.name) {
msg := c.table.is_same_method(imethod, method)
if msg.len > 0 {
sig := c.table.fn_signature(imethod, skip_receiver: true)
c.add_error_detail('$inter_sym.name has `$sig`')
c.error('`$styp` incorrectly implements method `$imethod.name` of interface `$inter_sym.name`: $msg',
pos)
2020-05-31 10:22:18 +02:00
return false
}
continue
}
c.error("`$styp` doesn't implement method `$imethod.name` of interface `$inter_sym.name`",
pos)
}
if mut inter_sym.info is table.Interface {
for ifield in inter_sym.info.fields {
if field := typ_sym.find_field(ifield.name) {
if ifield.typ != field.typ {
exp := c.table.type_to_str(ifield.typ)
got := c.table.type_to_str(field.typ)
c.error('`$styp` incorrectly implements field `$ifield.name` of interface `$inter_sym.name`, expected `$exp`, got `$got`',
pos)
return false
} else if ifield.is_mut && !(field.is_mut || field.is_global) {
c.error('`$styp` incorrectly implements interface `$inter_sym.name`, field `$ifield.name` must be mutable',
pos)
return false
}
continue
}
c.error("`$styp` doesn't implement field `$ifield.name` of interface `$inter_sym.name`",
pos)
}
if typ !in inter_sym.info.types && typ_sym.kind != .interface_ {
inter_sym.info.types << typ
}
}
2020-05-31 10:22:18 +02:00
return true
}
// return the actual type of the expression, once the optional is handled
pub fn (mut c Checker) check_expr_opt_call(expr ast.Expr, ret_type table.Type) table.Type {
2020-05-21 22:35:43 +02:00
if expr is ast.CallExpr {
if expr.return_type.has_flag(.optional) {
if expr.or_block.kind == .absent {
c.error('${expr.name}() returns an option, so it should have either an `or {}` block, or `?` at the end',
expr.pos)
} else {
c.check_or_expr(expr.or_block, ret_type, expr.return_type.clear_flag(.optional))
}
// remove optional flag
// return ret_type.clear_flag(.optional)
// TODO: currently unwrapped in assign, would need to refactor assign to unwrap here
return ret_type
} else if expr.or_block.kind == .block {
c.error('unexpected `or` block, the function `$expr.name` does not return an optional',
expr.or_block.pos)
} else if expr.or_block.kind == .propagate {
c.error('unexpected `?`, the function `$expr.name` does not return an optional',
expr.or_block.pos)
}
} else if expr is ast.IndexExpr {
if expr.or_expr.kind != .absent {
c.check_or_expr(expr.or_expr, ret_type, ret_type)
}
}
return ret_type
}
pub fn (mut c Checker) check_or_expr(or_expr ast.OrExpr, ret_type table.Type, expr_return_type table.Type) {
if or_expr.kind == .propagate {
if !c.cur_fn.return_type.has_flag(.optional) && c.cur_fn.name != 'main.main' {
c.error('to propagate the optional call, `$c.cur_fn.name` must return an optional',
or_expr.pos)
2020-05-23 08:51:15 +02:00
}
return
}
stmts_len := or_expr.stmts.len
if stmts_len == 0 {
if ret_type != table.void_type {
// x := f() or {}
c.error('assignment requires a non empty `or {}` block', or_expr.pos)
return
}
// allow `f() or {}`
return
}
last_stmt := or_expr.stmts[stmts_len - 1]
if ret_type != table.void_type {
match last_stmt {
ast.ExprStmt {
c.expected_type = ret_type
c.expected_or_type = ret_type.clear_flag(.optional)
last_stmt_typ := c.expr(last_stmt.expr)
c.expected_or_type = table.void_type
type_fits := c.check_types(last_stmt_typ, ret_type)
&& last_stmt_typ.nr_muls() == ret_type.nr_muls()
2020-06-18 20:38:59 +02:00
is_panic_or_exit := is_expr_panic_or_exit(last_stmt.expr)
if type_fits || is_panic_or_exit {
return
}
expected_type_name := c.table.type_to_str(ret_type.clear_flag(.optional))
if last_stmt.typ == table.void_type {
c.error('`or` block must provide a default value of type `$expected_type_name`, or return/exit/continue/break/panic',
last_stmt.pos)
} else {
type_name := c.table.type_to_str(last_stmt_typ)
c.error('wrong return type `$type_name` in the `or {}` block, expected `$expected_type_name`',
last_stmt.pos)
}
return
}
ast.BranchStmt {
if last_stmt.kind !in [.key_continue, .key_break] {
c.error('only break/continue is allowed as a branch statement in the end of an `or {}` block',
last_stmt.pos)
return
}
}
ast.Return {}
else {
expected_type_name := c.table.type_to_str(ret_type.clear_flag(.optional))
c.error('last statement in the `or {}` block should be an expression of type `$expected_type_name` or exit parent scope',
or_expr.pos)
return
}
}
} else {
match last_stmt {
ast.ExprStmt {
if last_stmt.typ == table.void_type {
return
}
if is_expr_panic_or_exit(last_stmt.expr) {
return
}
if c.check_types(last_stmt.typ, expr_return_type) {
return
}
// opt_returning_string() or { ... 123 }
type_name := c.table.type_to_str(last_stmt.typ)
expr_return_type_name := c.table.type_to_str(expr_return_type)
c.error('the default expression type in the `or` block should be `$expr_return_type_name`, instead you gave a value of type `$type_name`',
last_stmt.expr.position())
}
else {}
}
}
}
fn is_expr_panic_or_exit(expr ast.Expr) bool {
match expr {
2020-06-18 20:38:59 +02:00
ast.CallExpr { return expr.name in ['panic', 'exit'] }
2020-04-25 17:49:16 +02:00
else { return false }
}
}
2020-05-11 14:38:25 +02:00
pub fn (mut c Checker) selector_expr(mut selector_expr ast.SelectorExpr) table.Type {
prevent_sum_type_unwrapping_once := c.prevent_sum_type_unwrapping_once
c.prevent_sum_type_unwrapping_once = false
// T.name, typeof(expr).name
mut name_type := 0
match mut selector_expr.expr {
ast.Ident {
name := selector_expr.expr.name
valid_generic := util.is_generic_type_name(name)
&& c.cur_fn.generic_params.filter(it.name == name).len != 0
if valid_generic {
name_type = table.Type(c.table.find_type_idx(name)).set_flag(.generic)
}
}
// Note: in future typeof() should be a type known at compile-time
// sum types should not be handled dynamically
ast.TypeOf {
name_type = c.expr(selector_expr.expr.expr)
}
else {}
}
if name_type > 0 {
if selector_expr.field_name != 'name' {
c.error('invalid field `.$selector_expr.field_name` for type `$selector_expr.expr`',
selector_expr.pos)
}
selector_expr.name_type = name_type
return table.string_type
}
typ := c.expr(selector_expr.expr)
2020-03-11 05:56:15 +01:00
if typ == table.void_type_idx {
c.error('unknown selector expression', selector_expr.pos)
return table.void_type
}
2020-03-07 04:45:35 +01:00
selector_expr.expr_type = typ
if selector_expr.expr_type.has_flag(.optional) && !((selector_expr.expr is ast.Ident
&& (selector_expr.expr as ast.Ident).kind == .constant)) {
c.error('cannot access fields of an optional, handle the error with `or {...}` or propagate it with `?`',
selector_expr.pos)
}
2020-05-09 15:16:48 +02:00
field_name := selector_expr.field_name
utyp := c.unwrap_generic(typ)
sym := c.table.get_type_symbol(utyp)
if typ.has_flag(.variadic) || sym.kind == .array_fixed || sym.kind == .chan {
if field_name == 'len' || (sym.kind == .chan && field_name == 'cap') {
selector_expr.typ = table.int_type
2020-02-29 09:04:47 +01:00
return table.int_type
}
if sym.kind == .chan && field_name == 'closed' {
selector_expr.typ = table.bool_type
return table.bool_type
}
2020-02-29 09:04:47 +01:00
}
mut unknown_field_msg := 'type `$sym.name` has no field or method `$field_name`'
mut has_field := false
mut field := table.Field{}
if field_name.len > 0 && field_name[0].is_capital() && sym.info is table.Struct
&& sym.language == .v {
// x.Foo.y => access the embedded struct
sym_info := sym.info as table.Struct
for embed in sym_info.embeds {
embed_sym := c.table.get_type_symbol(embed)
if embed_sym.embed_name() == field_name {
selector_expr.typ = embed
return embed
}
}
} else {
if f := c.table.find_field(sym, field_name) {
has_field = true
field = f
} else {
// look for embedded field
if sym.info is table.Struct {
mut found_fields := []table.Field{}
mut embed_of_found_fields := []table.Type{}
for embed in sym.info.embeds {
embed_sym := c.table.get_type_symbol(embed)
if f := c.table.find_field(embed_sym, field_name) {
found_fields << f
embed_of_found_fields << embed
}
}
if found_fields.len == 1 {
field = found_fields[0]
has_field = true
selector_expr.from_embed_type = embed_of_found_fields[0]
} else if found_fields.len > 1 {
c.error('ambiguous field `$field_name`', selector_expr.pos)
}
}
if sym.kind == .aggregate {
unknown_field_msg = err
}
}
if !c.inside_unsafe {
if sym.info is table.Struct {
if sym.info.is_union && selector_expr.next_token !in token.assign_tokens {
c.warn('reading a union field (or its address) requires `unsafe`',
selector_expr.pos)
}
}
}
}
if has_field {
if sym.mod != c.mod && !field.is_pub && sym.language != .c {
c.error('field `${sym.name}.$field_name` is not public', selector_expr.pos)
2020-05-11 14:38:25 +02:00
}
field_sym := c.table.get_type_symbol(field.typ)
if field_sym.kind == .sum_type {
if !prevent_sum_type_unwrapping_once {
if scope_field := selector_expr.scope.find_struct_field(utyp, field_name) {
2020-11-23 16:16:13 +01:00
return scope_field.sum_type_casts.last()
}
}
}
selector_expr.typ = field.typ
return field.typ
}
if sym.kind !in [.struct_, .aggregate, .interface_] {
if sym.kind != .placeholder {
c.error('`$sym.name` is not a struct', selector_expr.pos)
}
} else {
if sym.info is table.Struct {
suggestion := util.new_suggestion(field_name, sym.info.fields.map(it.name))
c.error(suggestion.say(unknown_field_msg), selector_expr.pos)
}
c.error(unknown_field_msg, selector_expr.pos)
2020-02-08 16:59:57 +01:00
}
2020-02-22 12:58:16 +01:00
return table.void_type
}
// TODO: non deferred
2020-05-11 14:38:25 +02:00
pub fn (mut c Checker) return_stmt(mut return_stmt ast.Return) {
2020-05-23 08:51:15 +02:00
c.expected_type = c.cur_fn.return_type
expected_type := c.unwrap_generic(c.expected_type)
expected_type_sym := c.table.get_type_symbol(expected_type)
if return_stmt.exprs.len > 0 && c.cur_fn.return_type == table.void_type {
c.error('too many arguments to return, current function does not return anything',
2020-04-10 18:11:43 +02:00
return_stmt.pos)
return
} else if return_stmt.exprs.len == 0 && !(c.expected_type == table.void_type
|| expected_type_sym.kind == .void) {
2020-04-11 04:09:41 +02:00
c.error('too few arguments to return', return_stmt.pos)
2020-03-22 13:19:45 +01:00
return
}
2020-04-10 18:11:43 +02:00
if return_stmt.exprs.len == 0 {
return
}
2020-06-04 14:38:54 +02:00
exp_is_optional := expected_type.has_flag(.optional)
mut expected_types := [expected_type]
if expected_type_sym.info is table.MultiReturn {
expected_types = expected_type_sym.info.types
if c.cur_generic_types.len > 0 {
expected_types = expected_types.map(c.unwrap_generic(it))
}
}
mut got_types := []table.Type{}
for expr in return_stmt.exprs {
typ := c.expr(expr)
// Unpack multi return types
2020-05-28 05:50:57 +02:00
sym := c.table.get_type_symbol(typ)
if sym.kind == .multi_return {
for t in sym.mr_info().types {
got_types << t
2020-05-28 01:19:03 +02:00
}
} else {
got_types << typ
}
}
2020-03-16 07:42:45 +01:00
return_stmt.types = got_types
// allow `none` & `error (Option)` return types for function that returns optional
2020-04-25 09:08:53 +02:00
if exp_is_optional && got_types[0].idx() in [table.none_type_idx, c.table.type_idxs['Option']] {
return
}
if expected_types.len > 0 && expected_types.len != got_types.len {
2020-03-21 07:01:06 +01:00
c.error('wrong number of return arguments', return_stmt.pos)
return
}
for i, exp_type in expected_types {
got_typ := c.unwrap_generic(got_types[i])
if got_typ.has_flag(.optional) && (!exp_type.has_flag(.optional)
|| c.table.type_to_str(got_typ) != c.table.type_to_str(exp_type)) {
2020-06-04 10:40:32 +02:00
pos := return_stmt.exprs[i].position()
c.error('cannot use `${c.table.type_to_str(got_typ)}` as type `${c.table.type_to_str(exp_type)}` in return argument',
pos)
2020-06-04 10:40:32 +02:00
}
if !c.check_types(got_typ, exp_type) {
got_typ_sym := c.table.get_type_symbol(got_typ)
mut exp_typ_sym := c.table.get_type_symbol(exp_type)
2020-04-17 16:16:56 +02:00
pos := return_stmt.exprs[i].position()
2020-05-30 19:54:16 +02:00
if exp_typ_sym.kind == .interface_ {
c.type_implements(got_typ, exp_type, return_stmt.pos)
continue
}
c.error('cannot use `$got_typ_sym.name` as type `$exp_typ_sym.name` in return argument',
2020-04-25 17:49:16 +02:00
pos)
}
if (got_typ.is_ptr() || got_typ.is_pointer())
&& (!exp_type.is_ptr() && !exp_type.is_pointer()) {
pos := return_stmt.exprs[i].position()
c.error('fn `$c.cur_fn.name` expects you to return a non reference type `${c.table.type_to_str(exp_type)}`, but you are returning `${c.table.type_to_str(got_typ)}` instead',
pos)
}
if (exp_type.is_ptr() || exp_type.is_pointer())
&& (!got_typ.is_ptr() && !got_typ.is_pointer()) && got_typ != table.int_literal_type {
pos := return_stmt.exprs[i].position()
c.error('fn `$c.cur_fn.name` expects you to return a reference type `${c.table.type_to_str(exp_type)}`, but you are returning `${c.table.type_to_str(got_typ)}` instead',
pos)
}
}
if exp_is_optional && return_stmt.exprs.len > 0 {
expr0 := return_stmt.exprs[0]
if expr0 is ast.CallExpr {
if expr0.or_block.kind == .propagate {
c.error('`?` is not needed, use `return ${expr0.name}()`', expr0.pos)
}
}
}
}
2020-11-21 19:07:47 +01:00
pub fn (mut c Checker) const_decl(mut node ast.ConstDecl) {
mut field_names := []string{}
mut field_order := []int{}
for i, field in node.fields {
// TODO Check const name once the syntax is decided
if field.name in c.const_names {
c.error('duplicate const `$field.name`', field.pos)
}
c.const_names << field.name
field_names << field.name
field_order << i
}
mut needs_order := false
mut done_fields := []int{}
for i, field in node.fields {
c.const_decl = field.name
c.const_deps << field.name
mut typ := c.expr(field.expr)
if field.expr is ast.CallExpr {
if field.expr.or_block.kind != .absent {
typ = typ.clear_flag(.optional)
}
}
2020-11-21 19:07:47 +01:00
node.fields[i].typ = c.table.mktyp(typ)
for cd in c.const_deps {
for j, f in node.fields {
if j != i && cd in field_names && cd == f.name && j !in done_fields {
needs_order = true
x := field_order[j]
field_order[j] = field_order[i]
field_order[i] = x
break
}
}
}
done_fields << i
c.const_deps = []
}
if needs_order {
mut ordered_fields := []ast.ConstField{}
for order in field_order {
ordered_fields << node.fields[order]
}
node.fields = ordered_fields
}
}
2020-04-25 17:49:16 +02:00
pub fn (mut c Checker) enum_decl(decl ast.EnumDecl) {
2020-05-16 16:12:23 +02:00
c.check_valid_pascal_case(decl.name, 'enum name', decl.pos)
mut seen := []i64{}
if decl.fields.len == 0 {
c.error('enum cannot be empty', decl.pos)
}
for i, field in decl.fields {
2020-08-09 03:57:54 +02:00
if !c.pref.experimental && util.contains_capital(field.name) {
// TODO C2V uses hundreds of enums with capitals, remove -experimental check once it's handled
2020-05-16 16:12:23 +02:00
c.error('field name `$field.name` cannot contain uppercase letters, use snake_case instead',
field.pos)
}
2020-04-30 18:06:14 +02:00
for j in 0 .. i {
if field.name == decl.fields[j].name {
c.error('field name `$field.name` duplicate', field.pos)
}
}
2020-04-10 14:44:01 +02:00
if field.has_expr {
match field.expr {
ast.IntegerLiteral {
val := field.expr.val.i64()
2021-01-25 10:26:20 +01:00
if val < checker.int_min || val > checker.int_max {
c.error('enum value `$val` overflows int', field.expr.pos)
} else if !decl.is_multi_allowed && i64(val) in seen {
c.error('enum value `$val` already exists', field.expr.pos)
}
seen << i64(val)
}
2020-04-09 19:23:49 +02:00
ast.PrefixExpr {}
else {
2020-04-19 22:26:04 +02:00
if field.expr is ast.Ident {
if field.expr.language == .c {
2020-04-19 22:26:04 +02:00
continue
}
}
mut pos := field.expr.position()
2020-04-10 00:09:34 +02:00
if pos.pos == 0 {
pos = field.pos
}
2020-04-11 04:09:41 +02:00
c.error('default value for enum has to be an integer', pos)
2020-04-09 19:23:49 +02:00
}
}
} else {
if seen.len > 0 {
last := seen[seen.len - 1]
2021-01-25 10:26:20 +01:00
if last == checker.int_max {
c.error('enum value overflows', field.pos)
}
seen << last + 1
} else {
seen << 0
}
2020-04-09 19:23:49 +02:00
}
}
}
2020-05-11 14:38:25 +02:00
pub fn (mut c Checker) assign_stmt(mut assign_stmt ast.AssignStmt) {
2020-04-25 17:49:16 +02:00
c.expected_type = table.none_type // TODO a hack to make `x := if ... work`
2020-08-10 19:54:38 +02:00
defer {
c.expected_type = table.void_type
}
right_first := assign_stmt.right[0]
mut right_len := assign_stmt.right.len
mut right_type0 := table.void_type
for right in assign_stmt.right {
if right is ast.CallExpr || right is ast.IfExpr || right is ast.MatchExpr {
right_type0 = c.expr(right)
assign_stmt.right_types = [
c.check_expr_opt_call(right, right_type0),
]
right_type_sym0 := c.table.get_type_symbol(right_type0)
if right_type_sym0.kind == .multi_return {
if assign_stmt.right.len > 1 {
c.error('cannot use multi-value $right_type_sym0.name in signle-value context',
right.position())
}
assign_stmt.right_types = right_type_sym0.mr_info().types
right_len = assign_stmt.right_types.len
} else if right_type0 == table.void_type {
right_len = 0
}
}
}
if assign_stmt.left.len != right_len {
if right_first is ast.CallExpr {
c.error('assignment mismatch: $assign_stmt.left.len variable(s) but `${right_first.name}()` returns $right_len value(s)',
assign_stmt.pos)
} else {
c.error('assignment mismatch: $assign_stmt.left.len variable(s) $right_len value(s)',
assign_stmt.pos)
}
return
}
//
is_decl := assign_stmt.op == .decl_assign
for i, left in assign_stmt.left {
if left is ast.CallExpr {
c.error('cannot call function `${left.name}()` on the left side of an assignment',
left.pos)
}
is_blank_ident := left.is_blank_ident()
mut left_type := table.void_type
if !is_decl && !is_blank_ident {
if left is ast.Ident || left is ast.SelectorExpr {
c.prevent_sum_type_unwrapping_once = true
}
left_type = c.expr(left)
c.expected_type = c.unwrap_generic(left_type)
2021-01-30 11:55:10 +01:00
// `map = {}`
sym := c.table.get_type_symbol(left_type)
if sym.kind == .map && assign_stmt.right[i] is ast.StructInit {
assign_stmt.right[i] = ast.MapInit{}
}
}
if assign_stmt.right_types.len < assign_stmt.left.len { // first type or multi return types added above
right_type := c.expr(assign_stmt.right[i])
if assign_stmt.right_types.len == i {
assign_stmt.right_types << c.check_expr_opt_call(assign_stmt.right[i],
right_type)
}
}
right := if i < assign_stmt.right.len { assign_stmt.right[i] } else { assign_stmt.right[0] }
mut right_type := assign_stmt.right_types[i]
if is_decl {
left_type = c.table.mktyp(right_type)
if left_type == table.int_type {
if right is ast.IntegerLiteral {
mut is_large := right.val.len > 13
if !is_large && right.val.len > 8 {
val := right.val.i64()
2021-01-25 10:26:20 +01:00
is_large = val > checker.int_max || val < checker.int_min
}
if is_large {
c.error('overflow in implicit type `int`, use explicit type casting instead',
right.pos)
}
}
}
// we are unwrapping here instead if check_expr_opt_call currently
2020-06-17 04:05:13 +02:00
if left_type.has_flag(.optional) {
left_type = left_type.clear_flag(.optional)
}
} else {
// Make sure the variable is mutable
c.fail_if_immutable(left)
// left_type = c.expr(left)
}
assign_stmt.left_types << left_type
match mut left {
ast.Ident {
2020-06-18 20:38:59 +02:00
if left.kind == .blank_ident {
left_type = right_type
assign_stmt.left_types[i] = right_type
if assign_stmt.op !in [.assign, .decl_assign] {
2020-06-18 20:38:59 +02:00
c.error('cannot modify blank `_` identifier', left.pos)
}
} else if left.info !is ast.IdentVar {
c.error('cannot assign to $left.kind `$left.name`', left.pos)
2020-06-17 04:05:13 +02:00
} else {
if is_decl {
2020-06-18 20:38:59 +02:00
c.check_valid_snake_case(left.name, 'variable name', left.pos)
2020-06-17 04:05:13 +02:00
}
mut ident_var_info := left.info as ast.IdentVar
if ident_var_info.share == .shared_t {
left_type = left_type.set_flag(.shared_f)
if is_decl {
if left_type.nr_muls() > 1 {
c.error('shared cannot be multi level reference', left.pos)
}
left_type = left_type.set_nr_muls(1)
}
} else if left_type.has_flag(.shared_f) {
left_type = left_type.clear_flag(.shared_f)
}
if ident_var_info.share == .atomic_t {
left_type = left_type.set_flag(.atomic_f)
}
assign_stmt.left_types[i] = left_type
ident_var_info.typ = left_type
2020-06-18 20:38:59 +02:00
left.info = ident_var_info
if left_type != 0 {
match mut left.obj {
ast.Var { left.obj.typ = left_type }
ast.GlobalField { left.obj.typ = left_type }
else {}
}
/*
if left.obj is ast.Var as v {
v.typ = left_type
} else if left.obj is ast.GlobalDecl as v {
v.typ = left_type
}
*/
}
if is_decl {
full_name := '${left.mod}.$left.name'
if obj := c.file.global_scope.find(full_name) {
if obj is ast.ConstField {
c.warn('duplicate of a const name `$full_name`', left.pos)
}
}
}
}
}
ast.PrefixExpr {
// Do now allow `*x = y` outside `unsafe`
2020-06-18 20:38:59 +02:00
if left.op == .mul && !c.inside_unsafe {
c.error('modifying variables via dereferencing can only be done in `unsafe` blocks',
2020-06-17 04:05:13 +02:00
assign_stmt.pos)
}
if is_decl {
c.error('non-name on the left side of `:=`', left.pos)
}
}
else {
if is_decl {
c.error('non-name `$left` on left side of `:=`', left.position())
}
}
}
left_type_unwrapped := c.unwrap_generic(left_type)
right_type_unwrapped := c.unwrap_generic(right_type)
if right_type_unwrapped == 0 {
// right type was a generic `T`
continue
}
left_sym := c.table.get_type_symbol(left_type_unwrapped)
right_sym := c.table.get_type_symbol(right_type_unwrapped)
2020-08-10 19:54:38 +02:00
if c.pref.translated {
// TODO fix this in C2V instead, for example cast enums to int before using `|` on them.
2020-08-11 16:26:49 +02:00
// TODO replace all c.pref.translated checks with `$if !translated` for performance
2020-08-11 00:51:15 +02:00
continue
2020-08-10 19:54:38 +02:00
}
if left_sym.kind == .array && !c.inside_unsafe && assign_stmt.op in [.assign, .decl_assign]
&& right_sym.kind == .array && (left is ast.Ident && !left.is_blank_ident())
&& right is ast.Ident {
2020-12-20 10:42:46 +01:00
// Do not allow `a = b`, only `a = b.clone()`
2021-02-06 03:06:34 +01:00
c.error('use `array2 $assign_stmt.op.str() array1.clone()` instead of `array2 $assign_stmt.op.str() array1` (or use `unsafe`)',
assign_stmt.pos)
}
if left_sym.kind == .map && !c.inside_unsafe && assign_stmt.op in [.assign, .decl_assign]
&& right_sym.kind == .map && (left is ast.Ident && !left.is_blank_ident())
&& right is ast.Ident {
// Do not allow `a = b`, only `a = b.clone()`
c.error('use `map2 $assign_stmt.op.str() map1.clone()` instead of `map2 $assign_stmt.op.str() map1` (or use `unsafe`)',
2020-12-20 16:08:56 +01:00
assign_stmt.pos)
2020-12-20 10:42:46 +01:00
}
left_is_ptr := left_type.is_ptr() || left_sym.is_pointer()
if left_is_ptr && c.for_in_mut_val_name != left.str() && left.str() !in c.fn_mut_arg_names {
if !c.inside_unsafe && assign_stmt.op !in [.assign, .decl_assign] {
// ptr op=
c.warn('pointer arithmetic is only allowed in `unsafe` blocks', assign_stmt.pos)
}
right_is_ptr := right_type.is_ptr() || right_sym.is_pointer()
if !right_is_ptr && assign_stmt.op == .assign && right_type_unwrapped.is_number() {
c.error('cannot assign to `$left`: ' +
c.expected_msg(right_type_unwrapped, left_type_unwrapped), right.position())
}
if (right is ast.StructInit || !right_is_ptr) && !(right_sym.is_number()
|| left_type.has_flag(.shared_f)) {
left_name := c.table.type_to_str(left_type_unwrapped)
mut rtype := right_type_unwrapped
if rtype.is_ptr() {
rtype = rtype.deref()
}
right_name := c.table.type_to_str(rtype)
c.error('mismatched types `$left_name` and `$right_name`', assign_stmt.pos)
2020-11-07 20:19:46 +01:00
}
}
// Single side check
match assign_stmt.op {
.assign {} // No need to do single side check for =. But here put it first for speed.
.plus_assign, .minus_assign {
if left_type == table.string_type {
if assign_stmt.op != .plus_assign {
c.error('operator `$assign_stmt.op` not defined on left operand type `$left_sym.name`',
left.position())
}
if right_type != table.string_type {
c.error('invalid right operand: $left_sym.name $assign_stmt.op $right_sym.name',
right.position())
}
} else if !left_sym.is_number()
&& left_sym.kind !in [.byteptr, .charptr, .struct_, .alias] {
c.error('operator `$assign_stmt.op` not defined on left operand type `$left_sym.name`',
2020-06-17 04:05:13 +02:00
left.position())
} else if !right_sym.is_number()
&& left_sym.kind !in [.byteptr, .charptr, .struct_, .alias] {
c.error('invalid right operand: $left_sym.name $assign_stmt.op $right_sym.name',
2020-06-17 04:05:13 +02:00
right.position())
} else if right is ast.IntegerLiteral {
if right.val == '1' {
op := if assign_stmt.op == .plus_assign {
token.Kind.inc
} else {
token.Kind.dec
}
c.error('use `$op` instead of `$assign_stmt.op 1`', assign_stmt.pos)
}
}
}
.mult_assign, .div_assign {
if !left_sym.is_number()
&& !c.table.get_final_type_symbol(left_type_unwrapped).is_int()
&& left_sym.kind !in [.struct_, .alias] {
c.error('operator $assign_stmt.op.str() not defined on left operand type `$left_sym.name`',
left.position())
} else if !right_sym.is_number()
&& !c.table.get_final_type_symbol(left_type_unwrapped).is_int()
&& left_sym.kind !in [.struct_, .alias] {
c.error('operator $assign_stmt.op.str() not defined on right operand type `$right_sym.name`',
right.position())
}
}
.and_assign, .or_assign, .xor_assign, .mod_assign, .left_shift_assign, .right_shift_assign
{
if !left_sym.is_int()
&& !c.table.get_final_type_symbol(left_type_unwrapped).is_int() {
c.error('operator $assign_stmt.op.str() not defined on left operand type `$left_sym.name`',
left.position())
} else if !right_sym.is_int()
&& !c.table.get_final_type_symbol(right_type_unwrapped).is_int() {
c.error('operator $assign_stmt.op.str() not defined on right operand type `$right_sym.name`',
right.position())
}
}
else {}
}
if assign_stmt.op in [.plus_assign, .minus_assign, .mod_assign, .mult_assign, .div_assign]
&& ((left_sym.kind == .struct_ && right_sym.kind == .struct_)
|| left_sym.kind == .alias) {
left_name := c.table.type_to_str(left_type)
right_name := c.table.type_to_str(right_type)
parent_sym := c.table.get_final_type_symbol(left_type)
if left_sym.kind == .alias && right_sym.kind != .alias {
c.error('mismatched types `$left_name` and `$right_name`', assign_stmt.pos)
}
extracted_op := match assign_stmt.op {
.plus_assign { '+' }
.minus_assign { '-' }
.div_assign { '/' }
.mod_assign { '%' }
.mult_assign { '*' }
else { 'unknown op' }
}
if method := left_sym.find_method(extracted_op) {
if method.return_type != left_type {
c.error('operator `$extracted_op` must return `$left_name` to be used as an assignment operator',
assign_stmt.pos)
}
} else {
if parent_sym.is_primitive() {
c.error('cannot use operator methods on type alias for `$parent_sym.name`',
assign_stmt.pos)
}
if left_name == right_name {
c.error('undefined operation `$left_name` $extracted_op `$right_name`',
assign_stmt.pos)
} else {
c.error('mismatched types `$left_name` and `$right_name`', assign_stmt.pos)
}
}
}
if !is_blank_ident && right_sym.kind != .placeholder && left_sym.kind != .interface_ {
// Dual sides check (compatibility check)
c.check_expected(right_type_unwrapped, left_type_unwrapped) or {
c.error('cannot assign to `$left`: $err', right.position())
}
}
if left_sym.kind == .interface_ {
c.type_implements(right_type, left_type, right.position())
}
2020-02-27 16:51:39 +01:00
}
// this needs to run after the assign stmt left exprs have been run through checker so that ident.obj is set
// Check `x := &y` and `mut x := <-ch`
if right_first is ast.PrefixExpr {
node := right_first
left_first := assign_stmt.left[0]
if left_first is ast.Ident {
assigned_var := left_first
c.expr(node.right)
if node.right is ast.Ident {
if node.right.obj is ast.Var {
v := node.right.obj
right_type0 = v.typ
if node.op == .amp {
if !v.is_mut && assigned_var.is_mut && !c.inside_unsafe {
c.error('`$node.right.name` is immutable, cannot have a mutable reference to it',
node.pos)
}
}
}
}
if node.op == .arrow {
if assigned_var.is_mut {
right_sym := c.table.get_type_symbol(right_type0)
if right_sym.kind == .chan {
chan_info := right_sym.chan_info()
if chan_info.elem_type.is_ptr() && !chan_info.is_mut {
c.error('cannot have a mutable reference to object from `$right_sym.name`',
node.pos)
}
}
}
}
}
}
2020-12-20 10:42:46 +01:00
// right_sym := c.table.get_type_symbol(right_type_unwrapped)
2020-02-27 00:12:37 +01:00
}
2020-02-06 17:38:02 +01:00
fn scope_register_it(mut s ast.Scope, pos token.Position, typ table.Type) {
s.register(ast.Var{
name: 'it'
pos: pos
typ: typ
is_used: true
})
}
fn scope_register_ab(mut s ast.Scope, pos token.Position, typ table.Type) {
s.register(ast.Var{
name: 'a'
pos: pos
typ: typ.to_ptr()
is_used: true
})
s.register(ast.Var{
name: 'b'
pos: pos
typ: typ.to_ptr()
is_used: true
})
}
fn (mut c Checker) check_array_init_para_type(para string, expr ast.Expr, pos token.Position) {
sym := c.table.get_type_symbol(c.expr(expr))
if sym.kind !in [.int, .int_literal] {
c.error('array $para needs to be an int', pos)
}
}
2020-05-11 14:38:25 +02:00
pub fn (mut c Checker) array_init(mut array_init ast.ArrayInit) table.Type {
2020-03-24 16:18:37 +01:00
// println('checker: array init $array_init.pos.line_nr $c.file.path')
mut elem_type := table.void_type
2020-03-04 02:50:32 +01:00
// []string - was set in parser
if array_init.typ != table.void_type {
if array_init.exprs.len == 0 {
if array_init.has_cap {
c.check_array_init_para_type('cap', array_init.cap_expr, array_init.pos)
}
if array_init.has_len {
c.check_array_init_para_type('len', array_init.len_expr, array_init.pos)
}
}
sym := c.table.get_type_symbol(array_init.elem_type)
if array_init.has_default {
default_typ := c.expr(array_init.default_expr)
c.check_expected(default_typ, array_init.elem_type) or {
c.error(err, array_init.default_expr.position())
}
}
if sym.kind == .sum_type {
if array_init.has_len && !array_init.has_default {
c.error('cannot initalize sum type array without default value', array_init.elem_type_pos)
}
}
if sym.kind == .placeholder {
c.error('unknown type `$sym.name`', array_init.elem_type_pos)
}
2020-03-04 02:50:32 +01:00
return array_init.typ
}
// a = []
2020-03-04 02:50:32 +01:00
if array_init.exprs.len == 0 {
// a := fn_returing_opt_array() or { [] }
if c.expected_type == table.void_type && c.expected_or_type != table.void_type {
c.expected_type = c.expected_or_type
}
mut type_sym := c.table.get_type_symbol(c.expected_type)
if type_sym.kind != .array {
2020-06-24 14:44:06 +02:00
c.error('array_init: no type specified (maybe: `[]Type{}` instead of `[]`)',
array_init.pos)
return table.void_type
}
// TODO: seperate errors once bug is fixed with `x := if expr { ... } else { ... }`
// if c.expected_type == table.void_type {
// c.error('array_init: use `[]Type{}` instead of `[]`', array_init.pos)
// return table.void_type
// }
array_info := type_sym.array_info()
array_init.elem_type = array_info.elem_type
2020-03-04 02:50:32 +01:00
return c.expected_type
}
// [1,2,3]
if array_init.exprs.len > 0 && array_init.elem_type == table.void_type {
2020-04-29 12:20:22 +02:00
mut expected_value_type := table.void_type
mut expecting_interface_array := false
if c.expected_type != 0 {
expected_value_type = c.table.value_type(c.expected_type)
if c.table.get_type_symbol(expected_value_type).kind == .interface_ {
// Array of interfaces? (`[dog, cat]`) Save the interface type (`Animal`)
expecting_interface_array = true
}
}
// expecting_interface_array := c.expected_type != 0 &&
// c.table.get_type_symbol(c.table.value_type(c.expected_type)).kind == .interface_
//
// if expecting_interface_array {
// println('ex $c.expected_type')
// }
2020-03-04 02:50:32 +01:00
for i, expr in array_init.exprs {
typ := c.expr(expr)
array_init.expr_types << typ
// The first element's type
2020-04-29 12:20:22 +02:00
if expecting_interface_array {
if i == 0 {
elem_type = expected_value_type
c.expected_type = elem_type
c.type_implements(typ, elem_type, expr.position())
2020-04-29 12:20:22 +02:00
}
continue
}
2020-03-04 02:50:32 +01:00
// The first element's type
if i == 0 {
elem_type = c.table.mktyp(typ)
c.expected_type = elem_type
2020-03-04 02:50:32 +01:00
continue
}
c.check_expected(typ, elem_type) or {
c.error('invalid array element: $err', expr.position())
2020-03-04 02:50:32 +01:00
}
2020-01-19 13:52:34 +01:00
}
2020-04-25 18:26:38 +02:00
if array_init.is_fixed {
idx := c.table.find_or_register_array_fixed(elem_type, array_init.exprs.len)
2020-04-25 18:26:38 +02:00
array_init.typ = table.new_type(idx)
} else {
idx := c.table.find_or_register_array(elem_type)
2020-04-25 18:26:38 +02:00
array_init.typ = table.new_type(idx)
}
array_init.elem_type = elem_type
} else if array_init.is_fixed && array_init.exprs.len == 1
&& array_init.elem_type != table.void_type {
// [50]byte
mut fixed_size := 0
init_expr := array_init.exprs[0]
c.expr(init_expr)
match init_expr {
2020-03-04 02:50:32 +01:00
ast.IntegerLiteral {
fixed_size = init_expr.val.int()
2020-03-04 02:50:32 +01:00
}
2020-04-08 19:15:16 +02:00
ast.Ident {
if init_expr.obj is ast.ConstField {
if cint := const_int_value(init_expr.obj) {
2020-04-11 21:31:54 +02:00
fixed_size = cint
}
2020-04-08 19:15:16 +02:00
} else {
c.error('non existent integer const $init_expr.name while initializing the size of a static array',
2020-04-11 04:09:41 +02:00
array_init.pos)
2020-04-08 19:15:16 +02:00
}
2020-04-11 04:09:41 +02:00
}
else {
2020-03-04 02:50:32 +01:00
c.error('expecting `int` for fixed size', array_init.pos)
}
}
if fixed_size <= 0 {
c.error('fixed size cannot be zero or negative', init_expr.position())
}
idx := c.table.find_or_register_array_fixed(array_init.elem_type, fixed_size)
2020-02-19 14:34:44 +01:00
array_type := table.new_type(idx)
array_init.typ = array_type
if array_init.has_default {
c.expr(array_init.default_expr)
}
2020-02-19 14:34:44 +01:00
}
return array_init.typ
2020-01-19 13:52:34 +01:00
}
fn const_int_value(cfield ast.ConstField) ?int {
if cint := is_const_integer(cfield) {
return cint.val.int()
}
2020-04-11 21:31:54 +02:00
return none
}
fn is_const_integer(cfield ast.ConstField) ?ast.IntegerLiteral {
match cfield.expr {
ast.IntegerLiteral { return cfield.expr }
else {}
}
return none
}
[inline]
fn (mut c Checker) check_loop_label(label string, pos token.Position) {
if label.len == 0 {
// ignore
return
}
if c.loop_label.len != 0 {
c.error('nesting of labelled `for` loops is not supported', pos)
return
}
c.loop_label = label
}
2020-04-25 17:49:16 +02:00
fn (mut c Checker) stmt(node ast.Stmt) {
$if trace_checker ? {
stmt_pos := node.position()
eprintln('checking file: ${c.file.path:-30} | stmt pos: ${stmt_pos.str():-45} | stmt')
}
2020-02-27 00:12:37 +01:00
// c.expected_type = table.void_type
match mut node {
2020-03-26 10:27:46 +01:00
ast.AssertStmt {
cur_exp_typ := c.expected_type
2020-06-18 20:38:59 +02:00
assert_type := c.expr(node.expr)
if assert_type != table.bool_type_idx {
atype_name := c.table.get_type_symbol(assert_type).name
2020-06-18 20:38:59 +02:00
c.error('assert can be used only with `bool` expressions, but found `$atype_name` instead',
node.pos)
}
c.expected_type = cur_exp_typ
2020-03-26 10:27:46 +01:00
}
2020-02-06 17:38:02 +01:00
ast.AssignStmt {
2020-06-18 20:38:59 +02:00
c.assign_stmt(mut node)
2020-02-06 17:38:02 +01:00
}
2020-03-25 14:24:48 +01:00
ast.Block {
if node.is_unsafe {
assert !c.inside_unsafe
c.inside_unsafe = true
c.stmts(node.stmts)
c.inside_unsafe = false
} else {
c.stmts(node.stmts)
}
2020-03-25 14:24:48 +01:00
}
ast.BranchStmt {
if c.in_for_count == 0 {
c.error('$node.kind.str() statement not within a loop', node.pos)
}
if node.label.len > 0 {
if node.label != c.loop_label {
c.error('invalid label name `$node.label`', node.pos)
}
}
}
ast.CompFor {
// node.typ = c.expr(node.expr)
c.stmts(node.stmts)
}
ast.ConstDecl {
2020-11-21 19:07:47 +01:00
c.inside_const = true
c.const_decl(mut node)
c.inside_const = false
}
2020-04-09 19:23:49 +02:00
ast.DeferStmt {
if node.idx_in_fn < 0 {
node.idx_in_fn = c.cur_fn.defer_stmts.len
c.cur_fn.defer_stmts << &node
}
2020-06-18 20:38:59 +02:00
c.stmts(node.stmts)
2020-04-09 19:23:49 +02:00
}
ast.EnumDecl {
2020-06-18 20:38:59 +02:00
c.enum_decl(node)
2020-04-09 19:23:49 +02:00
}
ast.ExprStmt {
2020-06-18 20:38:59 +02:00
node.typ = c.expr(node.expr)
2020-03-19 15:18:29 +01:00
c.expected_type = table.void_type
2020-06-18 20:38:59 +02:00
c.check_expr_opt_call(node.expr, table.void_type)
// TODO This should work, even if it's prolly useless .-.
// node.typ = c.check_expr_opt_call(node.expr, table.void_type)
}
ast.FnDecl {
c.fn_decl(mut node)
}
ast.ForCStmt {
c.in_for_count++
prev_loop_label := c.loop_label
2020-06-18 20:38:59 +02:00
c.stmt(node.init)
c.expr(node.cond)
c.stmt(node.inc)
c.check_loop_label(node.label, node.pos)
2020-06-18 20:38:59 +02:00
c.stmts(node.stmts)
c.loop_label = prev_loop_label
c.in_for_count--
}
ast.ForInStmt {
c.in_for_count++
prev_loop_label := c.loop_label
2020-06-18 20:38:59 +02:00
typ := c.expr(node.cond)
2020-04-25 09:08:53 +02:00
typ_idx := typ.idx()
if node.key_var.len > 0 && node.key_var != '_' {
c.check_valid_snake_case(node.key_var, 'variable name', node.pos)
}
if node.val_var.len > 0 && node.val_var != '_' {
c.check_valid_snake_case(node.val_var, 'variable name', node.pos)
}
2020-06-18 20:38:59 +02:00
if node.is_range {
high_type := c.expr(node.high)
high_type_idx := high_type.idx()
2020-04-25 06:14:17 +02:00
if typ_idx in table.integer_type_idxs && high_type_idx !in table.integer_type_idxs {
2020-06-18 20:38:59 +02:00
c.error('range types do not match', node.cond.position())
2020-04-25 06:14:17 +02:00
} else if typ_idx in table.float_type_idxs || high_type_idx in table.float_type_idxs {
2020-06-18 20:38:59 +02:00
c.error('range type can not be float', node.cond.position())
2020-04-25 06:14:17 +02:00
} else if typ_idx == table.bool_type_idx || high_type_idx == table.bool_type_idx {
2020-06-18 20:38:59 +02:00
c.error('range type can not be bool', node.cond.position())
2020-04-25 06:14:17 +02:00
} else if typ_idx == table.string_type_idx || high_type_idx == table.string_type_idx {
2020-06-18 20:38:59 +02:00
c.error('range type can not be string', node.cond.position())
}
} else {
2020-03-21 10:22:16 +01:00
sym := c.table.get_type_symbol(typ)
if sym.kind == .struct_ {
// iterators
next_fn := sym.find_method('next') or {
c.error('a struct must have a `next()` method to be an iterator',
node.cond.position())
return
}
if !next_fn.return_type.has_flag(.optional) {
c.error('iterator method `next()` must return an optional', node.cond.position())
}
// the receiver
if next_fn.params.len != 1 {
c.error('iterator method `next()` must have 0 parameters', node.cond.position())
}
val_type := next_fn.return_type.clear_flag(.optional)
node.cond_type = typ
node.kind = sym.kind
node.val_type = val_type
node.scope.update_var_type(node.val_var, val_type)
} else {
if sym.kind == .map && !(node.key_var.len > 0 && node.val_var.len > 0) {
c.error(
'declare a key and a value variable when ranging a map: `for key, val in map {`\n' +
'use `_` if you do not need the variable', node.pos)
}
if node.key_var.len > 0 {
key_type := match sym.kind {
.map { sym.map_info().key_type }
else { table.int_type }
}
node.key_type = key_type
node.scope.update_var_type(node.key_var, key_type)
}
mut value_type := c.table.value_type(typ)
if value_type == table.void_type || typ.has_flag(.optional) {
if typ != table.void_type {
c.error('for in: cannot index `${c.table.type_to_str(typ)}`',
node.cond.position())
}
}
if node.val_is_mut {
value_type = value_type.to_ptr()
match node.cond {
ast.Ident {
if node.cond.obj is ast.Var {
obj := node.cond.obj as ast.Var
if !obj.is_mut {
c.error('`$obj.name` is immutable, it cannot be changed',
node.cond.pos)
}
}
}
ast.ArrayInit {
c.error('array literal is immutable, it cannot be changed',
node.cond.pos)
}
ast.MapInit {
c.error('map literal is immutable, it cannot be changed',
node.cond.pos)
}
else {}
}
}
node.cond_type = typ
node.kind = sym.kind
node.val_type = value_type
node.scope.update_var_type(node.val_var, value_type)
}
}
c.check_loop_label(node.label, node.pos)
if node.val_is_mut {
c.for_in_mut_val_name = node.val_var
}
c.stmts(node.stmts)
if node.val_is_mut {
c.for_in_mut_val_name = ''
}
c.loop_label = prev_loop_label
c.in_for_count--
}
ast.ForStmt {
c.for_stmt(mut node)
}
2020-05-16 16:12:23 +02:00
ast.GlobalDecl {
for field in node.fields {
c.check_valid_snake_case(field.name, 'global name', field.pos)
if field.name in c.global_names {
c.error('duplicate global `$field.name`', field.pos)
}
c.global_names << field.name
}
2020-05-16 16:12:23 +02:00
}
ast.GoStmt {
c.call_expr(mut node.call_expr)
// Make sure there are no mutable arguments
for arg in node.call_expr.args {
if arg.is_mut && !arg.typ.is_ptr() {
c.error('function in `go` statement cannot contain mutable non-reference arguments',
arg.expr.position())
}
}
if node.call_expr.is_method && node.call_expr.receiver_type.is_ptr()
&& !node.call_expr.left_type.is_ptr() {
c.error('method in `go` statement cannot have non-reference mutable receiver',
node.call_expr.left.position())
}
2020-04-03 15:18:17 +02:00
}
ast.GotoLabel {}
ast.GotoStmt {
if node.name !in c.cur_fn.label_names {
c.error('unknown label `$node.name`', node.pos)
}
// TODO: check label doesn't bypass variable declarations
}
ast.HashStmt {
c.hash_stmt(mut node)
}
ast.Import {
c.import_stmt(node)
}
ast.InterfaceDecl {
c.interface_decl(node)
}
ast.Module {
c.mod = node.name
2020-09-05 12:00:35 +02:00
c.is_builtin_mod = node.name in ['builtin', 'os', 'strconv']
c.check_valid_snake_case(node.name, 'module name', node.pos)
}
ast.Return {
// c.returns = true
c.return_stmt(mut node)
2020-04-29 12:31:18 +02:00
c.scope_returns = true
}
2020-06-24 14:32:14 +02:00
ast.SqlStmt {
c.sql_stmt(mut node)
}
ast.StructDecl {
c.struct_decl(mut node)
}
ast.TypeDecl {
c.type_decl(node)
}
}
}
fn (mut c Checker) hash_stmt(mut node ast.HashStmt) {
if c.skip_flags {
return
}
if c.pref.backend == .js {
if !c.file.path.ends_with('.js.v') {
c.error('Hash statements are only allowed in backend specific files such "x.js.v"',
node.pos)
}
if c.mod == 'main' {
c.error('Hash statements are not allowed in the main module. Please place them in a separate module.',
node.pos)
}
return
}
if node.kind == 'include' {
mut flag := node.main
if flag.contains('@VROOT') {
vroot := util.resolve_vroot(flag, c.file.path) or {
c.error(err, node.pos)
return
}
node.val = 'include $vroot'
2020-10-18 21:31:15 +02:00
node.main = vroot
flag = vroot
}
if flag.contains('\$env(') {
env := util.resolve_env_value(flag, true) or {
c.error(err, node.pos)
return
}
node.main = env
}
flag_no_comment := flag.all_before('//').trim_space()
if !((flag_no_comment.starts_with('"') && flag_no_comment.ends_with('"'))
|| (flag_no_comment.starts_with('<') && flag_no_comment.ends_with('>'))) {
c.error('including C files should use either `"header_file.h"` or `<header_file.h>` quoting',
node.pos)
}
} else if node.kind == 'pkgconfig' {
args := if node.main.contains('--') {
node.main.split(' ')
} else {
'--cflags --libs $node.main'.split(' ')
}
mut m := pkgconfig.main(args) or {
c.error(err, node.pos)
return
}
cflags := m.run() or {
c.error(err, node.pos)
return
}
c.table.parse_cflag(cflags, c.mod, c.pref.compile_defines_all) or {
c.error(err, node.pos)
return
}
} else if node.kind == 'flag' {
// #flag linux -lm
mut flag := node.main
// expand `@VROOT` to its absolute path
if flag.contains('@VROOT') {
flag = util.resolve_vroot(flag, c.file.path) or {
c.error(err, node.pos)
return
}
}
if flag.contains('\$env(') {
flag = util.resolve_env_value(flag, true) or {
c.error(err, node.pos)
return
}
}
for deprecated in ['@VMOD', '@VMODULE', '@VPATH', '@VLIB_PATH'] {
if flag.contains(deprecated) {
c.error('$deprecated had been deprecated, use @VROOT instead.', node.pos)
}
}
// println('adding flag "$flag"')
c.table.parse_cflag(flag, c.mod, c.pref.compile_defines_all) or { c.error(err, node.pos) }
} else {
if node.kind != 'define' {
2020-11-20 23:49:52 +01:00
c.error('expected `#define`, `#flag`, `#include` or `#pkgconfig` not $node.val',
node.pos)
}
}
}
fn (mut c Checker) import_stmt(imp ast.Import) {
for sym in imp.syms {
name := '${imp.mod}.$sym.name'
if sym.name[0].is_capital() {
if type_sym := c.table.find_type(name) {
if type_sym.kind != .placeholder {
if !type_sym.is_public {
c.error('module `$imp.mod` type `$sym.name` is private', sym.pos)
}
continue
}
}
c.error('module `$imp.mod` has no type `$sym.name`', sym.pos)
continue
}
if func := c.table.find_fn(name) {
if !func.is_pub {
c.error('module `$imp.mod` function `${sym.name}()` is private', sym.pos)
}
continue
}
if _ := c.file.global_scope.find_const(name) {
continue
}
c.error('module `$imp.mod` has no constant or function `$sym.name`', sym.pos)
}
}
2020-04-25 17:49:16 +02:00
fn (mut c Checker) stmts(stmts []ast.Stmt) {
2020-04-30 18:06:14 +02:00
mut unreachable := token.Position{
line_nr: -1
}
2020-03-18 12:18:48 +01:00
c.expected_type = table.void_type
2020-02-29 18:34:25 +01:00
for stmt in stmts {
2020-04-29 12:31:18 +02:00
if c.scope_returns {
if unreachable.line_nr == -1 {
unreachable = stmt.position()
}
}
2020-02-29 18:34:25 +01:00
c.stmt(stmt)
}
2020-04-29 12:31:18 +02:00
if unreachable.line_nr >= 0 {
c.error('unreachable code', unreachable)
2020-04-29 12:31:18 +02:00
}
c.scope_returns = false
2020-03-18 12:34:27 +01:00
c.expected_type = table.void_type
2020-02-29 18:34:25 +01:00
}
pub fn (c &Checker) unwrap_generic(typ table.Type) table.Type {
if typ.has_flag(.generic) {
sym := c.table.get_type_symbol(typ)
for i, generic_param in c.cur_fn.generic_params {
if generic_param.name == sym.name {
return c.cur_generic_types[i].derive(typ).clear_flag(.generic)
}
}
}
return typ
}
// TODO node must be mut
2020-04-25 17:49:16 +02:00
pub fn (mut c Checker) expr(node ast.Expr) table.Type {
c.expr_level++
2020-06-10 12:17:49 +02:00
defer {
c.expr_level--
}
if c.expr_level > 200 {
c.error('checker: too many expr levels: $c.expr_level ', node.position())
return table.void_type
}
match mut node {
ast.CTempVar {
return node.typ
}
2020-05-15 23:14:53 +02:00
ast.AnonFn {
c.inside_anon_fn = true
2020-05-23 08:51:15 +02:00
keep_fn := c.cur_fn
2020-06-18 20:38:59 +02:00
c.cur_fn = &node.decl
c.stmts(node.decl.stmts)
c.fn_decl(mut node.decl)
2020-05-23 08:51:15 +02:00
c.cur_fn = keep_fn
c.inside_anon_fn = false
return node.typ
2020-05-15 23:14:53 +02:00
}
ast.ArrayDecompose {
typ := c.expr(node.expr)
type_sym := c.table.get_type_symbol(typ)
if type_sym.kind != .array {
c.error('decomposition can only be used on arrays', node.expr.position())
return table.void_type
}
array_info := type_sym.info as table.Array
elem_type := array_info.elem_type.set_flag(.variadic)
node.expr_type = typ
node.arg_type = elem_type
return elem_type
}
ast.ArrayInit {
2020-06-18 20:38:59 +02:00
return c.array_init(mut node)
}
ast.AsCast {
2020-06-18 20:38:59 +02:00
node.expr_type = c.expr(node.expr)
expr_type_sym := c.table.get_type_symbol(node.expr_type)
type_sym := c.table.get_type_symbol(node.typ)
if expr_type_sym.kind == .sum_type {
if type_sym.kind == .placeholder {
// Unknown type used in the right part of `as`
c.error('unknown type `$type_sym.name`', node.pos)
}
if !c.table.sumtype_has_variant(node.expr_type, node.typ) {
c.error('cannot cast `$expr_type_sym.name` to `$type_sym.name`', node.pos)
// c.error('only $info.variants can be casted to `$typ`', node.pos)
}
} else {
mut s := 'cannot cast non-sum type `$expr_type_sym.name` using `as`'
if type_sym.kind == .sum_type {
s += ' - use e.g. `${type_sym.name}(some_expr)` instead.'
}
c.error(s, node.pos)
}
if expr_type_sym.kind == .sum_type {
return node.typ
}
2020-06-18 20:38:59 +02:00
return node.typ.to_ptr()
}
2020-02-29 18:07:29 +01:00
ast.Assoc {
v := node.scope.find_var(node.var_name) or { panic(err) }
2020-06-18 20:38:59 +02:00
for i, _ in node.fields {
c.expr(node.exprs[i])
2020-03-19 08:57:33 +01:00
}
2020-06-18 20:38:59 +02:00
node.typ = v.typ
2020-04-15 01:45:27 +02:00
return v.typ
2020-02-29 18:07:29 +01:00
}
ast.BoolLiteral {
return table.bool_type
}
ast.CastExpr {
return c.cast_expr(mut node)
}
ast.CallExpr {
2020-06-18 20:38:59 +02:00
return c.call_expr(mut node)
}
ast.GoExpr {
ret_type := c.call_expr(mut node.go_stmt.call_expr)
return c.table.find_or_register_gohandle(ret_type)
}
ast.ChanInit {
return c.chan_init(mut node)
}
ast.CharLiteral {
// return int_literal, not rune, so that we can do "bytes << `A`" without a cast etc
// return table.int_literal_type
2020-08-27 06:46:18 +02:00
return table.rune_type
// return table.byte_type
}
ast.Comment {
return table.void_type
}
ast.AtExpr {
return c.at_expr(mut node)
}
ast.ComptimeCall {
return c.comptime_call(mut node)
}
ast.ComptimeSelector {
node.left_type = c.unwrap_generic(c.expr(node.left))
expr_type := c.unwrap_generic(c.expr(node.field_expr))
expr_sym := c.table.get_type_symbol(expr_type)
if expr_type != table.string_type {
c.error('expected `string` instead of `$expr_sym.name` (e.g. `field.name`)',
node.field_expr.position())
}
if node.field_expr is ast.SelectorExpr {
left_pos := node.field_expr.expr.position()
if c.comptime_fields_type.len == 0 {
c.error('compile time field access can only be used when iterating over `T.fields`',
left_pos)
}
expr_name := node.field_expr.expr.str()
if expr_name in c.comptime_fields_type {
return c.comptime_fields_type[expr_name]
}
c.error('unknown `\$for` variable `$expr_name`', left_pos)
} else {
c.error('expected selector expression e.g. `$(field.name)`', node.field_expr.position())
}
return table.void_type
}
2020-05-15 23:14:53 +02:00
ast.ConcatExpr {
2020-06-18 20:38:59 +02:00
return c.concat_expr(mut node)
2020-05-15 23:14:53 +02:00
}
2020-02-25 15:02:34 +01:00
ast.EnumVal {
2020-06-18 20:38:59 +02:00
return c.enum_val(mut node)
2020-02-25 15:02:34 +01:00
}
2020-02-17 14:15:42 +01:00
ast.FloatLiteral {
return table.float_literal_type
}
ast.Ident {
// c.checked_ident = node.name
2020-06-18 20:38:59 +02:00
res := c.ident(mut node)
2020-04-08 19:15:16 +02:00
// c.checked_ident = ''
return res
2020-02-18 18:13:34 +01:00
}
ast.IfExpr {
2020-06-18 20:38:59 +02:00
return c.if_expr(mut node)
}
ast.IfGuardExpr {
2020-06-18 20:38:59 +02:00
node.expr_type = c.expr(node.expr)
if !node.expr_type.has_flag(.optional) {
c.error('expression should return an option', node.expr.position())
}
return table.bool_type
2020-02-18 18:13:34 +01:00
}
ast.IndexExpr {
2020-06-18 20:38:59 +02:00
return c.index_expr(mut node)
}
ast.InfixExpr {
2020-06-18 20:38:59 +02:00
return c.infix_expr(mut node)
}
ast.IntegerLiteral {
return table.int_literal_type
}
ast.LockExpr {
return c.lock_expr(mut node)
}
ast.MapInit {
2020-06-18 20:38:59 +02:00
return c.map_init(mut node)
}
ast.MatchExpr {
2020-06-18 20:38:59 +02:00
return c.match_expr(mut node)
}
ast.PostfixExpr {
return c.postfix_expr(mut node)
}
ast.PrefixExpr {
return c.prefix_expr(mut node)
}
ast.None {
return table.none_type
}
2020-06-17 00:59:33 +02:00
ast.OrExpr {
// never happens
return table.void_type
}
// ast.OrExpr2 {
// return node.typ
// }
ast.ParExpr {
return c.expr(node.expr)
}
2020-06-17 00:59:33 +02:00
ast.RangeExpr {
// never happens
return table.void_type
}
ast.SelectExpr {
return c.select_expr(mut node)
}
ast.SelectorExpr {
2020-06-18 20:38:59 +02:00
return c.selector_expr(mut node)
}
ast.SizeOf {
if !node.is_type {
node.typ = c.expr(node.expr)
}
return table.u32_type
}
2021-01-30 12:57:09 +01:00
ast.OffsetOf {
return c.offset_of(node)
}
2020-06-17 00:59:33 +02:00
ast.SqlExpr {
return c.sql_expr(mut node)
2020-06-17 00:59:33 +02:00
}
ast.StringLiteral {
2020-06-18 20:38:59 +02:00
if node.language == .c {
2020-04-01 17:14:17 +02:00
return table.byteptr_type
}
return table.string_type
}
2020-03-21 07:01:06 +01:00
ast.StringInterLiteral {
2020-06-18 20:38:59 +02:00
return c.string_inter_lit(mut node)
2020-03-21 07:01:06 +01:00
}
ast.StructInit {
if node.unresolved {
return c.expr(ast.resolve_init(node, c.unwrap_generic(node.typ), c.table))
}
2020-06-18 20:38:59 +02:00
return c.struct_init(mut node)
2020-02-29 05:36:39 +01:00
}
ast.Type {
2020-06-18 20:38:59 +02:00
return node.typ
2020-03-02 10:53:38 +01:00
}
2020-03-19 12:15:39 +01:00
ast.TypeOf {
2020-06-18 20:38:59 +02:00
node.expr_type = c.expr(node.expr)
2020-03-19 12:15:39 +01:00
return table.string_type
}
2020-07-12 12:58:33 +02:00
ast.UnsafeExpr {
return c.unsafe_expr(mut node)
}
ast.Likely {
2020-06-18 20:38:59 +02:00
ltype := c.expr(node.expr)
if !c.check_types(ltype, table.bool_type) {
ltype_sym := c.table.get_type_symbol(ltype)
2020-06-18 20:38:59 +02:00
lname := if node.is_likely { '_likely_' } else { '_unlikely_' }
c.error('`${lname}()` expects a boolean expression, instead it got `$ltype_sym.name`',
2020-06-18 20:38:59 +02:00
node.pos)
}
return table.bool_type
}
}
return table.void_type
}
pub fn (mut c Checker) cast_expr(mut node ast.CastExpr) table.Type {
2020-12-16 17:50:39 +01:00
node.expr_type = c.expr(node.expr) // type to be casted
from_type_sym := c.table.get_type_symbol(node.expr_type)
2020-12-16 17:50:39 +01:00
to_type_sym := c.table.get_type_symbol(node.typ) // type to be used as cast
expr_is_ptr := node.expr_type.is_ptr() || node.expr_type.idx() in table.pointer_type_idxs
if expr_is_ptr && to_type_sym.kind == .string && !node.in_prexpr {
if node.has_arg {
c.warn('to convert a C string buffer pointer to a V string, please use x.vstring_with_len(len) instead of string(x,len)',
node.pos)
} else {
c.warn('to convert a C string buffer pointer to a V string, please use x.vstring() instead of string(x)',
node.pos)
}
}
if node.expr_type == table.byte_type && to_type_sym.kind == .string {
c.error('can not cast type `byte` to string, use `${node.expr.str()}.str()` instead.',
node.pos)
}
if to_type_sym.kind == .sum_type {
if node.expr_type in [table.int_literal_type, table.float_literal_type] {
node.expr_type = c.promote_num(node.expr_type, if node.expr_type == table.int_literal_type {
table.int_type
} else {
table.f64_type
})
}
if !c.table.sumtype_has_variant(node.typ, node.expr_type) {
c.error('cannot cast `$from_type_sym.name` to `$to_type_sym.name`', node.pos)
}
} else if mut to_type_sym.info is table.Alias {
if !c.check_types(node.expr_type, to_type_sym.info.parent_type) {
parent_type_sym := c.table.get_type_symbol(to_type_sym.info.parent_type)
c.error('cannot convert type `$from_type_sym.name` to `$to_type_sym.name` (alias to `$parent_type_sym.name`)',
node.pos)
}
} else if node.typ == table.string_type
&& (from_type_sym.kind in [.int_literal, .int, .byte, .byteptr, .bool]
|| (from_type_sym.kind == .array && from_type_sym.name == 'array_byte')) {
type_name := c.table.type_to_str(node.expr_type)
c.error('cannot cast type `$type_name` to string, use `x.str()` instead', node.pos)
} else if node.expr_type == table.string_type {
if to_type_sym.kind != .alias {
mut error_msg := 'cannot cast a string'
if mut node.expr is ast.StringLiteral {
if node.expr.val.len == 1 {
error_msg += ", for denoting characters use `$node.expr.val` instead of '$node.expr.val'"
}
}
c.error(error_msg, node.pos)
}
} else if to_type_sym.kind == .byte && node.expr_type != table.voidptr_type
&& from_type_sym.kind != .enum_ && !node.expr_type.is_int() && !node.expr_type.is_float()
&& !node.expr_type.is_ptr() {
type_name := c.table.type_to_str(node.expr_type)
c.error('cannot cast type `$type_name` to `byte`', node.pos)
} else if to_type_sym.kind == .struct_ && !node.typ.is_ptr()
&& !(to_type_sym.info as table.Struct).is_typedef {
// For now we ignore C typedef because of `C.Window(C.None)` in vlib/clipboard
if from_type_sym.kind == .struct_ && !node.expr_type.is_ptr() {
c.warn('casting to struct is deprecated, use e.g. `Struct{...expr}` instead',
node.pos)
from_type_info := from_type_sym.info as table.Struct
to_type_info := to_type_sym.info as table.Struct
if !c.check_struct_signature(from_type_info, to_type_info) {
c.error('cannot convert struct `$from_type_sym.name` to struct `$to_type_sym.name`',
node.pos)
}
} else {
type_name := c.table.type_to_str(node.expr_type)
c.error('cannot cast `$type_name` to struct', node.pos)
}
} else if to_type_sym.kind == .interface_ {
c.type_implements(node.expr_type, node.typ, node.pos)
} else if node.typ == table.bool_type {
c.error('cannot cast to bool - use e.g. `some_int != 0` instead', node.pos)
} else if node.expr_type == table.none_type {
type_name := c.table.type_to_str(node.typ)
c.error('cannot cast `none` to `$type_name`', node.pos)
} else if from_type_sym.kind == .struct_ && !node.expr_type.is_ptr() {
if (node.typ.is_ptr() || to_type_sym.kind !in [.sum_type, .interface_]) && !c.is_builtin_mod {
type_name := c.table.type_to_str(node.typ)
c.error('cannot cast struct to `$type_name`', node.pos)
}
2020-12-16 17:50:39 +01:00
} else if node.expr_type.has_flag(.optional) || node.expr_type.has_flag(.variadic) {
// variadic case can happen when arrays are converted into variadic
msg := if node.expr_type.has_flag(.optional) { 'an optional' } else { 'a variadic' }
c.error('cannot type cast $msg', node.pos)
2021-01-05 19:26:48 +01:00
} else if !c.inside_unsafe && node.typ.is_ptr() && node.expr_type.is_ptr() {
ft := c.table.type_to_str(node.expr_type)
tt := c.table.type_to_str(node.typ)
2021-01-05 19:26:48 +01:00
c.warn('casting `$ft` to `$tt` is only allowed in `unsafe` code', node.pos)
}
if node.has_arg {
c.expr(node.arg)
}
node.typname = c.table.get_type_symbol(node.typ).name
return node.typ
}
fn (mut c Checker) comptime_call(mut node ast.ComptimeCall) table.Type {
node.sym = c.table.get_type_symbol(c.unwrap_generic(c.expr(node.left)))
if node.is_env {
env_value := util.resolve_env_value("\$env('$node.args_var')", false) or {
c.error(err, node.env_pos)
return table.string_type
}
node.env_value = env_value
return table.string_type
}
if node.is_embed {
c.file.embedded_files << node.embed_file
return c.table.find_type_idx('v.embed_file.EmbedFileData')
}
if node.is_vweb {
// TODO assoc parser bug
2021-01-22 23:24:48 +01:00
pref_ := *c.pref
pref2 := pref.Preferences{
...pref_
is_vweb: true
}
2021-02-05 08:27:14 +01:00
mut c2 := new_checker(c.table, pref2)
c2.check(node.vweb_tmpl)
mut i := 0 // tmp counter var for skipping first three tmpl vars
for k, _ in c2.file.scope.children[0].objects {
if i < 4 {
// Skip first three because they are tmpl vars see vlib/vweb/tmpl/tmpl.v
i++
continue
}
if k in c.fn_scope.objects && c.fn_scope.objects[k] is ast.Var {
mut vsc := c.fn_scope.objects[k] as ast.Var
vsc.is_used = true
c.fn_scope.objects[k] = vsc
}
}
c.warnings << c2.warnings
c.errors << c2.errors
c.nr_warnings += c2.nr_warnings
c.nr_errors += c2.nr_errors
}
if node.method_name == 'html' {
rtyp := c.table.find_type_idx('vweb.Result')
node.result_type = rtyp
return rtyp
}
if node.method_name == 'method' {
if node.args_var.len > 0 {
v := node.scope.find_var(node.args_var) or {
c.error('unknown identifier `$node.args_var`', node.method_pos)
return table.void_type
}
s := c.table.type_to_str(c.expr(v.expr))
if s != '[]string' {
c.error('expected `[]string`, not s', node.method_pos)
}
}
// assume string for now
return table.string_type
}
if node.is_vweb {
return table.string_type
}
// s.$my_str()
v := node.scope.find_var(node.method_name) or {
c.error('unknown identifier `$node.method_name`', node.method_pos)
return table.void_type
}
if v.typ != table.string_type {
s := c.expected_msg(v.typ, table.string_type)
c.error('invalid string method call: $s', node.method_pos)
return table.void_type
}
// note: we should use a compile-time evaluation function rather than handle here
// mut variables will not work after init
mut method_name := ''
if v.expr is ast.StringLiteral {
method_name = v.expr.val
} else {
c.error('todo: not a string literal', node.method_pos)
}
f := node.sym.find_method(method_name) or {
c.error('could not find method `$method_name`', node.method_pos)
return table.void_type
}
// println(f.name + ' ' + c.table.type_to_str(f.return_type))
node.result_type = f.return_type
return f.return_type
}
fn (mut c Checker) at_expr(mut node ast.AtExpr) table.Type {
match node.kind {
.fn_name {
node.val = c.cur_fn.name.all_after_last('.')
}
.method_name {
fname := c.cur_fn.name.all_after_last('.')
if c.cur_fn.is_method {
node.val = c.table.type_to_str(c.cur_fn.receiver.typ).all_after_last('.') + '.' +
fname
} else {
node.val = fname
}
}
.mod_name {
node.val = c.cur_fn.mod
}
.struct_name {
if c.cur_fn.is_method {
node.val = c.table.type_to_str(c.cur_fn.receiver.typ).all_after_last('.')
} else {
node.val = ''
}
}
.vexe_path {
node.val = pref.vexe_path()
}
.file_path {
node.val = os.real_path(c.file.path)
}
.line_nr {
node.val = (node.pos.line_nr + 1).str()
}
.column_nr {
_, column := util.filepath_pos_to_source_and_column(c.file.path, node.pos)
node.val = (column + 1).str()
}
.vhash {
node.val = util.vhash()
}
.vmod_file {
if c.vmod_file_content.len == 0 {
mut mcache := vmod.get_cache()
vmod_file_location := mcache.get_by_file(c.file.path)
if vmod_file_location.vmod_file.len == 0 {
c.error('@VMOD_FILE can be used only in projects, that have v.mod file',
node.pos)
}
vmod_content := os.read_file(vmod_file_location.vmod_file) or { '' }
$if windows {
c.vmod_file_content = vmod_content.replace('\r\n', '\n')
} $else {
c.vmod_file_content = vmod_content
}
}
node.val = c.vmod_file_content
}
.unknown {
c.error('unknown @ identifier: ${node.name}. Available identifiers: $token.valid_at_tokens',
node.pos)
}
}
return table.string_type
}
2020-05-11 14:38:25 +02:00
pub fn (mut c Checker) ident(mut ident ast.Ident) table.Type {
// TODO: move this
if c.const_deps.len > 0 {
mut name := ident.name
if !name.contains('.') && ident.mod != 'builtin' {
name = '${ident.mod}.$ident.name'
}
2020-04-04 08:05:26 +02:00
if name == c.const_decl {
c.error('cycle in constant `$c.const_decl`', ident.pos)
return table.void_type
}
c.const_deps << name
}
if ident.kind == .blank_ident {
if ident.tok_kind !in [.assign, .decl_assign] {
c.error('undefined ident: `_` (may only be used in assignments)', ident.pos)
}
return table.void_type
}
// second use
if ident.kind in [.constant, .global, .variable] {
info := ident.info as ast.IdentVar
2020-05-28 05:50:57 +02:00
// Got a var with type T, return current generic type
return info.typ
} else if ident.kind == .function {
info := ident.info as ast.IdentFn
return info.typ
} else if ident.kind == .unresolved {
// first use
2020-08-28 19:07:32 +02:00
if ident.tok_kind == .assign && ident.is_mut {
c.error('`mut` not allowed with `=` (use `:=` to declare a variable)', ident.pos)
}
if obj := ident.scope.find(ident.name) {
match mut obj {
ast.GlobalField {
2020-06-09 09:08:11 +02:00
ident.kind = .global
ident.info = ast.IdentVar{
2020-06-18 20:38:59 +02:00
typ: obj.typ
2020-06-09 09:08:11 +02:00
}
ident.obj = obj
2020-06-18 20:38:59 +02:00
return obj.typ
2020-06-09 09:08:11 +02:00
}
ast.Var {
// incase var was not marked as used yet (vweb tmpl)
obj.is_used = true
if ident.pos.pos < obj.pos.pos {
2020-06-24 14:44:06 +02:00
c.error('undefined variable `$ident.name` (used before declaration)',
ident.pos)
}
is_sum_type_cast := obj.sum_type_casts.len != 0
&& !c.prevent_sum_type_unwrapping_once
c.prevent_sum_type_unwrapping_once = false
2020-11-23 16:16:13 +01:00
mut typ := if is_sum_type_cast { obj.sum_type_casts.last() } else { obj.typ }
if typ == 0 {
if mut obj.expr is ast.Ident {
if obj.expr.kind == .unresolved {
c.error('unresolved variable: `$ident.name`', ident.pos)
return table.void_type
}
}
if mut obj.expr is ast.IfGuardExpr {
// new variable from if guard shouldn't have the optional flag for further use
// a temp variable will be generated which unwraps it
if_guard_var_type := c.expr(obj.expr.expr)
typ = if_guard_var_type.clear_flag(.optional)
} else {
typ = c.expr(obj.expr)
}
}
2020-06-04 14:38:54 +02:00
is_optional := typ.has_flag(.optional)
ident.kind = .variable
ident.info = ast.IdentVar{
typ: typ
is_optional: is_optional
}
// if typ == table.t_type {
// sym := c.table.get_type_symbol(c.cur_generic_type)
// println('IDENT T unresolved $ident.name typ=$sym.name')
// Got a var with type T, return current generic type
// typ = c.cur_generic_type
// }
2020-05-24 04:43:00 +02:00
// } else {
if !is_sum_type_cast {
obj.typ = typ
}
ident.obj = obj
// unwrap optional (`println(x)`)
if is_optional {
2020-06-04 14:38:54 +02:00
return typ.clear_flag(.optional)
}
return typ
}
else {}
}
}
mut name := ident.name
// check for imported symbol
if name in c.file.imported_symbols {
name = c.file.imported_symbols[name]
}
// prepend mod to look for fn call or const
else if !name.contains('.') && ident.mod != 'builtin' {
name = '${ident.mod}.$ident.name'
}
if obj := c.file.global_scope.find(name) {
match mut obj {
ast.ConstField {
if !(obj.is_pub || obj.mod == c.mod || c.pref.is_test) {
c.error('constant `$obj.name` is private', ident.pos)
}
2020-06-18 20:38:59 +02:00
mut typ := obj.typ
if typ == 0 {
c.inside_const = true
2020-06-18 20:38:59 +02:00
typ = c.expr(obj.expr)
c.inside_const = false
if obj.expr is ast.CallExpr {
if obj.expr.or_block.kind != .absent {
typ = typ.clear_flag(.optional)
}
}
}
ident.name = name
ident.kind = .constant
ident.info = ast.IdentVar{
typ: typ
}
2020-06-18 20:38:59 +02:00
obj.typ = typ
ident.obj = obj
return typ
}
else {}
}
}
// Non-anon-function object (not a call), e.g. `onclick(my_click)`
if func := c.table.find_fn(name) {
2020-06-24 14:44:06 +02:00
fn_type := table.new_type(c.table.find_or_register_fn_type(ident.mod, func,
false, true))
ident.name = name
ident.kind = .function
ident.info = ast.IdentFn{
typ: fn_type
}
return fn_type
}
}
if ident.language == .c {
if ident.name == 'C.NULL' {
return table.voidptr_type
}
2020-02-18 18:13:34 +01:00
return table.int_type
}
if c.inside_sql {
if field := c.table.find_field(c.cur_orm_ts, ident.name) {
return field.typ
}
}
if ident.kind == .unresolved && ident.mod != 'builtin' {
// search in the `builtin` idents, for example
// main.compare_f32 may actually be builtin.compare_f32
saved_mod := ident.mod
ident.mod = 'builtin'
builtin_type := c.ident(mut ident)
if builtin_type != table.void_type {
return builtin_type
}
ident.mod = saved_mod
}
if ident.tok_kind == .assign {
c.error('undefined ident: `$ident.name` (use `:=` to declare a variable)', ident.pos)
} else {
c.error('undefined ident: `$ident.name`', ident.pos)
2020-05-01 12:22:39 +02:00
}
2020-05-01 12:02:31 +02:00
if c.table.known_type(ident.name) {
// e.g. `User` in `json.decode(User, '...')`
return table.void_type
}
return table.void_type
}
2020-05-16 16:12:23 +02:00
pub fn (mut c Checker) concat_expr(mut concat_expr ast.ConcatExpr) table.Type {
2020-05-15 23:14:53 +02:00
mut mr_types := []table.Type{}
for expr in concat_expr.vals {
mr_types << c.expr(expr)
}
if concat_expr.vals.len == 1 {
typ := mr_types[0]
concat_expr.return_type = typ
return typ
} else {
typ := c.table.find_or_register_multi_return(mr_types)
table.new_type(typ)
concat_expr.return_type = typ
return typ
}
}
2020-05-11 14:38:25 +02:00
pub fn (mut c Checker) match_expr(mut node ast.MatchExpr) table.Type {
2020-03-18 12:18:48 +01:00
node.is_expr = c.expected_type != table.void_type
node.expected_type = c.expected_type
2020-03-18 12:23:32 +01:00
cond_type := c.expr(node.cond)
2020-08-17 20:19:21 +02:00
// we setting this here rather than at the end of the method
// since it is used in c.match_exprs() it saves checking twice
node.cond_type = cond_type
2020-03-18 12:23:32 +01:00
if cond_type == 0 {
c.error('compiler bug: match 0 cond type', node.pos)
}
2020-05-31 10:22:18 +02:00
cond_type_sym := c.table.get_type_symbol(cond_type)
if cond_type_sym.kind !in [.interface_, .sum_type] {
node.is_sum_type = false
}
2020-05-31 10:22:18 +02:00
c.match_exprs(mut node, cond_type_sym)
2020-03-18 12:23:32 +01:00
c.expected_type = cond_type
2020-03-03 17:29:16 +01:00
mut ret_type := table.void_type
mut nbranches_with_return := 0
mut nbranches_without_return := 0
2020-03-04 11:59:45 +01:00
for branch in node.branches {
c.stmts(branch.stmts)
if node.is_expr && branch.stmts.len > 0 {
// ignore last statement - workaround
// currently the last statement in a match branch does not have an
// expected value set, so e.g. IfExpr.is_expr is not set.
// probably any mismatch will be caught by not producing a value instead
for st in branch.stmts[0..branch.stmts.len - 1] {
// must not contain C statements
st.check_c_expr() or {
c.error('`match` expression branch has $err', st.position())
}
}
}
// If the last statement is an expression, return its type
2020-03-04 11:59:45 +01:00
if branch.stmts.len > 0 {
mut stmt := branch.stmts[branch.stmts.len - 1]
match mut stmt {
2020-03-03 17:29:16 +01:00
ast.ExprStmt {
expr_type := c.expr(stmt.expr)
if ret_type == table.void_type {
ret_type = expr_type
stmt.typ = ret_type
} else if node.is_expr && ret_type != expr_type {
if !c.check_types(ret_type, expr_type) {
ret_sym := c.table.get_type_symbol(ret_type)
c.error('return type mismatch, it should be `$ret_sym.name`',
stmt.expr.position())
}
}
2020-03-03 17:29:16 +01:00
}
else {
// TODO: ask alex about this
// typ := c.expr(stmt.expr)
// type_sym := c.table.get_type_symbol(typ)
// p.warn('match expr ret $type_sym.name')
// node.typ = typ
// return typ
}
}
}
if has_return := c.has_return(branch.stmts) {
if has_return {
nbranches_with_return++
} else {
nbranches_without_return++
}
}
}
if nbranches_with_return > 0 {
if nbranches_with_return == node.branches.len {
// an exhaustive match, and all branches returned
c.returns = true
}
if nbranches_without_return > 0 {
// some of the branches did not return
c.returns = false
}
}
// if ret_type != table.void_type {
// node.is_expr = c.expected_type != table.void_type
// node.expected_type = c.expected_type
// }
node.return_type = ret_type
2020-03-03 17:29:16 +01:00
return ret_type
}
fn (mut c Checker) match_exprs(mut node ast.MatchExpr, cond_type_sym table.TypeSymbol) {
// branch_exprs is a histogram of how many times
// an expr was used in the match
2020-05-04 21:03:18 +02:00
mut branch_exprs := map[string]int{}
for branch_i, _ in node.branches {
mut branch := node.branches[branch_i]
mut expr_types := []ast.Type{}
for expr in branch.exprs {
mut key := ''
if expr is ast.RangeExpr {
mut low := 0
mut high := 0
c.expected_type = node.expected_type
2020-07-16 19:40:14 +02:00
low_expr := expr.low
high_expr := expr.high
if low_expr is ast.IntegerLiteral {
if high_expr is ast.IntegerLiteral {
low = low_expr.val.int()
high = high_expr.val.int()
} else {
c.error('mismatched range types', low_expr.pos)
}
2020-07-16 19:40:14 +02:00
} else if low_expr is ast.CharLiteral {
if high_expr is ast.CharLiteral {
low = low_expr.val[0]
high = high_expr.val[0]
} else {
c.error('mismatched range types', low_expr.pos)
}
} else {
typ := c.table.type_to_str(c.expr(expr.low))
c.error('cannot use type `$typ` in match range', branch.pos)
}
for i in low .. high + 1 {
key = i.str()
val := if key in branch_exprs { branch_exprs[key] } else { 0 }
if val == 1 {
c.error('match case `$key` is handled more than once', branch.pos)
}
branch_exprs[key] = val + 1
}
continue
}
match expr {
ast.Type {
key = c.table.type_to_str(expr.typ)
expr_types << expr
}
ast.EnumVal {
key = expr.val
}
else {
key = expr.str()
}
}
2020-06-25 17:12:32 +02:00
val := if key in branch_exprs { branch_exprs[key] } else { 0 }
if val == 1 {
c.error('match case `$key` is handled more than once', branch.pos)
}
2020-08-17 20:19:21 +02:00
c.expected_type = node.cond_type
expr_type := c.expr(expr)
2020-12-12 13:52:22 +01:00
if expr_type.idx() == 0 {
// parser failed, stop checking
return
}
if cond_type_sym.kind == .interface_ {
// TODO
// This generates a memory issue with TCC
// Needs to be checked later when TCC errors are fixed
// Current solution is to move expr.position() to its own statement
// c.type_implements(expr_type, c.expected_type, expr.position())
expr_pos := expr.position()
c.type_implements(expr_type, c.expected_type, expr_pos)
} else if mut cond_type_sym.info is table.SumType {
if expr_type !in cond_type_sym.info.variants {
expr_str := c.table.type_to_str(expr_type)
expect_str := c.table.type_to_str(node.cond_type)
c.error('`$expect_str` has no variant `$expr_str`', expr.position())
}
} else if !c.check_types(expr_type, node.cond_type) {
2020-08-17 20:19:21 +02:00
expr_str := c.table.type_to_str(expr_type)
expect_str := c.table.type_to_str(node.cond_type)
c.error('cannot match `$expr_str` with `$expect_str` condition', expr.position())
2020-08-17 20:19:21 +02:00
}
branch_exprs[key] = val + 1
}
// when match is sum type matching, then register smart cast for every branch
if expr_types.len > 0 {
if cond_type_sym.kind == .sum_type {
mut expr_type := table.Type(0)
if expr_types.len > 1 {
mut agg_name := strings.new_builder(20)
mut agg_cname := strings.new_builder(20)
agg_name.write('(')
for i, expr in expr_types {
if i > 0 {
agg_name.write(' | ')
agg_cname.write('___')
}
type_str := c.table.type_to_str(expr.typ)
name := if c.is_builtin_mod { type_str } else { '${c.mod}.$type_str' }
agg_name.write(name)
agg_cname.write(util.no_dots(name))
}
agg_name.write(')')
name := agg_name.str()
2020-12-11 07:39:51 +01:00
existing_idx := c.table.type_idxs[name]
if existing_idx > 0 {
expr_type = existing_idx
} else {
expr_type = c.table.register_type_symbol(table.TypeSymbol{
name: name
cname: agg_cname.str()
kind: .aggregate
mod: c.mod
info: table.Aggregate{
types: expr_types.map(it.typ)
}
})
}
} else {
expr_type = expr_types[0].typ
}
c.smartcast_sumtype(node.cond, node.cond_type, expr_type, mut branch.scope)
}
}
}
// check that expressions are exhaustive
// this is achieved either by putting an else
// or, when the match is on a sum type or an enum
// by listing all variants or values
2020-04-25 21:51:44 +02:00
mut is_exhaustive := true
mut unhandled := []string{}
if node.cond_type == table.bool_type {
variants := ['true', 'false']
for v in variants {
if v !in branch_exprs {
is_exhaustive = false
unhandled << '`$v`'
}
}
} else {
match mut cond_type_sym.info {
table.SumType {
for v in cond_type_sym.info.variants {
v_str := c.table.type_to_str(v)
if v_str !in branch_exprs {
is_exhaustive = false
unhandled << '`$v_str`'
}
}
}
//
table.Enum {
for v in cond_type_sym.info.vals {
if v !in branch_exprs {
is_exhaustive = false
unhandled << '`.$v`'
}
}
}
else {
is_exhaustive = false
}
}
2020-04-25 21:51:44 +02:00
}
mut else_branch := node.branches[node.branches.len - 1]
mut has_else := else_branch.is_else
if !has_else {
for i, branch in node.branches {
if branch.is_else && i != node.branches.len - 1 {
c.error('`else` must be the last branch of `match`', branch.pos)
else_branch = branch
has_else = true
}
}
}
2020-04-25 21:51:44 +02:00
if is_exhaustive {
if has_else {
c.error('match expression is exhaustive, `else` is unnecessary', else_branch.pos)
}
return
}
if has_else {
return
}
mut err_details := 'match must be exhaustive'
if unhandled.len > 0 {
err_details += ' (add match branches for: '
2021-01-25 10:26:20 +01:00
if unhandled.len < checker.match_exhaustive_cutoff_limit {
err_details += unhandled.join(', ')
} else {
2021-01-25 10:26:20 +01:00
remaining := unhandled.len - checker.match_exhaustive_cutoff_limit
err_details += unhandled[0..checker.match_exhaustive_cutoff_limit].join(', ')
err_details += ', and $remaining others ...'
}
err_details += ' or `else {}` at the end)'
2020-04-25 21:51:44 +02:00
} else {
err_details += ' (add `else {}` at the end)'
}
c.error(err_details, node.pos)
}
// smartcast takes the expression with the current type which should be smartcasted to the target type in the given scope
fn (c Checker) smartcast_sumtype(expr ast.Expr, cur_type table.Type, to_type table.Type, mut scope ast.Scope) {
match mut expr {
ast.SelectorExpr {
mut is_mut := false
mut sum_type_casts := []table.Type{}
expr_sym := c.table.get_type_symbol(expr.expr_type)
mut orig_type := 0
if field := c.table.find_field(expr_sym, expr.field_name) {
if field.is_mut {
root_ident := expr.root_ident()
if v := scope.find_var(root_ident.name) {
is_mut = v.is_mut
}
}
if orig_type == 0 {
orig_type = field.typ
}
}
if field := scope.find_struct_field(expr.expr_type, expr.field_name) {
sum_type_casts << field.sum_type_casts
}
// smartcast either if the value is immutable or if the mut argument is explicitly given
if !is_mut || expr.is_mut {
sum_type_casts << to_type
scope.register_struct_field(ast.ScopeStructField{
struct_type: expr.expr_type
name: expr.field_name
typ: cur_type
sum_type_casts: sum_type_casts
pos: expr.pos
orig_type: orig_type
})
}
}
ast.Ident {
mut is_mut := false
mut sum_type_casts := []table.Type{}
mut is_already_casted := false
mut orig_type := 0
if mut expr.obj is ast.Var {
is_mut = expr.obj.is_mut
sum_type_casts << expr.obj.sum_type_casts
is_already_casted = expr.obj.pos.pos == expr.pos.pos
if orig_type == 0 {
orig_type = expr.obj.typ
}
}
// smartcast either if the value is immutable or if the mut argument is explicitly given
if (!is_mut || expr.is_mut) && !is_already_casted {
sum_type_casts << to_type
scope.register(ast.Var{
name: expr.name
typ: cur_type
pos: expr.pos
is_used: true
is_mut: expr.is_mut
sum_type_casts: sum_type_casts
orig_type: orig_type
})
}
}
else {}
}
}
pub fn (mut c Checker) select_expr(mut node ast.SelectExpr) table.Type {
node.is_expr = c.expected_type != table.void_type
node.expected_type = c.expected_type
for branch in node.branches {
c.stmt(branch.stmt)
match branch.stmt {
ast.ExprStmt {
if branch.is_timeout {
if !branch.stmt.typ.is_int() {
tsym := c.table.get_type_symbol(branch.stmt.typ)
c.error('invalid type `$tsym.name` for timeout - expected integer type aka `time.Duration`',
branch.stmt.pos)
}
} else {
if branch.stmt.expr is ast.InfixExpr {
if branch.stmt.expr.left !is ast.Ident
&& branch.stmt.expr.left !is ast.SelectorExpr
&& branch.stmt.expr.left !is ast.IndexExpr {
c.error('channel in `select` key must be predefined', branch.stmt.expr.left.position())
}
} else {
c.error('invalid expression for `select` key', branch.stmt.expr.position())
}
}
}
ast.AssignStmt {
expr := branch.stmt.right[0]
match expr {
ast.PrefixExpr {
if expr.right !is ast.Ident && expr.right !is ast.SelectorExpr
&& expr.right !is ast.IndexExpr {
c.error('channel in `select` key must be predefined', expr.right.position())
}
if expr.or_block.kind != .absent {
err_prefix := if expr.or_block.kind == .block {
'or block'
} else {
'error propagation'
}
c.error('$err_prefix not allowed in `select` key', expr.or_block.pos)
}
}
else {
c.error('`<-` receive expression expected', branch.stmt.right[0].position())
}
}
}
else {
if !branch.is_else {
c.error('receive or send statement expected as `select` key', branch.stmt.position())
}
}
}
c.stmts(branch.stmts)
}
return table.bool_type
}
pub fn (mut c Checker) lock_expr(mut node ast.LockExpr) table.Type {
if c.rlocked_names.len > 0 || c.locked_names.len > 0 {
c.error('nested `lock`/`rlock` not allowed', node.pos)
}
2020-09-27 16:08:11 +02:00
for i in 0 .. node.lockeds.len {
c.ident(mut node.lockeds[i])
id := node.lockeds[i]
if mut id.obj is ast.Var {
if id.obj.typ.share() != .shared_t {
c.error('`$id.name` must be declared `shared` to be locked', id.pos)
}
} else {
c.error('`$id.name` is not a variable and cannot be locked', id.pos)
}
if id.name in c.locked_names {
c.error('`$id.name` is already locked', id.pos)
} else if id.name in c.rlocked_names {
c.error('`$id.name` is already read-locked', id.pos)
}
if node.is_rlock[i] {
c.rlocked_names << id.name
} else {
c.locked_names << id.name
}
}
c.stmts(node.stmts)
c.rlocked_names = []
c.locked_names = []
// handle `x := rlock a { a.getval() }`
mut ret_type := table.void_type
if node.stmts.len > 0 {
last_stmt := node.stmts[node.stmts.len - 1]
if last_stmt is ast.ExprStmt {
ret_type = last_stmt.typ
}
}
if ret_type != table.void_type {
node.is_expr = true
}
node.typ = ret_type
return ret_type
}
2020-07-12 12:58:33 +02:00
pub fn (mut c Checker) unsafe_expr(mut node ast.UnsafeExpr) table.Type {
assert !c.inside_unsafe
c.inside_unsafe = true
t := c.expr(node.expr)
c.inside_unsafe = false
return t
2020-07-12 12:58:33 +02:00
}
fn (mut c Checker) for_stmt(mut node ast.ForStmt) {
c.in_for_count++
prev_loop_label := c.loop_label
c.expected_type = table.bool_type
typ := c.expr(node.cond)
if !node.is_inf && typ.idx() != table.bool_type_idx && !c.pref.translated {
c.error('non-bool used as for condition', node.pos)
}
if node.cond is ast.InfixExpr {
infix := node.cond
if infix.op == .key_is {
if (infix.left is ast.Ident || infix.left is ast.SelectorExpr)
&& infix.right is ast.Type {
right_expr := infix.right as ast.Type
is_variable := if mut infix.left is ast.Ident {
infix.left.kind == .variable
} else {
true
}
left_type := c.expr(infix.left)
left_sym := c.table.get_type_symbol(left_type)
if is_variable {
if left_sym.kind == .sum_type {
c.smartcast_sumtype(infix.left, infix.left_type, right_expr.typ, mut
node.scope)
}
}
}
}
}
// TODO: update loop var type
// how does this work currenly?
c.check_loop_label(node.label, node.pos)
c.stmts(node.stmts)
c.loop_label = prev_loop_label
c.in_for_count--
}
2020-05-11 14:38:25 +02:00
pub fn (mut c Checker) if_expr(mut node ast.IfExpr) table.Type {
2020-10-18 21:22:37 +02:00
if_kind := if node.is_comptime { '\$if' } else { 'if' }
expr_required := c.expected_type != table.void_type
former_expected_type := c.expected_type
node.typ = table.void_type
mut nbranches_with_return := 0
mut nbranches_without_return := 0
mut should_skip := false // Whether the current branch should be skipped
mut found_branch := false // Whether a matching branch was found- skip the rest
mut is_comptime_type_is_expr := false // if `$if T is string`
for i in 0 .. node.branches.len {
mut branch := node.branches[i]
if branch.cond is ast.ParExpr {
c.error('unnecessary `()` in `$if_kind` condition, use `$if_kind expr {` instead of `$if_kind (expr) {`.',
2020-04-25 17:49:16 +02:00
branch.pos)
}
if !node.has_else || i < node.branches.len - 1 {
2020-10-18 21:22:37 +02:00
if node.is_comptime {
should_skip = c.comp_if_branch(branch.cond, branch.pos)
} else {
// check condition type is boolean
c.expected_type = table.bool_type
cond_typ := c.expr(branch.cond)
if cond_typ.idx() !in [table.bool_type_idx, table.void_type_idx]
&& !c.pref.translated {
// void types are skipped, because they mean the var was initialized incorrectly
// (via missing function etc)
typ_sym := c.table.get_type_symbol(cond_typ)
c.error('non-bool type `$typ_sym.name` used as if condition', branch.pos)
}
}
}
2020-08-24 11:10:11 +02:00
// smartcast sumtypes and interfaces when using `is`
2020-10-18 21:22:37 +02:00
if !node.is_comptime && branch.cond is ast.InfixExpr {
2020-07-08 15:17:28 +02:00
infix := branch.cond as ast.InfixExpr
if infix.op == .key_is {
2020-07-08 15:17:28 +02:00
right_expr := infix.right as ast.Type
left_sym := c.table.get_type_symbol(infix.left_type)
expr_type := c.expr(infix.left)
if left_sym.kind == .interface_ {
c.type_implements(right_expr.typ, expr_type, branch.pos)
} else if !c.check_types(right_expr.typ, expr_type) {
expect_str := c.table.type_to_str(right_expr.typ)
expr_str := c.table.type_to_str(expr_type)
c.error('cannot use type `$expect_str` as type `$expr_str`', branch.pos)
}
if (infix.left is ast.Ident || infix.left is ast.SelectorExpr)
&& infix.right is ast.Type {
is_variable := if mut infix.left is ast.Ident {
infix.left.kind == .variable
} else {
true
}
if is_variable {
if left_sym.kind in [.interface_, .sum_type] {
if infix.left is ast.Ident && left_sym.kind == .interface_ {
// TODO: rewrite interface smartcast
left := infix.left as ast.Ident
mut is_mut := false
2020-11-23 16:16:13 +01:00
mut sum_type_casts := []table.Type{}
if v := branch.scope.find_var(left.name) {
is_mut = v.is_mut
2020-11-23 16:16:13 +01:00
sum_type_casts << v.sum_type_casts
}
branch.scope.register(ast.Var{
name: left.name
typ: right_expr.typ.to_ptr()
sum_type_casts: sum_type_casts
pos: left.pos
is_used: true
is_mut: is_mut
})
// TODO: needs to be removed
node.branches[i].smartcast = true
} else {
c.smartcast_sumtype(infix.left, infix.left_type, right_expr.typ, mut
branch.scope)
}
}
2020-07-08 15:17:28 +02:00
}
}
}
}
2020-10-18 21:22:37 +02:00
if node.is_comptime { // Skip checking if needed
// smartcast field type on comptime if
mut comptime_field_name := ''
if branch.cond is ast.InfixExpr {
if branch.cond.op == .key_is {
left := branch.cond.left
got_type := (branch.cond.right as ast.Type).typ
if left is ast.SelectorExpr {
comptime_field_name = left.expr.str()
c.comptime_fields_type[comptime_field_name] = got_type
is_comptime_type_is_expr = true
} else if left is ast.Type {
is_comptime_type_is_expr = true
left_type := c.unwrap_generic(left.typ)
if left_type != got_type {
should_skip = true
}
}
}
}
cur_skip_flags := c.skip_flags
if found_branch {
c.skip_flags = true
} else if should_skip {
c.skip_flags = true
should_skip = false // Reset the value of `should_skip` for the next branch
} else if !is_comptime_type_is_expr {
found_branch = true // If a branch wasn't skipped, the rest must be
}
if !c.skip_flags || c.pref.output_cross_c {
c.stmts(branch.stmts)
} else if !is_comptime_type_is_expr {
node.branches[i].stmts = []
}
if comptime_field_name.len > 0 {
c.comptime_fields_type.delete(comptime_field_name)
}
c.skip_flags = cur_skip_flags
} else {
c.stmts(branch.stmts)
}
if expr_required {
if branch.stmts.len > 0 && branch.stmts[branch.stmts.len - 1] is ast.ExprStmt {
mut last_expr := branch.stmts[branch.stmts.len - 1] as ast.ExprStmt
c.expected_type = former_expected_type
2020-05-21 22:35:43 +02:00
last_expr.typ = c.expr(last_expr.expr)
2020-08-27 06:46:18 +02:00
// if last_expr.typ != node.typ {
// if !c.check_types(node.typ, last_expr.typ) {
if !c.check_types(last_expr.typ, node.typ) {
if node.typ == table.void_type {
2020-05-21 22:35:43 +02:00
// first branch of if expression
node.is_expr = true
2020-05-21 22:35:43 +02:00
node.typ = last_expr.typ
continue
} else if node.typ in [table.float_literal_type, table.int_literal_type] {
if node.typ == table.int_literal_type {
if last_expr.typ.is_int() || last_expr.typ.is_float() {
node.typ = last_expr.typ
continue
2020-05-27 15:56:21 +02:00
}
} else { // node.typ == float_literal
if last_expr.typ.is_float() {
node.typ = last_expr.typ
continue
}
}
2020-04-30 12:17:31 +02:00
}
if last_expr.typ in [table.float_literal_type, table.int_literal_type] {
if last_expr.typ == table.int_literal_type {
if node.typ.is_int() || node.typ.is_float() {
continue
2020-05-27 15:56:21 +02:00
}
} else { // expr_type == float_literal
if node.typ.is_float() {
continue
}
}
}
c.error('mismatched types `${c.table.type_to_str(node.typ)}` and `${c.table.type_to_str(last_expr.typ)}`',
2020-05-28 05:50:57 +02:00
node.pos)
}
} else {
c.error('`$if_kind` expression requires an expression as the last statement of every branch',
branch.pos)
}
for st in branch.stmts {
// must not contain C statements
st.check_c_expr() or { c.error('`if` expression branch has $err', st.position()) }
}
}
// Also check for returns inside a comp.if's statements, even if its contents aren't parsed
if has_return := c.has_return(branch.stmts) {
if has_return {
nbranches_with_return++
} else {
nbranches_without_return++
}
}
}
if nbranches_with_return > 0 {
if nbranches_with_return == node.branches.len {
// if/else... where all branches returned
c.returns = true
}
if !node.has_else {
// `if cond { return ... }` means that when cond is false, execution continues
c.returns = false
}
if nbranches_without_return > 0 {
// some of the branches did not return
c.returns = false
}
}
// if only untyped literals were given default to int/f64
if node.typ == table.int_literal_type {
node.typ = table.int_type
} else if node.typ == table.float_literal_type {
node.typ = table.f64_type
}
if expr_required && !node.has_else {
d := if node.is_comptime { '$' } else { '' }
c.error('`$if_kind` expression needs `${d}else` clause', node.pos)
2020-05-11 14:38:25 +02:00
}
return node.typ
}
// comp_if_branch checks the condition of a compile-time `if` branch. It returns a `bool` that
// saying whether that branch's contents should be skipped (targets a different os for example)
fn (mut c Checker) comp_if_branch(cond ast.Expr, pos token.Position) bool {
// TODO: better error messages here
match cond {
ast.BoolLiteral {
return !cond.val
}
ast.ParExpr {
return c.comp_if_branch(cond.expr, pos)
}
ast.PrefixExpr {
if cond.op != .not {
c.error('invalid `\$if` condition', cond.pos)
}
return !c.comp_if_branch(cond.right, cond.pos)
}
ast.PostfixExpr {
if cond.op != .question {
c.error('invalid \$if postfix operator', cond.pos)
} else if cond.expr is ast.Ident {
return cond.expr.name !in c.pref.compile_defines_all
} else {
c.error('invalid `\$if` condition', cond.pos)
}
}
ast.InfixExpr {
match cond.op {
.and {
l := c.comp_if_branch(cond.left, cond.pos)
r := c.comp_if_branch(cond.right, cond.pos)
return l || r // skip (return true) if at least one should be skipped
}
.logical_or {
l := c.comp_if_branch(cond.left, cond.pos)
r := c.comp_if_branch(cond.right, cond.pos)
return l && r // skip (return true) only if both should be skipped
}
.key_is, .not_is {
if (cond.left is ast.SelectorExpr || cond.left is ast.Type)
&& cond.right is ast.Type {
// $if method.@type is string
} else {
c.error('invalid `\$if` condition: $cond.left', cond.pos)
}
}
.eq, .ne {
if cond.left is ast.SelectorExpr && cond.right is ast.IntegerLiteral {
// $if method.args.len == 1
} else if cond.left is ast.Ident {
// $if version == 2
left_type := c.expr(cond.left)
right_type := c.expr(cond.right)
expr := c.find_definition(cond.left) or {
c.error(err, cond.left.pos)
return false
}
if !c.check_types(right_type, left_type) {
left_name := c.table.type_to_str(left_type)
right_name := c.table.type_to_str(right_type)
c.error('mismatched types `$left_name` and `$right_name`',
cond.pos)
}
// :)
// until `v.eval` is stable, I can't think of a better way to do this
different := expr.str() != cond.right.str()
return if cond.op == .eq {
different
} else {
!different
}
} else {
c.error('invalid `\$if` condition: ${cond.left.type_name()}1',
cond.pos)
}
}
else {
c.error('invalid `\$if` condition', cond.pos)
}
}
}
ast.Ident {
2021-01-25 10:26:20 +01:00
if cond.name in checker.valid_comp_if_os {
return cond.name != c.pref.os.str().to_lower() // TODO hack
2021-01-25 10:26:20 +01:00
} else if cond.name in checker.valid_comp_if_compilers {
return pref.cc_from_string(cond.name) != c.pref.ccompiler_type
2021-01-25 10:26:20 +01:00
} else if cond.name in checker.valid_comp_if_platforms {
return false // TODO
2021-01-25 10:26:20 +01:00
} else if cond.name in checker.valid_comp_if_other {
// TODO: This should probably be moved
2020-11-24 17:55:24 +01:00
match cond.name {
'js' { return c.pref.backend != .js }
'debug' { return !c.pref.is_debug }
'test' { return !c.pref.is_test }
'glibc' { return false } // TODO
'prealloc' { return !c.pref.prealloc }
'no_bounds_checking' { return cond.name !in c.pref.compile_defines_all }
else { return false }
}
} else if cond.name !in c.pref.compile_defines_all {
// `$if some_var {}`
typ := c.expr(cond)
if cond.obj !is ast.Var && cond.obj !is ast.ConstField
&& cond.obj !is ast.GlobalField {
c.error('unknown var: `$cond.name`', pos)
return false
}
expr := c.find_obj_definition(cond.obj) or {
c.error(err, cond.pos)
return false
}
if !c.check_types(typ, table.bool_type) {
type_name := c.table.type_to_str(typ)
c.error('non-bool type `$type_name` used as \$if condition', cond.pos)
}
// :)
// until `v.eval` is stable, I can't think of a better way to do this
return !(expr as ast.BoolLiteral).val
}
}
else {
c.error('invalid `\$if` condition', pos)
}
}
return false
}
fn (mut c Checker) find_definition(ident ast.Ident) ?ast.Expr {
match ident.kind {
.unresolved, .blank_ident { return none }
.variable, .constant { return c.find_obj_definition(ident.obj) }
.global { return error('$ident.name is a global variable') }
.function { return error('$ident.name is a function') }
}
}
fn (mut c Checker) find_obj_definition(obj ast.ScopeObject) ?ast.Expr {
// TODO: remove once we have better type inference
mut name := ''
match obj {
ast.Var, ast.ConstField, ast.GlobalField { name = obj.name }
}
mut expr := ast.Expr{}
if obj is ast.Var {
if obj.is_mut {
return error('`$name` is mut and may have changed since its definition')
}
expr = obj.expr
} else if obj is ast.ConstField {
expr = obj.expr
} else {
return error('`$name` is a global variable and is unknown at compile time')
}
if expr is ast.Ident {
return c.find_definition(expr as ast.Ident) // TODO: smartcast
}
if !expr.is_lit() {
return error('definition of `$name` is unknown at compile time')
}
return expr
}
2020-08-10 23:59:38 +02:00
fn (c &Checker) has_return(stmts []ast.Stmt) ?bool {
// complexity means either more match or ifs
mut has_complexity := false
for s in stmts {
if s is ast.ExprStmt {
if s.expr is ast.IfExpr || s.expr is ast.MatchExpr {
has_complexity = true
break
}
}
}
// if the inner complexity covers all paths with returns there is no need for further checks
if !has_complexity || !c.returns {
return has_top_return(stmts)
}
return none
}
pub fn (mut c Checker) postfix_expr(mut node ast.PostfixExpr) table.Type {
2020-02-04 07:37:38 +01:00
typ := c.expr(node.expr)
2020-02-27 17:31:10 +01:00
typ_sym := c.table.get_type_symbol(typ)
is_non_void_pointer := (typ.is_ptr() || typ.is_pointer()) && typ_sym.kind != .voidptr
if !c.inside_unsafe && is_non_void_pointer {
c.warn('pointer arithmetic is only allowed in `unsafe` blocks', node.pos)
}
if !(typ_sym.is_number() || (c.inside_unsafe && is_non_void_pointer)) {
c.error('invalid operation: $node.op.str() (non-numeric type `$typ_sym.name`)',
2020-06-24 14:44:06 +02:00
node.pos)
2020-07-14 18:11:16 +02:00
} else {
node.auto_locked, _ = c.fail_if_immutable(node.expr)
2020-02-04 07:37:38 +01:00
}
return typ
}
pub fn (mut c Checker) prefix_expr(mut node ast.PrefixExpr) table.Type {
right_type := c.expr(node.right)
node.right_type = right_type
// TODO: testing ref/deref strategy
if node.op == .amp && !right_type.is_ptr() {
match node.right {
ast.IntegerLiteral {
c.error('cannot take the address of an int literal', node.pos)
}
ast.BoolLiteral {
c.error('cannot take the address of a bool literal', node.pos)
}
ast.StringLiteral, ast.StringInterLiteral {
c.error('cannot take the address of a string literal', node.pos)
}
ast.FloatLiteral {
c.error('cannot take the address of a float literal', node.pos)
}
ast.CharLiteral {
c.error('cannot take the address of a char literal', node.pos)
}
else {}
}
if mut node.right is ast.IndexExpr {
typ_sym := c.table.get_type_symbol(node.right.left_type)
mut is_mut := false
if mut node.right.left is ast.Ident {
ident := node.right.left
// TODO: temporary, remove this
ident_obj := ident.obj
if ident_obj is ast.Var {
is_mut = ident_obj.is_mut
}
}
if typ_sym.kind == .map {
c.error('cannot take the address of map values', node.right.pos)
}
if !c.inside_unsafe {
if typ_sym.kind == .array && is_mut {
c.error('cannot take the address of mutable array elements outside unsafe blocks',
node.right.pos)
}
}
}
return right_type.to_ptr()
} else if node.op == .amp && node.right !is ast.CastExpr {
return right_type.to_ptr()
}
if node.op == .mul {
if right_type.is_ptr() {
return right_type.deref()
}
if !right_type.is_pointer() {
s := c.table.type_to_str(right_type)
c.error('invalid indirect of `$s`', node.pos)
}
}
if node.op == .bit_not && !right_type.is_int() && !c.pref.translated {
c.error('operator ~ only defined on int types', node.pos)
}
if node.op == .not && right_type != table.bool_type_idx && !c.pref.translated {
c.error('! operator can only be used with bool types', node.pos)
}
if node.op == .arrow {
right := c.table.get_type_symbol(right_type)
if right.kind == .chan {
c.stmts(node.or_block.stmts)
return right.chan_info().elem_type
} else {
c.error('<- operator can only be used with `chan` types', node.pos)
}
}
return right_type
}
fn (mut c Checker) check_index(typ_sym &table.TypeSymbol, index ast.Expr, index_type table.Type, pos token.Position) {
index_type_sym := c.table.get_type_symbol(index_type)
// println('index expr left=$typ_sym.name $node.pos.line_nr')
// if typ_sym.kind == .array && (!(table.type_idx(index_type) in table.number_type_idxs) &&
// index_type_sym.kind != .enum_) {
if typ_sym.kind in [.array, .array_fixed, .string, .ustring] {
if !(index_type.is_number() || index_type_sym.kind == .enum_) {
type_str := if typ_sym.kind in [.string, .ustring] {
'non-integer string index `$index_type_sym.name`'
} else {
'non-integer index `$index_type_sym.name` (array type `$typ_sym.name`)'
}
c.error('$type_str', pos)
}
if index is ast.IntegerLiteral {
if index.val.starts_with('-') {
c.error('invalid index `$index.val` (index must be non-negative)', index.pos)
}
}
if index_type.has_flag(.optional) {
type_str := if typ_sym.kind in [.string, .ustring] {
'(type `$typ_sym.name`)'
} else {
'(array type `$typ_sym.name`)'
}
c.error('cannot use optional as index $type_str', pos)
}
}
}
2020-05-11 14:38:25 +02:00
pub fn (mut c Checker) index_expr(mut node ast.IndexExpr) table.Type {
mut typ := c.expr(node.left)
node.left_type = typ
typ_sym := c.table.get_final_type_symbol(typ)
if typ_sym.kind !in [.array, .array_fixed, .string, .map] && !typ.is_ptr()
&& typ !in [table.byteptr_type, table.charptr_type] && !typ.has_flag(.variadic) {
c.error('type `$typ_sym.name` does not support indexing', node.pos)
}
if typ_sym.kind == .string && !typ.is_ptr() && node.is_setter {
2020-08-13 23:21:11 +02:00
c.error('cannot assign to s[i] since V strings are immutable\n' +
'(note, that variables may be mutable but string values are always immutable, like in Go and Java)',
node.pos)
}
if !c.inside_unsafe && (typ.is_ptr() || typ.is_pointer()) {
mut is_ok := false
if mut node.left is ast.Ident {
if node.left.obj is ast.Var {
v := node.left.obj as ast.Var
// `mut param []T` function parameter
is_ok = ((v.is_mut && v.is_arg) || v.share == .shared_t) && !typ.deref().is_ptr()
}
}
2020-08-09 03:57:54 +02:00
if !is_ok && !c.pref.translated {
c.warn('pointer indexing is only allowed in `unsafe` blocks', node.pos)
}
}
if mut node.index is ast.RangeExpr { // [1..2]
if node.index.has_low {
index_type := c.expr(node.index.low)
c.check_index(typ_sym, node.index.low, index_type, node.pos)
}
if node.index.has_high {
index_type := c.expr(node.index.high)
c.check_index(typ_sym, node.index.high, index_type, node.pos)
}
2020-03-31 16:47:55 +02:00
// array[1..2] => array
// fixed_array[1..2] => array
if typ_sym.kind == .array_fixed {
elem_type := c.table.value_type(typ)
idx := c.table.find_or_register_array(elem_type)
typ = table.new_type(idx)
} else {
typ = typ.set_nr_muls(0)
2020-03-31 16:47:55 +02:00
}
} else { // [1]
index_type := c.expr(node.index)
if typ_sym.kind == .map {
info := typ_sym.info as table.Map
if !c.check_types(index_type, info.key_type) {
err := c.expected_msg(index_type, info.key_type)
c.error('invalid key: $err', node.pos)
}
} else {
c.check_index(typ_sym, node.index, index_type, node.pos)
}
value_type := c.table.value_type(typ)
if value_type != table.void_type {
typ = value_type
}
2020-03-31 16:47:55 +02:00
}
c.stmts(node.or_expr.stmts)
return typ
}
// `.green` or `Color.green`
// If a short form is used, `expected_type` needs to be an enum
// with this value.
2020-05-11 14:38:25 +02:00
pub fn (mut c Checker) enum_val(mut node ast.EnumVal) table.Type {
typ_idx := if node.enum_name == '' {
2020-04-25 09:08:53 +02:00
c.expected_type.idx()
2020-04-25 17:49:16 +02:00
} else { //
c.table.find_type_idx(node.enum_name)
}
2020-02-29 18:34:25 +01:00
// println('checker: enum_val: $node.enum_name typeidx=$typ_idx')
if typ_idx == 0 {
c.error('not an enum (name=$node.enum_name) (type_idx=0)', node.pos)
2020-05-08 15:00:04 +02:00
return table.void_type
2020-02-29 18:34:25 +01:00
}
mut typ := table.new_type(typ_idx)
2020-08-11 16:26:49 +02:00
if c.pref.translated {
// TODO make more strict
node.typ = typ
return typ
}
2020-04-26 19:59:03 +02:00
if typ == table.void_type {
c.error('not an enum', node.pos)
2020-05-08 15:00:04 +02:00
return table.void_type
2020-04-26 19:59:03 +02:00
}
mut typ_sym := c.table.get_type_symbol(typ)
// println('tname=$typ_sym.name $node.pos.line_nr $c.file.path')
if typ_sym.kind == .array && node.enum_name.len == 0 {
array_info := typ_sym.info as table.Array
typ = array_info.elem_type
typ_sym = c.table.get_type_symbol(typ)
}
2020-08-11 16:26:49 +02:00
if typ_sym.kind != .enum_ && !c.pref.translated {
// TODO in C int fields can be compared to enums, need to handle that in C2V
c.error('expected type is not an enum (`$typ_sym.name`)', node.pos)
2020-05-08 15:00:04 +02:00
return table.void_type
}
2020-07-08 15:46:58 +02:00
if typ_sym.info !is table.Enum {
2020-05-08 15:00:04 +02:00
c.error('not an enum', node.pos)
return table.void_type
}
2020-03-15 02:51:31 +01:00
// info := typ_sym.info as table.Enum
info := typ_sym.enum_info()
// rintln('checker: x = $info.x enum val $c.expected_type $typ_sym.name')
2020-02-27 00:12:37 +01:00
// println(info.vals)
if node.val !in info.vals {
c.error('enum `$typ_sym.name` does not have a value `$node.val`', node.pos)
}
2020-03-15 02:51:31 +01:00
node.typ = typ
return typ
}
pub fn (mut c Checker) chan_init(mut node ast.ChanInit) table.Type {
if node.typ != 0 {
info := c.table.get_type_symbol(node.typ).chan_info()
node.elem_type = info.elem_type
if node.has_cap {
c.check_array_init_para_type('cap', node.cap_expr, node.pos)
}
return node.typ
} else {
c.error('`chan` of unknown type', node.pos)
return node.typ
}
}
2021-01-30 12:57:09 +01:00
pub fn (mut c Checker) offset_of(node ast.OffsetOf) table.Type {
sym := c.table.get_final_type_symbol(node.struct_type)
if sym.kind != .struct_ {
c.error('first argument of __offsetof must be struct', node.pos)
return table.u32_type
}
if !c.table.struct_has_field(node.struct_type, node.field) {
c.error('struct `$sym.name` has no field called `$node.field`', node.pos)
}
return table.u32_type
}
pub fn (mut c Checker) check_dup_keys(node &ast.MapInit, i int) {
key_i := node.keys[i]
if key_i is ast.StringLiteral {
for j in 0 .. i {
key_j := node.keys[j]
if key_j is ast.StringLiteral {
if key_i.val == key_j.val {
c.error('duplicate key "$key_i.val" in map literal', key_i.pos)
}
}
}
} else if key_i is ast.IntegerLiteral {
for j in 0 .. i {
key_j := node.keys[j]
if key_j is ast.IntegerLiteral {
if key_i.val == key_j.val {
c.error('duplicate key "$key_i.val" in map literal', key_i.pos)
}
}
}
}
}
2020-05-11 14:38:25 +02:00
pub fn (mut c Checker) map_init(mut node ast.MapInit) table.Type {
2021-01-30 11:55:10 +01:00
// `map = {}`
if node.keys.len == 0 && node.vals.len == 0 && node.typ == 0 {
sym := c.table.get_type_symbol(c.expected_type)
if sym.kind == .map {
info := sym.map_info()
node.typ = c.expected_type
node.key_type = info.key_type
node.value_type = info.value_type
return node.typ
}
}
2020-10-03 06:59:30 +02:00
// `x := map[string]string` - set in parser
2020-03-07 08:13:00 +01:00
if node.typ != 0 {
info := c.table.get_type_symbol(node.typ).map_info()
key_sym := c.table.get_type_symbol(info.key_type)
value_sym := c.table.get_type_symbol(info.value_type)
if key_sym.kind == .placeholder {
c.error('unknown type `$key_sym.name`', node.pos)
}
if value_sym.kind == .placeholder {
c.error('unknown type `$value_sym.name`', node.pos)
}
2020-03-07 08:13:00 +01:00
node.key_type = info.key_type
node.value_type = info.value_type
return node.typ
}
// `{'age': 20}`
key0_type := c.table.mktyp(c.expr(node.keys[0]))
val0_type := c.table.mktyp(c.expr(node.vals[0]))
mut same_key_type := true
for i, key in node.keys {
if i == 0 {
continue
}
val := node.vals[i]
key_type := c.expr(key)
val_type := c.expr(val)
2020-05-24 04:43:00 +02:00
if !c.check_types(key_type, key0_type) {
msg := c.expected_msg(key_type, key0_type)
c.error('invalid map key: $msg', key.position())
same_key_type = false
}
2020-05-24 04:43:00 +02:00
if !c.check_types(val_type, val0_type) {
msg := c.expected_msg(val_type, val0_type)
c.error('invalid map value: $msg', val.position())
}
}
if same_key_type {
for i in 1 .. node.keys.len {
c.check_dup_keys(node, i)
}
}
map_type := table.new_type(c.table.find_or_register_map(key0_type, val0_type))
node.typ = map_type
2020-03-07 08:13:00 +01:00
node.key_type = key0_type
node.value_type = val0_type
return map_type
}
// call this *before* calling error or warn
pub fn (mut c Checker) add_error_detail(s string) {
c.error_details << s
}
2020-04-25 17:49:16 +02:00
pub fn (mut c Checker) warn(s string, pos token.Position) {
allow_warnings := !(c.pref.is_prod || c.pref.warns_are_errors) // allow warnings only in dev builds
2020-04-25 17:49:16 +02:00
c.warn_or_error(s, pos, allow_warnings) // allow warnings only in dev builds
}
2020-04-25 17:49:16 +02:00
pub fn (mut c Checker) error(message string, pos token.Position) {
2020-08-11 16:26:49 +02:00
if c.pref.translated && message.starts_with('mismatched types') {
// TODO move this
return
}
2020-04-16 15:32:11 +02:00
if c.pref.is_verbose {
print_backtrace()
}
msg := message.replace('`array_', '`[]')
c.warn_or_error(msg, pos, false)
}
// check `to` has all fields of `from`
fn (c Checker) check_struct_signature(from table.Struct, to table.Struct) bool {
// Note: `to` can have extra fields
if from.fields.len == 0 {
2020-08-14 14:57:08 +02:00
return false
}
for _, field in from.fields {
filtered := to.fields.filter(it.name == field.name)
if filtered.len != 1 {
// field doesn't exist
return false
}
counterpart := filtered[0]
if field.typ != counterpart.typ {
// field has different tye
return false
}
if field.is_pub != counterpart.is_pub {
// field is not public while the other one is
return false
}
if field.is_mut != counterpart.is_mut {
// field is not mutable while the other one is
return false
}
}
return true
}
2020-04-25 17:49:16 +02:00
fn (mut c Checker) warn_or_error(message string, pos token.Position, warn bool) {
2020-04-13 01:56:01 +02:00
// add backtrace to issue struct, how?
// if c.pref.is_verbose {
// print_backtrace()
2020-04-13 01:56:01 +02:00
// }
mut details := ''
if c.error_details.len > 0 {
details = c.error_details.join('\n')
c.error_details = []
}
if warn && !c.pref.skip_warnings {
c.nr_warnings++
wrn := errors.Warning{
reporter: errors.Reporter.checker
pos: pos
file_path: c.file.path
message: message
details: details
}
c.file.warnings << wrn
c.warnings << wrn
return
}
if !warn {
if c.pref.fatal_errors {
exit(1)
}
c.nr_errors++
if pos.line_nr !in c.error_lines {
err := errors.Error{
reporter: errors.Reporter.checker
2020-04-13 01:56:01 +02:00
pos: pos
file_path: c.file.path
message: message
details: details
2020-04-13 01:56:01 +02:00
}
c.file.errors << err
c.errors << err
2020-04-13 01:56:01 +02:00
c.error_lines << pos.line_nr
}
2020-02-19 19:54:36 +01:00
}
}
// for debugging only
fn (c &Checker) fileis(s string) bool {
return c.file.path.contains(s)
}
fn (mut c Checker) sql_expr(mut node ast.SqlExpr) table.Type {
c.inside_sql = true
defer {
c.inside_sql = false
}
sym := c.table.get_type_symbol(node.table_expr.typ)
2020-06-28 19:01:30 +02:00
if sym.kind == .placeholder {
c.error('orm: unknown type `$sym.name`', node.pos)
2020-06-28 19:01:30 +02:00
return table.void_type
}
c.cur_orm_ts = sym
info := sym.info as table.Struct
fields := c.fetch_and_verify_orm_fields(info, node.table_expr.pos, sym.name)
2021-02-04 20:28:33 +01:00
mut sub_structs := map[int]ast.SqlExpr{}
for f in fields.filter(c.table.types[int(it.typ)].kind == .struct_) {
mut n := ast.SqlExpr{
pos: node.pos
has_where: true
typ: f.typ
db_expr: node.db_expr
table_expr: ast.Type{
pos: node.table_expr.pos
typ: f.typ
}
}
tmp_inside_sql := c.inside_sql
c.sql_expr(mut n)
c.inside_sql = tmp_inside_sql
n.where_expr = ast.InfixExpr{
op: .eq
pos: n.pos
left: ast.Ident{
language: .v
tok_kind: .eq
scope: c.fn_scope
obj: ast.Var{}
mod: 'main'
name: 'id'
is_mut: false
kind: .unresolved
info: ast.IdentVar{}
}
right: ast.Ident{
language: .c
mod: 'main'
tok_kind: .eq
obj: ast.Var{}
is_mut: false
scope: c.fn_scope
info: ast.IdentVar{
typ: table.int_type
}
}
left_type: table.int_type
right_type: table.int_type
auto_locked: ''
or_block: ast.OrExpr{}
}
sub_structs[int(f.typ)] = n
}
node.fields = fields
2021-02-04 20:28:33 +01:00
node.sub_structs = sub_structs
2020-06-17 04:05:13 +02:00
if node.has_where {
c.expr(node.where_expr)
}
2020-06-27 16:41:29 +02:00
if node.has_offset {
c.expr(node.offset_expr)
}
if node.has_limit {
c.expr(node.limit_expr)
}
2020-07-02 19:29:22 +02:00
if node.has_order {
c.expr(node.order_expr)
}
c.expr(node.db_expr)
2020-06-17 04:05:13 +02:00
return node.typ
}
fn (mut c Checker) sql_stmt(mut node ast.SqlStmt) table.Type {
c.inside_sql = true
defer {
c.inside_sql = false
}
sym := c.table.get_type_symbol(node.table_expr.typ)
if node.table_expr.typ == 0 {
c.error('orm: unknown type `$sym.name`', node.pos)
2020-06-28 19:01:30 +02:00
}
if sym.kind == .placeholder {
c.error('orm: unknown type `$sym.name`', node.pos)
2020-06-28 19:01:30 +02:00
return table.void_type
}
c.cur_orm_ts = sym
info := sym.info as table.Struct
table_sym := c.table.get_type_symbol(node.table_expr.typ)
fields := c.fetch_and_verify_orm_fields(info, node.table_expr.pos, table_sym.name)
2021-02-04 20:28:33 +01:00
mut sub_structs := map[int]ast.SqlStmt{}
for f in fields.filter(c.table.types[int(it.typ)].kind == .struct_) {
mut n := ast.SqlStmt{
pos: node.pos
db_expr: node.db_expr
kind: node.kind
table_expr: ast.Type{
pos: node.table_expr.pos
typ: f.typ
}
object_var_name: '${node.object_var_name}.$f.name'
}
tmp_inside_sql := c.inside_sql
c.sql_stmt(mut n)
c.inside_sql = tmp_inside_sql
sub_structs[int(f.typ)] = n
}
node.fields = fields
2021-02-04 20:28:33 +01:00
node.sub_structs = sub_structs
c.expr(node.db_expr)
2020-06-27 16:41:29 +02:00
if node.kind == .update {
for expr in node.update_exprs {
c.expr(expr)
}
}
2020-06-25 22:37:36 +02:00
c.expr(node.where_expr)
2021-02-04 20:28:33 +01:00
2020-06-19 16:43:32 +02:00
return table.void_type
}
fn (mut c Checker) fetch_and_verify_orm_fields(info table.Struct, pos token.Position, table_name string) []table.Field {
fields := info.fields.filter((it.typ in [table.string_type, table.int_type, table.bool_type]
|| c.table.types[int(it.typ)].kind == .struct_) && !it.attrs.contains('skip'))
if fields.len == 0 {
c.error('V orm: select: empty fields in `$table_name`', pos)
return []table.Field{}
}
if fields[0].name != 'id' {
c.error('V orm: `id int` must be the first field in `$table_name`', pos)
}
return fields
}
fn (mut c Checker) post_process_generic_fns() {
// Loop thru each generic function concrete type.
// Check each specific fn instantiation.
for i in 0 .. c.file.generic_fns.len {
if c.table.fn_gen_types.len == 0 {
// no concrete types, so just skip:
continue
}
mut node := c.file.generic_fns[i]
c.mod = node.mod
for gen_types in c.table.fn_gen_types[node.name] {
c.cur_generic_types = gen_types
c.fn_decl(mut node)
if node.name in ['vweb.run_app', 'vweb.run'] {
c.vweb_gen_types << gen_types
}
}
c.cur_generic_types = []
}
}
fn (mut c Checker) fn_decl(mut node ast.FnDecl) {
c.returns = false
if node.generic_params.len > 0 && c.cur_generic_types.len == 0 {
// Just remember the generic function for now.
// It will be processed later in c.post_process_generic_fns,
// after all other normal functions are processed.
// This is done so that all generic function calls can
// have a chance to populate c.table.fn_gen_types with
// the correct concrete types.
c.file.generic_fns << node
return
}
if node.language == .v && !c.is_builtin_mod {
c.check_valid_snake_case(node.name, 'function name', node.pos)
}
2021-02-05 08:05:13 +01:00
if node.name == 'main.main' {
c.main_fn_decl_node = node
}
if node.return_type != table.void_type {
for attr in node.attrs {
2021-02-05 08:05:13 +01:00
if attr.is_comptime_define {
c.error('only functions that do NOT return values can have `[if $attr.name]` tags',
node.pos)
break
}
}
}
if node.is_method {
mut sym := c.table.get_type_symbol(node.receiver.typ)
if sym.kind == .array && !c.is_builtin_mod && node.name == 'map' {
// TODO `node.map in array_builtin_methods`
c.error('method overrides built-in array method', node.pos)
} else if sym.kind == .sum_type && node.name == 'type_name' {
c.error('method overrides built-in sum type method', node.pos)
}
2021-01-28 11:05:37 +01:00
if sym.name.len == 1 {
// One letter types are reserved for generics.
c.error('unknown type `$sym.name`', node.receiver_pos)
2021-01-28 11:05:37 +01:00
return
}
// if sym.has_method(node.name) {
// c.warn('duplicate method `$node.name`', node.pos)
// }
// needed for proper error reporting during vweb route checking
sym.methods[node.method_idx].source_fn = voidptr(node)
}
if node.language == .v {
// Make sure all types are valid
2020-09-27 03:32:56 +02:00
for arg in node.params {
sym := c.table.get_type_symbol(arg.typ)
if sym.kind == .placeholder
|| (sym.kind in [table.Kind.int_literal, .float_literal] && !c.is_builtin_mod) {
c.error('unknown type `$sym.name`', node.pos)
}
}
}
return_sym := c.table.get_type_symbol(node.return_type)
if node.language == .v && return_sym.kind in [.placeholder, .int_literal, .float_literal]
&& return_sym.language == .v {
c.error('unknown type `$return_sym.name`', node.pos)
}
if node.language == .v && node.is_method && node.name == 'str' {
if node.return_type != table.string_type {
c.error('.str() methods should return `string`', node.pos)
}
2020-09-27 03:32:56 +02:00
if node.params.len != 1 {
c.error('.str() methods should have 0 arguments', node.pos)
}
}
if node.language == .v && node.is_method && node.name in ['+', '-', '*', '%', '/', '<', '=='] {
if node.params.len != 2 {
c.error('operator methods should have exactly 1 argument', node.pos)
} else {
receiver_sym := c.table.get_type_symbol(node.receiver.typ)
param_sym := c.table.get_type_symbol(node.params[1].typ)
if param_sym.kind !in [.struct_, .alias] || receiver_sym.kind !in [.struct_, .alias] {
c.error('operator methods are only allowed for struct and type alias',
node.pos)
} else {
parent_sym := c.table.get_final_type_symbol(node.receiver.typ)
if node.rec_mut {
c.error('receiver cannot be `mut` for operator overloading', node.receiver_pos)
} else if node.params[1].is_mut {
c.error('argument cannot be `mut` for operator overloading', node.pos)
} else if node.receiver.typ != node.params[1].typ {
c.error('expected `$receiver_sym.name` not `$param_sym.name` - both operands must be the same type for operator overloading',
node.params[1].type_pos)
} else if node.name in ['<', '=='] && node.return_type != table.bool_type {
c.error('operator comparison methods should return `bool`', node.pos)
} else if parent_sym.is_primitive() {
c.error('cannot define operator methods on type alias for `$parent_sym.name`',
node.pos)
}
}
}
}
// TODO c.pref.is_vet
if node.language == .v && !node.is_method && node.params.len == 0
&& node.name.after('.').starts_with('test_') {
if !c.pref.is_test {
// simple heuristic
for st in node.stmts {
if st is ast.AssertStmt {
c.warn('tests will not be run, because filename does not end with `_test.v`',
node.pos)
break
}
}
}
// eprintln('> node.name: $node.name | node.return_type: $node.return_type')
if node.return_type != table.void_type_idx {
c.error('test functions should not return anything', node.pos)
}
}
c.expected_type = table.void_type
c.cur_fn = node
// Add return if `fn(...) ? {...}` have no return at end
if node.return_type != table.void_type && node.return_type.has_flag(.optional)
&& (node.stmts.len == 0 || node.stmts[node.stmts.len - 1] !is ast.Return) {
sym := c.table.get_type_symbol(node.return_type)
if sym.kind == .void {
node.stmts << ast.Return{
pos: node.pos
}
}
}
for param in node.params {
if param.is_mut {
c.fn_mut_arg_names << param.name
}
}
c.fn_scope = node.scope
c.stmts(node.stmts)
if c.fn_mut_arg_names.len > 0 {
c.fn_mut_arg_names.clear()
}
node.has_return = c.returns || has_top_return(node.stmts)
if node.language == .v && !node.no_body && node.return_type != table.void_type
&& !node.has_return && node.name !in ['panic', 'exit'] {
if c.inside_anon_fn {
c.error('missing return at the end of an anonymous function', node.pos)
} else {
c.error('missing return at end of function `$node.name`', node.pos)
}
}
c.returns = false
node.source_file = c.file
}
fn has_top_return(stmts []ast.Stmt) bool {
for stmt in stmts {
if stmt is ast.Return {
return true
} else if stmt is ast.Block {
if has_top_return(stmt.stmts) {
return true
}
} else if stmt is ast.ExprStmt {
if stmt.expr is ast.CallExpr {
if stmt.expr.name in ['panic', 'exit'] {
return true
}
}
}
}
return false
}
fn (mut c Checker) verify_vweb_params_for_method(m table.Fn) (bool, int, int) {
margs := m.params.len - 1 // first arg is the receiver/this
if m.attrs.len == 0 {
// allow non custom routed methods, with 1:1 mapping
return true, -1, margs
}
mut route_attributes := 0
for a in m.attrs {
if a.name.starts_with('/') {
route_attributes += a.name.count(':')
}
}
return route_attributes == margs, route_attributes, margs
}
fn (mut c Checker) verify_all_vweb_routes() {
if c.vweb_gen_types.len == 0 {
return
}
typ_vweb_result := c.table.find_type_idx('vweb.Result')
for vgt in c.vweb_gen_types {
sym_app := c.table.get_type_symbol(vgt)
for m in sym_app.methods {
if m.return_type == typ_vweb_result {
is_ok, nroute_attributes, nargs := c.verify_vweb_params_for_method(m)
if !is_ok {
2021-01-05 19:26:48 +01:00
f := &ast.FnDecl(m.source_fn)
if isnil(f) {
continue
}
if f.return_type == typ_vweb_result && f.receiver.typ == m.params[0].typ
&& f.name == m.name {
c.file = f.source_file // setup of file path for the warning
c.warn('mismatched parameters count between vweb method `${sym_app.name}.$m.name` ($nargs) and route attribute $m.attrs ($nroute_attributes)',
f.pos)
}
}
}
}
}
}
fn (mut c Checker) trace(fbase string, message string) {
if c.file.path_base == fbase {
println('> c.trace | ${fbase:-10s} | $message')
}
}