1 | |
---|
2 | /* |
---|
3 | * burst.c -- explode digests into individual messages |
---|
4 | * |
---|
5 | * $Id: burst.c,v 1.1.1.1 1999-02-07 18:14:13 danw Exp $ |
---|
6 | */ |
---|
7 | |
---|
8 | #include <h/mh.h> |
---|
9 | |
---|
10 | static struct swit switches[] = { |
---|
11 | #define INPLSW 0 |
---|
12 | { "inplace", 0 }, |
---|
13 | #define NINPLSW 1 |
---|
14 | { "noinplace", 0 }, |
---|
15 | #define QIETSW 2 |
---|
16 | { "quiet", 0 }, |
---|
17 | #define NQIETSW 3 |
---|
18 | { "noquiet", 0 }, |
---|
19 | #define VERBSW 4 |
---|
20 | { "verbose", 0 }, |
---|
21 | #define NVERBSW 5 |
---|
22 | { "noverbose", 0 }, |
---|
23 | #define VERSIONSW 6 |
---|
24 | { "version", 0 }, |
---|
25 | #define HELPSW 7 |
---|
26 | { "help", 4 }, |
---|
27 | { NULL, 0 } |
---|
28 | }; |
---|
29 | |
---|
30 | static char delim3[] = "-------"; |
---|
31 | |
---|
32 | struct smsg { |
---|
33 | long s_start; |
---|
34 | long s_stop; |
---|
35 | }; |
---|
36 | |
---|
37 | /* |
---|
38 | * static prototypes |
---|
39 | */ |
---|
40 | static int find_delim (int, struct smsg *); |
---|
41 | static void burst (struct msgs **, int, struct smsg *, int, int, int); |
---|
42 | static void cpybrst (FILE *, FILE *, char *, char *, int); |
---|
43 | |
---|
44 | |
---|
45 | int |
---|
46 | main (int argc, char **argv) |
---|
47 | { |
---|
48 | int inplace = 0, quietsw = 0, verbosw = 0; |
---|
49 | int msgp = 0, hi, msgnum, numburst; |
---|
50 | char *cp, *maildir, *folder = NULL, buf[BUFSIZ]; |
---|
51 | char **argp, **arguments, *msgs[MAXARGS]; |
---|
52 | struct smsg *smsgs; |
---|
53 | struct msgs *mp; |
---|
54 | |
---|
55 | #ifdef LOCALE |
---|
56 | setlocale(LC_ALL, ""); |
---|
57 | #endif |
---|
58 | invo_name = r1bindex (argv[0], '/'); |
---|
59 | |
---|
60 | /* read user profile/context */ |
---|
61 | context_read(); |
---|
62 | |
---|
63 | arguments = getarguments (invo_name, argc, argv, 1); |
---|
64 | argp = arguments; |
---|
65 | |
---|
66 | while ((cp = *argp++)) { |
---|
67 | if (*cp == '-') { |
---|
68 | switch (smatch (++cp, switches)) { |
---|
69 | case AMBIGSW: |
---|
70 | ambigsw (cp, switches); |
---|
71 | done (1); |
---|
72 | case UNKWNSW: |
---|
73 | adios (NULL, "-%s unknown\n", cp); |
---|
74 | |
---|
75 | case HELPSW: |
---|
76 | snprintf (buf, sizeof(buf), "%s [+folder] [msgs] [switches]", |
---|
77 | invo_name); |
---|
78 | print_help (buf, switches, 1); |
---|
79 | done (1); |
---|
80 | case VERSIONSW: |
---|
81 | print_version(invo_name); |
---|
82 | done (1); |
---|
83 | |
---|
84 | case INPLSW: |
---|
85 | inplace++; |
---|
86 | continue; |
---|
87 | case NINPLSW: |
---|
88 | inplace = 0; |
---|
89 | continue; |
---|
90 | |
---|
91 | case QIETSW: |
---|
92 | quietsw++; |
---|
93 | continue; |
---|
94 | case NQIETSW: |
---|
95 | quietsw = 0; |
---|
96 | continue; |
---|
97 | |
---|
98 | case VERBSW: |
---|
99 | verbosw++; |
---|
100 | continue; |
---|
101 | case NVERBSW: |
---|
102 | verbosw = 0; |
---|
103 | continue; |
---|
104 | } |
---|
105 | } |
---|
106 | if (*cp == '+' || *cp == '@') { |
---|
107 | if (folder) |
---|
108 | adios (NULL, "only one folder at a time!"); |
---|
109 | else |
---|
110 | folder = path (cp + 1, *cp == '+' ? TFOLDER : TSUBCWF); |
---|
111 | } else { |
---|
112 | msgs[msgp++] = cp; |
---|
113 | } |
---|
114 | } |
---|
115 | |
---|
116 | if (!context_find ("path")) |
---|
117 | free (path ("./", TFOLDER)); |
---|
118 | if (!msgp) |
---|
119 | msgs[msgp++] = "cur"; |
---|
120 | if (!folder) |
---|
121 | folder = getfolder (1); |
---|
122 | maildir = m_maildir (folder); |
---|
123 | |
---|
124 | if (chdir (maildir) == NOTOK) |
---|
125 | adios (maildir, "unable to change directory to"); |
---|
126 | |
---|
127 | /* read folder and create message structure */ |
---|
128 | if (!(mp = folder_read (folder))) |
---|
129 | adios (NULL, "unable to read folder %s", folder); |
---|
130 | |
---|
131 | /* check for empty folder */ |
---|
132 | if (mp->nummsg == 0) |
---|
133 | adios (NULL, "no messages in %s", folder); |
---|
134 | |
---|
135 | /* parse all the message ranges/sequences and set SELECTED */ |
---|
136 | for (msgnum = 0; msgnum < msgp; msgnum++) |
---|
137 | if (!m_convert (mp, msgs[msgnum])) |
---|
138 | done (1); |
---|
139 | seq_setprev (mp); /* set the previous-sequence */ |
---|
140 | |
---|
141 | smsgs = (struct smsg *) |
---|
142 | calloc ((size_t) (MAXFOLDER + 2), sizeof(*smsgs)); |
---|
143 | if (smsgs == NULL) |
---|
144 | adios (NULL, "unable to allocate burst storage"); |
---|
145 | |
---|
146 | hi = mp->hghmsg + 1; |
---|
147 | |
---|
148 | /* burst all the SELECTED messages */ |
---|
149 | for (msgnum = mp->lowsel; msgnum <= mp->hghsel; msgnum++) { |
---|
150 | if (is_selected (mp, msgnum)) { |
---|
151 | if ((numburst = find_delim (msgnum, smsgs)) >= 1) { |
---|
152 | if (verbosw) |
---|
153 | printf ("%d message%s exploded from digest %d\n", |
---|
154 | numburst, numburst > 1 ? "s" : "", msgnum); |
---|
155 | burst (&mp, msgnum, smsgs, numburst, inplace, verbosw); |
---|
156 | } else { |
---|
157 | if (numburst == 0) |
---|
158 | if (!quietsw) |
---|
159 | admonish (NULL, "message %d not in digest format", msgnum); |
---|
160 | else |
---|
161 | adios (NULL, "burst() botch -- you lose big"); |
---|
162 | } |
---|
163 | } |
---|
164 | } |
---|
165 | |
---|
166 | free ((char *) smsgs); |
---|
167 | context_replace (pfolder, folder); /* update current folder */ |
---|
168 | |
---|
169 | /* |
---|
170 | * If -inplace is given, then the first message burst becomes |
---|
171 | * the current message (which will now show a table of contents). |
---|
172 | * Otherwise, the first message extracted from the first digest |
---|
173 | * becomes the current message. |
---|
174 | */ |
---|
175 | if (inplace) { |
---|
176 | if (mp->lowsel != mp->curmsg) |
---|
177 | seq_setcur (mp, mp->lowsel); |
---|
178 | } else { |
---|
179 | if (hi <= mp->hghmsg) |
---|
180 | seq_setcur (mp, hi); |
---|
181 | } |
---|
182 | |
---|
183 | seq_save (mp); /* synchronize message sequences */ |
---|
184 | context_save (); /* save the context file */ |
---|
185 | folder_free (mp); /* free folder/message structure */ |
---|
186 | done (0); |
---|
187 | } |
---|
188 | |
---|
189 | |
---|
190 | /* |
---|
191 | * Scan the message and find the beginning and |
---|
192 | * end of all the messages in the digest. |
---|
193 | */ |
---|
194 | |
---|
195 | static int |
---|
196 | find_delim (int msgnum, struct smsg *smsgs) |
---|
197 | { |
---|
198 | int ld3, wasdlm, msgp; |
---|
199 | long pos; |
---|
200 | char c, *msgnam; |
---|
201 | int cc; |
---|
202 | char buffer[BUFSIZ]; |
---|
203 | FILE *in; |
---|
204 | |
---|
205 | ld3 = strlen (delim3); |
---|
206 | |
---|
207 | if ((in = fopen (msgnam = m_name (msgnum), "r")) == NULL) |
---|
208 | adios (msgnam, "unable to read message"); |
---|
209 | |
---|
210 | for (msgp = 0, pos = 0L; msgp <= MAXFOLDER;) { |
---|
211 | while (fgets (buffer, sizeof(buffer), in) && buffer[0] == '\n') |
---|
212 | pos += (long) strlen (buffer); |
---|
213 | if (feof (in)) |
---|
214 | break; |
---|
215 | fseek (in, pos, SEEK_SET); |
---|
216 | smsgs[msgp].s_start = pos; |
---|
217 | |
---|
218 | for (c = 0; fgets (buffer, sizeof(buffer), in); c = buffer[0]) { |
---|
219 | if (strncmp (buffer, delim3, ld3) == 0 |
---|
220 | && (msgp == 1 || c == '\n') |
---|
221 | && ((cc = peekc (in)) == '\n' || cc == EOF)) |
---|
222 | break; |
---|
223 | else |
---|
224 | pos += (long) strlen (buffer); |
---|
225 | } |
---|
226 | |
---|
227 | wasdlm = strncmp (buffer, delim3, ld3) == 0; |
---|
228 | if (smsgs[msgp].s_start != pos) |
---|
229 | smsgs[msgp++].s_stop = (c == '\n' && wasdlm) ? pos - 1 : pos; |
---|
230 | if (feof (in)) { |
---|
231 | #if 0 |
---|
232 | if (wasdlm) { |
---|
233 | smsgs[msgp - 1].s_stop -= ((long) strlen (buffer) + 1); |
---|
234 | msgp++; /* fake "End of XXX Digest" */ |
---|
235 | } |
---|
236 | #endif |
---|
237 | break; |
---|
238 | } |
---|
239 | pos += (long) strlen (buffer); |
---|
240 | } |
---|
241 | |
---|
242 | fclose (in); |
---|
243 | return (msgp - 1); /* toss "End of XXX Digest" */ |
---|
244 | } |
---|
245 | |
---|
246 | |
---|
247 | /* |
---|
248 | * Burst out the messages in the digest into the folder |
---|
249 | */ |
---|
250 | |
---|
251 | static void |
---|
252 | burst (struct msgs **mpp, int msgnum, struct smsg *smsgs, int numburst, |
---|
253 | int inplace, int verbosw) |
---|
254 | { |
---|
255 | int i, j, mode; |
---|
256 | char *msgnam; |
---|
257 | char f1[BUFSIZ], f2[BUFSIZ], f3[BUFSIZ]; |
---|
258 | FILE *in, *out; |
---|
259 | struct stat st; |
---|
260 | struct msgs *mp; |
---|
261 | |
---|
262 | if ((in = fopen (msgnam = m_name (msgnum), "r")) == NULL) |
---|
263 | adios (msgnam, "unable to read message"); |
---|
264 | |
---|
265 | mode = fstat (fileno(in), &st) != NOTOK ? (st.st_mode & 0777) : m_gmprot(); |
---|
266 | mp = *mpp; |
---|
267 | |
---|
268 | /* |
---|
269 | * See if we have enough space in the folder |
---|
270 | * structure for all the new messages. |
---|
271 | */ |
---|
272 | if ((mp->hghmsg + numburst > mp->hghoff) && |
---|
273 | !(mp = folder_realloc (mp, mp->lowoff, mp->hghmsg + numburst))) |
---|
274 | adios (NULL, "unable to allocate folder storage"); |
---|
275 | *mpp = mp; |
---|
276 | |
---|
277 | j = mp->hghmsg; /* old value */ |
---|
278 | mp->hghmsg += numburst; |
---|
279 | mp->nummsg += numburst; |
---|
280 | |
---|
281 | /* |
---|
282 | * If this is not the highest SELECTED message, then |
---|
283 | * increment mp->hghsel by numburst, since the highest |
---|
284 | * SELECTED is about to be slid down by that amount. |
---|
285 | */ |
---|
286 | if (msgnum < mp->hghsel) |
---|
287 | mp->hghsel += numburst; |
---|
288 | |
---|
289 | /* |
---|
290 | * If -inplace is given, renumber the messages after the |
---|
291 | * source message, to make room for each of the messages |
---|
292 | * contained within the digest. |
---|
293 | */ |
---|
294 | if (inplace) { |
---|
295 | for (i = mp->hghmsg; j > msgnum; i--, j--) { |
---|
296 | strncpy (f1, m_name (i), sizeof(f1)); |
---|
297 | strncpy (f2, m_name (j), sizeof(f2)); |
---|
298 | if (does_exist (mp, j)) { |
---|
299 | if (verbosw) |
---|
300 | printf ("message %d becomes message %d\n", j, i); |
---|
301 | |
---|
302 | if (rename (f2, f1) == NOTOK) |
---|
303 | admonish (f1, "unable to rename %s to", f2); |
---|
304 | copy_msg_flags (mp, i, j); |
---|
305 | clear_msg_flags (mp, j); |
---|
306 | mp->msgflags |= SEQMOD; |
---|
307 | } |
---|
308 | } |
---|
309 | } |
---|
310 | |
---|
311 | unset_selected (mp, msgnum); |
---|
312 | |
---|
313 | /* new hghmsg is hghmsg + numburst */ |
---|
314 | i = inplace ? msgnum + numburst : mp->hghmsg; |
---|
315 | for (j = numburst; j >= (inplace ? 0 : 1); i--, j--) { |
---|
316 | strncpy (f1, m_name (i), sizeof(f1)); |
---|
317 | strncpy (f2, m_scratch ("", invo_name), sizeof(f2)); |
---|
318 | if (verbosw && i != msgnum) |
---|
319 | printf ("message %d of digest %d becomes message %d\n", j, msgnum, i); |
---|
320 | |
---|
321 | if ((out = fopen (f2, "w")) == NULL) |
---|
322 | adios (f2, "unable to write message"); |
---|
323 | chmod (f2, mode); |
---|
324 | fseek (in, smsgs[j].s_start, SEEK_SET); |
---|
325 | cpybrst (in, out, msgnam, f2, |
---|
326 | (int) (smsgs[j].s_stop - smsgs[j].s_start)); |
---|
327 | fclose (out); |
---|
328 | |
---|
329 | if (i == msgnum) { |
---|
330 | strncpy (f3, m_backup (f1), sizeof(f3)); |
---|
331 | if (rename (f1, f3) == NOTOK) |
---|
332 | admonish (f3, "unable to rename %s to", f1); |
---|
333 | } |
---|
334 | if (rename (f2, f1) == NOTOK) |
---|
335 | admonish (f1, "unable to rename %s to", f2); |
---|
336 | copy_msg_flags (mp, i, msgnum); |
---|
337 | mp->msgflags |= SEQMOD; |
---|
338 | } |
---|
339 | |
---|
340 | fclose (in); |
---|
341 | } |
---|
342 | |
---|
343 | |
---|
344 | #define S1 0 |
---|
345 | #define S2 1 |
---|
346 | #define S3 2 |
---|
347 | |
---|
348 | /* |
---|
349 | * Copy a mesage which is being burst out of a digest. |
---|
350 | * It will remove any "dashstuffing" in the message. |
---|
351 | */ |
---|
352 | |
---|
353 | static void |
---|
354 | cpybrst (FILE *in, FILE *out, char *ifile, char *ofile, int len) |
---|
355 | { |
---|
356 | register int c, state; |
---|
357 | |
---|
358 | for (state = S1; (c = fgetc (in)) != EOF && len > 0; len--) { |
---|
359 | if (c == 0) |
---|
360 | continue; |
---|
361 | switch (state) { |
---|
362 | case S1: |
---|
363 | switch (c) { |
---|
364 | case '-': |
---|
365 | state = S3; |
---|
366 | break; |
---|
367 | |
---|
368 | default: |
---|
369 | state = S2; |
---|
370 | case '\n': |
---|
371 | fputc (c, out); |
---|
372 | break; |
---|
373 | } |
---|
374 | break; |
---|
375 | |
---|
376 | case S2: |
---|
377 | switch (c) { |
---|
378 | case '\n': |
---|
379 | state = S1; |
---|
380 | default: |
---|
381 | fputc (c, out); |
---|
382 | break; |
---|
383 | } |
---|
384 | break; |
---|
385 | |
---|
386 | case S3: |
---|
387 | switch (c) { |
---|
388 | case ' ': |
---|
389 | state = S2; |
---|
390 | break; |
---|
391 | |
---|
392 | default: |
---|
393 | state = (c == '\n') ? S1 : S2; |
---|
394 | fputc ('-', out); |
---|
395 | fputc (c, out); |
---|
396 | break; |
---|
397 | } |
---|
398 | break; |
---|
399 | } |
---|
400 | } |
---|
401 | |
---|
402 | if (ferror (in) && !feof (in)) |
---|
403 | adios (ifile, "error reading"); |
---|
404 | if (ferror (out)) |
---|
405 | adios (ofile, "error writing"); |
---|
406 | } |
---|