// Copyright 2016 the V8 project authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef V8_TEST_INSPECTOR_PROTOCOL_INSPECTOR_IMPL_H_
#define V8_TEST_INSPECTOR_PROTOCOL_INSPECTOR_IMPL_H_

#include "include/v8-inspector.h"
#include "include/v8.h"
#include "src/base/macros.h"
#include "src/base/platform/platform.h"
#include "test/inspector/task-runner.h"

class InspectorClientImpl : public v8_inspector::V8InspectorClient {
 public:
  class FrontendChannel {
   public:
    virtual ~FrontendChannel() = default;
    virtual void SendMessageToFrontend(
        const v8_inspector::StringView& message) = 0;
  };

  InspectorClientImpl(TaskRunner* task_runner,
                      FrontendChannel* frontend_channel,
                      v8::base::Semaphore* ready_semaphore);
  virtual ~InspectorClientImpl();

  void scheduleReconnect(v8::base::Semaphore* ready_semaphore);
  void scheduleCreateContextGroup(
      TaskRunner::SetupGlobalTasks setup_global_tasks,
      v8::base::Semaphore* ready_semaphore, int* context_group_id);

  static v8_inspector::V8Inspector* InspectorFromContext(
      v8::Local<v8::Context> context);
  static v8_inspector::V8InspectorSession* SessionFromContext(
      v8::Local<v8::Context> context);

  // context_group_id = 0 means default context group.
  v8_inspector::V8InspectorSession* session(int context_group_id = 0);

  void setCurrentTimeMSForTest(double time);
  void setMemoryInfoForTest(v8::Local<v8::Value> memory_info);
  void setLogConsoleApiMessageCalls(bool log);

 private:
  // V8InspectorClient implementation.
  bool formatAccessorsAsProperties(v8::Local<v8::Value>) override;
  v8::Local<v8::Context> ensureDefaultContextInGroup(
      int context_group_id) override;
  double currentTimeMS() override;
  v8::MaybeLocal<v8::Value> memoryInfo(v8::Isolate* isolate,
                                       v8::Local<v8::Context>) override;
  void runMessageLoopOnPause(int context_group_id) override;
  void quitMessageLoopOnPause() override;
  void consoleAPIMessage(int contextGroupId,
                         v8::Isolate::MessageErrorLevel level,
                         const v8_inspector::StringView& message,
                         const v8_inspector::StringView& url,
                         unsigned lineNumber, unsigned columnNumber,
                         v8_inspector::V8StackTrace*) override;
  friend class SendMessageToBackendTask;

  friend class ConnectTask;
  void connect(v8::Local<v8::Context> context);
  friend class DisconnectTask;
  void disconnect();
  friend class CreateContextGroupTask;
  int createContextGroup(
      const TaskRunner::SetupGlobalTasks& setup_global_tasks);

  std::unique_ptr<v8_inspector::V8Inspector> inspector_;
  std::unique_ptr<v8_inspector::V8Inspector::Channel> channel_;

  std::map<int, std::unique_ptr<v8_inspector::V8InspectorSession>> sessions_;
  std::map<int, std::unique_ptr<v8_inspector::StringBuffer>> states_;

  v8::Isolate* isolate_;
  v8::Global<v8::Value> memory_info_;

  TaskRunner* task_runner_;
  FrontendChannel* frontend_channel_;

  bool current_time_set_for_test_ = false;
  double current_time_ = 0.0;
  bool log_console_api_message_calls_ = false;

  DISALLOW_COPY_AND_ASSIGN(InspectorClientImpl);
};

class SendMessageToBackendExtension : public TaskRunner::SetupGlobalTask {
 public:
  void Run(v8::Isolate* isolate, v8::Local<v8::ObjectTemplate> global) override;

  static void set_backend_task_runner(TaskRunner* task_runner) {
    backend_task_runner_ = task_runner;
  }

 private:
  static void SendMessageToBackend(
      const v8::FunctionCallbackInfo<v8::Value>& args);

  static TaskRunner* backend_task_runner_;
};

#endif  //  V8_TEST_INSPECTOR_PROTOCOL_INSPECTOR_IMPL_H_