linux_power.hpp

#pragma once
 
#include <unistd.h>  // for geteuid()
 
#include <cstdlib>    // for system()
#include <stdexcept>  // for std::runtime_error
 
#include "libxr.hpp"
#include "logger.hpp"
#include "power.hpp"
 
namespace LibXR
{
 
class LinuxPowerManager : public PowerManager
{
 public:
  LinuxPowerManager() = default;
 
  void Reset() override
  {
    CheckRoot();
    int ret = std::system("reboot");
    if (ret != 0)
    {
      throw std::runtime_error("Failed to reboot system");
    }
  }
 
  void Shutdown() override
  {
    CheckRoot();
    int ret = std::system("poweroff");
    if (ret != 0)
    {
      throw std::runtime_error("Failed to shut down system");
    }
  }
 
 private:
  void CheckRoot()
  {
    if (geteuid() != 0)
    {
      XR_LOG_WARN("Must run as root");
    }
  }
};
 
}  // namespace LibXR