117 lines
2.6 KiB
C++
117 lines
2.6 KiB
C++
|
|
#include <Log.h>
|
|
|
|
#include <iostream>
|
|
#include <signal.h> //signal handler
|
|
#include <unistd.h>
|
|
#include <ctime>
|
|
|
|
#include "search.h"
|
|
|
|
static bool run = true;
|
|
|
|
void sig_handler(int sig_num) {
|
|
Log::info << "signalHandler triggered";
|
|
run = false;
|
|
::close(STDIN_FILENO); //interrupt input
|
|
(void) sig_num;
|
|
}
|
|
|
|
int main(int argc, const char** argv) {
|
|
Log::init();
|
|
Log::setConsoleLogLevel(Log::Level::TRACE);
|
|
Log::addLogfile("log.txt", Log::Level::TRACE);
|
|
#if __unix__
|
|
Log::setColoredOutput(true);
|
|
#endif
|
|
|
|
if(argc < 2) {
|
|
Log::error << "Usage: " << argv[0] << " <file> [<file2> ....]";
|
|
return 1;
|
|
}
|
|
|
|
time_t start = time(nullptr);
|
|
Search rootsearch;
|
|
for(int i = 1; i < argc; ++i) {
|
|
Log::info << "Load File: " << argv[i];
|
|
rootsearch.addFile(argv[i]);
|
|
}
|
|
rootsearch.finalize();
|
|
|
|
time_t end = time(nullptr);
|
|
Log::info << rootsearch.getChatCount() << " Chats with " << rootsearch.getMessageCount() << " Messages loaded in " << end-start << "s";
|
|
|
|
signal(SIGINT, sig_handler);
|
|
|
|
Search* search = nullptr;
|
|
|
|
while(run) {
|
|
if(!search)
|
|
search = &rootsearch;
|
|
|
|
Log::info << search->getMessageCount() << " Messages. Enter searchterm: ";
|
|
std::string searchterm;
|
|
std::getline(std::cin, searchterm);
|
|
Log::trace << "searchterm: \"" << searchterm << "\"";
|
|
|
|
if(!run) break;
|
|
|
|
Log::info << "Enter Flags: [RI]";
|
|
std::string flags;
|
|
std::cin >> flags;
|
|
Searchflags parsedflags = Search::fromString(flags);
|
|
|
|
if(!run) break;
|
|
|
|
time_t start = time(nullptr);
|
|
std::list<const Message*> results = search->search(searchterm, parsedflags);
|
|
time_t end = time(nullptr);
|
|
Log::info << results.size() << " results found in " << end-start << "s";
|
|
if(results.size()) {
|
|
Log::info << "Print results ?";
|
|
char c;
|
|
std::cin >> c;
|
|
|
|
if(!run) break;
|
|
|
|
if(c == 'y' || c == 'Y') {
|
|
//print results
|
|
for(const Message* m : results) {
|
|
Log::info << search->getShortChatname(m->chatid) << ": (" << m->messageid << ") " << m->text;
|
|
}
|
|
}
|
|
|
|
Log::info << "New Subfilter (y)? Clear Search (c)? Change current Filter (anykey)?";
|
|
std::cin >> c;
|
|
|
|
if(!run) break;
|
|
|
|
if(c == 'y' || c == 'Y') {
|
|
//filter
|
|
Search* subsearch = new Search(*search, &results);
|
|
|
|
//delete old subsearch (dont delete root)
|
|
if(search != &rootsearch)
|
|
delete search;
|
|
|
|
//set new search
|
|
search = subsearch;
|
|
} else if(c == 'c' || c == 'C') {
|
|
//move to root search
|
|
if(search != &rootsearch) {
|
|
delete search;
|
|
search = &rootsearch;
|
|
}
|
|
}
|
|
}
|
|
|
|
std::cin.ignore(std::numeric_limits<std::streamsize>::max(), '\n');
|
|
}
|
|
|
|
if(search != &rootsearch)
|
|
delete search;
|
|
|
|
Log::stop();
|
|
return 0;
|
|
}
|