Add/update copyright notice in all source code files.
[mmh] / uip / forw.c
1
2 /*
3  * forw.c -- forward a message, or group of messages.
4  *
5  * $Id$
6  *
7  * This code is Copyright (c) 2002, by the authors of nmh.  See the
8  * COPYRIGHT file in the root directory of the nmh distribution for
9  * complete copyright information.
10  */
11
12 #include <h/mh.h>
13 #include <fcntl.h>
14 #include <h/fmt_scan.h>
15 #include <h/tws.h>
16
17
18 #define IFORMAT "digest-issue-%s"
19 #define VFORMAT "digest-volume-%s"
20
21 static struct swit switches[] = {
22 #define ANNOSW                 0
23     { "annotate", 0 },
24 #define NANNOSW                1
25     { "noannotate", 0 },
26 #define DFOLDSW                2
27     { "draftfolder +folder", 0 },
28 #define DMSGSW                 3
29     { "draftmessage msg", 0 },
30 #define NDFLDSW                4
31     { "nodraftfolder", 0 },
32 #define EDITRSW                5
33     { "editor editor", 0 },
34 #define NEDITSW                6
35     { "noedit", 0 },
36 #define FILTSW                 7
37     { "filter filterfile", 0 },
38 #define FORMSW                 8
39     { "form formfile", 0 },
40 #define FRMTSW                 9
41     { "format", 5 },
42 #define NFRMTSW               10
43     { "noformat", 7 },
44 #define INPLSW                11
45     { "inplace", 0 },
46 #define NINPLSW               12
47     { "noinplace", 0 },
48 #define MIMESW                13
49     { "mime", 0 },
50 #define NMIMESW               14
51     { "nomime", 0 },
52 #define DGSTSW                15
53     { "digest list", 0 },
54 #define ISSUESW               16
55     { "issue number", 0 },
56 #define VOLUMSW               17
57     { "volume number", 0 },
58 #define WHATSW                18
59     { "whatnowproc program", 0 },
60 #define NWHATSW               19
61     { "nowhatnowproc", 0 },
62 #define BITSTUFFSW            20
63     { "dashstuffing", 0 },              /* interface to mhl */
64 #define NBITSTUFFSW           21
65     { "nodashstuffing", 0 },
66 #define VERSIONSW             22
67     { "version", 0 },
68 #define HELPSW                23
69     { "help", 0 },
70 #define FILESW                24
71     { "file file", 4 },                 /* interface from msh */
72
73 #ifdef MHE
74 #define BILDSW                25
75     { "build", 5 },                     /* interface from mhe */
76 #endif /* MHE */
77
78     { NULL, 0 }
79 };
80
81 static struct swit aqrnl[] = {
82 #define NOSW         0
83     { "quit", 0 },
84 #define YESW         1
85     { "replace", 0 },
86 #define LISTDSW      2
87     { "list", 0 },
88 #define REFILSW      3
89     { "refile +folder", 0 },
90 #define NEWSW        4
91     { "new", 0 },
92     { NULL, 0 }
93 };
94
95 static struct swit aqrl[] = {
96     { "quit", 0 },
97     { "replace", 0 },
98     { "list", 0 },
99     { "refile +folder", 0 },
100     { NULL, 0 }
101 };
102
103 static char drft[BUFSIZ];
104
105 static char delim3[] =
106     "\n------------------------------------------------------------\n\n";
107 static char delim4[] = "\n------------------------------\n\n";
108
109
110 static struct msgs *mp = NULL;          /* used a lot */
111
112
113 /*
114  * static prototypes
115  */
116 static void mhl_draft (int, char *, int, int, char *, char *, int);
117 static void copy_draft (int, char *, char *, int, int, int);
118 static void copy_mime_draft (int);
119 static int build_form (char *, char *, int, int);
120
121
122 int
123 main (int argc, char **argv)
124 {
125     int msgp = 0, anot = 0, inplace = 1, mime = 0;
126     int issue = 0, volume = 0, dashstuff = 0;
127     int nedit = 0, nwhat = 0, i, in;
128     int out, isdf = 0, msgnum;
129     char *cp, *cwd, *maildir, *dfolder = NULL;
130     char *dmsg = NULL, *digest = NULL, *ed = NULL;
131     char *file = NULL, *filter = NULL, *folder = NULL;
132     char *form = NULL, buf[BUFSIZ], value[10];
133     char **argp, **arguments, *msgs[MAXARGS];
134     struct stat st;
135
136 #ifdef  MHE
137     int buildsw = 0;
138 #endif  /* MHE */
139
140 #ifdef LOCALE
141     setlocale(LC_ALL, "");
142 #endif
143     invo_name = r1bindex (argv[0], '/');
144
145     /* read user profile/context */
146     context_read();
147
148     arguments = getarguments (invo_name, argc, argv, 1);
149     argp = arguments;
150
151     while ((cp = *argp++)) {
152         if (*cp == '-') {
153             switch (smatch (++cp, switches)) {
154                 case AMBIGSW: 
155                     ambigsw (cp, switches);
156                     done (1);
157                 case UNKWNSW: 
158                     adios (NULL, "-%s unknown", cp);
159
160                 case HELPSW: 
161                     snprintf (buf, sizeof(buf), "%s [+folder] [msgs] [switches]",
162                         invo_name);
163                     print_help (buf, switches, 1);
164                     done (1);
165                 case VERSIONSW:
166                     print_version(invo_name);
167                     done (1);
168
169                 case ANNOSW: 
170                     anot++;
171                     continue;
172                 case NANNOSW: 
173                     anot = 0;
174                     continue;
175
176                 case EDITRSW: 
177                     if (!(ed = *argp++) || *ed == '-')
178                         adios (NULL, "missing argument to %s", argp[-2]);
179                     nedit = 0;
180                     continue;
181                 case NEDITSW:
182                     nedit++;
183                     continue;
184
185                 case WHATSW: 
186                     if (!(whatnowproc = *argp++) || *whatnowproc == '-')
187                         adios (NULL, "missing argument to %s", argp[-2]);
188                     nwhat = 0;
189                     continue;
190 #ifdef MHE
191                 case BILDSW:
192                     buildsw++;  /* fall... */
193 #endif /* MHE */
194                 case NWHATSW: 
195                     nwhat++;
196                     continue;
197
198                 case FILESW: 
199                     if (file)
200                         adios (NULL, "only one file at a time!");
201                     if (!(cp = *argp++) || *cp == '-')
202                         adios (NULL, "missing argument to %s", argp[-2]);
203                     file = path (cp, TFILE);
204                     continue;
205                 case FILTSW:
206                     if (!(cp = *argp++) || *cp == '-')
207                         adios (NULL, "missing argument to %s", argp[-2]);
208                     filter = getcpy (etcpath (cp));
209                     mime = 0;
210                     continue;
211                 case FORMSW: 
212                     if (!(form = *argp++) || *form == '-')
213                         adios (NULL, "missing argument to %s", argp[-2]);
214                     continue;
215
216                 case FRMTSW:
217                     filter = getcpy (etcpath (mhlforward));
218                     continue;
219                 case NFRMTSW:
220                     filter = NULL;
221                     continue;
222
223                 case INPLSW: 
224                     inplace++;
225                     continue;
226                 case NINPLSW: 
227                     inplace = 0;
228                     continue;
229
230                 case MIMESW:
231                     mime++;
232                     filter = NULL;
233                     continue;
234                 case NMIMESW: 
235                     mime = 0;
236                     continue;
237
238                 case DGSTSW: 
239                     if (!(digest = *argp++) || *digest == '-')
240                         adios (NULL, "missing argument to %s", argp[-2]);
241                     mime = 0;
242                     continue;
243                 case ISSUESW:
244                     if (!(cp = *argp++) || *cp == '-')
245                         adios (NULL, "missing argument to %s", argp[-2]);
246                     if ((issue = atoi (cp)) < 1)
247                         adios (NULL, "bad argument %s %s", argp[-2], cp);
248                     continue;
249                 case VOLUMSW:
250                     if (!(cp = *argp++) || *cp == '-')
251                         adios (NULL, "missing argument to %s", argp[-2]);
252                     if ((volume = atoi (cp)) < 1)
253                         adios (NULL, "bad argument %s %s", argp[-2], cp);
254                     continue;
255
256                 case DFOLDSW: 
257                     if (dfolder)
258                         adios (NULL, "only one draft folder at a time!");
259                     if (!(cp = *argp++) || *cp == '-')
260                         adios (NULL, "missing argument to %s", argp[-2]);
261                     dfolder = path (*cp == '+' || *cp == '@' ? cp + 1 : cp,
262                                     *cp != '@' ? TFOLDER : TSUBCWF);
263                     continue;
264                 case DMSGSW:
265                     if (dmsg)
266                         adios (NULL, "only one draft message at a time!");
267                     if (!(dmsg = *argp++) || *dmsg == '-')
268                         adios (NULL, "missing argument to %s", argp[-2]);
269                     continue;
270                 case NDFLDSW: 
271                     dfolder = NULL;
272                     isdf = NOTOK;
273                     continue;
274
275                 case BITSTUFFSW: 
276                     dashstuff = 1;      /* trinary logic */
277                     continue;
278                 case NBITSTUFFSW: 
279                     dashstuff = -1;     /* trinary logic */
280                     continue;
281             }
282         }
283         if (*cp == '+' || *cp == '@') {
284             if (folder)
285                 adios (NULL, "only one folder at a time!");
286             else
287                 folder = path (cp + 1, *cp == '+' ? TFOLDER : TSUBCWF);
288         } else {
289             msgs[msgp++] = cp;
290         }
291     }
292
293     cwd = getcpy (pwd ());
294
295     if (!context_find ("path"))
296         free (path ("./", TFOLDER));
297     if (file && (msgp || folder))
298         adios (NULL, "can't mix files and folders/msgs");
299
300 try_it_again:
301
302 #ifdef MHE
303     strncpy (drft, buildsw ? m_maildir ("draft")
304                           : m_draft (dfolder, NULL, NOUSE, &isdf), sizeof(drft));
305
306     /* Check if a draft already exists */
307     if (!buildsw && stat (drft, &st) != NOTOK) {
308 #else
309     strncpy (drft, m_draft (dfolder, dmsg, NOUSE, &isdf), sizeof(drft));
310
311     /* Check if a draft already exists */
312     if (stat (drft, &st) != NOTOK) {
313 #endif  /* MHE */
314         printf ("Draft \"%s\" exists (%ld bytes).", drft, (long) st.st_size);
315         for (i = LISTDSW; i != YESW;) {
316             if (!(argp = getans ("\nDisposition? ", isdf ? aqrnl : aqrl)))
317                 done (1);
318             switch (i = smatch (*argp, isdf ? aqrnl : aqrl)) {
319                 case NOSW: 
320                     done (0);
321                 case NEWSW: 
322                     dmsg = NULL;
323                     goto try_it_again;
324                 case YESW: 
325                     break;
326                 case LISTDSW: 
327                     showfile (++argp, drft);
328                     break;
329                 case REFILSW: 
330                     if (refile (++argp, drft) == 0)
331                         i = YESW;
332                     break;
333                 default: 
334                     advise (NULL, "say what?");
335                     break;
336             }
337         }
338     }
339
340     if (file) {
341         /*
342          * Forwarding a file.
343          */
344         anot = 0;       /* don't want to annotate a file */
345     } else {
346         /*
347          * Forwarding a message.
348          */
349         if (!msgp)
350             msgs[msgp++] = "cur";
351         if (!folder)
352             folder = getfolder (1);
353         maildir = m_maildir (folder);
354
355         if (chdir (maildir) == NOTOK)
356             adios (maildir, "unable to change directory to");
357
358         /* read folder and create message structure */
359         if (!(mp = folder_read (folder)))
360             adios (NULL, "unable to read folder %s", folder);
361
362         /* check for empty folder */
363         if (mp->nummsg == 0)
364             adios (NULL, "no messages in %s", folder);
365
366         /* parse all the message ranges/sequences and set SELECTED */
367         for (msgnum = 0; msgnum < msgp; msgnum++)
368             if (!m_convert (mp, msgs[msgnum]))
369                 done (1);
370         seq_setprev (mp);       /* set the previous sequence */
371     }
372
373     if (filter && access (filter, R_OK) == NOTOK)
374         adios (filter, "unable to read");
375
376     /*
377      * Open form (component) file.
378      */
379     if (digest) {
380         if (issue == 0) {
381             snprintf (buf, sizeof(buf), IFORMAT, digest);
382             if (volume == 0
383                     && (cp = context_find (buf))
384                     && ((issue = atoi (cp)) < 0))
385                 issue = 0;
386             issue++;
387         }
388         if (volume == 0)
389             snprintf (buf, sizeof(buf), VFORMAT, digest);
390         if ((cp = context_find (buf)) == NULL || (volume = atoi (cp)) <= 0)
391             volume = 1;
392         if (!form)
393             form = digestcomps;
394         in = build_form (form, digest, volume, issue);
395     } else {
396         if (form) {
397             if ((in = open (etcpath (form), O_RDONLY)) == NOTOK)
398                 adios (form, "unable to open form file");
399         } else {
400             if ((in = open (etcpath (forwcomps), O_RDONLY)) == NOTOK)
401                 adios (forwcomps, "unable to open default components file");
402             form = forwcomps;
403         }
404     }
405
406     if ((out = creat (drft, m_gmprot ())) == NOTOK)
407         adios (drft, "unable to create");
408
409     /*
410      * copy the components into the draft
411      */
412     cpydata (in, out, form, drft);
413     close (in);
414
415     if (file) {
416         /* just copy the file into the draft */
417         if ((in = open (file, O_RDONLY)) == NOTOK)
418             adios (file, "unable to open");
419         cpydata (in, out, file, drft);
420         close (in);
421         close (out);
422     } else {
423         /*
424          * If filter file is defined, then format the
425          * messages into the draft using mhlproc.
426          */
427         if (filter)
428             mhl_draft (out, digest, volume, issue, drft, filter, dashstuff);
429         else if (mime)
430             copy_mime_draft (out);
431         else
432             copy_draft (out, digest, drft, volume, issue, dashstuff);
433         close (out);
434
435         if (digest) {
436             snprintf (buf, sizeof(buf), IFORMAT, digest);
437             snprintf (value, sizeof(value), "%d", issue);
438             context_replace (buf, getcpy (value));
439             snprintf (buf, sizeof(buf), VFORMAT, digest);
440             snprintf (value, sizeof(value), "%d", volume);
441             context_replace (buf, getcpy (value));
442         }
443
444         context_replace (pfolder, folder);      /* update current folder   */
445         seq_setcur (mp, mp->lowsel);            /* update current message  */
446         seq_save (mp);                          /* synchronize sequences   */
447         context_save ();                        /* save the context file   */
448     }
449
450     if (nwhat)
451         done (0);
452     what_now (ed, nedit, NOUSE, drft, NULL, 0, mp,
453         anot ? "Forwarded" : NULL, inplace, cwd);
454     return done (1);
455 }
456
457
458 /*
459  * Filter the messages you are forwarding, into the
460  * draft calling the mhlproc, and reading its output
461  * from a pipe.
462  */
463
464 static void
465 mhl_draft (int out, char *digest, int volume, int issue,
466             char *file, char *filter, int dashstuff)
467 {
468     pid_t child_id;
469     int i, msgnum, pd[2];
470     char *vec[MAXARGS];
471     char buf1[BUFSIZ];
472     char buf2[BUFSIZ];
473     
474     if (pipe (pd) == NOTOK)
475         adios ("pipe", "unable to create");
476
477     vec[0] = r1bindex (mhlproc, '/');
478
479     for (i = 0; (child_id = fork()) == NOTOK && i < 5; i++)
480         sleep (5);
481     switch (child_id) {
482         case NOTOK: 
483             adios ("fork", "unable to");
484
485         case OK: 
486             close (pd[0]);
487             dup2 (pd[1], 1);
488             close (pd[1]);
489
490             i = 1;
491             vec[i++] = "-forwall";
492             vec[i++] = "-form";
493             vec[i++] = filter;
494
495             if (digest) {
496                 vec[i++] = "-digest";
497                 vec[i++] = digest;
498                 vec[i++] = "-issue";
499                 snprintf (buf1, sizeof(buf1), "%d", issue);
500                 vec[i++] = buf1;
501                 vec[i++] = "-volume";
502                 snprintf (buf2, sizeof(buf2), "%d", volume);
503                 vec[i++] = buf2;
504             }
505
506             /*
507              * Are we dashstuffing (quoting) the lines that begin
508              * with `-'.  We use the mhl default (don't add any flag)
509              * unless the user has specified a specific flag.
510              */
511             if (dashstuff > 0)
512                 vec[i++] = "-dashstuffing";
513             else if (dashstuff < 0)
514                 vec[i++] = "-nodashstuffing";
515
516             if (mp->numsel >= MAXARGS - i)
517                 adios (NULL, "more than %d messages for %s exec",
518                         vec[0], MAXARGS - i);
519
520             /*
521              * Now add the message names to filter.  We can only
522              * handle about 995 messages (because vec is fixed size),
523              * but that should be plenty.
524              */
525             for (msgnum = mp->lowsel; msgnum <= mp->hghsel && i < sizeof(vec) - 1;
526                         msgnum++)
527                 if (is_selected (mp, msgnum))
528                     vec[i++] = getcpy (m_name (msgnum));
529             vec[i] = NULL;
530
531             execvp (mhlproc, vec);
532             fprintf (stderr, "unable to exec ");
533             perror (mhlproc);
534             _exit (-1);
535
536         default: 
537             close (pd[1]);
538             cpydata (pd[0], out, vec[0], file);
539             close (pd[0]);
540             pidXwait(child_id, mhlproc);
541             break;
542     }
543 }
544
545
546 /*
547  * Copy the messages into the draft.  The messages are
548  * not filtered through the mhlproc.  Do dashstuffing if
549  * necessary.
550  */
551
552 static void
553 copy_draft (int out, char *digest, char *file, int volume, int issue, int dashstuff)
554 {
555     int fd,i, msgcnt, msgnum;
556     int len, buflen;
557     register char *bp, *msgnam;
558     char buffer[BUFSIZ];
559
560     msgcnt = 1;
561     for (msgnum = mp->lowsel; msgnum <= mp->hghsel; msgnum++) {
562         if (is_selected (mp, msgnum)) {
563             if (digest) {
564                 strncpy (buffer, msgnum == mp->lowsel ? delim3 : delim4,
565                         sizeof(buffer));
566             } else {
567                 /* Get buffer ready to go */
568                 bp = buffer;
569                 buflen = sizeof(buffer);
570
571                 strncpy (bp, "\n-------", buflen);
572                 len = strlen (bp);
573                 bp += len;
574                 buflen -= len;
575
576                 if (msgnum == mp->lowsel) {
577                     snprintf (bp, buflen, " Forwarded Message%s",
578                         mp->numsel > 1 ? "s" : "");
579                 } else {
580                     snprintf (bp, buflen, " Message %d", msgcnt);
581                 }
582                 len = strlen (bp);
583                 bp += len;
584                 buflen -= len;
585
586                 strncpy (bp, "\n\n", buflen);
587             }
588             write (out, buffer, strlen (buffer));
589
590             if ((fd = open (msgnam = m_name (msgnum), O_RDONLY)) == NOTOK) {
591                 admonish (msgnam, "unable to read message");
592                 continue;
593             }
594
595             /*
596              * Copy the message.  Add RFC934 quoting (dashstuffing)
597              * unless given the -nodashstuffing flag.
598              */
599             if (dashstuff >= 0)
600                 cpydgst (fd, out, msgnam, file);
601             else
602                 cpydata (fd, out, msgnam, file);
603
604             close (fd);
605             msgcnt++;
606         }
607     }
608
609     if (digest) {
610         strncpy (buffer, delim4, sizeof(buffer));
611     } else {
612         snprintf (buffer, sizeof(buffer), "\n------- End of Forwarded Message%s\n\n",
613                 mp->numsel > 1 ? "s" : "");
614     }
615     write (out, buffer, strlen (buffer));
616
617     if (digest) {
618         snprintf (buffer, sizeof(buffer), "End of %s Digest [Volume %d Issue %d]\n",
619                 digest, volume, issue);
620         i = strlen (buffer);
621         for (bp = buffer + i; i > 1; i--)
622             *bp++ = '*';
623         *bp++ = '\n';
624         *bp = 0;
625         write (out, buffer, strlen (buffer));
626     }
627 }
628
629
630 /*
631  * Create a mhn composition file for forwarding message.
632  */
633
634 static void
635 copy_mime_draft (int out)
636 {
637     int msgnum;
638     char buffer[BUFSIZ];
639
640     snprintf (buffer, sizeof(buffer), "#forw [forwarded message%s] +%s",
641         mp->numsel == 1 ? "" : "s", mp->foldpath);
642     write (out, buffer, strlen (buffer));
643     for (msgnum = mp->lowsel; msgnum <= mp->hghsel; msgnum++)
644         if (is_selected (mp, msgnum)) {
645             snprintf (buffer, sizeof(buffer), " %s", m_name (msgnum));
646             write (out, buffer, strlen (buffer));
647         }
648     write (out, "\n", 1);
649 }
650
651
652 static int
653 build_form (char *form, char *digest, int volume, int issue)
654 {
655     int in;
656     int fmtsize;
657     register char *nfs;
658     char *line, tmpfil[BUFSIZ];
659     register FILE *tmp;
660     register struct comp *cptr;
661     struct format *fmt;
662     int dat[5];
663
664     /* Get new format string */
665     nfs = new_fs (form, NULL, NULL);
666     fmtsize = strlen (nfs) + 256;
667
668     /* Compile format string */
669     fmt_compile (nfs, &fmt);
670
671     FINDCOMP (cptr, "digest");
672     if (cptr)
673         cptr->c_text = digest;
674     FINDCOMP (cptr, "date");
675     if (cptr)
676         cptr->c_text = getcpy(dtimenow (0));
677
678     dat[0] = issue;
679     dat[1] = volume;
680     dat[2] = 0;
681     dat[3] = fmtsize;
682     dat[4] = 0;
683
684     strncpy (tmpfil, m_tmpfil (invo_name), sizeof(tmpfil));
685     if ((tmp = fopen (tmpfil, "w+")) == NULL)
686         adios (tmpfil, "unable to create");
687     unlink (tmpfil);
688     if ((in = dup (fileno (tmp))) == NOTOK)
689         adios ("dup", "unable to");
690
691     if ((line = malloc ((unsigned) fmtsize)) == NULL)
692         adios (NULL, "unable to allocate format line storage");
693     fmt_scan (fmt, line, fmtsize, dat);
694     fputs (line, tmp);
695     free (line);
696     if (fclose (tmp))
697         adios (tmpfil, "error writing");
698
699     lseek (in, (off_t) 0, SEEK_SET);
700     return in;
701 }