#pragma once

class CSystem
{
public:
    CSystem();
    ~CSystem();

    static int get_CPU_core_num();

    //程序休眠X秒
    static void my_sleep(int second);

    static std::wstring GetProgramDir()
    {
        wchar_t exeFullPath[MAX_PATH]; // Full path
        std::wstring strPath = L"";

        GetModuleFileName(NULL, exeFullPath, MAX_PATH);
        strPath = (wstring)exeFullPath;    // Get full path of the file

        int pos = strPath.find_last_of('\\', strPath.length());
        return strPath.substr(0, pos);  // Return the directory without the file name
    }

    // 程序开机自动启动
    static void autostart()
    {
        HKEY hKey;
        wstring strRegPath = L"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Run";

        //1、找到系统的启动项
        if(RegOpenKeyEx(HKEY_CURRENT_USER, strRegPath.c_str(), 0, KEY_ALL_ACCESS, &hKey) == ERROR_SUCCESS)  ///打开启动项
        {
            //2、得到本程序自身的全路径
            TCHAR strExeFullDir[MAX_PATH];
            GetModuleFileName(NULL, strExeFullDir, MAX_PATH);

            //3、判断注册表项是否已经存在
            TCHAR strDir[MAX_PATH] = {};
            DWORD nLength = MAX_PATH;
            long result = RegGetValue(hKey, nullptr, L"智铺子收银软件", RRF_RT_REG_SZ, 0, strDir, &nLength);

            //4、已经存在
            if(result != ERROR_SUCCESS || _tcscmp(strExeFullDir, strDir) != 0)
            {
                //5、添加一个子Key,并设置值，"GISRestart"是应用程序名字（不加后缀.exe）
                RegSetValueEx(hKey, L"智铺子收银软件", 0, REG_SZ, (LPBYTE)strExeFullDir, (lstrlen(strExeFullDir) + 1) * sizeof(TCHAR));

                //6、关闭注册表
                RegCloseKey(hKey);
            }
        }
        else
        {
            cout << "系统参数错误, 不能随系统启动";
        }
    }

    // 取消开机自动启动
    static void cancelAutoStart()
    {
        HKEY hKey;
        wstring strRegPath = L"SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Run";

        //1、找到系统的启动项
        if(RegOpenKeyEx(HKEY_CURRENT_USER, strRegPath.c_str(), 0, KEY_ALL_ACCESS, &hKey) == ERROR_SUCCESS)
        {
            //2、删除值
            RegDeleteValue(hKey, L"智铺子收银软件");

            //3、关闭注册表
            RegCloseKey(hKey);
        }
    }
};