Merge pull request #1361 from nspcc-dev/feature/iota

compiler: support `iota`

pkg/compiler/codegen.go

@@ -408,7 +408,12 @@ func (c *codegen) Visit(node ast.Node) ast.Visitor {
 			for _, spec := range n.Specs {
 				vs := spec.(*ast.ValueSpec)
 				for i := range vs.Names {
-					c.constMap[c.getIdentName("", vs.Names[i].Name)] = c.typeAndValueOf(vs.Values[i])
+					info := c.buildInfo.program.Package(c.currPkg.Path())
+					obj := info.Defs[vs.Names[i]]
+					c.constMap[c.getIdentName("", vs.Names[i].Name)] = types.TypeAndValue{
+						Type:  obj.Type(),
+						Value: obj.(*types.Const).Val(),
+					}
 				}
 			}
 			return nil

pkg/compiler/constant_test.go

@@ -72,3 +72,17 @@ func TestGlobalsWithFunctionParams(t *testing.T) {
 	`
 	eval(t, src, []byte("FOO"))
 }
+
+func TestIota(t *testing.T) {
+	src := `package foo
+	const (
+		a = 2 << iota
+		b
+
+		c = 11
+	)
+	func Main() int {
+		return a + b + c
+	}`
+	eval(t, src, big.NewInt(17))
+}