content/browser/renderer_host/scoped_view_transition_resources.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 CONTENT_BROWSER_RENDERER_HOST_SCOPED_VIEW_TRANSITION_RESOURCES_H_
#define CONTENT_BROWSER_RENDERER_HOST_SCOPED_VIEW_TRANSITION_RESOURCES_H_

#include "base/functional/callback_helpers.h"
#include "content/common/content_export.h"
#include "content/public/browser/render_process_host.h"
#include "third_party/blink/public/common/tokens/tokens.h"

namespace content {
class RenderFrameHostImpl;

// For cross-document navigations associated with a ViewTransition, textures
// generated by the old Document are cached in the GPU process. The ownership of
// these textures is transferred from the old renderer to the browser and
// finally to the new renderer process.
//
// This class manages the lifetime of these resources to ensure they are
// released when ownership is with the browser process or if the renderer
// process owning the resources is terminated before the transition animations
// start in the new renderer process.
class CONTENT_EXPORT ScopedViewTransitionResources {
 public:
  explicit ScopedViewTransitionResources(
      const blink::ViewTransitionToken& transition_token,
      RenderProcessHost& render_process_host,
      bool delay_layer_tree_view_deletion);
  ~ScopedViewTransitionResources();

  const blink::ViewTransitionToken& transition_token() const {
    return transition_token_;
  }

  bool delay_layer_tree_view_deletion() const {
    return delay_layer_tree_view_deletion_;
  }

  void set_delay_layer_tree_view_deletion(bool delay_layer_tree_view_deletion) {
    delay_layer_tree_view_deletion_ = delay_layer_tree_view_deletion;
  }

  void MaybeDelayProcessShutdown(const base::TimeDelta& shutdown_delay,
                                 RenderFrameHostImpl& render_frame_host);

 private:
  void OnViewTransitionResourcesCaptured();
  void CancelDelayProcessShutdown();

  const blink::ViewTransitionToken transition_token_;
  bool is_resources_captured_ = false;
  ChildProcessId render_process_host_id_;
  bool delay_layer_tree_view_deletion_;
  base::ScopedClosureRunner process_shutdown_delay_runner_;
  base::WeakPtrFactory<ScopedViewTransitionResources> weak_factory_{this};
};

}  // namespace content

#endif  // CONTENT_BROWSER_RENDERER_HOST_SCOPED_VIEW_TRANSITION_RESOURCES_H_