process.hxx

/*
 * Copyright (C) 2024-2026 David C. Manuelda (StormBytePP)
 *
 * This file is part of StormByte-System.
 *
 * StormByte-System is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License version 3
 * or later, as published by the Free Software Foundation.
 *
 * StormByte-System is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with StormByte-System. If not, see
 * <https://www.gnu.org/licenses/lgpl-3.0.html>.
 */
 
#pragma once
 
#include <StormByte/system/visibility.h>
 
#include <filesystem>
#include <iostream>
#include <memory>
#include <thread>
#ifdef WINDOWS
#include <windows.h>
#else
#include <unistd.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 UNIX
            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;
 
            std::string& Stderr(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 UNIX
            pid_t m_pid;                                        
            #else
            STARTUPINFOW m_siStartInfo;                         
            PROCESS_INFORMATION m_piProcInfo;                   
            #endif
            std::unique_ptr<Pipe> m_pstdout;                    
            std::unique_ptr<Pipe> m_pstdin;                     
            std::unique_ptr<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& exec);
 
            void ReleaseOwnership() noexcept;
 
            #ifdef WINDOWS
            std::wstring FullCommand() const;
            #endif
    };
 
    STORMBYTE_SYSTEM_PUBLIC std::ostream& operator<<(std::ostream& ostream, const Process& proc);
}