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 << "Collector::collect()\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=\"http://computers.douglasthrift.net/zoe.xml\">Zoe</a>."
                "</font> :-*", "<font face=\"Tahoma\" size=2>Mommy told me not to talk to strangers.</font> O:-)", capabilities);

        // figure out whether or not we logged in

        while (true) sleep(Minute(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
        {
                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;

                cerr << bright << red << "Collector::status(" << blue << buddy << red << ", " << reset << message << bright << red
                        << ")\n" << reset;

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

void Collector::receive(const ext::String& buddy, const ext::String& message)
{
        cerr << bright << red << "Collector::receive(" << blue << Buddy(buddy) << red << ", " << reset << message << bright
                << red << ")\n" << reset;

        if (buddies.find(buddy) != buddies.end())
        {
                Stamp stamp;
                
                sleep(1);

                icbm->Simple(buddy, ext::String(stamp) + " " + lexical_cast<ext::String>(std::time(NULL)) + " "
                        + lexical_cast<ext::String>(api::GetWallTimerSeconds()));
        }
        else
        {
                sleep(1);

                icbm->Simple(buddy, "<font face=\"Tahoma\" size=2>Mommy told me not to talk to strangers.</font> O:-)");
        }
}
Revision:	35
Committed:	2004-07-21T20:43:31-07:00 (20 years, 11 months ago) by douglas
File size:	3652 byte(s)
Log Message:	Adding Atom support, other stuff.
#	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	23	cerr << bright << green << "Collector::collect()\n" << reset;
34	douglas	15
35			net::Oscar::Session session;
36			net::Oscar::AuthTool auth(session);
37			net::Oscar::BuddyTool buddy(session);
38			net::Oscar::ChatTool chat(session);
39			net::Oscar::IcbmTool icbm(session);
40			net::Oscar::InfoTool info(session);
41
42			auth.Login(login, password);
43
44	douglas	18	if (prompted) password.Clear();
45
46	douglas	17	this->buddy = &buddy;
47			this->icbm = &icbm;
48	douglas	22	this->info = &info;
49	douglas	16
50	douglas	22	buddy.OnOnline += etl::Bind<0>(&Collector::status, this);
51			icbm.OnReceive += etl::Bind<0>(&Collector::receive, this);
52	douglas	16
53	douglas	20	net::Oscar::StringSet buddies;
54	douglas	15
55	douglas	34	for (std::set<Buddy>::const_iterator buddy(this->buddies.begin()); buddy != this->buddies.end(); ++buddy)
56			buddies.Insert(*buddy);
57	douglas	15
58	douglas	20	buddy.Insert(buddies);
59	douglas	15
60			net::Oscar::Capabilities capabilities;
61
62			capabilities.Insert(net::Oscar::ChatCapability);
63	douglas	34	info.SetProfile("<font face=\"Tahoma\" size=2>I am <a href=\"http://computers.douglasthrift.net/zoe.xml\">Zoe</a>."
64	douglas	35	"</font> :-*", "<font face=\"Tahoma\" size=2>Mommy told me not to talk to strangers.</font> O:-)", capabilities);
65	douglas	15
66			// figure out whether or not we logged in
67
68	douglas	31	while (true) sleep(Minute(1));
69	douglas	15	}
70
71			void Collector::prompt()
72			{
73	douglas	18	if (!prompted) prompted = true;
74
75	douglas	16	ext::String prompt(login + "'s AIM Password: ");
76			char* password(getpass(prompt.NullTerminate()));
77	douglas	15
78			this->password = password;
79
80	douglas	34	for (size_t index(std::strlen(password)); index > 0; --index) password[index - 1] = '\0';
81	douglas	15	}
82	douglas	16
83	douglas	22	void Collector::status(const net::Oscar::UserInfo& user)
84	douglas	16	{
85	douglas	25	Buddy buddy(*buddies.find(user));
86	douglas	30	AwayMessage message(buddy, *info);
87	douglas	22
88	douglas	30	if (!ext::String(message).IsEmpty()) // XXX
89			{
90	douglas	34	ext::Handle<dbi::Connection> db(dbi::Connect(database.driver, database.host, database.user, database.password,
91			database.db));
92			ext::Handle<dbi::ResultSet> previous(db->Execute("SELECT message FROM messages WHERE stamp=(SELECT MAX(stamp) "
93			"FROM messages WHERE id='" + lexical_cast<ext::String>(buddy.getId()) + "') AND id='"
94			+ lexical_cast<ext::String>(buddy.getId()) + "'"));
95	douglas	31
96	douglas	34	if (previous->MoveNext()) if (ext::String(message) == previous->GetString("message")) return;
97	douglas	30
98	douglas	34	cerr << bright << red << "Collector::status(" << blue << buddy << red << ", " << reset << message << bright << red
99			<< ")\n" << reset;
100
101			db->Execute("INSERT INTO messages (stamp, message, id) VALUES ('" + ext::String(message.getStamp()) + "', '"
102			+ db->Escape(message) + "', '" + lexical_cast<ext::String>(buddy.getId()) + "')");
103	douglas	30	}
104	douglas	22	}
105
106			void Collector::receive(const ext::String& buddy, const ext::String& message)
107			{
108	douglas	34	cerr << bright << red << "Collector::receive(" << blue << Buddy(buddy) << red << ", " << reset << message << bright
109			<< red << ")\n" << reset;
110	douglas	31
111	douglas	20	if (buddies.find(buddy) != buddies.end())
112			{
113	douglas	22	Stamp stamp;
114
115			sleep(1);
116
117	douglas	35	icbm->Simple(buddy, ext::String(stamp) + " " + lexical_cast<ext::String>(std::time(NULL)) + " "
118			+ lexical_cast<ext::String>(api::GetWallTimerSeconds()));
119	douglas	20	}
120			else
121			{
122			sleep(1);
123
124	douglas	35	icbm->Simple(buddy, "<font face=\"Tahoma\" size=2>Mommy told me not to talk to strangers.</font> O:-)");
125	douglas	20	}
126	douglas	16	}