ZYM/BH-Bookstore-2023
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: add check for quit and exit

Browse Source
This commit is contained in:
ZhuangYumin
2023-12-14 06:42:41 +00:00
parent be4903a0fa
commit 159cacaa21
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
backend/src/engine.cpp
View File

@ -46,6 +46,10 @@ std::vector<std::string> BookStoreEngineClass::Execute(
std::string head = ""; std::string head = "";
ss >> head; ss >> head;
if (head == "quit" || head == "exit") { if (head == "quit" || head == "exit") {
static std::basic_regex<char> quit_regex(R"(^ *(quit|exit) *$)",
std::regex_constants::optimize);
if (!std::regex_match(cmd, quit_regex))
return std::vector<std::string>({"Invalid"});
while (login_stack.size()) { while (login_stack.size()) {
login_count[login_stack.top().first]--; login_count[login_stack.top().first]--;
login_stack.pop(); login_stack.pop();
@ -313,7 +317,7 @@ std::vector<std::string> BookStoreEngineClass::ExecuteMOdify(
return std::vector<std::string>({"Invalid"}); return std::vector<std::string>({"Invalid"});
// debugPrint("successfully lexed modify"); // debugPrint("successfully lexed modify");
// debugPrint("modify", new_ISBN, ' ', name, ' ', author, ' ', keyword, ' ', // debugPrint("modify", new_ISBN, ' ', name, ' ', author, ' ', keyword, ' ',
// price); // price);
if (login_stack.empty() || if (login_stack.empty() ||
user_data_base.GetPrevilege(login_stack.top().first) < 3) user_data_base.GetPrevilege(login_stack.top().first) < 3)
return std::vector<std::string>({"Invalid"}); return std::vector<std::string>({"Invalid"});
@ -392,9 +396,9 @@ std::vector<std::string> BookStoreEngineClass::ExecuteShowFinance(
std::vector<std::string> BookStoreEngineClass::ExecuteLog( std::vector<std::string> BookStoreEngineClass::ExecuteLog(
const std::string &cmd, const std::string &cmd,
std::stack<std::pair<std::string, int>> &login_stack) { std::stack<std::pair<std::string, int>> &login_stack) {
static std::basic_regex<char> log_regex( static std::basic_regex<char> log_regex(R"(^ *log *$)",
R"(^ *log *$)", std::regex_constants::optimize); std::regex_constants::optimize);
if(!std::regex_match(cmd, log_regex)) if (!std::regex_match(cmd, log_regex))
return std::vector<std::string>({"Invalid"}); return std::vector<std::string>({"Invalid"});
return std::vector<std::string>(); return std::vector<std::string>();
} }