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 38 by douglas, 2004-07-22T22:42: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 == "-color")
44	+	{
45	+	if (!Zoe::color) Zoe::color = true;
46	+	}
47		else if (arg == "-D")
48		{
49		if (!Zoe::debug) Zoe::debug = true;
50		}
51		else
52		{
53	<	cerr << "Usage: " << Zoe::program << " [-config=config] [-D]\n";
53	>	cerr << "Usage: " << Zoe::program << " [-config=config] "
54	>	"[-collector] [-publisher] [-color] [-D]\n";
55
56		return 1;
57		}
#	Line 37 | Line 62 | int main(int argc, char* argv[])
62		return 0;
63		}
64
65	<	Zoe::Zoe() : rss("zoe.rss")
65	>	Zoe::Zoe()
66		{
67		configure();
68	+	initialize();
69
70	<	// This is just to figure out Jay's XML output, seems to have worked!
71	<	// It should really go in Rss.cpp later
72	<	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");
70	>	Collector collector(login, password, buddies, database, Zoe::collector);
71	>	Publisher publisher(buddies, database, Zoe::publisher);
72	>	}
73
74	<	rss.SetAttribute("version", "2.0");
74	>	bool Zoe::debug(false), Zoe::collector(false), Zoe::publisher(false),
75	>	Zoe::color(false);
76	>	ext::String Zoe::program, Zoe::config("zoe.xml");
77
78	<	xml::ScopeElement channel(rss, "channel");
78	>	ext::String Zoe::generator(Generator generator)
79	>	{
80	>	ext::String generator_(generator == all || generator == agent ? "Zoe" : "");
81
82	<	rss.OpenElement("title");
83	<	rss.OutputText("Zoe's Away Messages");
84	<	rss.CloseElement();
85	<	rss.OpenElement("link");
86	<	rss.OutputText("http://computers.douglasthrift.net/zoe.xml");
87	<	rss.CloseElement();
88	<	}
82	>	switch (generator)
83	>	{
84	>	case all:
85	>	generator_.InsertAllLast("/0.9");
86	>	case agent:
87	>	{
88	>	utsname system;
89
90	<	bool Zoe::debug(false);
91	<	ext::String Zoe::program, Zoe::config("zoe.xml");
90	>	uname(&system);
91	>
92	>	generator_.InsertAllLast(_S(" (") + system.sysname);
93	>	}
94	>
95	>	if (generator == agent)
96	>	{
97	>	generator_.InsertAllLast(")");
98	>
99	>	break;
100	>	}
101	>	else generator_.InsertAllLast("; ");
102	>	case url:
103	>	generator_.InsertAllLast("http://computers.douglasthrift.net/zoe.xml");
104	>
105	>	if (generator == all) generator_.InsertAllLast(")");
106	>
107	>	break;
108	>	case version:
109	>	generator_ = "0.9";
110	>	}
111	>
112	>	return generator_;
113	>	}
114
115		void Zoe::configure()
116		{
117	+	if (debug) cerr << "config = " << config << '\n';
118	+
119		ext::Handle<xml::Document> document(xml::Parse(config));
120		ext::Handle<xml::Node> zoe(*document/"zoe");
121
122	<	login = *zoe/"login";
123	<	password = *zoe/"password";
122	>	login = std::string(*zoe/"login");
123	>	password = std::string(*zoe/"password");
124	>	database.driver = std::string(*zoe/"database"/"driver");
125	>	database.host = std::string(*zoe/"database"/"host");
126	>	database.user = std::string(*zoe/"database"/"user");
127	>
128	>	if ((*zoe/"database"/"password").IsEmpty())
129	>	{
130	>	ext::String prompt(database.user + (!database.host.IsEmpty() ? "@" : "")
131	>	+ database.host + (!database.user.IsEmpty() ? "'s " : "")
132	>	+ "Database Password: ");
133	>	char* password(getpass(prompt.NullTerminate()));
134	>
135	>	database.password = password;
136	>
137	>	for (size_t index(std::strlen(password)); index > 0; --index)
138	>	password[index - 1] = '\0';
139	>	}
140	>	else database.password = std::string(*zoe/"database"/"password");
141	>
142	>	database.db = std::string(*zoe/"database"/"db");
143	>
144	>	if (debug) cerr << "login = " << login << "\npassword = "
145	>	<< std::string(password.GetSize(), '*') << "\ndatabase = {\n"
146	>	<< "   driver = " << database.driver << "\n   host=" << database.host
147	>	<< "\n   user = " << database.user << '\n'
148	>	<< "   password = " << std::string(database.password.GetSize(), '*')
149	>	<< "\n   db = " << database.db << "\n}\n";
150	>
151	>	xml::NodeSet buddies(*zoe/"buddy");
152	>
153	>	for (xml::NodeSet::Iterator buddy(buddies.Begin()); buddy != buddies.End();
154	>	++buddy)
155	>	{
156	>	this->buddies.insert(Buddy(**buddy/"login", **buddy/"rss",
157	>	**buddy/"atom"));
158	>	}
159	>
160	>	if (debug)
161	>	{
162	>	cerr << "buddies = {\n";
163	>
164	>	for (std::set<Buddy>::const_iterator buddy(this->buddies.begin());
165	>	buddy != this->buddies.end(); ++buddy)
166	>	{
167	>	cerr << "   " << *buddy << " = {\n      rss = " << buddy->getRss()
168	>	<< "\n      atom = " << buddy->getAtom() << "\n   }\n";
169	>	}
170	>
171	>	cerr << "}\n";
172	>	}
173	>
174	>	if (!collector && !publisher)
175	>	{
176	>	collector = true;
177	>	publisher = true;
178	>	}
179	>
180	>	if (debug) cerr << "collector = " << lexical_cast<ext::String>(collector)
181	>	<< "\npublisher = " << lexical_cast<ext::String>(publisher)
182	>	<< "\ncolor = " << lexical_cast<ext::String>(color) << '\n';
183	>	}
184	>
185	>	void Zoe::initialize()
186	>	{
187	>	ext::Handle<dbi::Connection> db(dbi::Connect(database.driver, database.host,
188	>	database.user, database.password, database.db));
189	>	ext::Handle<dbi::ResultSet> buddies(db->Execute("SELECT * FROM buddies"));
190	>	std::set<Buddy> buddies_;
191	>
192	>	while (buddies->MoveNext())
193	>	{
194	>	Buddy buddy(*this->buddies.find(buddies->GetString("buddy")));
195	>
196	>	buddy.setId(lexical_cast<ext::Uuid>(buddies->GetString("id")));
197	>	buddies_.insert(buddy);
198	>	}
199	>
200	>	std::vector<Buddy> difference;
201	>
202	>	set_difference(this->buddies.begin(), this->buddies.end(), buddies_.begin(),
203	>	buddies_.end(), std::insert_iterator<std::vector<Buddy> >(difference,
204	>	difference.begin()));
205	>
206	>	for (std::vector<Buddy>::iterator buddy(difference.begin());
207	>	buddy != difference.end(); ++buddy)
208	>	{
209	>	ext::Uuid id;
210	>
211	>	api::Uuid::CreateSequential(id);
212	>
213	>	buddy->setId(id);
214	>	db->Execute("INSERT INTO buddies (id, buddy) VALUES ('"
215	>	+ lexical_cast<ext::String>(id) + "', '" + ext::String(*buddy)
216	>	+ "')");
217	>
218	>	buddies_.insert(*buddy);
219	>	}
220
221	<	if (!(*zoe/"rss").IsEmpty()) rss = *zoe/"rss";
221	>	this->buddies = buddies_;
222		}

Diff Legend

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