This commit is contained in:
2025-08-21 13:43:42 +08:00
parent 84ffd021a2
commit eb20c59347
6 changed files with 162 additions and 3 deletions

124
main.c Normal file
View File

@@ -0,0 +1,124 @@
#include <windows.h>
#include <stdio.h>
#include "embedded.h"
// 简化路径拼接宏
#define APPEND_PATH(base, file) (lstrcpyA(lstrcatA(lstrcpyA((base), (base)), "\\"), (file)))
// 检查文件是否存在
static BOOL file_exists(const char* filename) {
DWORD attrs = GetFileAttributesA(filename);
return (attrs != INVALID_FILE_ATTRIBUTES && !(attrs & FILE_ATTRIBUTE_DIRECTORY));
}
// 仅当文件不存在时保存资源
static BOOL save_resource_if_missing(const unsigned char* data, int len, const char* filepath) {
if (file_exists(filepath)) return TRUE;
HANDLE hFile = CreateFileA(filepath, GENERIC_WRITE, 0, NULL, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, NULL);
if (hFile == INVALID_HANDLE_VALUE) return FALSE;
DWORD written;
BOOL result = WriteFile(hFile, data, len, &written, NULL);
CloseHandle(hFile);
return result && (written == len);
}
// 检测 Python 是否可用
static BOOL CheckPythonAvailable() {
char python_path[MAX_PATH];
if (SearchPathA(NULL, "python.exe", NULL, MAX_PATH, python_path, NULL) == 0) {
return FALSE;
}
CHAR cmd[MAX_PATH + 32];
sprintf_s(cmd, sizeof(cmd), "\"%s\" --version", python_path);
STARTUPINFOA si = { sizeof(si) };
PROCESS_INFORMATION pi = { 0 };
if (!CreateProcessA(NULL, cmd, NULL, NULL, FALSE, CREATE_NO_WINDOW, NULL, NULL, &si, &pi)) {
return FALSE;
}
WaitForSingleObject(pi.hProcess, 5000);
CloseHandle(pi.hProcess);
CloseHandle(pi.hThread);
return TRUE;
}
// 安全地从完整路径中提取目录部分(不包含最后的反斜杠)
void GetDirectoryFromPath(char* path) {
char* last_slash = strrchr(path, '\\');
if (last_slash != NULL) {
*last_slash = '\0'; // 截断文件名
} else {
// 如果没有反斜杠,说明只有文件名,保留原路径或设为 "."
lstrcpyA(path, ".");
}
}
int WINAPI WinMain(HINSTANCE hInst, HINSTANCE h0, LPSTR cmd, int nShow) {
// 检查 Python
if (!CheckPythonAvailable()) {
MessageBoxA(NULL, "Python is not available. Please install Python first.", "Error", MB_OK | MB_ICONERROR);
return 1;
}
CHAR exe_dir[MAX_PATH] = { 0 };
CHAR cash_dir[MAX_PATH] = { 0 };
CHAR command[MAX_PATH * 2] = { 0 };
// 获取当前目录
GetModuleFileNameA(NULL, exe_dir, MAX_PATH);
GetDirectoryFromPath(exe_dir);
lstrcpyA(cash_dir, exe_dir);
lstrcatA(cash_dir, "\\cash");
// 创建 cash 目录
if (!CreateDirectoryA(cash_dir, NULL) && GetLastError() != ERROR_ALREADY_EXISTS) {
MessageBoxA(NULL, "Failed to create 'cash' directory.", "Error", MB_OK | MB_ICONERROR);
return 1;
}
// 定义资源项结构
struct {
const unsigned char* data;
int len;
const char* filename;
} resources[] = {
{ script_main, script_main_len, "\\init.py" },
{ data_zip, data_zip_len, "\\data.db.zip" },
{ cyzg_zip, cyzg_zip_len, "\\cyzg.zip" }
};
// 提取所有嵌入资源(如果不存在)
for (int i = 0; i < _countof(resources); ++i) {
CHAR filepath[MAX_PATH];
lstrcpyA(filepath, cash_dir);
lstrcatA(filepath, resources[i].filename);
if (!save_resource_if_missing(resources[i].data, resources[i].len, filepath)) {
MessageBoxA(NULL, "Failed to extract embedded resource.", "Error", MB_OK | MB_ICONERROR);
return 1;
}
}
// 判断是否已完成初始化
CHAR main_py[MAX_PATH], data_db[MAX_PATH];
lstrcpyA(main_py, cash_dir); lstrcatA(main_py, "\\main.py");
lstrcpyA(data_db, cash_dir); lstrcatA(data_db, "\\data.db");
const char* script_to_run = file_exists(main_py) && file_exists(data_db) ? "main.py" : "init.py";
// 构造并执行命令
snprintf(command, sizeof(command),
"cmd /c \"cd /d \"%s\" && python \"%s\"", cash_dir, script_to_run);
WinExec(command, nShow);
return 0;
}