136 lines
3.6 KiB
C++
136 lines
3.6 KiB
C++
#ifndef CLIPROCESS_H
|
|
#define CLIPROCESS_H
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include <mutex>
|
|
#include <thread>
|
|
#include <map>
|
|
|
|
#ifdef _WIN32
|
|
#include <windows.h>
|
|
#else
|
|
#include <unistd.h>
|
|
#include <sys/wait.h>
|
|
#include <signal.h>
|
|
#include <fcntl.h>
|
|
#include <errno.h>
|
|
#endif
|
|
|
|
// 输出编码枚举
|
|
enum class OutputEncoding {
|
|
AUTO_DETECT = 0,
|
|
UTF8,
|
|
#ifdef _WIN32
|
|
GBK,
|
|
GB2312,
|
|
BIG5,
|
|
SHIFT_JIS,
|
|
#else
|
|
// Unix/Linux 常见编码
|
|
ISO_8859_1,
|
|
GB18030,
|
|
BIG5,
|
|
EUC_JP,
|
|
#endif
|
|
};
|
|
|
|
class CLIProcess {
|
|
public:
|
|
CLIProcess();
|
|
~CLIProcess();
|
|
|
|
void SetMaxLogLines(int max_lines);
|
|
void SetStopCommand(const std::string& command, int timeout_ms = 5000);
|
|
void SetEnvironmentVariables(const std::map<std::string, std::string>& env_vars);
|
|
void SetOutputEncoding(OutputEncoding encoding);
|
|
|
|
// 新增：工作目录设置
|
|
void SetWorkingDirectory(const std::string& working_dir);
|
|
std::string GetWorkingDirectory() const;
|
|
|
|
void Start(const std::string& command);
|
|
void Stop();
|
|
void Restart(const std::string& command);
|
|
|
|
void ClearLogs();
|
|
void AddLog(const std::string& log);
|
|
const std::vector<std::string>& GetLogs() const;
|
|
|
|
bool SendCommand(const std::string& command);
|
|
void CopyLogsToClipboard() const;
|
|
|
|
bool IsRunning() const;
|
|
|
|
// 环境变量管理接口
|
|
const std::map<std::string, std::string>& GetEnvironmentVariables() const;
|
|
void AddEnvironmentVariable(const std::string& key, const std::string& value);
|
|
void RemoveEnvironmentVariable(const std::string& key);
|
|
void ClearEnvironmentVariables();
|
|
|
|
// 编码相关接口
|
|
OutputEncoding GetOutputEncoding() const;
|
|
static std::string GetEncodingName(OutputEncoding encoding);
|
|
static std::vector<std::pair<OutputEncoding, std::string>> GetSupportedEncodings();
|
|
|
|
// 新增：工作目录相关静态方法
|
|
static std::string ExtractDirectoryFromCommand(const std::string& command);
|
|
static std::string GetAbsolutePath(const std::string& path);
|
|
static bool DirectoryExists(const std::string& path);
|
|
|
|
private:
|
|
void ReadOutput();
|
|
void CloseProcessHandles();
|
|
void CleanupResources();
|
|
|
|
// 编码转换相关方法
|
|
static std::string ConvertToUTF8(std::string& input, OutputEncoding encoding);
|
|
std::string DetectAndConvertToUTF8(std::string& input);
|
|
|
|
#ifdef _WIN32
|
|
static UINT GetCodePageFromEncoding(OutputEncoding encoding);
|
|
static bool IsValidUTF8(const std::string& str);
|
|
|
|
PROCESS_INFORMATION pi_{};
|
|
HANDLE hReadPipe_{};
|
|
HANDLE hWritePipe_{};
|
|
HANDLE hReadPipe_stdin_{};
|
|
HANDLE hWritePipe_stdin_;
|
|
#else
|
|
// Unix/Linux 进程管理
|
|
pid_t process_pid_;
|
|
int pipe_stdout_[2];
|
|
int pipe_stdin_[2];
|
|
bool process_running_;
|
|
|
|
// Unix 编码转换辅助函数
|
|
std::string ConvertUnixEncoding(const std::string& input, const std::string& from_encoding);
|
|
static std::string GetUnixEncodingName(OutputEncoding encoding);
|
|
#endif
|
|
|
|
mutable std::mutex logs_mutex_;
|
|
std::vector<std::string> logs_;
|
|
int max_log_lines_;
|
|
|
|
std::thread output_thread_;
|
|
|
|
// 停止命令相关
|
|
mutable std::mutex stop_mutex_;
|
|
std::string stop_command_;
|
|
int stop_timeout_ms_;
|
|
|
|
// 环境变量相关
|
|
mutable std::mutex env_mutex_;
|
|
std::map<std::string, std::string> environment_variables_;
|
|
|
|
// 编码相关
|
|
mutable std::mutex encoding_mutex_;
|
|
OutputEncoding output_encoding_;
|
|
|
|
// 新增：工作目录相关
|
|
mutable std::mutex working_dir_mutex_;
|
|
std::string working_directory_;
|
|
bool use_auto_working_dir_;
|
|
};
|
|
|
|
#endif // CLIPROCESS_H
|