Update shioricaller.cpp

src/shioricaller.cpp

@@ -9,26 +9,61 @@
 int wmain(int argc, wchar_t* argv[]){
 	InstallExceptionFilter();
 
+	//获取标准输出原有颜色
+	CONSOLE_SCREEN_BUFFER_INFO csbi;
+	GetConsoleScreenBufferInfo(GetStdHandle(STD_OUTPUT_HANDLE), &csbi);
+
+	auto reset_output_color=[&](){
+		//恢复标准输出颜色
+		SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), csbi.wAttributes);
+	};
+	auto set_output_color=[&](WORD color){
+		//设置输出颜色
+		SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), color);
+	};
+
 	if(argc != 2) {
-		std::cout << "Usage: shioricaller shiori.dll < request.txt > response.txt" << std::endl;
+		std::cout << "Usage 1: \t";
+
+		//命令部分使用绿色显示
+		set_output_color(FOREGROUND_GREEN | FOREGROUND_INTENSITY);
+		std::cout << "shioricaller shiori.dll < request.txt > response.txt" << std::endl;
+		//恢复标准输出颜色
+		reset_output_color();
+		std::cout << "Usage 2: \t";
+		//命令部分使用绿色显示
+		set_output_color(FOREGROUND_GREEN | FOREGROUND_INTENSITY);
+		std::cout << "shioricaller shiori.dll" << std::endl;
+		//恢复标准输出颜色
+		reset_output_color();
+		std::cout << "(In Usage 2, you need to:" << std::endl
+				  << "\t1. input request manually" << std::endl
+				  << "\t2. press Enter*2 to send request" << std::endl
+				  << "\t3. when you want to quit, press Ctrl+Z or something to send EOF" << std::endl
+				  << ")" << std::endl;
 		return 0;
 	}
 
 	Cshiori shiori{argv[1]};
 	if(not shiori.All_OK()) {
+		//cerr设置颜色到红色
+		SetConsoleTextAttribute(GetStdHandle(STD_ERROR_HANDLE), FOREGROUND_RED | FOREGROUND_INTENSITY);
 		std::cerr << "Error: something fucked up." << std::endl;
 		return 1;
 	}
 
 	std::string req;
 	std::string req_line;
 	while(true) {
+		//恢复标准输出颜色
+		reset_output_color();
 		std::getline(std::cin, req_line);
-		if(std::cin.eof())
-			break;
-		req += req_line + "\x0d\x0a";
+		if(std::cin.eof())break;
+		req += req_line + "\r\n";
 		if(req_line.empty()) {
 			auto res = shiori(req);
+			//设置输出颜色到淡黄色
+			set_output_color(FOREGROUND_GREEN | FOREGROUND_RED | FOREGROUND_INTENSITY);
 			std::cout << res;
 			req.clear();
 		}