/* $Id$ */

/*
 * Copyright (c) 2007 Nicholas Marriott <nicholas.marriott@gmail.com>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF MIND, USE, DATA OR PROFITS, WHETHER
 * IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING
 * OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <sys/types.h>

#include <string.h>

#include "fdm.h"
#include "deliver.h"

int	 deliver_remove_header_deliver(struct deliver_ctx *, struct actitem *);
void	 deliver_remove_header_desc(struct actitem *, char *, size_t);

struct deliver deliver_remove_header = {
	"remove-header",
	DELIVER_INCHILD,
	deliver_remove_header_deliver,
	deliver_remove_header_desc
};

int
deliver_remove_header_deliver(struct deliver_ctx *dctx, struct actitem *ti)
{
	struct account				*a = dctx->account;
	struct mail				*m = dctx->mail;
	struct deliver_remove_header_data	*data = ti->data;
	char					*ptr, *hdr;
	size_t					 len, off, wrap;
	u_int					 i, j;


	for (j = 0; j < ARRAY_LENGTH(data->hdrs); j++) {
		hdr = replacestr(
		    &ARRAY_ITEM(data->hdrs, j), m->tags, m, &m->rml);
		if (hdr == NULL || *hdr == '\0') {
			if (hdr != NULL)
				xfree(hdr);
			log_warnx("%s: empty header", a->name);
			return (DELIVER_FAILURE);
		}
		log_debug2("%s: removing header: %s", a->name, hdr);

		ARRAY_FREE(&m->wrapped);
		m->wrapchar = '\0';
		fill_wrapped(m);

		set_wrapped(m, ' ');

		while ((ptr = match_header(m, hdr, &len, 0)) != NULL) {
			log_debug3("%s: found header to remove: %.*s", a->name,
			    (int) len, ptr);

			/* Remove the header. */
			memmove(
			    ptr, ptr + len, m->size - len - (ptr - m->data));
			m->size -= len;
			m->body -= len;

			/* Fix up the wrapped array. */
			off = ptr - m->data;
			i = 0;
			while (i < ARRAY_LENGTH(&m->wrapped)) {
				wrap = ARRAY_ITEM(&m->wrapped, i);
				if (wrap >= off + len) {
					ARRAY_SET(&m->wrapped, i, wrap - len);
					i++;
				} else if (wrap >= off)
					ARRAY_REMOVE(&m->wrapped, i);
				else
					i++;
			}
		}
	}

	/* Invalidate the match data since stuff may have moved. */
	m->rml.valid = 0;

	set_wrapped(m, '\n');
	return (DELIVER_SUCCESS);
}

void
deliver_remove_header_desc(struct actitem *ti, char *buf, size_t len)
{
	struct deliver_remove_header_data	*data = ti->data;
	char					*hdrs;

	hdrs = fmt_replstrs("remove-header ", data->hdrs);
	strlcpy(buf, hdrs, len);
	xfree(hdrs);
}