repos/Smersh/Smersh.cpp

// Smersh
//
// Douglas Thrift
//
// $Id$

#include "Smersh.hpp"
#include "Matcher.hpp"
#include "Person.hpp"
#include "Daemon.hpp"
#include "Redirector.hpp"

string program;
bool debug(false);

int main(int argc, char* argv[])
{
        program = argv[0];

        int port(8080);
        bool daemon(false), redirector(false);
        string redirect;

        for (int index(1); index < argc; ++index)
        {
                string arg(argv[index]);
                Matcher matcher;

                if (arg == "-D")
                {
                        if (!debug) debug = true;
                }
                else if (arg == "-daemon")
                {
                        if (!daemon) daemon = true;
                }
                else if (arg == matcher("^-redirector=(.+)$"))
                {
                        if (!redirector) redirector = true;

                        redirect = matcher[1];
                }
                else if (arg == matcher("^-port=([0-9]+)$"))
                {
                        port = lexical_cast<int>(matcher[1]);
                }
        }

        if (daemon)
        {
                Daemon daemon(port);
        }
        else if (redirector)
        {
                Redirector redirector(port, redirect);
        }
        else
        {
                Smersh smersh;
        }

        return 0;
}

Smersh::Smersh(istream& sin, ostream& sout)
{
        parse(sin);
        smersh(sout);
}

void Smersh::parse(istream& sin)
{
        string query(sgetenv("QUERY_STRING"));

        if (sgetenv("REQUEST_METHOD") == "POST") getline(sin, query);
        if (query == "") return;

        istringstream input(query);

        do
        {
                string name, value;

                getline(input, name, '=');
                getline(input, value, '&');

                cgi.insert(pair<string, string>(name, value));
        }
        while (input.good());
}

void Smersh::smersh(ostream& sout)
{
        sout << "Content-Type: text/html\n\n";

        vector<Person> people;

        for (multimap<string, string>::iterator itor(cgi.lower_bound("sn")); itor
                != cgi.upper_bound("sn"); ++itor)
        {
                string sn(itor->second);

                for (string::size_type index(0); index < sn.length(); ++index)
                {
                        while (!isalnum(sn[index]) && index < sn.length())
                        {
                                sn.erase(index, 1);
                        }

                        if (index < sn.length()) sn[index] = tolower(sn[index]);
                }

                Person person(sn);

                people.push_back(person);

                if (person.isMultiple())
                {
                        people.insert(people.end(), person.beginMultiple(),
                                person.endMultiple());
                        person.clearMultiple();
                }
        }

        sout << "<html><head><title>Hello, " << flush;

        output(sout, people);

        sout << "!</title></head><body><p style=\"font-variant: small-caps\"><stro"
                << "ng><font face=\"Comic Sans MS\" size=\"4\">Hello, " << flush;

        output(sout, people);

        sout << "!</font></strong></p></body></html>\n";
}

void Smersh::output(ostream& sout, const vector<Person>& people)
{
        for (vector<Person>::const_iterator person(people.begin()); person !=
                people.end(); ++person)
        {
                sout << *person;

                if (person + 2 == people.end())
                {
                        sout << (people.size() > 2 ? "," : "") << " and ";
                }
                else if (person + 1 != people.end())
                {
                        sout << ", ";
                }

                sout << flush;
        }
}
Revision:	166
Committed:	2004-06-18T22:24:07-07:00 (21 years ago) by Douglas Thrift
File size:	2696 byte(s)
Log Message:	... with good friends and good times ...
#	User	Rev	Content
1	Douglas Thrift	162	// Smersh
2			//
3			// Douglas Thrift
4			//
5			// $Id$
6
7			#include "Smersh.hpp"
8	Douglas Thrift	164	#include "Matcher.hpp"
9	Douglas Thrift	165	#include "Person.hpp"
10			#include "Daemon.hpp"
11			#include "Redirector.hpp"
12	Douglas Thrift	162
13	Douglas Thrift	165	string program;
14			bool debug(false);
15
16	Douglas Thrift	162	int main(int argc, char* argv[])
17			{
18	Douglas Thrift	165	program = argv[0];
19	Douglas Thrift	162
20	Douglas Thrift	165	int port(8080);
21			bool daemon(false), redirector(false);
22			string redirect;
23
24			for (int index(1); index < argc; ++index)
25			{
26			string arg(argv[index]);
27			Matcher matcher;
28
29			if (arg == "-D")
30			{
31			if (!debug) debug = true;
32			}
33			else if (arg == "-daemon")
34			{
35			if (!daemon) daemon = true;
36			}
37			else if (arg == matcher("^-redirector=(.+)$"))
38			{
39			if (!redirector) redirector = true;
40
41			redirect = matcher[1];
42			}
43			else if (arg == matcher("^-port=([0-9]+)$"))
44			{
45			port = lexical_cast<int>(matcher[1]);
46			}
47			}
48
49			if (daemon)
50			{
51			Daemon daemon(port);
52			}
53			else if (redirector)
54			{
55			Redirector redirector(port, redirect);
56			}
57			else
58			{
59			Smersh smersh;
60			}
61
62	Douglas Thrift	162	return 0;
63			}
64
65	Douglas Thrift	166	Smersh::Smersh(istream& sin, ostream& sout)
66	Douglas Thrift	162	{
67	Douglas Thrift	166	parse(sin);
68			smersh(sout);
69	Douglas Thrift	162	}
70
71	Douglas Thrift	166	void Smersh::parse(istream& sin)
72	Douglas Thrift	162	{
73			string query(sgetenv("QUERY_STRING"));
74
75	Douglas Thrift	166	if (sgetenv("REQUEST_METHOD") == "POST") getline(sin, query);
76	Douglas Thrift	162	if (query == "") return;
77
78			istringstream input(query);
79
80			do
81			{
82			string name, value;
83
84			getline(input, name, '=');
85			getline(input, value, '&');
86
87			cgi.insert(pair<string, string>(name, value));
88			}
89			while (input.good());
90			}
91	Douglas Thrift	166
92			void Smersh::smersh(ostream& sout)
93			{
94			sout << "Content-Type: text/html\n\n";
95
96			vector<Person> people;
97
98			for (multimap<string, string>::iterator itor(cgi.lower_bound("sn")); itor
99			!= cgi.upper_bound("sn"); ++itor)
100			{
101			string sn(itor->second);
102
103			for (string::size_type index(0); index < sn.length(); ++index)
104			{
105			while (!isalnum(sn[index]) && index < sn.length())
106			{
107			sn.erase(index, 1);
108			}
109
110			if (index < sn.length()) sn[index] = tolower(sn[index]);
111			}
112
113			Person person(sn);
114
115			people.push_back(person);
116
117			if (person.isMultiple())
118			{
119			people.insert(people.end(), person.beginMultiple(),
120			person.endMultiple());
121			person.clearMultiple();
122			}
123			}
124
125			sout << "<html><head><title>Hello, " << flush;
126
127			output(sout, people);
128
129			sout << "!</title></head><body><p style=\"font-variant: small-caps\"><stro"
130			<< "ng><font face=\"Comic Sans MS\" size=\"4\">Hello, " << flush;
131
132			output(sout, people);
133
134			sout << "!</font></strong></p></body></html>\n";
135			}
136
137			void Smersh::output(ostream& sout, const vector<Person>& people)
138			{
139			for (vector<Person>::const_iterator person(people.begin()); person !=
140			people.end(); ++person)
141			{
142			sout << *person;
143
144			if (person + 2 == people.end())
145			{
146			sout << (people.size() > 2 ? "," : "") << " and ";
147			}
148			else if (person + 1 != people.end())
149			{
150			sout << ", ";
151			}
152
153			sout << flush;
154			}
155			}