shader: Bug fix: Checking unused variables defined with var

internal/shader/shader.go

@@ -322,9 +322,11 @@ func (cs *compileState) parseDecl(b *block, d ast.Decl) ([]shaderir.Stmt, bool)
 					continue
 				}
 
-				base := b.totalLocalVariableNum()
-				b.vars = append(b.vars, vs...)
+				for _, v := range vs {
+					b.addNamedLocalVariable(v.name, v.typ, d.Pos())
+				}
 
+				base := b.totalLocalVariableNum()
 				if len(inits) > 0 {
 					for i := range vs {
 						stmts = append(stmts, shaderir.Stmt{

shader_test.go

@@ -588,6 +588,26 @@ func Fragment(position vec4, texCoord vec2, color vec4) vec4 {
 `)); err != nil {
 		t.Error(err)
 	}
+
+	if _, err := NewShader([]byte(`package main
+
+func Fragment(position vec4, texCoord vec2, color vec4) vec4 {
+	var a int
+	return vec4(0)
+}
+`)); err == nil {
+		t.Errorf("error must be non-nil but was nil")
+	}
+
+	if _, err := NewShader([]byte(`package main
+
+func Fragment(position vec4, texCoord vec2, color vec4) vec4 {
+	var a, b int
+	return vec4(0)
+}
+`)); err == nil {
+		t.Errorf("error must be non-nil but was nil")
+	}
 }
 
 func TestShaderBlankLhs(t *testing.T) {