Function variablesForArgMode

analysis/validation.go:490–504

variablesForArgMode returns variable arguments that match an argmode mask.

(atom ast.Atom, mode ast.Mode, mask ast.ArgMode)

// variablesForArgMode returns variable arguments that match an argmode mask.
func variablesForArgMode(atom ast.Atom, mode ast.Mode, mask ast.ArgMode) []ast.Variable {
	var boundVars []ast.Variable

	for i, argMode := range mode {
		if argMode&mask == 0 {
			continue
		}

		if v, ok := atom.Args[i].(ast.Variable); ok {
			boundVars = append(boundVars, v)
		}
	}

	return boundVars
}

// CheckRule checks arity and that every variable appearing is bound.
// A variable is bound when: