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);
        icbm.RequestParams();

        net::Oscar::Capabilities capabilities;

        capabilities.Insert(net::Oscar::ChatCapability);
        info.SetProfile("<font size=3>I am <a href=\"http://computers.douglasthrift.net/zoe.xml\">Zoe</a>.</font> :-*", "<font size=3>Mommy told me not to talk to strangers.</font> O:-)", capabilities);

        // figure out whether or not we logged in

        while (true) sleep(60);
}

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));

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

void Collector::receive(const ext::String& buddy, const ext::String& message)
{
        if (buddies.find(buddy) != buddies.end())
        {
                Stamp stamp;
                
                sleep(1);

                icbm->Simple(buddy, stamp);
        }
        else
        {
                sleep(1);

                icbm->Simple(buddy, "<font size=3>Mommy told me not to talk to strangers. O:-)");
        }
}
Revision:	30
Committed:	2004-07-19T21:24:52-07:00 (20 years, 11 months ago) by douglas
File size:	2804 byte(s)
Log Message:	Cleaned up stuff, initial Away Message collection.
#	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			Collector::Collector(const ext::String& login, ext::String& password, const
20	douglas	21	std::set<Buddy>& buddies, const Database& database, bool start) :
21			login(login), password(password), buddies(buddies), database(database),
22			start(start), prompted(false)
23	douglas	13	{
24	douglas	15	if (start)
25			{
26			if (password.IsEmpty()) prompt();
27
28	douglas	20	collector.Spawn(etl::Bind<0>(&Collector::collect, this));
29	douglas	15	}
30	douglas	13	}
31	douglas	15
32	douglas	20	int Collector::collect()
33	douglas	15	{
34	douglas	23	cerr << bright << green << "Collector::collect()\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	20	for (std::set<Buddy>::const_iterator buddy(this->buddies.begin()); buddy !=
57			this->buddies.end(); ++buddy) buddies.Insert(*buddy);
58	douglas	15
59	douglas	20	buddy.Insert(buddies);
60	douglas	15	icbm.RequestParams();
61
62			net::Oscar::Capabilities capabilities;
63
64			capabilities.Insert(net::Oscar::ChatCapability);
65	douglas	20	info.SetProfile("<font size=3>I am <a href=\"http://computers.douglasthrift.net/zoe.xml\">Zoe</a>.</font> :-*", "<font size=3>Mommy told me not to talk to strangers.</font> O:-)", capabilities);
66	douglas	15
67			// figure out whether or not we logged in
68
69	douglas	16	while (true) sleep(60);
70	douglas	15	}
71
72			void Collector::prompt()
73			{
74	douglas	18	if (!prompted) prompted = true;
75
76	douglas	16	ext::String prompt(login + "'s AIM Password: ");
77			char* password(getpass(prompt.NullTerminate()));
78	douglas	15
79			this->password = password;
80
81	douglas	21	for (size_t index(std::strlen(password)); index > 0; --index)
82			password[index - 1] = '\0';
83	douglas	15	}
84	douglas	16
85	douglas	22	void Collector::status(const net::Oscar::UserInfo& user)
86	douglas	16	{
87	douglas	25	Buddy buddy(*buddies.find(user));
88	douglas	30	AwayMessage message(buddy, *info);
89	douglas	22
90	douglas	30	if (!ext::String(message).IsEmpty()) // XXX
91			{
92			ext::Handle<dbi::Connection> db(dbi::Connect(database.driver,
93			database.host, database.user, database.password, database.db));
94
95			db->Execute("INSERT INTO messages (stamp, message, id) VALUES ('" +
96			message.getStamp() + "', '" + db->Escape(message) + "', '" +
97			buddy.getId() + "')");
98			}
99	douglas	22	}
100
101			void Collector::receive(const ext::String& buddy, const ext::String& message)
102			{
103	douglas	20	if (buddies.find(buddy) != buddies.end())
104			{
105	douglas	22	Stamp stamp;
106
107			sleep(1);
108
109			icbm->Simple(buddy, stamp);
110	douglas	20	}
111			else
112			{
113			sleep(1);
114
115			icbm->Simple(buddy, "<font size=3>Mommy told me not to talk to strangers. O:-)");
116			}
117	douglas	16	}