zoe/trunk/Collector.cpp

// Zoe AIM Away Message RSS Feed Generator
//
// Seth King and Douglas Thrift
//
// $Id$

#include "AwayMessage.hpp"
#include "Collector.hpp"
#include "Stamp.hpp"

#include <cstring>

extern "C"
{
#include <pwd.h>
#include <unistd.h>
}

Collector::Collector(const ext::String& login, ext::String& password, const     std::set<Buddy>& buddies,
        const Database& database, bool start) : login(login), password(password), buddies(buddies), database(database),
        start(start), prompted(false)
{
        if (start)
        {
                if (password.IsEmpty()) prompt();

                collector.Spawn(etl::Bind<0>(&Collector::collect, this));
        }
}

int Collector::collect()
{
        cerr << bright << green << "Started collector daemon at " << Stamp()
                << ".\n" << reset;

        net::Oscar::Session session;
        net::Oscar::AuthTool auth(session);
        net::Oscar::BuddyTool buddy(session);
        net::Oscar::ChatTool chat(session);
        net::Oscar::IcbmTool icbm(session);
        net::Oscar::InfoTool info(session);

        auth.Login(login, password);

        if (prompted) password.Clear();

        this->buddy = &buddy;
        this->icbm = &icbm;
        this->info = &info;

        buddy.OnOnline += etl::Bind<0>(&Collector::status, this);
        icbm.OnReceive += etl::Bind<0>(&Collector::receive, this);

        net::Oscar::StringSet buddies;

        for (std::set<Buddy>::const_iterator buddy(this->buddies.begin());
                buddy != this->buddies.end(); ++buddy) buddies.Insert(*buddy);

        buddy.Insert(buddies);

        net::Oscar::Capabilities capabilities;

        capabilities.Insert(net::Oscar::ChatCapability);
        info.SetProfile("<font face=\"Tahoma\" size=2>I am <a href=\""
                + Zoe::generator(Zoe::url) + "\">Zoe</a>.</font> :-*", "<font face=\""
                "Tahoma\" size=2>Mommy told me not to talk to strangers.</font> O:-)",
                capabilities);

        // XXX figure out whether or not we logged in

        while (true) sleep(Hour(1));
}

void Collector::prompt()
{
        if (!prompted) prompted = true;

        ext::String prompt(login + "'s AIM Password: ");
        char* password(getpass(prompt.NullTerminate()));

        this->password = password;

        for (size_t index(std::strlen(password)); index > 0; --index)
                password[index - 1] = '\0';
}

void Collector::status(const net::Oscar::UserInfo& user)
{
        Buddy buddy(*buddies.find(user));
        AwayMessage message(buddy, *info);

        if (!ext::String(message).IsEmpty()) // XXX
        {
                if (Zoe::debug) cerr << "buddy = " << buddy << "\nmessage = " << message
                        << '\n';

                ext::Handle<dbi::Connection> db(dbi::Connect(database.driver,
                        database.host, database.user, database.password, database.db));
                ext::Handle<dbi::ResultSet> previous(db->Execute("SELECT message "
                        "FROM messages WHERE stamp=(SELECT MAX(stamp) FROM messages "
                        "WHERE id='" + lexical_cast<ext::String>(buddy.getId())
                        + "') AND id='" + lexical_cast<ext::String>(buddy.getId()) + "'"));

                if (previous->MoveNext())
                {
                        if (ext::String(message) == previous->GetString("message")) return;
                }

                db->Execute("INSERT INTO messages (stamp, message, id) VALUES ('"
                        + ext::String(message.getStamp()) + "', '" + db->Escape(message)
                        + "', '" + lexical_cast<ext::String>(buddy.getId()) + "')");

                cerr << bright << red << "Collected away message from " << blue << buddy
                        << red << " at " << Stamp() << ".\n" << reset;
        }
}

void Collector::receive(const ext::String& buddy, const ext::String& message)
{
        if (Zoe::debug) cerr << "buddy = " << Buddy(buddy) << "\nmessage = "
                << message << '\n';

        if (buddies.find(buddy) != buddies.end())
        {
                Buddy buddy_(*buddies.find(buddy));
                
                sleep(1);

                icbm->Simple(buddy, "<font face=\"Tahoma\" size=2>Hello! I'm Zoe. I am "
                        "supposed to be collecting your <a href=\"" + buddy_.getLink()
                        + "\">Away Messages</a> (<a href=\"" + buddy_.getRssLink() + "\">"
                        "RSS</a>) (<a href=\"" + buddy_.getAtomLink() + "\">Atom</a>)."
                        "</font> :-)");
        }
        else
        {
                sleep(1);

                icbm->Simple(buddy, "<font face=\"Tahoma\" size=2>Mommy told me not to "
                        "talk to strangers.</font> O:-)");
        }
}
Revision:	46
Committed:	2004-08-16T18:52:59-07:00 (20 years, 10 months ago) by douglas
File size:	3886 byte(s)
Log Message:	Made some drastic changes! Maybe not so drastic.
#	User	Rev	Content
1	douglas	13	// Zoe AIM Away Message RSS Feed Generator
2			//
3			// Seth King and Douglas Thrift
4			//
5			// $Id$
6
7	douglas	21	#include "AwayMessage.hpp"
8	douglas	13	#include "Collector.hpp"
9	douglas	21	#include "Stamp.hpp"
10	douglas	13
11	douglas	16	#include <cstring>
12	douglas	21
13			extern "C"
14			{
15	douglas	15	#include <pwd.h>
16			#include <unistd.h>
17	douglas	21	}
18	douglas	15
19	douglas	34	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	douglas	21	start(start), prompted(false)
22	douglas	13	{
23	douglas	15	if (start)
24			{
25			if (password.IsEmpty()) prompt();
26
27	douglas	20	collector.Spawn(etl::Bind<0>(&Collector::collect, this));
28	douglas	15	}
29	douglas	13	}
30	douglas	15
31	douglas	20	int Collector::collect()
32	douglas	15	{
33	douglas	38	cerr << bright << green << "Started collector daemon at " << Stamp()
34			<< ".\n" << reset;
35	douglas	15
36			net::Oscar::Session session;
37			net::Oscar::AuthTool auth(session);
38			net::Oscar::BuddyTool buddy(session);
39			net::Oscar::ChatTool chat(session);
40			net::Oscar::IcbmTool icbm(session);
41			net::Oscar::InfoTool info(session);
42
43			auth.Login(login, password);
44
45	douglas	18	if (prompted) password.Clear();
46
47	douglas	17	this->buddy = &buddy;
48			this->icbm = &icbm;
49	douglas	22	this->info = &info;
50	douglas	16
51	douglas	22	buddy.OnOnline += etl::Bind<0>(&Collector::status, this);
52			icbm.OnReceive += etl::Bind<0>(&Collector::receive, this);
53	douglas	16
54	douglas	20	net::Oscar::StringSet buddies;
55	douglas	15
56	douglas	38	for (std::set<Buddy>::const_iterator buddy(this->buddies.begin());
57			buddy != this->buddies.end(); ++buddy) buddies.Insert(*buddy);
58	douglas	15
59	douglas	20	buddy.Insert(buddies);
60	douglas	15
61			net::Oscar::Capabilities capabilities;
62
63			capabilities.Insert(net::Oscar::ChatCapability);
64	douglas	38	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	douglas	15
69	douglas	40	// XXX figure out whether or not we logged in
70	douglas	15
71	douglas	38	while (true) sleep(Hour(1));
72	douglas	15	}
73
74			void Collector::prompt()
75			{
76	douglas	18	if (!prompted) prompted = true;
77
78	douglas	16	ext::String prompt(login + "'s AIM Password: ");
79			char* password(getpass(prompt.NullTerminate()));
80	douglas	15
81			this->password = password;
82
83	douglas	38	for (size_t index(std::strlen(password)); index > 0; --index)
84			password[index - 1] = '\0';
85	douglas	15	}
86	douglas	16
87	douglas	22	void Collector::status(const net::Oscar::UserInfo& user)
88	douglas	16	{
89	douglas	25	Buddy buddy(*buddies.find(user));
90	douglas	30	AwayMessage message(buddy, *info);
91	douglas	22
92	douglas	30	if (!ext::String(message).IsEmpty()) // XXX
93			{
94	douglas	38	if (Zoe::debug) cerr << "buddy = " << buddy << "\nmessage = " << message
95			<< '\n';
96	douglas	31
97	douglas	38	ext::Handle<dbi::Connection> db(dbi::Connect(database.driver,
98			database.host, database.user, database.password, database.db));
99			ext::Handle<dbi::ResultSet> previous(db->Execute("SELECT message "
100			"FROM messages WHERE stamp=(SELECT MAX(stamp) FROM messages "
101			"WHERE id='" + lexical_cast<ext::String>(buddy.getId())
102			+ "') AND id='" + lexical_cast<ext::String>(buddy.getId()) + "'"));
103
104	douglas	37	if (previous->MoveNext())
105			{
106			if (ext::String(message) == previous->GetString("message")) return;
107			}
108	douglas	30
109	douglas	37	db->Execute("INSERT INTO messages (stamp, message, id) VALUES ('"
110			+ ext::String(message.getStamp()) + "', '" + db->Escape(message)
111			+ "', '" + lexical_cast<ext::String>(buddy.getId()) + "')");
112	douglas	34
113	douglas	37	cerr << bright << red << "Collected away message from " << blue << buddy
114	douglas	38	<< red << " at " << Stamp() << ".\n" << reset;
115	douglas	30	}
116	douglas	22	}
117
118			void Collector::receive(const ext::String& buddy, const ext::String& message)
119			{
120	douglas	37	if (Zoe::debug) cerr << "buddy = " << Buddy(buddy) << "\nmessage = "
121			<< message << '\n';
122	douglas	31
123	douglas	20	if (buddies.find(buddy) != buddies.end())
124			{
125	douglas	37	Buddy buddy_(*buddies.find(buddy));
126	douglas	22
127			sleep(1);
128
129	douglas	46	icbm->Simple(buddy, "<font face=\"Tahoma\" size=2>Hello! I'm Zoe. I am "
130			"supposed to be collecting your <a href=\"" + buddy_.getLink()
131			+ "\">Away Messages</a> (<a href=\"" + buddy_.getRssLink() + "\">"
132			"RSS</a>) (<a href=\"" + buddy_.getAtomLink() + "\">Atom</a>)."
133	douglas	40	"</font> :-)");
134	douglas	20	}
135			else
136			{
137			sleep(1);
138
139	douglas	37	icbm->Simple(buddy, "<font face=\"Tahoma\" size=2>Mommy told me not to "
140			"talk to strangers.</font> O:-)");
141	douglas	20	}
142	douglas	16	}