anno: Undid the ``preserve argument hack''.
[mmh] / uip / anno.c
1 /*
2 ** anno.c -- annotate messages
3 **
4 ** This code is Copyright (c) 2002, by the authors of nmh.  See the
5 ** COPYRIGHT file in the root directory of the nmh distribution for
6 ** complete copyright information.
7 **
8 ** Three new options have been added: delete, list, and number.
9 ** Message header fields are used by the new MIME attachment code in
10 ** the send command.  Adding features to generalize the anno command
11 ** seemed to be a better approach than the creation of a new command
12 ** whose features would overlap with those of the anno command.
13 **
14 ** The -delete option deletes header elements that match the -component
15 ** field name.  If -delete is used without the -text option, the first
16 ** header field whose field name matches the component name is deleted.
17 ** If the -delete is used with the -text option, and the -text argument
18 ** begins with a /, the first header field whose field name matches the
19 ** component name and whose field body matches the text is deleted.  If
20 ** the -text argument does not begin with a /, then the text is assumed
21 ** to be the last component of a path name, and the first header field
22 ** whose field name matches the component name and a field body whose
23 ** last path name component matches the text is deleted.  If the -delete
24 ** option is used with the new -number option described below, the nth
25 ** header field whose field name matches the component name is deleted.
26 ** No header fields are deleted if none of the above conditions are met.
27 **
28 ** The -list option outputs the field bodies from each header field whose
29 ** field name matches the component name, one per line.  If no -text
30 ** option is specified, only the last path name component of each field
31 ** body is output.  The entire field body is output if the -text option
32 ** is used; the contents of the -text argument are ignored.  If the -list
33 ** option is used in conjuction with the new -number option described
34 ** below, each line is numbered starting with 1.  A tab separates the
35 ** number from the field body.
36 **
37 ** The -number option works with both the -delete and -list options as
38 ** described above.  The -number option takes an optional argument.  A
39 ** value of 1 is assumed if this argument is absent.
40 */
41
42 #include <h/mh.h>
43 #include <h/utils.h>
44
45 static struct swit switches[] = {
46 #define COMPSW 0
47         { "component field", 0 },
48 #define DATESW 1
49         { "date", 0 },
50 #define NDATESW 2
51         { "nodate", 0 },
52 #define TEXTSW 3
53         { "text body", 0 },
54 #define VERSIONSW 4
55         { "version", 0 },
56 #define HELPSW 5
57         { "help", 0 },
58 #define LISTSW 6
59         { "list", 1 },
60 #define DELETESW 7
61         { "delete", 2 },
62 #define NUMBERSW 8
63         { "number", 2 },
64 #define APPENDSW 9
65         { "append", 1 },
66 #define PRESERVESW 10
67         { "preserve", 1 },
68 #define NOPRESERVESW 11
69         { "nopreserve", 3 },
70         { NULL, 0 }
71 };
72
73 /*
74 ** static prototypes
75 */
76 static void make_comp(unsigned char **);
77
78
79 int
80 main(int argc, char **argv)
81 {
82         int datesw = 1;
83         int preserve = 0;
84         int msgnum;
85         char *cp, *maildir;
86         unsigned char *comp = NULL;
87         char *text = NULL, *folder = NULL, buf[BUFSIZ];
88         char **argp, **arguments;
89         struct msgs_array msgs = { 0, 0, NULL };
90         struct msgs *mp;
91         int append = 0;  /* append annotations instead of default prepend */
92         int delete = -2;  /* delete header element if set */
93         int list = 0;  /* list header elements if set */
94         int number = 0; /* delete specific number of like elements if set */
95
96 #ifdef LOCALE
97         setlocale(LC_ALL, "");
98 #endif
99         invo_name = mhbasename(argv[0]);
100
101         /* read user profile/context */
102         context_read();
103
104         arguments = getarguments(invo_name, argc, argv, 1);
105         argp = arguments;
106
107         while ((cp = *argp++)) {
108                 if (*cp == '-') {
109                         switch (smatch(++cp, switches)) {
110                         case AMBIGSW:
111                                 ambigsw(cp, switches);
112                                 done(1);
113                         case UNKWNSW:
114                                 adios(NULL, "-%s unknown", cp);
115
116                         case HELPSW:
117                                 snprintf(buf, sizeof(buf),
118                                         "%s [+folder] [msgs] [switches]",
119                                         invo_name);
120                                 print_help(buf, switches, 1);
121                                 done(1);
122                         case VERSIONSW:
123                                 print_version(invo_name);
124                                 done(1);
125
126                         case COMPSW:
127                                 if (comp)
128                                         adios(NULL, "only one component at a time!");
129                                 if (!(comp = *argp++) || *comp == '-')
130                                         adios(NULL, "missing argument to %s",
131                                                         argp[-2]);
132                                 continue;
133
134                         case DATESW:
135                                 datesw++;
136                                 continue;
137                         case NDATESW:
138                                 datesw = 0;
139                                 continue;
140
141                         case TEXTSW:
142                                 if (text)
143                                         adios(NULL, "only one body at a time!");
144                                 if (!(text = *argp++) || *text == '-')
145                                         adios(NULL, "missing argument to %s",
146                                                         argp[-2]);
147                                 continue;
148
149                         case DELETESW:  /* delete annotations */
150                                 delete = 0;
151                                 continue;
152
153                         case LISTSW:  /* produce a listing */
154                                 list = 1;
155                                 continue;
156
157                         case NUMBERSW: /* number listing or delete by number */
158                                 if (number != 0)
159                                         adios(NULL, "only one number at a time!");
160
161                                 if (argp-arguments == argc-1 || **argp == '-')
162                                         number = 1;
163
164                                 else {
165                                                 if (strcmp(*argp, "all") == 0)
166                                                         number = -1;
167
168                                                 else if (!(number = atoi(*argp)))
169                                                 adios(NULL, "missing argument to %s", argp[-2]);
170
171                                         argp++;
172                                 }
173
174                                 delete = number;
175                                 continue;
176
177                         case APPENDSW:  /* append annotations instead of default prepend */
178                                 append = 1;
179                                 continue;
180
181                         case PRESERVESW:  /* preserve access and modification times on annotated message */
182                                 preserve = 1;
183                                 continue;
184
185                         case NOPRESERVESW:  /* don't preserve access and modification times on annotated message (default) */
186                                 preserve = 0;
187                                 continue;
188                         }
189                 }
190                 if (*cp == '+' || *cp == '@') {
191                         if (folder)
192                                 adios(NULL, "only one folder at a time!");
193                         else
194                                 folder = getcpy(expandfol(cp));
195                 } else
196                         app_msgarg(&msgs, cp);
197         }
198
199         if (!msgs.size)
200                 app_msgarg(&msgs, seq_cur);
201         if (!folder)
202                 folder = getcurfol();
203         maildir = toabsdir(folder);
204
205         if (chdir(maildir) == NOTOK)
206                 adios(maildir, "unable to change directory to");
207
208         /* read folder and create message structure */
209         if (!(mp = folder_read(folder)))
210                 adios(NULL, "unable to read folder %s", folder);
211
212         /* check for empty folder */
213         if (mp->nummsg == 0)
214                 adios(NULL, "no messages in %s", folder);
215
216         /* parse all the message ranges/sequences and set SELECTED */
217         for (msgnum = 0; msgnum < msgs.size; msgnum++)
218                 if (!m_convert(mp, msgs.msgs[msgnum]))
219                         done(1);
220
221         make_comp(&comp);
222
223         /* annotate all the SELECTED messages */
224         for (msgnum = mp->lowsel; msgnum <= mp->hghsel; msgnum++) {
225                 if (is_selected(mp, msgnum)) {
226                         if (list)
227                                 annolist(m_name(msgnum), comp, text, number);
228                         else
229                                 annotate(m_name(msgnum), comp, text, datesw, delete, append, preserve);
230                 }
231         }
232
233         context_replace(curfolder, folder);  /* update current folder  */
234         seq_setcur(mp, mp->lowsel);  /* update current message */
235         seq_save(mp);  /* synchronize message sequences */
236         folder_free(mp);  /* free folder/message structure */
237         context_save();  /* save the context file */
238         done(0);
239         return 1;
240 }
241
242 static void
243 make_comp(unsigned char **ap)
244 {
245         register unsigned char *cp;
246         char buffer[BUFSIZ];
247
248         if (*ap == NULL) {
249                 printf("Enter component name: ");
250                 fflush(stdout);
251
252                 if (fgets(buffer, sizeof buffer, stdin) == NULL)
253                         done(1);
254                 *ap = trimcpy(buffer);
255         }
256
257         if ((cp = *ap + strlen(*ap) - 1) > *ap && *cp == ':')
258                 *cp = 0;
259         if (strlen(*ap) == 0)
260                 adios(NULL, "null component name");
261         if (**ap == '-')
262                 adios(NULL, "invalid component name %s", *ap);
263         if (strlen(*ap) >= NAMESZ)
264                 adios(NULL, "too large component name %s", *ap);
265
266         for (cp = *ap; *cp; cp++)
267                 if (!isalnum(*cp) && *cp != '-')
268                         adios(NULL, "invalid component name %s", *ap);
269 }