| 4 |
|
// |
| 5 |
|
// $Id$ |
| 6 |
|
|
| 7 |
+ |
#include "AwayMessage.hpp" |
| 8 |
|
#include "Collector.hpp" |
| 9 |
+ |
#include "Stamp.hpp" |
| 10 |
|
|
| 11 |
|
#include <cstring> |
| 12 |
+ |
|
| 13 |
+ |
extern "C" |
| 14 |
+ |
{ |
| 15 |
|
#include <pwd.h> |
| 16 |
|
#include <unistd.h> |
| 17 |
+ |
} |
| 18 |
|
|
| 19 |
< |
Collector::Collector(const ext::String& login, ext::String& password, const |
| 20 |
< |
std::set<Buddy>& buddies, bool start) : login(login), password(password), |
| 21 |
< |
start(start) |
| 19 |
> |
Collector::Collector(const ext::String& login, ext::String& password, const std::set<Buddy>& buddies, |
| 20 |
> |
const Database& database, bool start) : login(login), password(password), buddies(buddies), database(database), |
| 21 |
> |
start(start), prompted(false) |
| 22 |
|
{ |
| 23 |
|
if (start) |
| 24 |
|
{ |
| 25 |
|
if (password.IsEmpty()) prompt(); |
| 26 |
|
|
| 27 |
< |
collector.Spawn(etl::BindAll(&Collector::collect, this, buddies)); |
| 27 |
> |
collector.Spawn(etl::Bind<0>(&Collector::collect, this)); |
| 28 |
|
} |
| 29 |
|
} |
| 30 |
|
|
| 31 |
< |
net::Oscar::BuddyTool* Collector::buddy(NULL); |
| 26 |
< |
net::Oscar::IcbmTool* Collector::icbm(NULL); |
| 27 |
< |
|
| 28 |
< |
int Collector::collect(const std::set<Buddy>& buddies) |
| 31 |
> |
int Collector::collect() |
| 32 |
|
{ |
| 33 |
< |
cerr << "Collector::collect()\n"; |
| 33 |
> |
cerr << bright << green << "Started collector daemon at " << Stamp() |
| 34 |
> |
<< ".\n" << reset; |
| 35 |
|
|
| 36 |
|
net::Oscar::Session session; |
| 37 |
|
net::Oscar::AuthTool auth(session); |
| 42 |
|
|
| 43 |
|
auth.Login(login, password); |
| 44 |
|
|
| 45 |
< |
Collector::buddy = &buddy; |
| 42 |
< |
Collector::icbm = &icbm; |
| 45 |
> |
if (prompted) password.Clear(); |
| 46 |
|
|
| 47 |
< |
icbm.OnIncomingIcbm += &Collector::reply; |
| 47 |
> |
this->buddy = &buddy; |
| 48 |
> |
this->icbm = &icbm; |
| 49 |
> |
this->info = &info; |
| 50 |
|
|
| 51 |
< |
// for (size_t index(password.GetSize()); index > 0; --index) password[index - |
| 52 |
< |
// 1] = '\0'; |
| 51 |
> |
buddy.OnOnline += etl::Bind<0>(&Collector::status, this); |
| 52 |
> |
icbm.OnReceive += etl::Bind<0>(&Collector::receive, this); |
| 53 |
|
|
| 54 |
< |
net::Oscar::StringSet buddies_; |
| 54 |
> |
net::Oscar::StringSet buddies; |
| 55 |
|
|
| 56 |
< |
for (std::set<Buddy>::const_iterator buddy(buddies.begin()); buddy != |
| 57 |
< |
buddies.end(); ++buddy) buddies_.Insert(*buddy); |
| 56 |
> |
for (std::set<Buddy>::const_iterator buddy(this->buddies.begin()); |
| 57 |
> |
buddy != this->buddies.end(); ++buddy) buddies.Insert(*buddy); |
| 58 |
|
|
| 59 |
< |
buddy.Insert(buddies_); |
| 55 |
< |
icbm.RequestParams(); |
| 59 |
> |
buddy.Insert(buddies); |
| 60 |
|
|
| 61 |
|
net::Oscar::Capabilities capabilities; |
| 62 |
|
|
| 63 |
|
capabilities.Insert(net::Oscar::ChatCapability); |
| 64 |
< |
info.SetProfile("<font size=3>I am <a href=\"http://computers.douglasthrift.net/zoe.xml\">Zoe</a>.</font> :-*", "", capabilities); |
| 64 |
> |
info.SetProfile("<font face=\"Tahoma\" size=2>I am <a href=\"" |
| 65 |
> |
+ Zoe::generator(Zoe::url) + "\">Zoe</a>.</font> :-*", "<font face=\"" |
| 66 |
> |
"Tahoma\" size=2>Mommy told me not to talk to strangers.</font> O:-)", |
| 67 |
> |
capabilities); |
| 68 |
|
|
| 69 |
< |
// figure out whether or not we logged in |
| 69 |
> |
// XXX figure out whether or not we logged in |
| 70 |
|
|
| 71 |
< |
while (true) sleep(60); |
| 71 |
> |
while (true) sleep(Hour(1)); |
| 72 |
|
} |
| 73 |
|
|
| 74 |
|
void Collector::prompt() |
| 75 |
|
{ |
| 76 |
+ |
if (!prompted) prompted = true; |
| 77 |
+ |
|
| 78 |
|
ext::String prompt(login + "'s AIM Password: "); |
| 79 |
|
char* password(getpass(prompt.NullTerminate())); |
| 80 |
|
|
| 81 |
|
this->password = password; |
| 82 |
|
|
| 83 |
< |
for (size_t index(strlen(password)); index > 0; --index) password[index - |
| 84 |
< |
1] = '\0'; |
| 83 |
> |
for (size_t index(std::strlen(password)); index > 0; --index) |
| 84 |
> |
password[index - 1] = '\0'; |
| 85 |
> |
} |
| 86 |
> |
|
| 87 |
> |
void Collector::status(const net::Oscar::UserInfo& user) |
| 88 |
> |
{ |
| 89 |
> |
Buddy buddy(*buddies.find(user)); |
| 90 |
> |
AwayMessage message(buddy, *info); |
| 91 |
> |
|
| 92 |
> |
if (!ext::String(message).IsEmpty()) // XXX |
| 93 |
> |
{ |
| 94 |
> |
if (Zoe::debug) cerr << "buddy = " << buddy << "\nmessage = " << message |
| 95 |
> |
<< '\n'; |
| 96 |
> |
|
| 97 |
> |
ext::Handle<dbi::Driver> driver(dbi::GetDriver(database.driver)); |
| 98 |
> |
ext::Handle<dbi::Connection> db(driver->Connect(database.host, |
| 99 |
> |
database.user, database.password, database.db)); |
| 100 |
> |
ext::Handle<dbi::ResultSet> previous(db->Execute("SELECT message " |
| 101 |
> |
"FROM messages WHERE stamp=(SELECT MAX(stamp) FROM messages " |
| 102 |
> |
"WHERE id='" + lexical_cast<ext::String>(buddy.getId()) |
| 103 |
> |
+ "') AND id='" + lexical_cast<ext::String>(buddy.getId()) + "'")); |
| 104 |
> |
|
| 105 |
> |
if (previous->MoveNext()) |
| 106 |
> |
{ |
| 107 |
> |
if (ext::String(message) == previous->GetString("message")) return; |
| 108 |
> |
} |
| 109 |
> |
|
| 110 |
> |
db->Execute("INSERT INTO messages (stamp, message, id) VALUES ('" |
| 111 |
> |
+ ext::String(message.getStamp()) + "', '" + db->Escape(message) |
| 112 |
> |
+ "', '" + lexical_cast<ext::String>(buddy.getId()) + "')"); |
| 113 |
> |
|
| 114 |
> |
cerr << bright << red << "Collected away message from " << blue << buddy |
| 115 |
> |
<< red << " at " << Stamp() << ".\n" << reset; |
| 116 |
> |
} |
| 117 |
> |
|
| 118 |
> |
if (user.screenname != buddy.getDisplay(database)) |
| 119 |
> |
{ |
| 120 |
> |
buddy.setDisplay(user.screenname, database); |
| 121 |
> |
|
| 122 |
> |
sleep(1); |
| 123 |
> |
|
| 124 |
> |
icbm->Simple(buddy, "<font face=\"Tahoma\" size=2>So sexy!</font> :-*"); |
| 125 |
> |
} |
| 126 |
|
} |
| 127 |
|
|
| 128 |
< |
void Collector::reply(const ext::String& login, const ext::String& reply) |
| 128 |
> |
void Collector::receive(const ext::String& buddy, const ext::String& message) |
| 129 |
|
{ |
| 130 |
< |
icbm->Simple(login, reply); |
| 130 |
> |
if (Zoe::debug) cerr << "buddy = " << Buddy(buddy) << "\nmessage = " |
| 131 |
> |
<< message << '\n'; |
| 132 |
> |
|
| 133 |
> |
if (buddies.find(buddy) != buddies.end()) |
| 134 |
> |
{ |
| 135 |
> |
Buddy buddy_(*buddies.find(buddy)); |
| 136 |
> |
|
| 137 |
> |
sleep(1); |
| 138 |
> |
|
| 139 |
> |
icbm->Simple(buddy, "<font face=\"Tahoma\" size=2>Hello! I am Zoe. I am" |
| 140 |
> |
" supposed to be collecting your <a href=\"" + buddy_.getLink() |
| 141 |
> |
+ "\">Away Messages</a> (<a href=\"" + buddy_.getRssLink() + "\">" |
| 142 |
> |
"RSS 2.0</a>) (<a href=\"" + buddy_.getAtomLink() + "\">Atom</a>)." |
| 143 |
> |
"</font> :-*"); |
| 144 |
> |
} |
| 145 |
> |
else |
| 146 |
> |
{ |
| 147 |
> |
sleep(1); |
| 148 |
> |
|
| 149 |
> |
icbm->Simple(buddy, "<font face=\"Tahoma\" size=2>Mommy told me not to " |
| 150 |
> |
"talk to strangers.</font> O:-)"); |
| 151 |
> |
} |
| 152 |
|
} |
