diff --git a/compiler.go b/compiler.go index 6591603..0d46a4d 100644 --- a/compiler.go +++ b/compiler.go @@ -692,12 +692,15 @@ func (c *Compiler) compileAssign( return c.errorf(node, "operator ':=' not allowed with selector") } + _, isFunc := rhs[0].(*parser.FuncLit) symbol, depth, exists := c.symbolTable.Resolve(ident, false) if op == token.Define { if depth == 0 && exists { return c.errorf(node, "'%s' redeclared in this block", ident) } - symbol = c.symbolTable.Define(ident) + if isFunc { + symbol = c.symbolTable.Define(ident) + } } else { if !exists { return c.errorf(node, "unresolved reference '%s'", ident) @@ -718,6 +721,10 @@ func (c *Compiler) compileAssign( } } + if op == token.Define && !isFunc { + symbol = c.symbolTable.Define(ident) + } + switch op { case token.AddAssign: c.emit(node, parser.OpBinaryOp, int(token.Add)) diff --git a/compiler_test.go b/compiler_test.go index fd1d422..d525b47 100644 --- a/compiler_test.go +++ b/compiler_test.go @@ -1023,6 +1023,8 @@ func TestCompilerErrorReport(t *testing.T) { expectCompileError(t, `a = 1`, "Compile Error: unresolved reference 'a'\n\tat test:1:1") + expectCompileError(t, `a := a`, + "Compile Error: unresolved reference 'a'\n\tat test:1:6") expectCompileError(t, `a, b := 1, 2`, "Compile Error: tuple assignment not allowed\n\tat test:1:1") expectCompileError(t, `a.b := 1`,