Added mhmail.in. start of a replacement for compiled mhmail.
[mmh] / uip / refile.c
1
2 /*
3  * refile.c -- move or link message(s) from a source folder
4  *          -- into one or more destination folders
5  *
6  * This code is Copyright (c) 2002, by the authors of nmh.  See the
7  * COPYRIGHT file in the root directory of the nmh distribution for
8  * complete copyright information.
9  */
10
11 #include <h/mh.h>
12 #include <h/utils.h>
13 #include <fcntl.h>
14 #include <errno.h>
15
16 static struct swit switches[] = {
17 #define DRAFTSW          0
18     { "draft", 0 },
19 #define LINKSW           1
20     { "link", 0 },
21 #define NLINKSW          2
22     { "nolink", 0 },
23 #define PRESSW           3
24     { "preserve", 0 },
25 #define NPRESSW          4
26     { "nopreserve", 0 },
27 #define UNLINKSW         5
28     { "unlink", 0 },
29 #define NUNLINKSW        6
30     { "nounlink", 0 },
31 #define SRCSW            7
32     { "src +folder", 0 },
33 #define FILESW           8
34     { "file file", 0 },
35 #define RPROCSW          9
36     { "rmmproc program", 0 },
37 #define NRPRCSW         10
38     { "normmproc", 0 },
39 #define VERSIONSW       11
40     { "version", 0 },
41 #define HELPSW          12
42     { "help", 0 },
43     { NULL, 0 }
44 };
45
46 static char maildir[BUFSIZ];
47
48 struct st_fold {
49     char *f_name;
50     struct msgs *f_mp;
51 };
52
53 /*
54  * static prototypes
55  */
56 static void opnfolds (struct st_fold *, int);
57 static void clsfolds (struct st_fold *, int);
58 static void remove_files (int, char **);
59 static int m_file (char *, struct st_fold *, int, int, int);
60
61
62 int
63 main (int argc, char **argv)
64 {
65     int linkf = 0, preserve = 0, filep = 0;
66     int foldp = 0, isdf = 0, unlink_msgs = 0;
67     int i, msgnum;
68     char *cp, *folder = NULL, buf[BUFSIZ];
69     char **argp, **arguments;
70     char *filevec[NFOLDERS + 2];
71     char **files = &filevec[1];    /* leave room for remove_files:vec[0] */
72     struct st_fold folders[NFOLDERS + 1];
73     struct msgs_array msgs = { 0, 0, NULL };
74     struct msgs *mp;
75
76 #ifdef LOCALE
77     setlocale(LC_ALL, "");
78 #endif
79     invo_name = r1bindex (argv[0], '/');
80
81     /* read user profile/context */
82     context_read();
83
84     arguments = getarguments (invo_name, argc, argv, 1);
85     argp = arguments;
86
87     /*
88      * Parse arguments
89      */
90     while ((cp = *argp++)) {
91         if (*cp == '-') {
92             switch (smatch (++cp, switches)) {
93             case AMBIGSW: 
94                 ambigsw (cp, switches);
95                 done (1);
96             case UNKWNSW: 
97                 adios (NULL, "-%s unknown\n", cp);
98
99             case HELPSW: 
100                 snprintf (buf, sizeof(buf), "%s [msgs] [switches] +folder ...",
101                           invo_name);
102                 print_help (buf, switches, 1);
103                 done (0);
104             case VERSIONSW:
105                 print_version(invo_name);
106                 done (0);
107
108             case LINKSW: 
109                 linkf++;
110                 continue;
111             case NLINKSW: 
112                 linkf = 0;
113                 continue;
114
115             case PRESSW: 
116                 preserve++;
117                 continue;
118             case NPRESSW: 
119                 preserve = 0;
120                 continue;
121
122             case UNLINKSW:
123                 unlink_msgs++;
124                 continue;
125             case NUNLINKSW:
126                 unlink_msgs = 0;
127                 continue;
128
129             case SRCSW: 
130                 if (folder)
131                     adios (NULL, "only one source folder at a time!");
132                 if (!(cp = *argp++) || *cp == '-')
133                     adios (NULL, "missing argument to %s", argp[-2]);
134                 folder = path (*cp == '+' || *cp == '@' ? cp + 1 : cp,
135                                *cp != '@' ? TFOLDER : TSUBCWF);
136                 continue;
137             case DRAFTSW:
138                 if (filep > NFOLDERS)
139                     adios (NULL, "only %d files allowed!", NFOLDERS);
140                 isdf = 0;
141                 files[filep++] = getcpy (m_draft (NULL, NULL, 1, &isdf));
142                 continue;
143             case FILESW: 
144                 if (filep > NFOLDERS)
145                     adios (NULL, "only %d files allowed!", NFOLDERS);
146                 if (!(cp = *argp++) || *cp == '-')
147                     adios (NULL, "missing argument to %s", argp[-2]);
148                 files[filep++] = path (cp, TFILE);
149                 continue;
150
151             case RPROCSW: 
152                 if (!(rmmproc = *argp++) || *rmmproc == '-')
153                     adios (NULL, "missing argument to %s", argp[-2]);
154                 continue;
155             case NRPRCSW: 
156                 rmmproc = NULL;
157                 continue;
158             }
159         }
160         if (*cp == '+' || *cp == '@') {
161             if (foldp > NFOLDERS)
162                 adios (NULL, "only %d folders allowed!", NFOLDERS);
163             folders[foldp++].f_name =
164                 pluspath (cp);
165         } else
166                 app_msgarg(&msgs, cp);
167     }
168
169     if (!context_find ("path"))
170         free (path ("./", TFOLDER));
171     if (foldp == 0)
172         adios (NULL, "no folder specified");
173
174 #ifdef WHATNOW
175     if (!msgs.size && !foldp && !filep && (cp = getenv ("mhdraft")) && *cp)
176         files[filep++] = cp;
177 #endif /* WHATNOW */
178
179     /*
180      * We are refiling a file to the folders
181      */
182     if (filep > 0) {
183         if (folder || msgs.size)
184             adios (NULL, "use -file or some messages, not both");
185         opnfolds (folders, foldp);
186         for (i = 0; i < filep; i++)
187             if (m_file (files[i], folders, foldp, preserve, 0))
188                 done (1);
189         /* If -nolink, then "remove" files */
190         if (!linkf)
191             remove_files (filep, filevec);
192         done (0);
193     }
194
195     if (!msgs.size)
196         app_msgarg(&msgs, "cur");
197     if (!folder)
198         folder = getfolder (1);
199     strncpy (maildir, m_maildir (folder), sizeof(maildir));
200
201     if (chdir (maildir) == NOTOK)
202         adios (maildir, "unable to change directory to");
203
204     /* read source folder and create message structure */
205     if (!(mp = folder_read (folder)))
206         adios (NULL, "unable to read folder %s", folder);
207
208     /* check for empty folder */
209     if (mp->nummsg == 0)
210         adios (NULL, "no messages in %s", folder);
211
212     /* parse the message range/sequence/name and set SELECTED */
213     for (msgnum = 0; msgnum < msgs.size; msgnum++)
214         if (!m_convert (mp, msgs.msgs[msgnum]))
215             done (1);
216     seq_setprev (mp);   /* set the previous-sequence */
217
218     /* create folder structures for each destination folder */
219     opnfolds (folders, foldp);
220
221     /* Link all the selected messages into destination folders.
222      *
223      * This causes the add hook to be run for messages that are
224      * linked into another folder.  The refile hook is run for
225      * messages that are moved to another folder.
226      */
227     for (msgnum = mp->lowsel; msgnum <= mp->hghsel; msgnum++) {
228         if (is_selected (mp, msgnum)) {
229             cp = getcpy (m_name (msgnum));
230             if (m_file (cp, folders, foldp, preserve, !linkf))
231                 done (1);
232             free (cp);
233         }
234     }
235
236     /*
237      * This is a hack.  If we are using an external rmmproc,
238      * then save the current folder to the context file,
239      * so the external rmmproc will remove files from the correct
240      * directory.  This should be moved to folder_delmsgs().
241      */
242     if (rmmproc) {
243         context_replace (pfolder, folder);
244         context_save ();
245         fflush (stdout);
246     }
247
248     /* If -nolink, then "remove" messages from source folder.
249      *
250      * Note that folder_delmsgs does not call the delete hook
251      * because the message has already been handled above.
252      */
253     if (!linkf) {
254         folder_delmsgs (mp, unlink_msgs, 1);
255     }
256
257     clsfolds (folders, foldp);
258
259     if (mp->hghsel != mp->curmsg
260         && (mp->numsel != mp->nummsg || linkf))
261         seq_setcur (mp, mp->hghsel);
262     seq_save (mp);      /* synchronize message sequences */
263
264     context_replace (pfolder, folder);  /* update current folder   */
265     context_save ();                    /* save the context file   */
266     folder_free (mp);                   /* free folder structure   */
267     done (0);
268     return 1;
269 }
270
271
272 /*
273  * Read all the destination folders and
274  * create folder structures for all of them.
275  */
276
277 static void
278 opnfolds (struct st_fold *folders, int nfolders)
279 {
280     char nmaildir[BUFSIZ];
281     register struct st_fold *fp, *ep;
282     register struct msgs *mp;
283
284     for (fp = folders, ep = folders + nfolders; fp < ep; fp++) {
285         chdir (m_maildir (""));
286         strncpy (nmaildir, m_maildir (fp->f_name), sizeof(nmaildir));
287
288     create_folder (nmaildir, 0, done);
289
290         if (chdir (nmaildir) == NOTOK)
291             adios (nmaildir, "unable to change directory to");
292         if (!(mp = folder_read (fp->f_name)))
293             adios (NULL, "unable to read folder %s", fp->f_name);
294         mp->curmsg = 0;
295
296         fp->f_mp = mp;
297
298         chdir (maildir);
299     }
300 }
301
302
303 /*
304  * Set the Previous-Sequence and then sychronize the
305  * sequence file, for each destination folder.
306  */
307
308 static void
309 clsfolds (struct st_fold *folders, int nfolders)
310 {
311     register struct st_fold *fp, *ep;
312     register struct msgs *mp;
313
314     for (fp = folders, ep = folders + nfolders; fp < ep; fp++) {
315         mp = fp->f_mp;
316         seq_setprev (mp);
317         seq_save (mp);
318     }
319 }
320
321
322 /*
323  * If you have a "rmmproc" defined, we called that
324  * to remove all the specified files.  If "rmmproc"
325  * is not defined, then just unlink the files.
326  */
327
328 static void
329 remove_files (int filep, char **files)
330 {
331     int i;
332     char **vec;
333
334     /* If rmmproc is defined, we use that */
335     if (rmmproc) {
336         vec = files++;          /* vec[0] = filevec[0] */
337         files[filep] = NULL;    /* NULL terminate list */
338
339         fflush (stdout);
340         vec[0] = r1bindex (rmmproc, '/');
341         execvp (rmmproc, vec);
342         adios (rmmproc, "unable to exec");
343     }
344
345     /* Else just unlink the files */
346     files++;    /* advance past filevec[0] */
347     for (i = 0; i < filep; i++) {
348         if (unlink (files[i]) == NOTOK)
349             admonish (files[i], "unable to unlink");
350     }
351 }
352
353
354 /*
355  * Link (or copy) the message into each of
356  * the destination folders.
357  */
358
359 static int
360 m_file (char *msgfile, struct st_fold *folders, int nfolders, int preserve, int refile)
361 {
362     int msgnum;
363     struct st_fold *fp, *ep;
364
365     for (fp = folders, ep = folders + nfolders; fp < ep; fp++) {
366         if ((msgnum = folder_addmsg (&fp->f_mp, msgfile, 1, 0, preserve, nfolders == 1 && refile, maildir)) == -1)
367             return 1;
368     }
369     return 0;
370 }