lewaimai_pos_windows/lewaimai_pos_windows_server/helper/CSystem.h

#pragma once

#include "../pch/pch.h"

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

	static int get_CPU_core_num();

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

#ifdef _WIN32
	// 程序开机自动启动
	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、添加一个子Key,并设置值，"GISRestart"是应用程序名字（不加后缀.exe）
			RegSetValueEx(hKey, L"乐外卖接单软件", 0, REG_SZ, (LPBYTE)strExeFullDir, (lstrlen(strExeFullDir) + 1) * sizeof(TCHAR));

			//4、关闭注册表
			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);
		}
	}

	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 BOOL IsFileExist(const wstring& csFile)
	{
		DWORD dwAttrib = GetFileAttributes(csFile.c_str());
		return INVALID_FILE_ATTRIBUTES != dwAttrib && 0 == (dwAttrib & FILE_ATTRIBUTE_DIRECTORY);
	}

	// 判断文件夹是否存在
	static BOOL IsDirExist(const wstring& csDir)
	{
		DWORD dwAttrib = GetFileAttributes(csDir.c_str());
		return INVALID_FILE_ATTRIBUTES != dwAttrib && 0 != (dwAttrib & FILE_ATTRIBUTE_DIRECTORY);
	}

	// 判断文件或文件夹是否存在
	static BOOL IsPathExist(const wstring& csPath)
	{
		DWORD dwAttrib = GetFileAttributes(csPath.c_str());
		return INVALID_FILE_ATTRIBUTES != dwAttrib;
	}
#else
	static std::string getAbsopath()
	{
		char buf[1024];
		size_t size = 1024;

		size_t i;
		size_t len = readlink("/proc/self/exe", buf, size - 1);

		if (len < 0 || (len >= size - 1))
		{
			return NULL;
		}

		buf[len] = '\0';
		for (i = len; i >= 0; i--)
		{
			if (buf[i] == '/')
			{
				buf[i + 1] = '\0';
				break;
			}
		}

		return buf;
	}
#endif
};