/*
 *	aegis - project change supervisor
 *	Copyright (C) 1999, 2001 Peter Miller;
 *	All rights reserved.
 *
 *	This program is free software; you can redistribute it and/or modify
 *	it under the terms of the GNU General Public License as published by
 *	the Free Software Foundation; either version 2 of the License, or
 *	(at your option) any later version.
 *
 *	This program is distributed in the hope that it will be useful,
 *	but WITHOUT ANY WARRANTY; without even the implied warranty of
 *	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *	GNU General Public License for more details.
 *
 *	You should have received a copy of the GNU General Public License
 *	along with this program; if not, write to the Free Software
 *	Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111, USA.
 *
 * MANIFEST: functions to manipulate mains
 */

#include <ac/stdio.h>
#include <ac/stdlib.h>

#include <arglex.h>
#include <error.h>
#include <input/base64.h>
#include <input/crlf.h>
#include <input/file.h>
#include <input/quoted_print.h>
#include <input/uudecode.h>
#include <os.h>
#include <output/base64.h>
#include <output/file.h>
#include <output/quoted_print.h>
#include <output/uuencode.h>
#include <progname.h>
#include <str.h>


enum
{
	arglex_token_input,
	arglex_token_output,
	arglex_token_quoted_printable,
	arglex_token_unix_to_unix
};


static arglex_table_ty argtab[] =
{
	{ "-Quoted_Printable", arglex_token_quoted_printable, },
	{ "-Input", arglex_token_input, },
	{ "-Output", arglex_token_output, },
	{ "-Unix_to_Unix_Encode", arglex_token_unix_to_unix, },
	{ "-Unix_to_Unix_Decode", arglex_token_unix_to_unix, },
	{ "-Unix_to_Unix", arglex_token_unix_to_unix, },
	{ 0 } /* end marker */
};


static void usage _((void));

static void
usage()
{
	const char	*progname;

	progname = progname_get();
	fprintf(stderr, "Usage: %s [ -i | -o ][ <infile> [ <outfile> ]]\n", progname);
	exit(1);
}


static void test_input_base64 _((string_ty *, string_ty *));

static void
test_input_base64(ifn, ofn)
	string_ty	*ifn;
	string_ty	*ofn;
{
	input_ty	*ifp;
	output_ty	*ofp;

	ifp = input_file_open(ifn);
	ofp = output_file_text_open(ofn);

	/*
	 * skip lines until we find a blank line
	 * as a crude way to pass over the rfc822  header
	 */
	for (;;)
	{
		string_ty	*s;

		s = input_one_line(ifp);
		if (!s)
			break;
		if (!s->str_length)
		{
			str_free(s);
			break;
		}
		str_free(s);
	}

	/*
	 * filter the rest
	 */
	ifp = input_base64(ifp, 1);
	input_to_output(ifp, ofp);
	input_delete(ifp);
	output_delete(ofp);
}


static void test_input_qp _((string_ty *, string_ty *));

static void
test_input_qp(ifn, ofn)
	string_ty	*ifn;
	string_ty	*ofn;
{
	input_ty	*ifp;
	output_ty	*ofp;

	ifp = input_file_open(ifn);
	ofp = output_file_text_open(ofn);

	/*
	 * skip lines until we find a blank line
	 * as a crude way to pass over the rfc822  header
	 */
	for (;;)
	{
		string_ty	*s;

		s = input_one_line(ifp);
		if (!s)
			break;
		if (!s->str_length)
		{
			str_free(s);
			break;
		}
		str_free(s);
	}

	ifp = input_quoted_printable(ifp, 1);
	input_to_output(ifp, ofp);
	input_delete(ifp);
	output_delete(ofp);
}


static void test_input_uu _((string_ty *, string_ty *));

static void
test_input_uu(ifn, ofn)
	string_ty	*ifn;
	string_ty	*ofn;
{
	input_ty	*ifp;
	output_ty	*ofp;

	ifp = input_file_open(ifn);
	ofp = output_file_text_open(ofn);

	/*
	 * skip lines until we find a blank line
	 * as a crude way to pass over the rfc822  header
	 */
	for (;;)
	{
		string_ty	*s;

		s = input_one_line(ifp);
		if (!s)
			break;
		if (!s->str_length)
		{
			str_free(s);
			break;
		}
		str_free(s);
	}

	ifp = input_uudecode(ifp, 1);
	input_to_output(ifp, ofp);
	input_delete(ifp);
	output_delete(ofp);
}


static void test_output_base64 _((string_ty *, string_ty *));

static void
test_output_base64(ifn, ofn)
	string_ty	*ifn;
	string_ty	*ofn;
{
	input_ty	*ifp;
	output_ty	*ofp;

	ifp = input_file_open(ifn);
	ifp = input_crlf(ifp, 1);
	ofp = output_file_text_open(ofn);
	output_fprintf(ofp, "Content-Type: application/x-aegis-test\n");
	output_fprintf(ofp, "Content-Transfer-Encoding: base64\n");
	output_fprintf(ofp, "\n");
	ofp = output_base64(ofp, 1);
	input_to_output(ifp, ofp);
	input_delete(ifp);
	output_delete(ofp);
}


static void test_output_qp _((string_ty *, string_ty *));

static void
test_output_qp(ifn, ofn)
	string_ty	*ifn;
	string_ty	*ofn;
{
	input_ty	*ifp;
	output_ty	*ofp;

	ifp = input_file_open(ifn);
	ifp = input_crlf(ifp, 1);
	ofp = output_file_text_open(ofn);
	output_fprintf(ofp, "Content-Type: application/x-aegis-test\n");
	output_fprintf(ofp, "Content-Transfer-Encoding: quoted-printable\n");
	output_fprintf(ofp, "\n");
	ofp = output_quoted_printable(ofp, 1, 0);
	input_to_output(ifp, ofp);
	input_delete(ifp);
	output_delete(ofp);
}


static void test_output_uu _((string_ty *, string_ty *));

static void
test_output_uu(ifn, ofn)
	string_ty	*ifn;
	string_ty	*ofn;
{
	input_ty	*ifp;
	output_ty	*ofp;

	ifp = input_file_open(ifn);
	ifp = input_crlf(ifp, 1);
	ofp = output_file_text_open(ofn);
	output_fprintf(ofp, "Content-Type: application/x-aegis-test\n");
	output_fprintf(ofp, "Content-Transfer-Encoding: uuencode\n");
	output_fprintf(ofp, "\n");
	ofp = output_uuencode(ofp, 1);
	input_to_output(ifp, ofp);
	input_delete(ifp);
	output_delete(ofp);
}


int main _((int, char **));

int
main(argc, argv)
	int		argc;
	char		**argv;
{
	string_ty	*ifn;
	string_ty	*ofn;
	void		(*ifunc)_((string_ty *, string_ty *));
	void		(*ofunc)_((string_ty *, string_ty *));
	void		(*func)_((string_ty *, string_ty *));

	ifunc = test_input_base64;
	ofunc = test_output_base64;
	arglex_init(argc, argv, argtab);
	str_initialize();
	arglex();
	os_become_init_mortal();

	ifn = 0;
	ofn = 0;
	func = 0;
	while (arglex_token != arglex_token_eoln)
	{
		switch (arglex_token)
		{
		default:
			usage();

		case arglex_token_quoted_printable:
			if (func)
				usage();
			ifunc = test_input_qp;
			ofunc = test_output_qp;
			break;

		case arglex_token_unix_to_unix:
			if (func)
				usage();
			ifunc = test_input_uu;
			ofunc = test_output_uu;
			break;

		case arglex_token_stdio:
			if (!ifn)
				ifn = str_from_c("");
			else if (!ofn)
				ofn = str_from_c("");
			else
				usage();
			break;

		case arglex_token_string:
			if (!ifn)
				ifn = str_from_c(arglex_value.alv_string);
			else if (!ofn)
				ofn = str_from_c(arglex_value.alv_string);
			else
				usage();
			break;

		case arglex_token_input:
			if (func)
			{
				too_many:
				error_raw("too many test functions specified");
				usage();
			}
			func = ifunc;
			break;

		case arglex_token_output:
			if (func)
				goto too_many;
			func = ofunc;
			break;
		}
		arglex();
	}
	if (!func)
	{
		error_raw("no test function specified");
		usage();
	}

	os_become_orig();
	func(ifn, ofn);
	exit(0);
	return 0;
}