text: /*
   * Copyright (c) 2021, Idan Horowitz <idan.horowitz@serenityos.org>
   *
   * SPDX-License-Identifier: BSD-2-Clause
   */
  
  #include <LibJS/Runtime/FinalizationRegistry.h>
  
  namespace JS {
  
  FinalizationRegistry* FinalizationRegistry::create(GlobalObject& global_object, FunctionObject& cleanup_callback)
  {
      return global_object.heap().allocate<FinalizationRegistry>(global_object, cleanup_callback, *global_object.finalization_registry_prototype());
  }
  
  FinalizationRegistry::FinalizationRegistry(FunctionObject& cleanup_callback, Object& prototype)
      : Object(prototype)
      , WeakContainer(heap())
      , m_cleanup_callback(&cleanup_callback)
  {
  }
  
  FinalizationRegistry::~FinalizationRegistry()
  {
  }
  
  void FinalizationRegistry::add_finalization_record(Cell& target, Value held_value, Object* unregister_token)
  {
      VERIFY(!held_value.is_empty());
      m_records.append({ &target, held_value, unregister_token });
  }
  
  bool FinalizationRegistry::remove_by_token(Object& unregister_token)
  {
      auto removed = false;
      for (auto it = m_records.begin(); it != m_records.end(); ++it) {
          if (it->unregister_token == &unregister_token) {
              it.remove(m_records);
              removed = true;
          }
      }
      return removed;
  }
  
  void FinalizationRegistry::remove_sweeped_cells(Badge<Heap>, Vector<Cell*>& cells)
  {
      auto any_cells_were_sweeped = false;
      for (auto cell : cells) {
          for (auto& record : m_records) {
              if (record.target != cell)
                  continue;
              record.target = nullptr;
              any_cells_were_sweeped = true;
              break;
          }
      }
      if (any_cells_were_sweeped)
          vm().enqueue_finalization_registry_cleanup_job(*this);
  }
  
  // 9.13 CleanupFinalizationRegistry ( finalizationRegistry ), https://tc39.es/ecma262/#sec-cleanup-finalization-registry
  void FinalizationRegistry::cleanup(FunctionObject* callback)
  {
      auto& vm = this->vm();
      auto cleanup_callback = callback ?: m_cleanup_callback;
      for (auto it = m_records.begin(); it != m_records.end(); ++it) {
          if (it->target != nullptr)
              continue;
          (void)vm.call(*cleanup_callback, js_undefined(), it->held_value);
          it.remove(m_records);
          if (vm.exception())
              return;
      }
  }
  
  void FinalizationRegistry::visit_edges(Cell::Visitor& visitor)
  {
      Object::visit_edges(visitor);
      visitor.visit(m_cleanup_callback);
      for (auto& record : m_records) {
          visitor.visit(record.held_value);
          visitor.visit(record.unregister_token);
      }
  }
  
  }
  