X-Git-Url: https://git.notmuchmail.org/git?p=notmuch;a=blobdiff_plain;f=gmime-filter-reply.c;fp=gmime-filter-reply.c;h=3e298e1b647d4bf34ed91c6d033b6914f67fc50b;hp=0000000000000000000000000000000000000000;hb=357aba3ec8177c11a7ce22cbe26d92482f6a5e53;hpb=50144f95cababfb73027ca95ad1fb303c235a893

diff --git a/gmime-filter-reply.c b/gmime-filter-reply.c
new file mode 100644
index 00000000..3e298e1b
--- /dev/null
+++ b/gmime-filter-reply.c
@@ -0,0 +1,208 @@
+/*
+ * Copyright © 2009 Keith Packard <keithp@keithp.com>
+ *
+ * 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-reply.h"
+
+/**
+ * SECTION: gmime-filter-reply
+ * @title: GMimeFilterReply
+ * @short_description: Add/remove reply markers
+ *
+ * A #GMimeFilter for adding or removing reply markers
+ **/
+
+
+static void g_mime_filter_reply_class_init (GMimeFilterReplyClass *klass);
+static void g_mime_filter_reply_init (GMimeFilterReply *filter, GMimeFilterReplyClass *klass);
+static void g_mime_filter_reply_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_reply_get_type (void)
+{
+	static GType type = 0;
+
+	if (!type) {
+		static const GTypeInfo info = {
+			sizeof (GMimeFilterReplyClass),
+			NULL, /* base_class_init */
+			NULL, /* base_class_finalize */
+			(GClassInitFunc) g_mime_filter_reply_class_init,
+			NULL, /* class_finalize */
+			NULL, /* class_data */
+			sizeof (GMimeFilterReply),
+			0,    /* n_preallocs */
+			(GInstanceInitFunc) g_mime_filter_reply_init,
+			NULL	/* value_table */
+		};
+
+		type = g_type_register_static (GMIME_TYPE_FILTER, "GMimeFilterReply", &info, (GTypeFlags) 0);
+	}
+
+	return type;
+}
+
+
+static void
+g_mime_filter_reply_class_init (GMimeFilterReplyClass *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_reply_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_reply_init (GMimeFilterReply *filter, GMimeFilterReplyClass *klass)
+{
+	(void) klass;
+	filter->saw_nl = TRUE;
+	filter->saw_angle = FALSE;
+}
+
+static void
+g_mime_filter_reply_finalize (GObject *object)
+{
+	G_OBJECT_CLASS (parent_class)->finalize (object);
+}
+
+
+static GMimeFilter *
+filter_copy (GMimeFilter *filter)
+{
+	GMimeFilterReply *reply = (GMimeFilterReply *) filter;
+
+	return g_mime_filter_reply_new (reply->encode);
+}
+
+static void
+filter_filter (GMimeFilter *filter, char *inbuf, size_t inlen, size_t prespace,
+	       char **outbuf, size_t *outlen, size_t *outprespace)
+{
+	GMimeFilterReply *reply = (GMimeFilterReply *) filter;
+	register const char *inptr = inbuf;
+	const char *inend = inbuf + inlen;
+	char *outptr;
+
+	(void) prespace;
+	if (reply->encode) {
+		g_mime_filter_set_size (filter, 3 * inlen, FALSE);
+
+		outptr = filter->outbuf;
+		while (inptr < inend) {
+			if (reply->saw_nl) {
+				*outptr++ = '>';
+				*outptr++ = ' ';
+				reply->saw_nl = FALSE;
+			}
+			if (*inptr == '\n')
+				reply->saw_nl = TRUE;
+			else
+				reply->saw_nl = FALSE;
+
+			*outptr++ = *inptr++;
+		}
+	} else {
+		g_mime_filter_set_size (filter, inlen + 1, FALSE);
+
+		outptr = filter->outbuf;
+		while (inptr < inend) {
+			if (reply->saw_nl) {
+				if (*inptr == '>')
+					reply->saw_angle = TRUE;
+				else
+					*outptr++ = *inptr;
+				reply->saw_nl = FALSE;
+			} else if (reply->saw_angle) {
+				if (*inptr == ' ')
+					;
+				else
+					*outptr++ = *inptr;
+				reply->saw_angle = FALSE;
+			} else {
+				if (*inptr == '\n')
+					reply->saw_nl = TRUE;
+				*outptr++ = *inptr;
+			}
+
+			inptr++;
+		}
+	}
+
+	*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)
+{
+	GMimeFilterReply *reply = (GMimeFilterReply *) filter;
+
+	reply->saw_nl = TRUE;
+	reply->saw_angle = FALSE;
+}
+
+
+/**
+ * g_mime_filter_reply_new:
+ * @encode: %TRUE if the filter should encode or %FALSE otherwise
+ * @dots: encode/decode dots (as for SMTP)
+ *
+ * Creates a new #GMimeFilterReply filter.
+ *
+ * If @encode is %TRUE, then all lines will be prefixed by "> ",
+ * otherwise any lines starting with "> " will have that removed
+ *
+ * Returns: a new #GMimeFilterReply filter.
+ **/
+GMimeFilter *
+g_mime_filter_reply_new (gboolean encode)
+{
+	GMimeFilterReply *new_reply;
+
+	new_reply = (GMimeFilterReply *) g_object_newv (GMIME_TYPE_FILTER_REPLY, 0, NULL);
+	new_reply->encode = encode;
+
+	return (GMimeFilter *) new_reply;
+}
+