/*
 * booth: the voting booth for a STV vote
 *
 * very dumb and quick
 */
#include <stdio.h>
#include <string.h>
#include <malloc.h>
#include <stdlib.h>
#include <unistd.h>
#include <ctype.h>
#include <dbm.h>

char line[1000];
char from[200];
#define RANKINGS 100
char *votes[RANKINGS];


char *
TRIM(char *s)
{
    char *p;
    while (isspace(*s) && *s)
	++s;
    for (p=s; *p; ++p)
	;
    while (p>s && isspace(*p))
	--p;
    *p = 0;
    return s;
} /* TRIM */


main()
{
    FILE *mail;
    FILE *audit;
    FILE *spool;
    char *ptr;

    enum {BEGIN, FROM, VOTE, END} state = BEGIN;
    int i;

    for (i=0; i<RANKINGS; i++)
	votes[i] = 0;

    spool = tmpfile();

    while (gets(line) && state != END) {
	fprintf(spool, "%s\n", line);
	ptr = line;
	if (*ptr == '>')
	    ++ptr;
	if (state == BEGIN && strncmp("From:", line, 5) == 0) {
	    strncpy(from, TRIM(line+5), 80);
	    state = FROM;
	}
	else if (state == FROM && strcmp("--vote--", ptr) == 0) {
	    state = VOTE;
	}
	else if (state == VOTE && strcmp("--done--", ptr) == 0) {
	    state = END;
	}
	if (state == VOTE) {
	    /* vote format: [.*] what */
	    char *p;
	    int thisrank;

	    if (ptr[0] != '[')
		continue;	/* invalid line */
	    thisrank = atoi(1+ptr);
	    if (thisrank < 1 || thisrank > RANKINGS)
		continue;
	    if (votes[thisrank-1])
		exit(1);
	    for (p=ptr; *p && *p != ']'; p++)
		;
	    if (*p) {
		p = strdup(TRIM(p+1));
		for (i=0; i<RANKINGS; i++) {
		    /* no duplicate votes */
		    if (votes[i] && strcmp(votes[i], p) == 0) {
			free(p);
			goto nextone;
		    }
		}
		if (p && *p)
		    votes[thisrank-1] = p;
	    }
    nextone:;
	}
    }
    if (state == END) {
	datum key, data;
	int j;

	key.dptr = from;
	key.dsize = strlen(from);

	line[0] = 0;
	for (j=0,i=0; i<RANKINGS; i++)
	    if (votes[i]) {
		if (j > 0)
		    strcat(line, "|");
		strcat(line, votes[i]);
		j++;
	    }
	if (audit = fopen("/usr/orc/vote/audit", "a")) {
	    fprintf(audit, "[%s][%s]\n", from, line);
	    fclose(audit);
	}

	data.dptr = line;
	data.dsize = strlen(line);

	dbminit("/usr/orc/vote/ballots");
	i = delete(key);
	if (data.dsize > 0)
	    store(key, data);
	dbmclose();

	sprintf(line, "/usr/lib/sendmail '%s'", from);

	if (mail = popen(line, "w")) {
	    int j;

	    fprintf(mail, "To: %s\n", from);
	    fprintf(mail, "From: vote@gehenna.apple.com (the voting machine)\n"
			  "Subject: Your vote\n"
                          "\n"
	                  "Your vote has been received and is saved as:\n\n");
	    if (data.dsize == 0)
		fprintf(mail,"\t>>please cancel my vote<<\n");
	    else {
		fprintf(mail, "\t--vote--\n");
		for (j=1,i=0; i<RANKINGS; i++)
		    if (votes[i])
			fprintf(mail, "\t[%d] %s\n", j++, votes[i]);
		fprintf(mail, "\t--done--\n");
	    }
	    fprintf(mail, "\n"
			  "If you wish to change your vote, send another\n"
	                  "one in and it will replace this one.  If you wish\n"
	                  "to cancel your vote, send a vote that contains nothing\n"
			  "between the --vote-- and --done-- lines.\n"
			  "\n"
			  " -the voting machine\n");
	    pclose(mail);
	}
    }
    else {
	sprintf(line, "/usr/lib/sendmail '%s' orc", from);
	if (mail = popen(line, "w")) {
	    fprintf(mail, "To: %s\n", from);
	    fprintf(mail, "From: vote@gehenna.apple.com (the voting machine)\n"
			  "Subject: Your vote\n"
                          "\n"
			  "I can't read your vote -- I'm forwarding it to\n"
			  "my supervisor, but it may not be counted\n"
			  "\n"
			  "---[mail message follows]---\n");
	    rewind(spool);
	    while (fgets(line, 1000, spool))
		fprintf(mail, ">%s", line);
	    pclose(mail);
	}
    }
}