Source repositoryprocess.h

Go to the documentation of this file.

#pragma once
 
#ifndef __kernel_entry
  #define __kernel_entry
#endif
 
// standard includes
#include <optional>
#include <unordered_map>
 
// local includes
#include "config.h"
#include "platform/common.h"
#include "rtsp.h"
#include "src/boost_process_compat.h"
#include "utility.h"

#define DEFAULT_APP_IMAGE_PATH SUNSHINE_ASSETS_DIR "/box.png"
 
namespace proc {
  using file_t = util::safe_ptr_v2<FILE, int, &fclose>;

  using cmd_t = config::prep_cmd_t;

  struct ctx_t {
    std::vector<cmd_t> prep_cmds;  

    std::vector<std::string> detached;
 
    std::string name;  
    std::string cmd;  
    std::string working_dir;  
    std::string output;  
    std::string image_path;  
    std::string id;  
    bool elevated;  
    bool auto_detach;  
    bool wait_all;  
    std::chrono::seconds exit_timeout;  
  };

  class proc_t {
  public:
    KITTY_DEFAULT_CONSTR_MOVE_THROW(proc_t)
 
    
    proc_t(
      boost::process::v1::environment &&env,
      std::vector<ctx_t> &&apps
    ):
        _app_id(0),
        _env(std::move(env)),
        _apps(std::move(apps)) {
    }

    int execute(int app_id, std::shared_ptr<rtsp_stream::launch_session_t> launch_session);

    int running();
 
    ~proc_t();

    const std::vector<ctx_t> &get_apps() const;
    std::vector<ctx_t> &get_apps();
    std::string get_app_image(int app_id);
    std::string get_last_run_app_name();
    void terminate();

    void update_apps_and_env(proc_t &&other);
 
#ifdef SUNSHINE_TESTS
    const boost::process::v1::environment &get_env() const {
      return _env;
    }
#endif
 
  private:
    int _app_id;
 
    boost::process::v1::environment _env;
    std::vector<ctx_t> _apps;
    ctx_t _app;
    std::chrono::steady_clock::time_point _app_launch_time;
 
    // If no command associated with _app_id, yet it's still running
    bool placebo {};
 
    boost::process::v1::child _process;
    boost::process::v1::group _process_group;
 
    file_t _pipe;
    std::vector<cmd_t>::const_iterator _app_prep_it;
    std::vector<cmd_t>::const_iterator _app_prep_begin;
  };

  std::tuple<std::string, std::string> calculate_app_id(const std::string &app_name, std::string app_image_path, int index);

  std::string prepare_command(const std::string &command);
 
  bool check_valid_png(const std::filesystem::path &path);
  std::string validate_app_image_path(std::string app_image_path);
  void refresh(const std::string &file_name);
  std::optional<proc::proc_t> parse(const std::string &file_name);

  std::unique_ptr<platf::deinit_t> init();

  void terminate_process_group(boost::process::v1::child &child_process, boost::process::v1::group &group, std::chrono::seconds exit_timeout);
 
  extern proc_t proc;
}  // namespace proc