| 1 |
// Zoe AIM Away Message RSS Feed Generator |
| 2 |
// |
| 3 |
// Seth King and Douglas Thrift |
| 4 |
// |
| 5 |
// $Id$ |
| 6 |
|
| 7 |
#include "Publisher.hpp" |
| 8 |
#include "Rss.hpp" |
| 9 |
#include "Atom.hpp" |
| 10 |
|
| 11 |
Publisher::Publisher(const std::set<Buddy>& buddies, const Database& database, bool start) : buddies(buddies), |
| 12 |
database(database), start(start) |
| 13 |
{ |
| 14 |
if (start) publish(); |
| 15 |
} |
| 16 |
|
| 17 |
void Publisher::publish() |
| 18 |
{ |
| 19 |
api::Cerr << bright << yellow << "Started updating feeds at " << Stamp() |
| 20 |
<< ".\n" << reset; |
| 21 |
|
| 22 |
Stamp stamp; |
| 23 |
ext::Handle<dbi::Driver> driver(dbi::GetDriver(database.driver)); |
| 24 |
ext::Handle<dbi::Connection> db(driver->Connect(database.host, |
| 25 |
database.user, database.password, database.db)); |
| 26 |
|
| 27 |
db->Execute("DELETE FROM messages WHERE stamp<'" |
| 28 |
+ ext::String(stamp - Day(30)) + "'"); |
| 29 |
|
| 30 |
for (std::set<Buddy>::const_iterator buddy(buddies.begin()); |
| 31 |
buddy != buddies.end(); ++buddy) |
| 32 |
{ |
| 33 |
ext::Handle<dbi::ResultSet> messages(db->Execute(ios::String() |
| 34 |
<< "SELECT stamp, message FROM messages WHERE id='" |
| 35 |
<< buddy->getId() << "'")); |
| 36 |
std::set<AwayMessage, std::greater<AwayMessage> > messages_; |
| 37 |
|
| 38 |
while (messages->MoveNext()) |
| 39 |
messages_.insert(AwayMessage(messages->GetString("message"), |
| 40 |
messages->GetString("stamp"))); |
| 41 |
|
| 42 |
ext::String display(buddy->getDisplay(database)); |
| 43 |
Rss rss(*buddy, display, messages_.begin(), messages_.end(), stamp); |
| 44 |
Atom atom(*buddy, display, messages_.begin(), messages_.end(), stamp); |
| 45 |
|
| 46 |
api::Cerr << bright << yellow << "Updated feeds for " << blue << *buddy |
| 47 |
<< yellow << ".\n" << reset; |
| 48 |
} |
| 49 |
|
| 50 |
api::Cerr << bright << yellow << "Finished updating feeds at " << Stamp() |
| 51 |
<< ".\n" << reset; |
| 52 |
} |
