#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);

    std::wstring GetPid() const;

    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