Function EnvironmentFile

tools/compiler/compiler.go:224–268

EnvironmentFile returns an EnvOption which loads a serialized CEL environment from a file. The file must be in one of the following formats: - Textproto - YAML - Binarypb

(path string)

// - YAML
// - Binarypb
func EnvironmentFile(path string) cel.EnvOption {
	return func(e *cel.Env) (*cel.Env, error) {
		format := InferFileFormat(path)
		if format != TextProto && format != TextYAML && format != BinaryProto {
			return nil, fmt.Errorf("file extension must be one of .textproto, .yaml, .binarypb: found %v", format)
		}
		var envConfig *env.Config
		var fileDescriptorSet *descpb.FileDescriptorSet
		switch format {
		case TextProto, BinaryProto:
			pbEnv := &configpb.Environment{}
			var err error
			if err = loadProtoFile(path, format, pbEnv); err != nil {
				return nil, err
			}
			envConfig, fileDescriptorSet, err = envProtoToConfig(pbEnv)
			if err != nil {
				return nil, err
			}
		case TextYAML:
			envConfig = &env.Config{}
			data, err := loadFile(path)
			if err != nil {
				return nil, err
			}
			err = yaml.Unmarshal(data, envConfig)
			if err != nil {
				return nil, fmt.Errorf("yaml.Unmarshal failed to map CEL environment: %w", err)
			}
		default:
			return nil, fmt.Errorf("unsupported format: %v, file: %s", format, path)
		}
		var envOpts []cel.EnvOption
		if fileDescriptorSet != nil {
			envOpts = append(envOpts, cel.TypeDescs(fileDescriptorSet))
		}
		envOpts = append(envOpts, cel.FromConfig(envConfig, ext.ExtensionOptionFactory))
		var err error
		e, err = e.Extend(envOpts...)
		if err != nil {
			return nil, fmt.Errorf("e.Extend() with env options %v failed: %w", envOpts, err)
		}
		return e, nil
	}
}

func envProtoToConfig(pbEnv *configpb.Environment) (*env.Config, *descpb.FileDescriptorSet, error) {
	if pbEnv == nil {