more charset improvements.
[apps/madmutt.git] / state.c
1 /*
2  * Copyright notice from original mutt:
3  * Copyright (C) 1996-2000 Michael R. Elkins <me@mutt.org>
4  *
5  * This file is part of mutt-ng, see http://www.muttng.org/.
6  * It's licensed under the GNU General Public License,
7  * please see the file GPL in the top level source directory.
8  */
9 #if HAVE_CONFIG_H
10 #include "config.h"
11 #endif
12
13 #include <stdlib.h>
14 #include <stdio.h>
15 #include <stdarg.h>
16
17 #include <lib-mime/rfc3676.h>
18
19 #include "mutt.h"
20 #include "state.h"
21
22 static void state_prefix_put (const char *d, ssize_t dlen, STATE * s)
23 {
24   if (s->prefix)
25     while (dlen--)
26       state_prefix_putc (*d++, s);
27   else
28     fwrite (d, dlen, 1, s->fpout);
29 }
30
31 void mutt_convert_to_state (iconv_t cd, char *bufi, ssize_t * l, STATE * s)
32 {
33   char bufo[BUFO_SIZE];
34   const char *ib;
35   char *ob;
36   ssize_t ibl, obl;
37
38   if (!bufi) {
39     if (cd != (iconv_t) (-1)) {
40       ob = bufo, obl = sizeof (bufo);
41       my_iconv(cd, 0, 0, &ob, &obl);
42       if (ob != bufo)
43         state_prefix_put (bufo, ob - bufo, s);
44     }
45     if (Quotebuf[0] != '\0')
46       state_prefix_putc ('\n', s);
47     return;
48   }
49
50   if (cd == (iconv_t) (-1)) {
51     state_prefix_put (bufi, *l, s);
52     *l = 0;
53     return;
54   }
55
56   ib = bufi, ibl = *l;
57   for (;;) {
58     ob = bufo, obl = sizeof (bufo);
59     mutt_iconv (cd, &ib, &ibl, &ob, &obl, 0, "?");
60     if (ob == bufo)
61       break;
62     state_prefix_put (bufo, ob - bufo, s);
63   }
64   memmove (bufi, ib, ibl);
65   *l = ibl;
66 }
67
68 void state_prefix_putc (char c, STATE * s)
69 {
70   if (s->flags & M_PENDINGPREFIX) {
71     int i;
72
73     i = m_strlen(Quotebuf);
74     Quotebuf[i++] = c;
75     Quotebuf[i] = '\0';
76     if (i == sizeof (Quotebuf) - 1 || c == '\n') {
77       char buf[2 * SHORT_STRING];
78       int j = 0, offset = 0;
79       regmatch_t pmatch[1];
80
81       state_reset_prefix (s);
82       while (regexec
83              ((regex_t *) QuoteRegexp.rx, &Quotebuf[offset], 1, pmatch,
84               0) == 0)
85         offset += pmatch->rm_eo;
86
87       if (!option (OPTQUOTEEMPTY) && Quotebuf[offset] == '\n') {
88         buf[0] = '\n';
89         buf[1] = '\0';
90       }
91       else if (!option (OPTTEXTFLOWED) && option (OPTQUOTEQUOTED) && offset) {
92         for (i = 0; i < offset; i++)
93           if (Quotebuf[i] != ' ')
94             j = i;
95         strncpy (buf, Quotebuf, j + 1);
96         strcpy (buf + j + 1, Quotebuf + j);
97       }
98       else
99         snprintf (buf, sizeof (buf), "%s%s", NONULL (s->prefix), Quotebuf);
100
101       state_puts (buf, s);
102     }
103   }
104   else
105     state_putc (c, s);
106
107   if (c == '\n') {
108     state_set_prefix (s);
109     Quotebuf[0] = '\0';
110   }
111 }
112
113 int state_printf (STATE * s, const char *fmt, ...)
114 {
115   int rv;
116   va_list ap;
117
118   va_start (ap, fmt);
119   rv = vfprintf (s->fpout, fmt, ap);
120   va_end (ap);
121
122   return rv;
123 }
124
125 void state_mark_attach (STATE * s)
126 {
127   if ((s->flags & M_DISPLAY) && !m_strcmp(Pager, "builtin"))
128     state_puts (AttachmentMarker, s);
129 }
130
131 void state_attach_puts (const char *t, STATE * s)
132 {
133   if (*t != '\n')
134     state_mark_attach (s);
135   while (*t) {
136     state_putc (*t, s);
137     if (*t++ == '\n' && *t)
138       if (*t != '\n')
139         state_mark_attach (s);
140   }
141 }