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 |
13 |
|
10 |
douglas |
34 |
Publisher::Publisher(const std::set<Buddy>& buddies, const Database& database, bool start) : buddies(buddies), |
11 |
|
|
database(database), start(start) |
12 |
douglas |
13 |
{ |
13 |
douglas |
34 |
if (start) publisher.Spawn(etl::Bind<0>(&Publisher::publish, this)); |
14 |
douglas |
13 |
} |
15 |
douglas |
15 |
|
16 |
douglas |
21 |
int Publisher::publish() |
17 |
douglas |
15 |
{ |
18 |
douglas |
23 |
cerr << bright << green << "Publisher::publish()\n" << reset; |
19 |
douglas |
15 |
|
20 |
|
|
// publish |
21 |
douglas |
25 |
|
22 |
douglas |
31 |
while (true) |
23 |
|
|
{ |
24 |
douglas |
34 |
double next(api::GetWallTimerSeconds() + Hour(1)), now; |
25 |
douglas |
31 |
|
26 |
|
|
update(); |
27 |
|
|
|
28 |
|
|
do |
29 |
|
|
{ |
30 |
|
|
sleep(Minute(1)); |
31 |
|
|
|
32 |
douglas |
34 |
now = api::GetWallTimerSeconds(); |
33 |
douglas |
31 |
} |
34 |
douglas |
34 |
while (now < next); |
35 |
douglas |
31 |
} |
36 |
douglas |
27 |
} |
37 |
douglas |
25 |
|
38 |
douglas |
27 |
void Publisher::update() |
39 |
|
|
{ |
40 |
douglas |
31 |
cerr << bright << yellow << "Publisher::update()\n" << reset; |
41 |
|
|
|
42 |
douglas |
27 |
Stamp stamp; |
43 |
douglas |
34 |
ext::Handle<dbi::Connection> db(dbi::Connect(database.driver, database.host, database.user, database.password, |
44 |
|
|
database.db)); |
45 |
douglas |
25 |
|
46 |
douglas |
34 |
db->Execute("DELETE FROM messages WHERE stamp<'" + ext::String(stamp - Day(30)) + "'"); |
47 |
douglas |
27 |
|
48 |
douglas |
34 |
for (std::set<Buddy>::const_iterator buddy(buddies.begin()); buddy != buddies.end(); ++buddy) |
49 |
douglas |
27 |
{ |
50 |
douglas |
34 |
ext::Handle<dbi::ResultSet> messages(db->Execute("SELECT stamp, message FROM messages WHERE id='" |
51 |
|
|
+ lexical_cast<ext::String>(buddy->getId()) + "'")); |
52 |
douglas |
27 |
std::vector<AwayMessage> messages_; |
53 |
|
|
|
54 |
douglas |
34 |
while (messages->MoveNext()) messages_.push_back(AwayMessage(messages->GetString("message"), |
55 |
douglas |
27 |
messages->GetString("stamp"))); |
56 |
|
|
|
57 |
|
|
Rss rss(*buddy, messages_, stamp); |
58 |
|
|
} |
59 |
douglas |
15 |
} |