acetone
/
ircabot
mirror of https://notabug.org/acetone/ircabot.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

длина поискового запроса

master
const an teen 2021-07-16 19:52:18 +03:00
parent 9dad525525
commit d5be76618d
1 changed files with 11 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
main.cpp
View File

@ -35,6 +35,7 @@ std::string cFINDZERO; // команда поиска без параметро
std::string cLINKS; // ссылки на лог (в конце выдачи в ЛС) std::string cLINKS; // ссылки на лог (в конце выдачи в ЛС)
std::string cTRYLATER; // "перегрузка, попробуйте позже" std::string cTRYLATER; // "перегрузка, попробуйте позже"
std::string cHELP; // подсказка std::string cHELP; // подсказка
uint8_t cMINLEN; // минимальная длина искомого слова
std::map<std::string, std::string> conf; std::map<std::string, std::string> conf;
std::map<std::string, std::string> custom; std::map<std::string, std::string> custom;
@ -173,11 +174,6 @@ std::string search(std::string text)
values += matches[i]; values += matches[i];
} }
for (auto it = values.begin(), end = values.end(); it != end; ++it)
{ // Замена тире на точку
if (*it == '-') *it = '-';
}
values += "."; values += ".";
} }
return values; return values;
@ -197,7 +193,6 @@ bool read_config()
{ {
conf[child.first] = child.second.get_value<std::string>(); conf[child.first] = child.second.get_value<std::string>();
} }
cADMIN = conf["admin"]; cADMIN = conf["admin"];
cERROR = conf["error"]; cERROR = conf["error"];
cSUCCESS = conf["success"]; cSUCCESS = conf["success"];
@ -208,6 +203,7 @@ bool read_config()
cLINKS = conf["links"]; cLINKS = conf["links"];
cTRYLATER = conf["trylater"]; cTRYLATER = conf["trylater"];
cHELP = conf["help"]; cHELP = conf["help"];
cMINLEN = std::stoi(conf["minlen"]);
for (auto child: pt.get_child("custom")) for (auto child: pt.get_child("custom"))
{ {
@ -287,18 +283,20 @@ void handler()
if (tsc->get_msg_nick() == cADMIN && (msg.find("reload") == 0)) //// Reload if (tsc->get_msg_nick() == cADMIN && (msg.find("reload") == 0)) //// Reload
{ {
if (read_config()) tsc->write_to_channel(cSUCCESS); if (read_config()) tsc->write_to_channel(cSUCCESS);
else tsc->write_to_channel(cERROR); else tsc->write_to_channel(cERROR); // FIXME - падение в случае неудачи
} }
else if (msg.find(cFIND) == 0) //// Поиск else if (msg.find(cFIND) == 0) //// Поиск
{ {
std::regex date_check(cFIND + " [0-9]{4}.[0-9]{2}.[0-9]{2}.*", std::regex_constants::egrep); std::regex date_check(cFIND + " [0-9]{4}.[0-9]{2}.[0-9]{2}.*", std::regex_constants::egrep);
if (msg.find('*') != std::string::npos || msg.find('.') != std::string::npos) { if (msg.length() < cFIND.length() + cMINLEN)
// Защита от хитрой регулярки { // Защита от коротких запросов
tsc->write_to_channel(tsc->get_msg_nick() + ", " + cERROR); tsc->write_to_channel(tsc->get_msg_nick() + ", " + cERROR
+ " (" + std::to_string(cMINLEN) + ")");
} }
else if (msg.find(' ') == std::string::npos) { else if (msg.find(' ') == std::string::npos)
{ // Пустой запрос (без пробела после ключевого слова поиска)
tsc->write_to_channel(tsc->get_msg_nick() + ", " + cFINDZERO); tsc->write_to_channel(tsc->get_msg_nick() + ", " + cFINDZERO);
} }
@ -369,11 +367,11 @@ void handler()
} }
} }
} }
else tsc->write_to_channel(tsc->get_msg_nick() + ", " + cNOTFOUND); else tsc->write_to_channel (tsc->get_msg_nick() + ", " + cNOTFOUND);
} }
} }
else // Общий обработчик else // Обработчик кастомных ответов
{ {
handled = false; handled = false;
for (auto value: custom) for (auto value: custom)