Merge branch 'master' of https://github.com/Fluorohydride/ygopro

gframe/gframe.cpp

@@ -12,9 +12,6 @@ bool open_file = false;
 wchar_t open_file_name[256] = L"";
 bool bot_mode = false;
 
-void GetParameterW(wchar_t* param, const char* arg) {
-	BufferIO::DecodeUTF8(arg, param);
-}
 void ClickButton(irr::gui::IGUIElement* btn) {
 	irr::SEvent event;
 	event.EventType = irr::EET_GUI_EVENT;
@@ -60,166 +57,113 @@ int main(int argc, char* argv[]) {
 		return 0;
 
 #ifdef _WIN32
-	wchar_t* command = GetCommandLineW();
-	char buffer[2048];
-	BufferIO::EncodeUTF8(command, buffer);
-	char* ptr = buffer;
-	argc = 0;
-	int j = 0;
-	bool in_QM = false, in_TEXT = false, in_SPACE = true;
-	while(*ptr) {
-		if(in_QM) {
-			if(*ptr == '\"')
-				in_QM = false;
-			else
-				++j;
-		} else {
-			switch(*ptr) {
-			case '\"': {
-				in_QM = true;
-				in_TEXT = true;
-				if(in_SPACE) {
-					argv[argc] = ptr + 1;
-					j = 0;
-				}
-				in_SPACE = false;
-				break;
-			}
-			case ' ':
-			case '\t':
-			case '\n':
-			case '\r': {
-				if(in_TEXT) {
-					argv[argc][j] = '\0';
-					++argc;
-				}
-				in_TEXT = false;
-				in_SPACE = true;
-				break;
-			}
-			default: {
-				in_TEXT = true;
-				if(in_SPACE) {
-					argv[argc] = ptr;
-					j = 1;
-				} else
-					++j;
-				in_SPACE = false;
-				break;
-			}
-			}
-		}
-		++ptr;
-	}
-	if(in_TEXT) {
-		argv[argc][j] = '\0';
-		++argc;
+	int wargc;
+	wchar_t** wargv = CommandLineToArgvW(GetCommandLineW(), &wargc);
+#else
+	int wargc = argc;
+	wchar_t (*wargv)[256] = new wchar_t[wargc][256];
+	for(int i = 0; i < argc; ++i) {
+		BufferIO::DecodeUTF8(argv[i], wargv[i]);
 	}
-	argv[argc] = NULL;
-#endif // _WIN32
+#endif //_WIN32
 
 	bool keep_on_return = false;
-	for(int i = 1; i < argc; ++i) {
-		if(argv[i][0] == '-' && argv[i][1] == 'e' && argv[i][2]) {
-			ygo::dataManager.LoadDB(&argv[i][2]);
+	for(int i = 1; i < wargc; ++i) {
+		if(wargv[i][0] == L'-' && wargv[i][1] == L'e' && wargv[i][2] != L'\0') {
+			char param[128];
+			BufferIO::EncodeUTF8(&wargv[i][2], param);
+			ygo::dataManager.LoadDB(param);
 			continue;
 		}
-		if(!strcmp(argv[i], "-e")) { // extra database
+		if(!wcscmp(wargv[i], L"-e")) { // extra database
 			++i;
-			if(i < argc)
-				ygo::dataManager.LoadDB(&argv[i][0]);
+			if(i < wargc) {
+				char param[128];
+				BufferIO::EncodeUTF8(wargv[i], param);
+				ygo::dataManager.LoadDB(param);
+			}
 			continue;
-		} else if(!strcmp(argv[i], "-n")) { // nickName
+		} else if(!wcscmp(wargv[i], L"-n")) { // nickName
 			++i;
-			if(i < argc) {
-				wchar_t param[128];
-				GetParameterW(param, &argv[i][0]);
-				ygo::mainGame->ebNickName->setText(param);
-			}
+			if(i < wargc)
+				ygo::mainGame->ebNickName->setText(wargv[i]);
 			continue;
-		} else if(!strcmp(argv[i], "-h")) { // Host address
+		} else if(!wcscmp(wargv[i], L"-h")) { // Host address
 			++i;
-			if(i < argc) {
-				wchar_t param[128];
-				GetParameterW(param, &argv[i][0]);
-				ygo::mainGame->ebJoinHost->setText(param);
-			}
+			if(i < wargc)
+				ygo::mainGame->ebJoinHost->setText(wargv[i]);
 			continue;
-		} else if(!strcmp(argv[i], "-p")) { // host Port
+		} else if(!wcscmp(wargv[i], L"-p")) { // host Port
 			++i;
-			if(i < argc) {
-				wchar_t param[128];
-				GetParameterW(param, &argv[i][0]);
-				ygo::mainGame->ebJoinPort->setText(param);
-			}
+			if(i < wargc)
+				ygo::mainGame->ebJoinPort->setText(wargv[i]);
 			continue;
-		} else if(!strcmp(argv[i], "-w")) { // host passWord
+		} else if(!wcscmp(wargv[i], L"-w")) { // host passWord
 			++i;
-			if(i < argc) {
-				wchar_t param[128];
-				GetParameterW(param, &argv[i][0]);
-				ygo::mainGame->ebJoinPass->setText(param);
-			}
+			if(i < wargc)
+				ygo::mainGame->ebJoinPass->setText(wargv[i]);
 			continue;
-		} else if(!strcmp(argv[i], "-k")) { // Keep on return
+		} else if(!wcscmp(wargv[i], L"-k")) { // Keep on return
 			exit_on_return = false;
 			keep_on_return = true;
-		} else if(!strcmp(argv[i], "-d")) { // Deck
-			if(i + 2 < argc) { // select deck
+		} else if(!wcscmp(wargv[i], L"-d")) { // Deck
 			++i;
-				GetParameterW(ygo::mainGame->gameConf.lastdeck, &argv[i][0]);
+			if(i + 1 < wargc) { // select deck
+				wcscpy(ygo::mainGame->gameConf.lastdeck, wargv[i]);
 				continue;
 			} else { // open deck
 				exit_on_return = !keep_on_return;
-				if(i + 1 < argc) {
+				if(i < wargc) {
 					open_file = true;
-					GetParameterW(open_file_name, &argv[i + 1][0]);
+					wcscpy(open_file_name, wargv[i]);
 				}
 				ClickButton(ygo::mainGame->btnDeckEdit);
 				break;
 			}
-		} else if(!strcmp(argv[i], "-c")) { // Create host
+		} else if(!wcscmp(wargv[i], L"-c")) { // Create host
 			exit_on_return = !keep_on_return;
 			ygo::mainGame->HideElement(ygo::mainGame->wMainMenu);
 			ClickButton(ygo::mainGame->btnHostConfirm);
 			break;
-		} else if(!strcmp(argv[i], "-j")) { // Join host
+		} else if(!wcscmp(wargv[i], L"-j")) { // Join host
 			exit_on_return = !keep_on_return;
 			ygo::mainGame->HideElement(ygo::mainGame->wMainMenu);
 			ClickButton(ygo::mainGame->btnJoinHost);
 			break;
-		} else if(!strcmp(argv[i], "-r")) { // Replay
+		} else if(!wcscmp(wargv[i], L"-r")) { // Replay
 			exit_on_return = !keep_on_return;
-			if(i + 1 < argc) {
+			++i;
+			if(i < wargc) {
 				open_file = true;
-				GetParameterW(open_file_name, &argv[i + 1][0]);
+				wcscpy(open_file_name, wargv[i]);
 			}
 			ClickButton(ygo::mainGame->btnReplayMode);
 			if(open_file)
 				ClickButton(ygo::mainGame->btnLoadReplay);
 			break;
-		} else if(!strcmp(argv[i], "-s")) { // Single
+		} else if(!wcscmp(wargv[i], L"-s")) { // Single
 			exit_on_return = !keep_on_return;
-			if(i + 1 < argc) {
+			++i;
+			if(i < wargc) {
 				open_file = true;
-				GetParameterW(open_file_name, &argv[i + 1][0]);
+				wcscpy(open_file_name, wargv[i]);
 			}
 			ClickButton(ygo::mainGame->btnSingleMode);
 			if(open_file)
 				ClickButton(ygo::mainGame->btnLoadSinglePlay);
 			break;
-		} else if(argc == 2 && strlen(argv[1]) >= 4) {
-			char* pstrext = argv[1] + strlen(argv[1]) - 4;
-			if(!mystrncasecmp(pstrext, ".ydk", 4)) {
+		} else if(wargc == 2 && wcslen(wargv[1]) >= 4) {
+			wchar_t* pstrext = wargv[1] + wcslen(wargv[1]) - 4;
+			if(!mywcsncasecmp(pstrext, L".ydk", 4)) {
 				open_file = true;
-				GetParameterW(open_file_name, &argv[1][0]);
+				wcscpy(open_file_name, wargv[i]);
 				exit_on_return = !keep_on_return;
 				ClickButton(ygo::mainGame->btnDeckEdit);
 				break;
 			}
-			if(!mystrncasecmp(pstrext, ".yrp", 4)) {
+			if(!mywcsncasecmp(pstrext, L".yrp", 4)) {
 				open_file = true;
-				GetParameterW(open_file_name, &argv[1][0]);
+				wcscpy(open_file_name, wargv[i]);
 				exit_on_return = !keep_on_return;
 				ClickButton(ygo::mainGame->btnReplayMode);
 				ClickButton(ygo::mainGame->btnLoadReplay);
@@ -227,6 +171,11 @@ int main(int argc, char* argv[]) {
 			}
 		}
 	}
+#ifdef _WIN32
+	LocalFree(wargv);
+#else
+	delete[] wargv;
+#endif
 	ygo::mainGame->MainLoop();
 #ifdef _WIN32
 	WSACleanup();