/*
 * Copyright 2025 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
 *
 *      http://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_PYTHON_PY_DESCRIPTOR_DATABASE_H_
#define THIRD_PARTY_CEL_PYTHON_PY_DESCRIPTOR_DATABASE_H_

#include <Python.h>  // IWYU pragma: keep - Needed for PyObject

#include <string_view>  // IWYU pragma: keep - Needed for string_view in OSS

#include "google/protobuf/descriptor.pb.h"
#include "absl/strings/string_view.h"
#include "google/protobuf/descriptor_database.h"

namespace cel_python {

// A DescriptorDatabase that uses a Python DescriptorPool to find descriptors.
class PyDescriptorDatabase : public google::protobuf::DescriptorDatabase {
 private:
  using StringViewArg = std::string_view;
 public:
  explicit PyDescriptorDatabase(PyObject* py_descriptor_pool);
  ~PyDescriptorDatabase() override;

  bool FindFileByName(StringViewArg filename,
                      google::protobuf::FileDescriptorProto* output) override;
  bool FindFileContainingSymbol(StringViewArg symbol_name,
                                google::protobuf::FileDescriptorProto* output) override;
  bool FindFileContainingExtension(
      StringViewArg containing_type, int field_number,
      google::protobuf::FileDescriptorProto* output) override;

 private:
  PyObject* py_descriptor_pool_;
  google::protobuf::DescriptorPool standard_pool_;
};

}  // namespace cel_python

#endif  // THIRD_PARTY_CEL_PYTHON_PY_DESCRIPTOR_DATABASE_H_