allow short struct initialization with Config structs

pull/3450/head
Alexander Medvednikov 2020-01-14 15:34:47 +01:00
parent 0b235cc893
commit 6b3f54692a
3 changed files with 11 additions and 1 deletions

View File

@ -902,6 +902,13 @@ fn (p mut Parser) factor() string {
if p.peek() == .str {
return p.map_init()
}
peek2 := p.tokens[p.token_idx + 1]
if p.peek() == .name && peek2.tok == .colon {
if !p.expected_type.ends_with('Config') {
p.error('short struct initialization syntax only works with structs that end with `Config`')
}
return p.struct_init(p.expected_type)
}
// { user | name :'new name' }
return p.assoc()
}

View File

@ -488,7 +488,9 @@ fn (p mut Parser) gen_struct_init(typ string, t &Type) bool {
if typ == 'tm' {
p.cgen.lines[p.cgen.lines.len - 1] = ''
}
p.next()
if p.tok != .lcbr {
p.next()
}
p.check(.lcbr)
ptr := typ.contains('*')
// `user := User{foo:bar}` => `User user = (User){ .foo = bar}`

View File

@ -337,6 +337,7 @@ fn (p mut Parser) struct_decl(generic_param_types []string) {
// p.fgenln('//kek')
}
// `User{ foo: bar }`
// tok == struct name
fn (p mut Parser) struct_init(typ_ string) string {
p.is_struct_init = true
mut typ := typ_