twisterarmy
/
twister-core
mirror of https://github.com/twisterarmy/twister-core.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
Browse Source

adding favorites to the search scopes

miguelfreitas
erqan 9 years ago
parent
73dc8acc72
commit
391b25993d
2 changed files with 96 additions and 24 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 117
      src/twister.cpp
  2. 3
      src/twister.h

117
src/twister.cpp
Unescape View File

@ -3303,21 +3303,32 @@ lazy_entry const* TextSearch::matchRawMessage(string const &rawMessage, lazy_ent
if( post ) { if( post ) {
lazy_entry const* rt = post->dict_find_dict("rt"); lazy_entry const* rt = post->dict_find_dict("rt");
lazy_entry const* p = rt ? rt : post; lazy_entry const* p = rt ? rt : post;
string comment;
if (rt)
comment = post->dict_find_string_value("msg");
if( username.length() ) { if( username.length() ) {
string user = p->dict_find_string_value("n"); string user = p->dict_find_string_value("n");
if( user != username ) { string rtuser;
if (rt)
rtuser = post->dict_find_string_value("n");
if( user != username && (!comment.size() || rtuser != username)) {
return 0; return 0;
} }
} }
int64_t time = p->dict_find_int_value("time"); int64_t time = p->dict_find_int_value("time");
if( !matchTime(time) ) { int64_t rttime = 0;
if (comment.size())
rttime = post->dict_find_int_value("time");
if( !matchTime(time) && (!rttime || !matchTime(rttime)) ) {
return 0; return 0;
} }
string msg = p->dict_find_string_value("msg"); string msg = p->dict_find_string_value("msg");
return matchText( msg ) ? p : 0; if (matchText(msg) || matchText(comment))
//for RTable results, it returns signed post instead of userpost
return vv ? vv : &v;
} }
} }
return 0; return 0;
@ -3329,7 +3340,7 @@ Value search(const Array& params, bool fHelp)
throw runtime_error( throw runtime_error(
"search <scope> <text> <count> ['{\"username\":username,\"mode\":\"exact\"|\"all\"|\"any\",\"case\":\"sensitive\"|\"insensitive\",\"agemin\":agemin,\"agemax\":agemin}']\n" "search <scope> <text> <count> ['{\"username\":username,\"mode\":\"exact\"|\"all\"|\"any\",\"case\":\"sensitive\"|\"insensitive\",\"agemin\":agemin,\"agemax\":agemin}']\n"
"search text in available data\n" "search text in available data\n"
"<scope> is data area: messages, directmsgs, profiles, users, hashtags\n" "<scope> is data area: messages, directmsgs, profiles, users, hashtags, favorites\n"
"<text> is a phrase to search\n" "<text> is a phrase to search\n"
"up to <count> entries are returned\n" "up to <count> entries are returned\n"
"<username> in messages scope is optional and allows to search in username's messages only\n" "<username> in messages scope is optional and allows to search in username's messages only\n"
@ -3377,20 +3388,24 @@ Value search(const Array& params, bool fHelp)
BOOST_FOREACH(const PAIRTYPE(std::string,torrent_handle)& item, users) { BOOST_FOREACH(const PAIRTYPE(std::string,torrent_handle)& item, users) {
std::vector<std::string> pieces; std::vector<std::string> pieces;
item.second.get_pieces(pieces, std::numeric_limits<int>::max(), std::numeric_limits<int>::max(), -1, ~USERPOST_FLAG_DM, 0); item.second.get_pieces(pieces, std::numeric_limits<int>::max(), std::numeric_limits<int>::max(), -1, USERPOST_FLAG_HOME, 0);
BOOST_FOREACH(string const& piece, pieces) { BOOST_FOREACH(string const& piece, pieces) {
lazy_entry const* p = searcher.matchRawMessage(piece, v); lazy_entry const* p = searcher.matchRawMessage(piece, v);
if( p ) { if( p ) {
string n = p->dict_find_string_value("n"); const lazy_entry *up = p->dict_find_dict("userpost");
int k = p->dict_find_int_value("k"); if (up)
int64 time = p->dict_find_int_value("time",-1); {
string n = up->dict_find_string_value("n");
int k = up->dict_find_int_value("k");
int64 time = up->dict_find_int_value("time",-1);
entry vEntry; entry vEntry;
vEntry = *p; vEntry = *p;
hexcapePost(vEntry); hexcapePost(vEntry);
posts[pair<std::string,int>(n,k)] = pair<int64,entry>(time,vEntry); posts[pair<std::string,int>(n,k)] = pair<int64,entry>(time,vEntry);
}
} }
} }
} }
@ -3413,17 +3428,21 @@ Value search(const Array& params, bool fHelp)
string str_p = key_p->string(); string str_p = key_p->string();
lazy_entry const* p = searcher.matchRawMessage(str_p, v); lazy_entry const* p = searcher.matchRawMessage(str_p, v);
if( p ) { if( p ) {
string n = p->dict_find_string_value("n"); const lazy_entry *up = p->dict_find_dict("userpost");
int k = p->dict_find_int_value("k"); if (up)
pair<std::string,int> post_id(n,k); {
if( posts.count(post_id) == 0 ) { string n = up->dict_find_string_value("n");
int64 time = p->dict_find_int_value("time",-1); int k = up->dict_find_int_value("k");
pair<std::string,int> post_id(n,k);
entry vEntry; if( posts.count(post_id) == 0 ) {
vEntry = *p; int64 time = up->dict_find_int_value("time",-1);
hexcapePost(vEntry);
entry vEntry;
vEntry = *p;
hexcapePost(vEntry);
posts[post_id] = pair<int64,entry>(time,vEntry); posts[post_id] = pair<int64,entry>(time,vEntry);
}
} }
} }
} }
@ -3583,7 +3602,59 @@ Value search(const Array& params, bool fHelp)
ret.push_back( entryToJson(it->second) ); ret.push_back( entryToJson(it->second) );
} }
} else { }
else if (scope == "favorites")
{
std::multimap<int64_t,Value> postsByTime;
TextSearch searcher(keyword, options);
set<string> users;
if (username.size())
users.insert(username);
else
{
for (map<string,torrent_handle>::const_iterator it = m_userTorrent.begin(); it != m_userTorrent.end(); ++it)
users.insert(it->first);
}
BOOST_FOREACH(string user, users)
{
Array params;
params.push_back(user);
params.push_back(INT_MAX);
Array favs = getfavs(params, false).get_array();
for (int i = 0; i < favs.size(); i++)
{
entry favp = jsonToEntry(favs[i]);
entry *favu = favp.find_key("userpost");
if (favu && favu->type() == entry::dictionary_t)
{
entry* commEnt = favu->find_key("msg");
string comnt;
if (commEnt && commEnt->type() == entry::string_t)
comnt = commEnt->string();
entry *favEnt = favu->find_key("fav");
string msg;
if (favEnt)
msg = favEnt->find_key("msg")->string();
if( searcher.matchText(msg) || searcher.matchText(comnt) )
{
int64_t ft = favu->find_key("time")->integer();
if(searcher.matchTime(ft))
postsByTime.insert(pair<int64_t,Value>(ft, favs[i]));
}
}
}
}
std::multimap<int64_t,Value>::reverse_iterator rit;
for (rit = postsByTime.rbegin(); rit != postsByTime.rend() && ret.size() < count; ++rit)
ret.push_back(rit->second);
}
else {
throw runtime_error("Unknown <scope> value"); throw runtime_error("Unknown <scope> value");
} }

3
src/twister.h
Unescape View File

@ -13,7 +13,8 @@
#define USERPOST_FLAG_FAV 0x04 #define USERPOST_FLAG_FAV 0x04
#define USERPOST_FLAG_P_FAV 0x0C #define USERPOST_FLAG_P_FAV 0x0C
#define USERPOST_FLAG_HOME (~USERPOST_FLAG_DM & ~USERPOST_FLAG_FAV & ~USERPOST_FLAG_P_FAV) //only no flagged posts and RTs are displayed at home postboard
#define USERPOST_FLAG_HOME USERPOST_FLAG_RT
#define BLOCK_AGE_TO_EXPIRE_DHT_ENTRY (2016) // about 2 weeks #define BLOCK_AGE_TO_EXPIRE_DHT_ENTRY (2016) // about 2 weeks
#define BLOCK_AGE_TO_EXPIRE_DHT_POSTS (4320*2) // about 2 months #define BLOCK_AGE_TO_EXPIRE_DHT_POSTS (4320*2) // about 2 months

Write Preview
Loading…
Cancel
Save