// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef EXTENSIONS_RENDERER_EXTENSION_JS_RUNNER_H_
#define EXTENSIONS_RENDERER_EXTENSION_JS_RUNNER_H_

#include "base/memory/raw_ptr.h"
#include "base/memory/weak_ptr.h"
#include "base/time/time.h"
#include "extensions/renderer/bindings/js_runner.h"
#include "v8/include/v8-forward.h"

namespace extensions {
class ScriptContext;

// Implementation of a JSRunner that handles possible JS suspension.
class ExtensionJSRunner : public JSRunner {
 public:
  explicit ExtensionJSRunner(ScriptContext* script_context);

  ExtensionJSRunner(const ExtensionJSRunner&) = delete;
  ExtensionJSRunner& operator=(const ExtensionJSRunner&) = delete;

  ~ExtensionJSRunner() override;

  // JSRunner:
  void RunJSFunction(v8::Local<v8::Function> function,
                     v8::Local<v8::Context> context,
                     base::span<v8::Local<v8::Value>> args,
                     ResultCallback callback) override;
  v8::MaybeLocal<v8::Value> RunJSFunctionSync(
      v8::Local<v8::Function> function,
      v8::Local<v8::Context> context,
      base::span<v8::Local<v8::Value>> args) override;

 private:
  // Called with the result of executing the JS function.
  void OnFunctionComplete(ResultCallback,
                          std::optional<base::Value> value,
                          base::TimeTicks start_time);

  // The associated ScriptContext. Guaranteed to outlive this object.
  const raw_ptr<ScriptContext, DanglingUntriaged> script_context_;

  base::WeakPtrFactory<ExtensionJSRunner> weak_factory_{this};
};

}  // namespace extensions

#endif  // EXTENSIONS_RENDERER_EXTENSION_JS_RUNNER_H_