content/browser/wake_lock/wake_lock_context_host.h

// 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 CONTENT_BROWSER_WAKE_LOCK_WAKE_LOCK_CONTEXT_HOST_H_
#define CONTENT_BROWSER_WAKE_LOCK_WAKE_LOCK_CONTEXT_HOST_H_

#include "base/memory/raw_ptr.h"
#include "content/public/browser/web_contents.h"
#include "mojo/public/cpp/bindings/remote.h"
#include "services/device/public/mojom/wake_lock_context.mojom.h"
#include "services/device/public/mojom/wake_lock_provider.mojom.h"
#include "ui/gfx/native_widget_types.h"

namespace content {

// On Android, WakeLockContext requires the NativeView associated with the
// context in order to lock the screen. WakeLockContextHost provides this
// functionality by mapping WakeLockContext IDs to the WebContents associated
// with those IDs.
class WakeLockContextHost {
 public:
  explicit WakeLockContextHost(WebContents* web_contents);

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

  ~WakeLockContextHost();

  // This callback is passed into the DeviceService constructor in order to
  // enable WakeLockContext to map a context ID to a Native View as necessary.
  static gfx::NativeView GetNativeViewForContext(int context_id);

  // Returns the WakeLockContext* to which this instance is connected.
  device::mojom::WakeLockContext* GetWakeLockContext() {
    return wake_lock_context_ ? wake_lock_context_.get() : nullptr;
  }

 private:
  // This instance's ID.
  int id_;

  // The WebContents that owns this instance.
  raw_ptr<WebContents> web_contents_;

  // The WakeLockContext instance that is connected to this instance.
  mojo::Remote<device::mojom::WakeLockContext> wake_lock_context_;
};

}  // namespace content

#endif  // CONTENT_BROWSER_WAKE_LOCK_WAKE_LOCK_CONTEXT_HOST_H_