[ViewVC] Diff of: zoe/trunk/Zoe.cpp

Comparing trunk/Zoe.cpp (file contents):
Revision 5 by douglas, 2004-07-09T15:04:18-07:00 vs.
Revision 39 by douglas, 2004-07-23T17:52:53-07:00

+//
+// $Id$
-–
+#include "Zoe.hpp"
+#include "Matcher.hpp"
-+
+#include "Collector.hpp"
-+
+#include "Publisher.hpp"
-+
-+
+#include <algorithm>
-+
+#include <cstring>
-+
+#include <iterator>
-+
-+
+extern "C"
-+
+{
-+
+#include <pwd.h>
-+
+#include <sys/utsname.h>
-+
+#include <unistd.h>
-+
+}
+int main(int argc, char* argv[])
-<
+        //
->
+        Zoe::program = argv[0];
->
->
+        for (int index(1); index < argc; ++index)
->
+        {
->
+                ext::String arg(argv[index]);
->
+                Matcher matcher;
->
->
+                if (arg == matcher("^-config=(.*)$"))
->
+                {
->
+                        Zoe::config = matcher[1];
->
+                }
->
+                else if (arg == "-collector")
->
+                {
->
+                        if (!Zoe::collector) Zoe::collector = true;
->
+                }
->
+                else if (arg == "-publisher")
->
+                {
->
+                        if (!Zoe::publisher) Zoe::publisher = true;
->
+                }
->
+                else if (arg == "-color")
->
+                {
->
+                        if (!Zoe::color) Zoe::color = true;
->
+                }
->
+                else if (arg == "-D")
->
+                {
->
+                        if (!Zoe::debug) Zoe::debug = true;
->
+                }
->
+                else
->
+                {
->
+                        cerr << "Usage: " << Zoe::program << " [-config=config] "
->
+                                "[-collector] [-publisher] [-color] [-D]\n";
->
->
+                        return 1;
->
+                }
->
+        }
->
->
+        Zoe zoe;
+        return 0;
-+
-+
+Zoe::Zoe()
-+
+{
-+
+        configure();
-+
+        initialize();
-+
-+
+        Collector collector(login, password, buddies, database, Zoe::collector);
-+
+        Publisher publisher(buddies, database, Zoe::publisher);
-+
+}
-+
-+
+bool Zoe::debug(false), Zoe::collector(false), Zoe::publisher(false),
-+
+         Zoe::color(false);
-+
+ext::String Zoe::program, Zoe::config("zoe.xml");
-+
-+
+ext::String Zoe::generator(Generator generator)
-+
+{
-+
+        ext::String generator_(generator == all || generator == agent ? "Zoe" : "");
-+
-+
+        switch (generator)
-+
+        {
-+
+        case all:
-+
+                generator_.InsertAllLast("/0.9");
-+
+        case agent:
-+
+                {
-+
+                        utsname system;
-+
-+
+                        uname(&system);
-+
-+
+                        generator_.InsertAllLast(_S(" (") + system.sysname);
-+
+                }
-+
-+
+                if (generator == agent)
-+
+                {
-+
+                        generator_.InsertAllLast(")");
-+
-+
+                        break;
-+
+                }
-+
+                else generator_.InsertAllLast("; ");
-+
+        case url:
-+
+                generator_.InsertAllLast("http://computers.douglasthrift.net/zoe.xml");
-+
-+
+                if (generator == all) generator_.InsertAllLast(")");
-+
-+
+                break;
-+
+        case version:
-+
+                generator_ = "0.9";
-+
+        }
-+
-+
+        return generator_;
-+
+}
-+
-+
+void Zoe::configure()
-+
+{
-+
+        if (debug) cerr << "config = " << config << '\n';
-+
-+
+        ext::Handle<xml::Document> document(xml::Parse(config));
-+
+        ext::Handle<xml::Node> zoe(*document/"zoe");
-+
-+
+        login = std::string(*zoe/"login");
-+
+        password = std::string(*zoe/"password");
-+
+        database.driver = std::string(*zoe/"database"/"driver");
-+
+        database.host = std::string(*zoe/"database"/"host");
-+
+        database.user = std::string(*zoe/"database"/"user");
-+
-+
+        if ((*zoe/"database"/"password").IsEmpty())
-+
+        {
-+
+                ext::String prompt(database.user + (!database.host.IsEmpty() ? "@" : "")
-+
+                        + database.host + (!database.user.IsEmpty() ? "'s " : "")
-+
+                        + "Database Password: ");
-+
+                char* password(getpass(prompt.NullTerminate()));
-+
-+
+                database.password = password;
-+
-+
+                for (size_t index(std::strlen(password)); index > 0; --index)
-+
+                        password[index - 1] = '\0';
-+
+        }
-+
+        else database.password = std::string(*zoe/"database"/"password");
-+
-+
+        database.db = std::string(*zoe/"database"/"db");
-+
-+
+        if (debug) cerr << "login = " << login << "\npassword = "
-+
+                << std::string(password.GetSize(), '*') << "\ndatabase = {\n"
-+
+                << "   driver = " << database.driver << "\n   host=" << database.host
-+
+                << "\n   user = " << database.user << '\n'
-+
+                << "   password = " << std::string(database.password.GetSize(), '*')
-+
+                << "\n   db = " << database.db << "\n}\n";
-+
-+
+        std::string link(*zoe/"link");
-+
+        xml::NodeSet buddies(*zoe/"buddy");
-+
-+
+        for (xml::NodeSet::Iterator buddy(buddies.Begin()); buddy != buddies.End();
-+
+                ++buddy)
-+
+        {
-+
+                Buddy buddy_(**buddy/"login", **buddy/"rss", **buddy/"atom",
-+
+                        !(**buddy/"link").IsEmpty() ? **buddy/"link" : link);
-+
-+
+                this->buddies.insert(buddy_);
-+
+        }
-+
-+
+        if (debug)
-+
+        {
-+
+                cerr << "buddies = {\n";
-+
-+
+                for (std::set<Buddy>::const_iterator buddy(this->buddies.begin());
-+
+                        buddy != this->buddies.end(); ++buddy)
-+
+                {
-+
+                        cerr << "   " << *buddy << " = {\n      rss = " << buddy->getRss()
-+
+                                << "\n      atom = " << buddy->getAtom() << "\n      link = "
-+
+                                << buddy->getLink() << "\n   }\n";
-+
+                }
-+
-+
+                cerr << "}\n";
-+
+        }
-+
-+
+        if (!collector && !publisher) { collector = true; publisher = true; }
-+
-+
+        if (debug) cerr << "collector = " << lexical_cast<ext::String>(collector)
-+
+                << "\npublisher = " << lexical_cast<ext::String>(publisher)
-+
+                << "\ncolor = " << lexical_cast<ext::String>(color) << '\n';
-+
+}
-+
-+
+void Zoe::initialize()
-+
+{
-+
+        ext::Handle<dbi::Connection> db(dbi::Connect(database.driver, database.host,
-+
+                database.user, database.password, database.db));
-+
+        ext::Handle<dbi::ResultSet> buddies(db->Execute("SELECT * FROM buddies"));
-+
+        std::set<Buddy> buddies_;
-+
-+
+        while (buddies->MoveNext())
-+
+        {
-+
+                Buddy buddy(*this->buddies.find(buddies->GetString("buddy")));
-+
-+
+                buddy.setId(lexical_cast<ext::Uuid>(buddies->GetString("id")));
-+
+                buddies_.insert(buddy);
-+
+        }
-+
-+
+        std::vector<Buddy> difference;
-+
-+
+        set_difference(this->buddies.begin(), this->buddies.end(), buddies_.begin(),
-+
+                buddies_.end(), std::insert_iterator<std::vector<Buddy> >(difference,
-+
+                difference.begin()));
-+
-+
+        for (std::vector<Buddy>::iterator buddy(difference.begin());
-+
+                buddy != difference.end(); ++buddy)
-+
+        {
-+
+                ext::Uuid id;
-+
-+
+                api::Uuid::CreateSequential(id);
-+
-+
+                buddy->setId(id);
-+
+                db->Execute("INSERT INTO buddies (id, buddy) VALUES ('"
-+
+                        + lexical_cast<ext::String>(id) + "', '" + ext::String(*buddy)
-+
+                        + "')");
-+
-+
+                buddies_.insert(*buddy);
-+
+        }
-+
-+
+        this->buddies = buddies_;
-+
+}

Diff Legend

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

Comparing trunk/Zoe.cpp (file contents): Revision 5 by douglas, 2004-07-09T15:04:18-07:00 vs. Revision 39 by douglas, 2004-07-23T17:52:53-07:00

Diff Legend

Comparing trunk/Zoe.cpp (file contents):
Revision 5 by douglas, 2004-07-09T15:04:18-07:00 vs.
Revision 39 by douglas, 2004-07-23T17:52:53-07:00