/*
 * Copyright © 2009 Keith Packard <keithp@keithp.com>
 * Copyright © 2010 Michal Sojka <sojkam1@fel.cvut.cz>
 *
 * 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 3 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-1307 USA.
 */

#include "gmime-filter-headers.h"
#include <string.h>
#include <gmime/gmime-utils.h>
#include <glib/gprintf.h>
#include <stdlib.h>
#include <xutil.h>

/**
 * SECTION: gmime-filter-headers
 * @title: GMimeFilterHeaders
 * @short_description: Add/remove headers markers
 *
 * A #GMimeFilter for decoding rfc2047 encoded headers to UTF-8
 **/


static void g_mime_filter_headers_class_init (GMimeFilterHeadersClass *klass);
static void g_mime_filter_headers_init (GMimeFilterHeaders *filter, GMimeFilterHeadersClass *klass);
static void g_mime_filter_headers_finalize (GObject *object);

static GMimeFilter *filter_copy (GMimeFilter *filter);
static void filter_filter (GMimeFilter *filter, char *in, size_t len, size_t prespace,
			   char **out, size_t *outlen, size_t *outprespace);
static void filter_complete (GMimeFilter *filter, char *in, size_t len, size_t prespace,
			     char **out, size_t *outlen, size_t *outprespace);
static void filter_reset (GMimeFilter *filter);


static GMimeFilterClass *parent_class = NULL;

GType
g_mime_filter_headers_get_type (void)
{
	static GType type = 0;

	if (!type) {
		static const GTypeInfo info = {
			sizeof (GMimeFilterHeadersClass),
			NULL, /* base_class_init */
			NULL, /* base_class_finalize */
			(GClassInitFunc) g_mime_filter_headers_class_init,
			NULL, /* class_finalize */
			NULL, /* class_data */
			sizeof (GMimeFilterHeaders),
			0,    /* n_preallocs */
			(GInstanceInitFunc) g_mime_filter_headers_init,
			NULL	/* value_table */
		};

		type = g_type_register_static (GMIME_TYPE_FILTER, "GMimeFilterHeaders", &info, (GTypeFlags) 0);
	}

	return type;
}


static void
g_mime_filter_headers_class_init (GMimeFilterHeadersClass *klass)
{
	GObjectClass *object_class = G_OBJECT_CLASS (klass);
	GMimeFilterClass *filter_class = GMIME_FILTER_CLASS (klass);

	parent_class = (GMimeFilterClass *) g_type_class_ref (GMIME_TYPE_FILTER);

	object_class->finalize = g_mime_filter_headers_finalize;

	filter_class->copy = filter_copy;
	filter_class->filter = filter_filter;
	filter_class->complete = filter_complete;
	filter_class->reset = filter_reset;
}

static void
g_mime_filter_headers_init (GMimeFilterHeaders *filter, GMimeFilterHeadersClass *klass)
{
	(void) klass;
	filter->saw_nl = TRUE;
	filter->line = NULL;
	filter->line_size = 0;
	filter->lineptr = NULL;
}

static void
g_mime_filter_headers_finalize (GObject *object)
{
	free (GMIME_FILTER_HEADERS (object)->line);
	G_OBJECT_CLASS (parent_class)->finalize (object);
}


static GMimeFilter *
filter_copy (GMimeFilter *filter)
{
	(void) filter;
	return g_mime_filter_headers_new ();
}

static void
output_decoded_header (GMimeFilterHeaders *headers, char **outptr)
{
	char *colon, *name, *s, *decoded_value;
	size_t offset;
	gint ret;

	colon = strchr (headers->line, ':');
	if (colon == NULL)
		return;

	name = headers->line;
	*colon = '\0';
	s = colon + 1;
	while (*s == ' ' || *s == '\t')
		s++;
	decoded_value = g_mime_utils_header_decode_text(s);
	if (decoded_value == NULL)
		return;
	offset = *outptr - GMIME_FILTER (headers)->outbuf;
	g_mime_filter_set_size (GMIME_FILTER (headers), strlen(name) + 2 +
			       strlen(decoded_value) + 2, TRUE);
	*outptr = GMIME_FILTER (headers)->outbuf + offset;
	ret = g_sprintf (*outptr, "%s: %s\n", name, decoded_value);
	if (ret > 0)
		*outptr += ret;
	free (decoded_value);
}

static void
output_final_newline (GMimeFilterHeaders *headers, char **outptr)
{
	size_t offset;

	offset = *outptr - GMIME_FILTER (headers)->outbuf;
	g_mime_filter_set_size (GMIME_FILTER (headers), 1, TRUE);
	*outptr = GMIME_FILTER (headers)->outbuf + offset;
	*(*outptr)++ = '\n';
}

static void
filter_filter (GMimeFilter *filter, char *inbuf, size_t inlen, size_t prespace,
	       char **outbuf, size_t *outlen, size_t *outprespace)
{
	GMimeFilterHeaders *headers = (GMimeFilterHeaders *) filter;
	register const char *inptr = inbuf;
	const char *inend = inbuf + inlen;
	char *lineptr, *lineend, *outptr;

	(void) prespace;
	if (headers->line == NULL) {
		headers->line_size = 200;
		headers->lineptr = headers->line = malloc (headers->line_size);
	}
	lineptr = headers->lineptr;
	lineend = headers->line + headers->line_size - 1;
	if (lineptr == NULL)
		return;
	outptr = filter->outbuf;
	while (inptr < inend) {
		if (*inptr == '\n') {
			if (headers->saw_nl)
				output_final_newline(headers, &outptr);
			headers->saw_nl = TRUE;
			inptr++;
			continue;
		}

		if (lineptr == lineend) {
			headers->line_size *= 2;
			headers->line = xrealloc (headers->line, headers->line_size);
			lineptr = headers->line + (headers->line_size / 2) - 1;
			lineend = headers->line + headers->line_size - 1;
		}

		if (headers->saw_nl && *inptr != ' ' && *inptr != '\t') {
			*lineptr = '\0';
			output_decoded_header (headers, &outptr);
			lineptr = headers->line;
		}
		if (headers->saw_nl && (*inptr == ' ' || *inptr == '\t')) {
			*lineptr = ' ';
			lineptr++;
			while (inptr < inend && (*inptr == ' ' || *inptr == '\t'))
				inptr++;
			headers->saw_nl = FALSE;
			continue;
		}
		headers->saw_nl = FALSE;

		if (*inptr != '\r')
			*lineptr++ = *inptr;
		inptr++;
	}
	if (headers->saw_nl) {
		*lineptr = '\0';
		output_decoded_header (headers, &outptr);
		lineptr = headers->line;
	}
	headers->lineptr = lineptr;
	*outlen = outptr - filter->outbuf;
	*outprespace = filter->outpre;
	*outbuf = filter->outbuf;
}

static void
filter_complete (GMimeFilter *filter, char *inbuf, size_t inlen, size_t prespace,
		 char **outbuf, size_t *outlen, size_t *outprespace)
{
	if (inbuf && inlen)
		filter_filter (filter, inbuf, inlen, prespace, outbuf, outlen, outprespace);
}

static void
filter_reset (GMimeFilter *filter)
{
	GMimeFilterHeaders *headers = (GMimeFilterHeaders *) filter;

	headers->saw_nl = TRUE;
	free(headers->line);
	headers->line = NULL;
	headers->line_size = 0;
}


/**
 * g_mime_filter_headers_new:
 * @encode: %TRUE if the filter should encode or %FALSE otherwise
 * @dots: encode/decode dots (as for SMTP)
 *
 * Creates a new #GMimeFilterHeaders filter.
 *
 * If @encode is %TRUE, then all lines will be prefixed by "> ",
 * otherwise any lines starting with "> " will have that removed
 *
 * Returns: a new #GMimeFilterHeaders filter.
 **/
GMimeFilter *
g_mime_filter_headers_new (void)
{
	GMimeFilterHeaders *new_headers;

	new_headers = (GMimeFilterHeaders *) g_object_newv (GMIME_TYPE_FILTER_HEADERS, 0, NULL);

	return (GMimeFilter *) new_headers;
}