process.hxx

#pragma once
 
#include <StormByte/system/visibility.h>
 
#include <filesystem>
#include <iostream>
#include <thread>
#ifdef LINUX
#include <unistd.h>
#else
#include <windows.h>
#endif
#include <vector>
 
namespace StormByte::System {
    class Pipe;                                             
    struct {} typedef _EoF;
 
    static constexpr const _EoF EoF = {};
 
    class STORMBYTE_SYSTEM_PUBLIC Process {
        public:
            Process(const std::filesystem::path& prog, const std::vector<std::string>& args = std::vector<std::string>());
 
            Process(std::filesystem::path&& prog, std::vector<std::string>&& args = std::vector<std::string>());
 
            Process(const Process& proc)                        = delete;
 
            Process(Process&& proc) noexcept;
 
            Process& operator=(const Process& proc)             = delete;
 
            Process& operator=(Process&& proc) noexcept;
 
            virtual ~Process() noexcept;
            #ifdef LINUX
            int                                                 Wait() noexcept;
 
            pid_t                                               Pid() noexcept;
            #else
            DWORD                                               Wait() noexcept;
 
            PROCESS_INFORMATION                                 Pid();
            #endif
            void                                                Suspend();
 
            void                                                Resume();
 
            Process&                                            operator>>(Process& proc);
 
            std::string&                                        operator>>(std::string& str) const;
 
            friend STORMBYTE_SYSTEM_PUBLIC std::ostream&        operator<<(std::ostream& ostream, const Process& proc);
 
            Process&                                            operator<<(const std::string& str);
 
            void                                                operator<<(const System::_EoF& eof);
 
            enum class Status:unsigned short {
                RUNNING,    
                SUSPENDED,  
                TERMINATED  
            };
 
        protected:
            Status m_status;                                    
            #ifdef LINUX
            pid_t m_pid;                                        
            #else
            STARTUPINFO m_siStartInfo;                          
            PROCESS_INFORMATION m_piProcInfo;                   
            #endif
            Pipe* m_pstdout;                                    
            Pipe* m_pstdin;                                     
            Pipe* m_pstderr;                                    
            std::filesystem::path m_program;                    
            std::vector<std::string> m_arguments;               
            std::unique_ptr<std::thread> m_forwarder;           
            
        private:
            void                                                Send(const std::string& str);
 
            void                                                Run();
 
            void                                                ConsumeAndForward(Process&);
            #ifdef WINDOWS
            std::wstring                                        FullCommand() const;
            #endif
    };
    STORMBYTE_SYSTEM_PUBLIC std::ostream&                       operator<<(std::ostream&, const Process&);
}