third_party/blink/renderer/modules/ai/ai_rewriter.h

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

#ifndef THIRD_PARTY_BLINK_RENDERER_MODULES_AI_AI_REWRITER_H_
#define THIRD_PARTY_BLINK_RENDERER_MODULES_AI_AI_REWRITER_H_

#include "mojo/public/cpp/bindings/pending_remote.h"
#include "third_party/blink/public/mojom/ai/ai_rewriter.mojom-blink.h"
#include "third_party/blink/renderer/bindings/core/v8/script_promise.h"
#include "third_party/blink/renderer/bindings/modules/v8/v8_ai_rewriter_length.h"
#include "third_party/blink/renderer/bindings/modules/v8/v8_ai_rewriter_tone.h"
#include "third_party/blink/renderer/core/execution_context/execution_context_lifecycle_observer.h"
#include "third_party/blink/renderer/platform/bindings/script_wrappable.h"
#include "third_party/blink/renderer/platform/mojo/heap_mojo_remote.h"
#include "third_party/blink/renderer/platform/wtf/text/wtf_string.h"

namespace base {
class SequencedTaskRunner;
}  // namespace base

namespace blink {

class AIRewriterRewriteOptions;
class ExecutionContext;
class ReadableStream;

// The class that represents a rewriter object.
class AIRewriter final : public ScriptWrappable, public ExecutionContextClient {
  DEFINE_WRAPPERTYPEINFO();

 public:
  AIRewriter(ExecutionContext* execution_context,
             scoped_refptr<base::SequencedTaskRunner> task_runner,
             mojo::PendingRemote<mojom::blink::AIRewriter> pending_remote,
             const String& shared_context_string,
             const V8AIRewriterTone& tone,
             const V8AIRewriterLength& length);
  void Trace(Visitor* visitor) const override;

  // ai_rewriter.idl implementation.
  ScriptPromise<IDLString> rewrite(ScriptState* script_state,
                                   const String& input,
                                   const AIRewriterRewriteOptions* options,
                                   ExceptionState& exception_state);
  ReadableStream* rewriteStreaming(ScriptState* script_state,
                                   const String& input,
                                   const AIRewriterRewriteOptions* options,
                                   ExceptionState& exception_state);
  void destroy(ScriptState* script_state, ExceptionState& exception_state);
  String sharedContext() const { return shared_context_string_; }
  const V8AIRewriterTone& tone() const { return tone_; }
  const V8AIRewriterLength& length() const { return length_; }

 private:
  scoped_refptr<base::SequencedTaskRunner> task_runner_;
  HeapMojoRemote<mojom::blink::AIRewriter> remote_;
  const String shared_context_string_;
  const V8AIRewriterTone tone_;
  const V8AIRewriterLength length_;
};

}  // namespace blink

#endif  // THIRD_PARTY_BLINK_RENDERER_MODULES_AI_AI_REWRITER_H_