Added APIs to control device sleep

These APIs can be used to prevent device into sleep when sending supper big files.

PiperOrigin-RevId: 556809236

internal/platform/device_info.h

@@ -52,6 +52,9 @@ class DeviceInfo {
   virtual void UnregisterScreenLockedListener(
       absl::string_view listener_name) = 0;
 
+  virtual bool PreventSleep() = 0;
+  virtual bool AllowSleep() = 0;
+
   // Returns localized device name depends on device type.
   std::u16string GetDeviceTypeName() const {
     // TODO(b/230132370): return localized device name.

internal/platform/device_info_impl.cc

@@ -108,4 +108,10 @@ void DeviceInfoImpl::UnregisterScreenLockedListener(
   device_info_impl_->UnregisterScreenLockedListener(listener_name);
 }
 
+bool DeviceInfoImpl::PreventSleep() {
+  return device_info_impl_->PreventSleep();
+}
+
+bool DeviceInfoImpl::AllowSleep() { return device_info_impl_->AllowSleep(); }
+
 }  // namespace nearby

internal/platform/device_info_impl.h

@@ -53,6 +53,9 @@ class DeviceInfoImpl : public DeviceInfo {
       std::function<void(api::DeviceInfo::ScreenStatus)> callback) override;
   void UnregisterScreenLockedListener(absl::string_view listener_name) override;
 
+  bool PreventSleep() override;
+  bool AllowSleep() override;
+
  private:
   std::unique_ptr<api::DeviceInfo> device_info_impl_;
 };

internal/platform/implementation/apple/device_info.h

@@ -58,6 +58,20 @@ class DeviceInfo : public api::DeviceInfo {
       std::function<void(api::DeviceInfo::ScreenStatus)> callback) override;
 
   void UnregisterScreenLockedListener(absl::string_view listener_name) override;
+
+  // Request the system to prevent the device from going to sleep.
+  //
+  // To allow the system to sleep again, call @c AllowSleep().
+  //
+  // Returns @c true on success or if the platform does not allow preventing
+  // sleep. Returns @c false on any other error.
+  bool PreventSleep() override;
+
+  // Release any requests preventing the device from going to sleep.
+  //
+  // Returns @c true on success or if the system is already allowed to sleep.
+  // Returns @c false if an error occurred.
+  bool AllowSleep() override;
 };
 
 }  // namespace apple

internal/platform/implementation/apple/device_info.mm

@@ -168,5 +168,9 @@ void DeviceInfo::RegisterScreenLockedListener(
 
 void DeviceInfo::UnregisterScreenLockedListener(absl::string_view listener_name) {}
 
+bool DeviceInfo::PreventSleep() { return true; }
+
+bool DeviceInfo::AllowSleep() { return true; }
+
 }  // namespace apple
 }  // namespace nearby

internal/platform/implementation/device_info.h

@@ -59,6 +59,10 @@ class DeviceInfo {
       std::function<void(ScreenStatus)> callback) = 0;
   virtual void UnregisterScreenLockedListener(
       absl::string_view listener_name) = 0;
+
+  // Control device sleep
+  virtual bool PreventSleep() = 0;
+  virtual bool AllowSleep() = 0;
 };
 
 }  // namespace api

internal/platform/implementation/g3/device_info.h

@@ -38,7 +38,7 @@ class DeviceInfo : public api::DeviceInfo {
     return api::DeviceInfo::DeviceType::kLaptop;
   }
 
-  api::DeviceInfo::OsType GetOsType() const override{
+  api::DeviceInfo::OsType GetOsType() const override {
     return api::DeviceInfo::OsType::kChromeOs;
   }
 
@@ -92,6 +92,10 @@ class DeviceInfo : public api::DeviceInfo {
     screen_locked_listeners_.erase(listener_name);
   }
 
+  bool PreventSleep() override { return true; }
+
+  bool AllowSleep() override { return true; }
+
  private:
   absl::flat_hash_map<std::string,
                       std::function<void(api::DeviceInfo::ScreenStatus)>>

internal/platform/implementation/windows/device_info.cc

@@ -353,5 +353,15 @@ void DeviceInfo::UnregisterScreenLockedListener(
   session_manager_.UnregisterSessionListener(listener_name);
 }
 
+bool DeviceInfo::PreventSleep() {
+  absl::MutexLock lock(&mutex_);
+  return session_manager_.PreventSleep();
+}
+
+bool DeviceInfo::AllowSleep() {
+  absl::MutexLock lock(&mutex_);
+  return session_manager_.AllowSleep();
+}
+
 }  // namespace windows
 }  // namespace nearby

internal/platform/implementation/windows/device_info.h

@@ -52,6 +52,9 @@ class DeviceInfo : public api::DeviceInfo {
       std::function<void(api::DeviceInfo::ScreenStatus)> callback) override;
   void UnregisterScreenLockedListener(absl::string_view listener_name) override;
 
+  bool PreventSleep() override;
+  bool AllowSleep() override;
+
  private:
   mutable absl::Mutex mutex_;
   SessionManager session_manager_ ABSL_GUARDED_BY(mutex_);

internal/platform/implementation/windows/device_info_test.cc

@@ -77,6 +77,14 @@ TEST(DeviceInfo, DISABLED_IsScreenLocked) {
   EXPECT_FALSE(DeviceInfo().IsScreenLocked());
 }
 
+TEST(DeviceInfo, DISABLED_PreventSleep) {
+  EXPECT_TRUE(DeviceInfo().PreventSleep());
+}
+
+TEST(DeviceInfo, DISABLED_AllowSleep) {
+  EXPECT_TRUE(DeviceInfo().AllowSleep());
+}
+
 }  // namespace
 }  // namespace windows
 }  // namespace nearby

internal/test/fake_device_info.h

@@ -86,6 +86,10 @@ class FakeDeviceInfo : public DeviceInfo {
     screen_locked_listeners_.erase(listener_name);
   }
 
+  bool PreventSleep() override { return true; }
+
+  bool AllowSleep() override { return true; }
+
   int GetScreenLockedListenerCount() { return screen_locked_listeners_.size(); }
 
   // Mock methods.

internal/test/fake_device_info_test.cc

@@ -188,5 +188,15 @@ TEST(FakeDeviceInfo, UpdateScreenLockedListener) {
   EXPECT_EQ(screen_locked_tracker_2, api::DeviceInfo::ScreenStatus::kUnlocked);
 }
 
+TEST(FakeDeviceInfo, PreventSleep) {
+  FakeDeviceInfo device_info;
+  EXPECT_TRUE(device_info.PreventSleep());
+}
+
+TEST(FakeDeviceInfo, AllowSleep) {
+  FakeDeviceInfo device_info;
+  EXPECT_TRUE(device_info.AllowSleep());
+}
+
 }  // namespace
 }  // namespace nearby