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