Function envToStdLib

tools/compiler/compiler.go:361–404

(pbEnv *configpb.Environment)

}

func envToStdLib(pbEnv *configpb.Environment) (*env.LibrarySubset, error) {
	stdLib := env.NewLibrarySubset()
	pbEnvStdLib := pbEnv.GetStdlib()
	if pbEnvStdLib == nil {
		if pbEnv.GetDisableStandardCelDeclarations() {
			stdLib.SetDisabled(true)
			return stdLib, nil
		}
		return nil, nil
	}
	if !stdLib.Disabled {
		stdLib.SetDisabled(pbEnvStdLib.GetDisabled())
	}
	stdLib.SetDisableMacros(pbEnvStdLib.GetDisableMacros())
	stdLib.AddIncludedMacros(pbEnvStdLib.GetIncludeMacros()...)
	stdLib.AddExcludedMacros(pbEnvStdLib.GetExcludeMacros()...)
	if pbEnvStdLib.GetIncludeFunctions() != nil {
		for _, includeFn := range pbEnvStdLib.GetIncludeFunctions() {
			if includeFn.GetFunction() != nil {
				fn, err := protoDeclToFunction(includeFn)
				if err != nil {
					return nil, err
				}
				stdLib.AddIncludedFunctions(fn)
			} else {
				return nil, fmt.Errorf("IncludeFunctions must specify a function decl")
			}
		}
	}
	if pbEnvStdLib.GetExcludeFunctions() != nil {
		for _, excludeFn := range pbEnvStdLib.GetExcludeFunctions() {
			if excludeFn.GetFunction() != nil {
				fn, err := protoDeclToFunction(excludeFn)
				if err != nil {
					return nil, err
				}
				stdLib.AddExcludedFunctions(fn)
			} else {
				return nil, fmt.Errorf("ExcludeFunctions must specify a function decl")
			}
		}
	}
	return stdLib, nil
}

func protoDeclToFunction(decl *celpb.Decl) (*env.Function, error) {
	declFn := decl.GetFunction()