// Copyright 2022 Google LLC

//

// Licensed under the Apache License, Version 2.0 (the "License");

// you may not use this file except in compliance with the License.

// You may obtain a copy of the License at

//

//     https://www.apache.org/licenses/LICENSE-2.0

//

// Unless required by applicable law or agreed to in writing, software

// distributed under the License is distributed on an "AS IS" BASIS,

// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

// See the License for the specific language governing permissions and

// limitations under the License.

#ifndef THIRD_PARTY_CEL_CPP_EVAL_INTERNAL_ADAPTER_ACTIVATION_IMPL_H_

#define THIRD_PARTY_CEL_CPP_EVAL_INTERNAL_ADAPTER_ACTIVATION_IMPL_H_

#include <vector>

#include "absl/base/nullability.h"

#include "absl/status/statusor.h"

#include "absl/strings/string_view.h"

#include "absl/types/span.h"

#include "base/attribute.h"

#include "common/value.h"

#include "eval/public/base_activation.h"

#include "runtime/activation_interface.h"

#include "runtime/function_overload_reference.h"

#include "runtime/internal/attribute_matcher.h"

#include "google/protobuf/arena.h"

#include "google/protobuf/descriptor.h"

#include "google/protobuf/message.h"

namespace cel::interop_internal {

// An Activation implementation that adapts the legacy version (based on

// expr::CelValue) to the new cel::Handle based version. This implementation

// must be scoped to an evaluation.

class AdapterActivationImpl : public ActivationInterface {

 public:

  explicit AdapterActivationImpl(

      const google::api::expr::runtime::BaseActivation& legacy_activation)

      : legacy_activation_(legacy_activation) {}

  absl::StatusOr<bool> FindVariable(

      absl::string_view name,

      const google::protobuf::DescriptorPool* absl_nonnull descriptor_pool,

      google::protobuf::MessageFactory* absl_nonnull message_factory,

      google::protobuf::Arena* absl_nonnull arena,

      Value* absl_nonnull result) const override;

  std::vector<FunctionOverloadReference> FindFunctionOverloads(

      absl::string_view name) const override;

  absl::Span<const cel::AttributePattern> GetUnknownAttributes() const override;

  absl::Span<const cel::AttributePattern> GetMissingAttributes() const override;

 private:

  const runtime_internal::AttributeMatcher* absl_nullable GetAttributeMatcher()

      const override;

  const google::api::expr::runtime::BaseActivation& legacy_activation_;

};

}  // namespace cel::interop_internal

#endif  // THIRD_PARTY_CEL_CPP_EVAL_INTERNAL_ADAPTER_ACTIVATION_IMPL_H_