ViewVC Help

View File | Revision Log | Show Annotations | Download File | View Changeset | Root Listing

root/zoe/trunk/Zoe.cpp

Comparing trunk/Zoe.cpp (file contents):
Revision 11 by douglas, 2004-07-10T22:30:54-07:00 vs.
Revision 36 by douglas, 2004-07-22T01:19:40-07:00

#	Line 4 | Line 4
4		//
5		// $Id$
6
7	–	#include "Zoe.hpp"
7		#include "Matcher.hpp"
8	+	#include "Collector.hpp"
9	+	#include "Publisher.hpp"
10	+
11	+	#include <algorithm>
12	+	#include <cstring>
13	+	#include <iterator>
14	+
15	+	extern "C"
16	+	{
17	+	#include <pwd.h>
18	+	#include <sys/utsname.h>
19	+	#include <unistd.h>
20	+	}
21
22		int main(int argc, char* argv[])
23		{
#	Line 20 | Line 32 | int main(int argc, char* argv[])
32		{
33		Zoe::config = matcher[1];
34		}
35	+	else if (arg == "-collector")
36	+	{
37	+	if (!Zoe::collector) Zoe::collector = true;
38	+	}
39	+	else if (arg == "-publisher")
40	+	{
41	+	if (!Zoe::publisher) Zoe::publisher = true;
42	+	}
43		else if (arg == "-D")
44		{
45		if (!Zoe::debug) Zoe::debug = true;
46		}
47		else
48		{
49	<	cerr << "Usage: " << Zoe::program << " [-config=config] [-D]\n";
49	>	cerr << bright << "Usage: " << cyan << Zoe::program << reset << " ["
50	>	<< bright << blue <<  "-config=" << yellow << "config" << reset
51	>	<< "] [" << bright << blue << "-collector" << reset << "] ["
52	>	<< bright << blue << "-publisher" << reset << "] [" << bright
53	>	<< blue << "-D" << reset << "]\n";
54
55		return 1;
56		}
#	Line 37 | Line 61 | int main(int argc, char* argv[])
61		return 0;
62		}
63
64	<	Zoe::Zoe() : rss("zoe.rss")
64	>	Zoe::Zoe()
65		{
66		configure();
67	+	initialize();
68
69	<	// This is just to figure out Jay's XML output, seems to have worked!
70	<	// It should really go in Rss.cpp later
71	<	api::FileOutputStream fout(this->rss != "-" ? this->rss : "/dev/null");
47	<	xml::TextWriter rss(this->rss != "-" ? fout : api::Cout.GetStream());
48	<	xml::ScopeElement root(rss, "rss");
69	>	Collector collector(login, password, buddies, database, Zoe::collector);
70	>	Publisher publisher(buddies, database, Zoe::publisher);
71	>	}
72
73	<	rss.SetAttribute("version", "2.0");
73	>	bool Zoe::debug(false), Zoe::collector(false), Zoe::publisher(false);
74	>	ext::String Zoe::program, Zoe::config("zoe.xml");
75
76	<	xml::ScopeElement channel(rss, "channel");
76	>	ext::String Zoe::generator(Generator generator)
77	>	{
78	>	ext::String generator_(generator == all || generator == agent ? "Zoe" : "");
79
80	<	rss.OpenElement("title");
81	<	rss.OutputText("Zoe's Away Messages");
82	<	rss.CloseElement();
83	<	rss.OpenElement("link");
84	<	rss.OutputText("http://computers.douglasthrift.net/zoe.xml");
85	<	rss.CloseElement();
86	<	}
80	>	switch (generator)
81	>	{
82	>	case all:
83	>	generator_.InsertAllLast("/0.9");
84	>	case agent:
85	>	{
86	>	utsname system;
87
88	<	bool Zoe::debug(false);
89	<	ext::String Zoe::program, Zoe::config("zoe.xml");
88	>	uname(&system);
89	>
90	>	generator_.InsertAllLast(_S(" (") + system.sysname);
91	>	}
92	>
93	>	if (generator == agent)
94	>	{
95	>	generator_.InsertAllLast(")");
96	>
97	>	break;
98	>	}
99	>	else generator_.InsertAllLast("; ");
100	>	case url:
101	>	generator_.InsertAllLast("http://computers.douglasthrift.net/zoe.xml");
102	>
103	>	if (generator == all) generator_.InsertAllLast(")");
104	>
105	>	break;
106	>	case version:
107	>	generator_ = "0.9";
108	>	}
109	>
110	>	return generator_;
111	>	}
112
113		void Zoe::configure()
114		{
115	+	if (debug) cerr << "config = " << config << '\n';
116	+
117		ext::Handle<xml::Document> document(xml::Parse(config));
118		ext::Handle<xml::Node> zoe(*document/"zoe");
119
120	<	login = *zoe/"login";
121	<	password = *zoe/"password";
120	>	login = std::string(*zoe/"login");
121	>	password = std::string(*zoe/"password");
122	>	database.driver = std::string(*zoe/"database"/"driver");
123	>	database.host = std::string(*zoe/"database"/"host");
124	>	database.user = std::string(*zoe/"database"/"user");
125	>
126	>	if ((*zoe/"database"/"password").IsEmpty())
127	>	{
128	>	ext::String prompt(database.user + (!database.host.IsEmpty() ? "@" : "") + database.host
129	>	+ (!database.user.IsEmpty() ? "'s " : "") + "Database Password: ");
130	>	char* password(getpass(prompt.NullTerminate()));
131	>
132	>	database.password = password;
133	>
134	>	for (size_t index(std::strlen(password)); index > 0; --index) password[index - 1] = '\0';
135	>	}
136	>	else database.password = std::string(*zoe/"database"/"password");
137	>
138	>	database.db = std::string(*zoe/"database"/"db");
139	>
140	>	if (debug) cerr << "login = " << login << "\npassword = " << std::string(password.GetSize(), '*') << "\ndatabase = {\n"
141	>	<< "   driver = " << database.driver << "\n   host=" << database.host << "\n   user = " << database.user << '\n'
142	>	<< "   password = " << std::string(database.password.GetSize(), '*') << "\n   db = " << database.db << "\n}\n";
143	>
144	>	xml::NodeSet buddies(*zoe/"buddy");
145	>
146	>	for (xml::NodeSet::Iterator buddy(buddies.Begin()); buddy != buddies.End(); ++buddy)
147	>	this->buddies.insert(Buddy(**buddy/"login", **buddy/"rss"));
148	>
149	>	if (debug)
150	>	{
151	>	cerr << "buddies = {\n";
152	>
153	>	for (std::set<Buddy>::const_iterator buddy(this->buddies.begin()); buddy != this->buddies.end(); ++buddy)
154	>	cerr << "   " << *buddy << " = " << buddy->getRss() << '\n';
155	>
156	>	cerr << "}\n";
157	>	}
158	>
159	>	if (!collector && !publisher)
160	>	{
161	>	collector = true;
162	>	publisher = true;
163	>	}
164	>
165	>	if (debug) cerr << "collector = " << lexical_cast<ext::String>(collector)
166	>	<< "\npublisher = " << lexical_cast<ext::String>(publisher) << '\n';
167	>	}
168	>
169	>	void Zoe::initialize()
170	>	{
171	>	ext::Handle<dbi::Connection> db(dbi::Connect(database.driver, database.host, database.user, database.password,
172	>	database.db));
173	>	ext::Handle<dbi::ResultSet> buddies(db->Execute("SELECT * FROM buddies"));
174	>	std::set<Buddy> buddies_;
175	>
176	>	while (buddies->MoveNext())
177	>	{
178	>	Buddy buddy(*this->buddies.find(buddies->GetString("buddy")));
179	>
180	>	buddy.setId(lexical_cast<ext::Uuid>(buddies->GetString("id")));
181	>	buddies_.insert(buddy);
182	>	}
183	>
184	>	std::vector<Buddy> difference;
185	>
186	>	set_difference(this->buddies.begin(), this->buddies.end(), buddies_.begin(), buddies_.end(),
187	>	std::insert_iterator<std::vector<Buddy> >(difference, difference.begin()));
188	>
189	>	for (std::vector<Buddy>::iterator buddy(difference.begin()); buddy != difference.end(); ++buddy)
190	>	{
191	>	ext::Uuid id;
192	>
193	>	api::Uuid::CreateSequential(id);
194	>
195	>	buddy->setId(id);
196	>	db->Execute("INSERT INTO buddies (id, buddy) VALUES ('" + lexical_cast<ext::String>(id) + "', '"
197	>	+ ext::String(*buddy) + "')");
198	>
199	>	buddies_.insert(*buddy);
200	>	}
201
202	<	if (!(*zoe/"rss").IsEmpty()) rss = *zoe/"rss";
202	>	this->buddies = buddies_;
203		}

Diff Legend

–	Removed lines
+	Added lines
<	Changed lines
>	Changed lines