Browse Source

adding favorites to the search scopes

miguelfreitas
erqan 9 years ago
parent
commit
391b25993d
  1. 95
      src/twister.cpp
  2. 3
      src/twister.h

95
src/twister.cpp

@ -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,14 +3388,17 @@ 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;
@ -3395,6 +3409,7 @@ Value search(const Array& params, bool fHelp)
} }
} }
} }
}
// search messages in dht // search messages in dht
boost::shared_ptr<session> ses(m_ses); boost::shared_ptr<session> ses(m_ses);
@ -3413,11 +3428,14 @@ 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)
{
string n = up->dict_find_string_value("n");
int k = up->dict_find_int_value("k");
pair<std::string,int> post_id(n,k); pair<std::string,int> post_id(n,k);
if( posts.count(post_id) == 0 ) { if( posts.count(post_id) == 0 ) {
int64 time = p->dict_find_int_value("time",-1); int64 time = up->dict_find_int_value("time",-1);
entry vEntry; entry vEntry;
vEntry = *p; vEntry = *p;
@ -3431,6 +3449,7 @@ Value search(const Array& params, bool fHelp)
} }
} }
} }
}
std::multimap<int64,entry> postsByTime; std::multimap<int64,entry> postsByTime;
BOOST_FOREACH(const PAIRTYPE(PAIRTYPE(std::string,int),PAIRTYPE(int64,entry))& item, posts) { BOOST_FOREACH(const PAIRTYPE(PAIRTYPE(std::string,int),PAIRTYPE(int64,entry))& item, posts) {
@ -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

@ -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

Loading…
Cancel
Save