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 5 by douglas, 2004-07-09T15:04:18-07:00 vs.
Revision 44 by douglas, 2004-08-15T23:40:19-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	<	int main(int argc, char* argv[])
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 zoe(const std::vector<const char*> args)
23		{
24	<	//
24	>	Zoe::program = args[0];
25	>
26	>	for (int index(1); index < args.size(); ++index)
27	>	{
28	>	ext::String arg(args[index]);
29	>	Matcher matcher;
30	>
31	>	if (arg == matcher("^-config=(.*)$"))
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 Zoe::usage();
52	>	}
53	>
54	>	Zoe zoe;
55
56		return 0;
57		}
58	+
59	+	int main(int argc, char* argv[])
60	+	{
61	+	std::vector<const char*> args(argv, argv + argc);
62	+
63	+	return app::be::ThreadMain_(etl::BindAll(&zoe, args));
64	+	}
65	+
66	+	Zoe::Zoe()
67	+	{
68	+	if (!(collector || publisher)) usage();
69	+
70	+	configure();
71	+	initialize();
72	+
73	+	Collector collector(login, password, buddies, database, Zoe::collector);
74	+	Publisher publisher(buddies, database, Zoe::publisher);
75	+	}
76	+
77	+	bool Zoe::debug(false), Zoe::collector(false), Zoe::publisher(false),
78	+	Zoe::color(false);
79	+	ext::String Zoe::program, Zoe::config("zoe.xml");
80	+
81	+	void Zoe::usage()
82	+	{
83	+	cerr << "Usage: " << Zoe::program << " [-config=config] [-collector] "
84	+	"[-publisher] [-color] [-D]\n";
85	+
86	+	exit(1);
87	+	}
88	+
89	+	ext::String Zoe::generator(Generator generator)
90	+	{
91	+	ext::String generator_(generator == all || generator == agent ? "Zoe" : "");
92	+
93	+	switch (generator)
94	+	{
95	+	case all:
96	+	generator_.InsertAllLast("/0.9");
97	+	case agent:
98	+	{
99	+	utsname system;
100	+
101	+	uname(&system);
102	+
103	+	generator_.InsertAllLast(_S(" (") + system.sysname);
104	+	}
105	+
106	+	if (generator == agent)
107	+	{
108	+	generator_.InsertAllLast(")");
109	+
110	+	break;
111	+	}
112	+	else generator_.InsertAllLast("; ");
113	+	case url:
114	+	generator_.InsertAllLast("http://computers.douglasthrift.net/zoe.xml");
115	+
116	+	if (generator == all) generator_.InsertAllLast(")");
117	+
118	+	break;
119	+	case version:
120	+	generator_ = "0.9";
121	+	}
122	+
123	+	return generator_;
124	+	}
125	+
126	+	void Zoe::configure()
127	+	{
128	+	if (debug) cerr << "config = " << config << '\n';
129	+
130	+	ext::Handle<xml::Document> document(xml::Parse(config));
131	+	ext::Handle<xml::Node> zoe(*document/"zoe");
132	+
133	+	login = std::string(*zoe/"login");
134	+	password = std::string(*zoe/"password");
135	+	database.driver = std::string(*zoe/"database"/"driver");
136	+	database.host = std::string(*zoe/"database"/"host");
137	+	database.user = std::string(*zoe/"database"/"user");
138	+
139	+	if ((*zoe/"database"/"password").IsEmpty())
140	+	{
141	+	ext::String prompt(database.user + (!database.host.IsEmpty() ? "@" : "")
142	+	+ database.host + (!database.user.IsEmpty() ? "'s " : "")
143	+	+ "Database Password: ");
144	+	char* password(getpass(prompt.NullTerminate()));
145	+
146	+	database.password = password;
147	+
148	+	for (size_t index(std::strlen(password)); index > 0; --index)
149	+	password[index - 1] = '\0';
150	+	}
151	+	else database.password = std::string(*zoe/"database"/"password");
152	+
153	+	database.db = std::string(*zoe/"database"/"db");
154	+
155	+	if (debug) cerr << "login = " << login << "\npassword = "
156	+	<< std::string(password.GetSize(), '*') << "\ndatabase = {\n"
157	+	<< "   driver = " << database.driver << "\n   host=" << database.host
158	+	<< "\n   user = " << database.user << '\n'
159	+	<< "   password = " << std::string(database.password.GetSize(), '*')
160	+	<< "\n   db = " << database.db << "\n}\n";
161	+
162	+	std::string link(*zoe/"link");
163	+	xml::NodeSet buddies(*zoe/"buddy");
164	+
165	+	for (xml::NodeSet::Iterator buddy(buddies.Begin()); buddy != buddies.End();
166	+	++buddy)
167	+	{
168	+	Buddy buddy_(**buddy/"login", **buddy/"rss", **buddy/"atom",
169	+	!(**buddy/"link").IsEmpty() ? **buddy/"link" : link);
170	+
171	+	this->buddies.insert(buddy_);
172	+	}
173	+
174	+	if (debug)
175	+	{
176	+	cerr << "buddies = {\n";
177	+
178	+	for (std::set<Buddy>::const_iterator buddy(this->buddies.begin());
179	+	buddy != this->buddies.end(); ++buddy)
180	+	{
181	+	cerr << "   " << *buddy << " = {\n      rss = " << buddy->getRss()
182	+	<< "\n      atom = " << buddy->getAtom() << "\n      link = "
183	+	<< buddy->getLink() << "\n   }\n";
184	+	}
185	+
186	+	cerr << "}\n";
187	+	}
188	+
189	+	if (debug) cerr << "collector = " << lexical_cast<ext::String>(collector)
190	+	<< "\npublisher = " << lexical_cast<ext::String>(publisher)
191	+	<< "\ncolor = " << lexical_cast<ext::String>(color) << '\n';
192	+	}
193	+
194	+	void Zoe::initialize()
195	+	{
196	+	ext::Handle<dbi::Connection> db(dbi::Connect(database.driver, database.host,
197	+	database.user, database.password, database.db));
198	+	ext::Handle<dbi::ResultSet> buddies(db->Execute("SELECT * FROM buddies"));
199	+	std::set<Buddy> buddies_;
200	+
201	+	while (buddies->MoveNext())
202	+	{
203	+	Buddy buddy(*this->buddies.find(buddies->GetString("buddy")));
204	+
205	+	buddy.setId(lexical_cast<ext::Uuid>(buddies->GetString("id")));
206	+	buddies_.insert(buddy);
207	+	}
208	+
209	+	std::vector<Buddy> difference;
210	+
211	+	set_difference(this->buddies.begin(), this->buddies.end(), buddies_.begin(),
212	+	buddies_.end(), std::insert_iterator<std::vector<Buddy> >(difference,
213	+	difference.begin()));
214	+
215	+	for (std::vector<Buddy>::iterator buddy(difference.begin());
216	+	buddy != difference.end(); ++buddy)
217	+	{
218	+	ext::Uuid id;
219	+
220	+	api::Uuid::CreateSequential(id);
221	+
222	+	buddy->setId(id);
223	+	db->Execute("INSERT INTO buddies (id, buddy) VALUES ('"
224	+	+ lexical_cast<ext::String>(id) + "', '" + ext::String(*buddy)
225	+	+ "')");
226	+
227	+	buddies_.insert(*buddy);
228	+	}
229	+
230	+	this->buddies = buddies_;
231	+	}

Diff Legend

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