Function RegoVarPath

coderd/rbac/regosql/sqltypes/variable.go:38–68

RegoVarPath will consume the following terms from the given rego Ref and return the remaining terms. If the path does not fully match, an error is returned. The first term must always be a Var.

(path []string, terms []*ast.Term)

// return the remaining terms. If the path does not fully match, an error is
// returned. The first term must always be a Var.
func RegoVarPath(path []string, terms []*ast.Term) ([]*ast.Term, error) {
	if len(terms) < len(path) {
		return nil, xerrors.Errorf("path %s longer than rego path %s", path, terms)
	}

	if len(terms) == 0 || len(path) == 0 {
		return nil, xerrors.Errorf("path %s and rego path %s must not be empty", path, terms)
	}

	varTerm, ok := terms[0].Value.(ast.Var)
	if !ok {
		return nil, xerrors.Errorf("expected var, got %T", terms[0])
	}

	if string(varTerm) != path[0] {
		return nil, xerrors.Errorf("expected var %s, got %s", path[0], varTerm)
	}

	for i := 1; i < len(path); i++ {
		nextTerm, ok := terms[i].Value.(ast.String)
		if !ok {
			return nil, xerrors.Errorf("expected ast.string, got %T", terms[i])
		}

		if string(nextTerm) != path[i] {
			return nil, xerrors.Errorf("expected string %s, got %s", path[i], nextTerm)
		}
	}

	return terms[len(path):], nil
}

var (
	_ VariableMatcher = astStringVar{}