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 12 by douglas, 2004-07-11T20:45:38-07:00 vs.
Revision 39 by douglas, 2004-07-23T17:52:53-07:00

#	Line 5 | Line 5
5		// $Id$
6
7		#include "Matcher.hpp"
8	<	#include "Buddy.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 40 | Line 65 | int main(int argc, char* argv[])
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 = 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	+	std::string link(*zoe/"link");
152		xml::NodeSet buddies(*zoe/"buddy");
153
154		for (xml::NodeSet::Iterator buddy(buddies.Begin()); buddy != buddies.End();
155		++buddy)
156		{
157	<	this->buddies.insert(Buddy(**buddy/"login", **buddy/"rss"));
157	>	Buddy buddy_(**buddy/"login", **buddy/"rss", **buddy/"atom",
158	>	!(**buddy/"link").IsEmpty() ? **buddy/"link" : link);
159	>
160	>	this->buddies.insert(buddy_);
161		}
162	+
163	+	if (debug)
164	+	{
165	+	cerr << "buddies = {\n";
166	+
167	+	for (std::set<Buddy>::const_iterator buddy(this->buddies.begin());
168	+	buddy != this->buddies.end(); ++buddy)
169	+	{
170	+	cerr << "   " << *buddy << " = {\n      rss = " << buddy->getRss()
171	+	<< "\n      atom = " << buddy->getAtom() << "\n      link = "
172	+	<< buddy->getLink() << "\n   }\n";
173	+	}
174	+
175	+	cerr << "}\n";
176	+	}
177	+
178	+	if (!collector && !publisher) { collector = true; publisher = true; }
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	+	this->buddies = buddies_;
222		}

Diff Legend

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