Simplifications.
[apps/madmutt.git] / editmsg.c
1 /*
2  * Copyright notice from original mutt:
3  * Copyright (C) 1999-2000 Thomas Roessler <roessler@does-not-exist.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
10 /* simple, editor-based message editing */
11
12 #include <lib-lib/lib-lib.h>
13 #include <lib-ui/lib-ui.h>
14 #include <lib-mx/mx.h>
15
16 #include "mutt.h"
17 #include "copy.h"
18
19 /*
20  * return value:
21  * 
22  * 1    message not modified
23  * 0    message edited successfully
24  * -1   error
25  */
26
27 static int edit_one_message (CONTEXT * ctx, HEADER * cur)
28 {
29   char tmp[_POSIX_PATH_MAX];
30   char buff[STRING];
31   int omagic;
32   int oerrno;
33   int rc;
34
35   unsigned short o_read;
36   unsigned short o_old;
37
38   int of, cf;
39
40   CONTEXT tmpctx;
41   MESSAGE *msg;
42
43   FILE *fp = NULL;
44
45   struct stat sb;
46   time_t mtime = 0;
47   ssize_t size = 0;
48
49   mutt_mktemp (tmp);
50
51   omagic = DefaultMagic;
52   DefaultMagic = M_MBOX;
53
54   rc = (mx_open_mailbox (tmp, M_NEWFOLDER, &tmpctx) == NULL) ? -1 : 0;
55
56   DefaultMagic = omagic;
57
58   if (rc == -1) {
59     mutt_error (_("could not create temporary folder: %s"), strerror (errno));
60     return -1;
61   }
62
63   rc = mutt_append_message (&tmpctx, ctx, cur, 0, CH_NOLEN |
64                             (ctx->magic == M_MBOX ? 0 : CH_NOSTATUS));
65   oerrno = errno;
66
67   mx_close_mailbox (&tmpctx, NULL);
68
69   if (rc == -1) {
70     mutt_error (_("could not write temporary mail folder: %s"),
71                 strerror (oerrno));
72     goto bail;
73   }
74
75   if (stat (tmp, &sb) == 0) {
76     mtime = sb.st_mtime;
77     size = sb.st_size;
78   }
79
80   /*
81    * 2002-09-05 me@sigpipe.org
82    * The file the user is going to edit is not a real mbox, so we need to
83    * truncate the last newline in the temp file, which is logically part of
84    * the message separator, and not the body of the message.  If we fail to
85    * remove it, the message will grow by one line each time the user edits
86    * the message.
87    */
88   if (size != 0 && truncate (tmp, --size) == -1) {
89     mutt_error (_("could not truncate temporary mail folder: %s"),
90                 strerror (errno));
91     goto bail;
92   }
93
94   mutt_edit_file(tmp);
95
96   if ((rc = stat (tmp, &sb)) == -1) {
97     mutt_error (_("Can't stat %s: %s"), tmp, strerror (errno));
98     goto bail;
99   }
100
101   if (sb.st_size == 0) {
102     mutt_message (_("Message file is empty!"));
103     rc = 1;
104     goto bail;
105   }
106
107   if (sb.st_mtime == mtime && sb.st_size == size) {
108     mutt_message (_("Message not modified!"));
109     rc = 1;
110     goto bail;
111   }
112
113   if ((fp = fopen (tmp, "r")) == NULL) {
114     rc = -1;
115     mutt_error (_("Can't open message file: %s"), strerror (errno));
116     goto bail;
117   }
118
119   if (mx_open_mailbox (ctx->path, M_APPEND, &tmpctx) == NULL) {
120     rc = -1;
121     mutt_error (_("Can't append to folder: %s"), strerror (errno));
122     goto bail;
123   }
124
125   of = 0;
126   cf = (tmpctx.magic == M_MBOX ? 0 : CH_NOSTATUS);
127
128   if (fgets (buff, sizeof (buff), fp) && is_from (buff, NULL, 0, NULL)) {
129     if (tmpctx.magic == M_MBOX)
130       cf = CH_FROM | CH_FORCE_FROM;
131   } else {
132     of = M_ADD_FROM;
133   }
134
135   /* 
136    * XXX - we have to play games with the message flags to avoid
137    * problematic behaviour with maildir folders.
138    *
139    */
140
141   o_read = cur->read;
142   o_old = cur->old;
143   cur->read = cur->old = 0;
144   msg = mx_open_new_message (&tmpctx, cur, of);
145   cur->read = o_read;
146   cur->old = o_old;
147
148   if (msg == NULL) {
149     mutt_error (_("Can't append to folder: %s"), strerror (errno));
150     mx_close_mailbox (&tmpctx, NULL);
151     goto bail;
152   }
153
154   if ((rc =
155        mutt_copy_hdr (fp, msg->fp, 0, sb.st_size, CH_NOLEN | cf,
156                       NULL)) == 0) {
157     fputc ('\n', msg->fp);
158     rc = mutt_copy_stream (fp, msg->fp);
159   }
160
161   rc = mx_commit_message (msg, &tmpctx);
162   mx_close_message (&msg);
163
164   mx_close_mailbox (&tmpctx, NULL);
165
166 bail:
167   m_fclose(&fp);
168
169   if (rc >= 0)
170     unlink (tmp);
171
172   if (rc == 0) {
173     mutt_set_flag (Context, cur, M_DELETE, 1);
174     mutt_set_flag (Context, cur, M_READ, 1);
175
176     if (option (OPTDELETEUNTAG))
177       mutt_set_flag (Context, cur, M_TAG, 0);
178   }
179   else if (rc == -1)
180     mutt_message (_("Error. Preserving temporary file: %s"), tmp);
181
182
183   return rc;
184 }
185
186 int mutt_edit_message (CONTEXT * ctx, HEADER * hdr)
187 {
188   int i, j;
189
190   if (hdr)
191     return edit_one_message (ctx, hdr);
192
193
194   for (i = 0; i < ctx->vcount; i++) {
195     j = ctx->v2r[i];
196     if (ctx->hdrs[j]->tagged) {
197       if (edit_one_message (ctx, ctx->hdrs[j]) == -1)
198         return -1;
199     }
200   }
201
202   return 0;
203 }