Renamed r1bindex() to mhbasename(), to make its function becomes clear.
[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 INPLSW 1
49         { "inplace", 0 },
50 #define NINPLSW 2
51         { "noinplace", 0 },
52 #define DATESW 3
53         { "date", 0 },
54 #define NDATESW 4
55         { "nodate", 0 },
56 #define TEXTSW 5
57         { "text body", 0 },
58 #define VERSIONSW 6
59         { "version", 0 },
60 #define HELPSW 7
61         { "help", 0 },
62 #define LISTSW 8
63         { "list", 1 },
64 #define DELETESW 9
65         { "delete", 2 },
66 #define NUMBERSW 10
67         { "number", 2 },
68 #define APPENDSW 11
69         { "append", 1 },
70 #define PRESERVESW 12
71         { "preserve", 1 },
72 #define NOPRESERVESW 13
73         { "nopreserve", 3 },
74         { NULL, 0 }
75 };
76
77 /*
78 ** static prototypes
79 */
80 static void make_comp(unsigned char **);
81
82
83 int
84 main(int argc, char **argv)
85 {
86         int inplace = 1, datesw = 1;
87         int msgnum;
88         char *cp, *maildir;
89         unsigned char *comp = NULL;
90         char *text = NULL, *folder = NULL, buf[BUFSIZ];
91         char **argp, **arguments;
92         struct msgs_array msgs = { 0, 0, NULL };
93         struct msgs *mp;
94         int append = 0;  /* append annotations instead of default prepend */
95         int delete = -2;  /* delete header element if set */
96         int list = 0;  /* list header elements if set */
97         int number = 0; /* delete specific number of like elements if set */
98
99 #ifdef LOCALE
100         setlocale(LC_ALL, "");
101 #endif
102         invo_name = mhbasename(argv[0]);
103
104         /* read user profile/context */
105         context_read();
106
107         arguments = getarguments(invo_name, argc, argv, 1);
108         argp = arguments;
109
110         while ((cp = *argp++)) {
111                 if (*cp == '-') {
112                         switch (smatch(++cp, switches)) {
113                                 case AMBIGSW:
114                                         ambigsw(cp, switches);
115                                         done(1);
116                                 case UNKWNSW:
117                                         adios(NULL, "-%s unknown", cp);
118
119                                 case HELPSW:
120                                         snprintf(buf, sizeof(buf), "%s [+folder] [msgs] [switches]",
121                                                 invo_name);
122                                         print_help(buf, switches, 1);
123                                         done(1);
124                                 case VERSIONSW:
125                                         print_version(invo_name);
126                                         done(1);
127
128                                 case COMPSW:
129                                         if (comp)
130                                                 adios(NULL, "only one component at a time!");
131                                         if (!(comp = *argp++) || *comp == '-')
132                                                 adios(NULL, "missing argument to %s", argp[-2]);
133                                         continue;
134
135                                 case DATESW:
136                                         datesw++;
137                                         continue;
138                                 case NDATESW:
139                                         datesw = 0;
140                                         continue;
141
142                                 case INPLSW:
143                                         inplace++;
144                                         continue;
145                                 case NINPLSW:
146                                         inplace = 0;
147                                         continue;
148
149                                 case TEXTSW:
150                                         if (text)
151                                                 adios(NULL, "only one body at a time!");
152                                         if (!(text = *argp++) || *text == '-')
153                                                 adios(NULL, "missing argument to %s", argp[-2]);
154                                         continue;
155
156                                 case DELETESW:  /* delete annotations */
157                                         delete = 0;
158                                         continue;
159
160                                 case LISTSW:  /* produce a listing */
161                                         list = 1;
162                                         continue;
163
164                                 case NUMBERSW:  /* number listing or delete by number */
165                                         if (number != 0)
166                                                 adios(NULL, "only one number at a time!");
167
168                                         if (argp - arguments == argc - 1 || **argp == '-')
169                                                 number = 1;
170
171                                         else {
172                                                         if (strcmp(*argp, "all") == 0)
173                                                                 number = -1;
174
175                                                         else if (!(number = atoi(*argp)))
176                                                         adios(NULL, "missing argument to %s", argp[-2]);
177
178                                                 argp++;
179                                         }
180
181                                         delete = number;
182                                         continue;
183
184                                 case APPENDSW:  /* append annotations instead of default prepend */
185                                         append = 1;
186                                         continue;
187
188                                 case PRESERVESW:  /* preserve access and modification times on annotated message */
189                                         annopreserve(1);
190                                         continue;
191
192                                 case NOPRESERVESW:  /* don't preserve access and modification times on annotated message (default) */
193                                         annopreserve(0);
194                                         continue;
195                         }
196                 }
197                 if (*cp == '+' || *cp == '@') {
198                         if (folder)
199                                 adios(NULL, "only one folder at a time!");
200                         else
201                                 folder = pluspath(cp);
202                 } else
203                         app_msgarg(&msgs, cp);
204         }
205
206 #ifdef UCI
207         if (strcmp(invo_name, "fanno") == 0)  /* ugh! */
208                 datesw = 0;
209 #endif /* UCI */
210
211         if (!context_find("path"))
212                 free(path("./", TFOLDER));
213         if (!msgs.size)
214                 app_msgarg(&msgs, "cur");
215         if (!folder)
216                 folder = getfolder(1);
217         maildir = m_maildir(folder);
218
219         if (chdir(maildir) == NOTOK)
220                 adios(maildir, "unable to change directory to");
221
222         /* read folder and create message structure */
223         if (!(mp = folder_read(folder)))
224                 adios(NULL, "unable to read folder %s", folder);
225
226         /* check for empty folder */
227         if (mp->nummsg == 0)
228                 adios(NULL, "no messages in %s", folder);
229
230         /* parse all the message ranges/sequences and set SELECTED */
231         for (msgnum = 0; msgnum < msgs.size; msgnum++)
232                 if (!m_convert(mp, msgs.msgs[msgnum]))
233                         done(1);
234
235         make_comp(&comp);
236
237         /* annotate all the SELECTED messages */
238         for (msgnum = mp->lowsel; msgnum <= mp->hghsel; msgnum++) {
239                 if (is_selected(mp, msgnum)) {
240                         if (list)
241                                 annolist(m_name(msgnum), comp, text, number);
242                         else
243                                 annotate(m_name(msgnum), comp, text, inplace, datesw, delete, append);
244                 }
245         }
246
247         context_replace(pfolder, folder);  /* update current folder  */
248         seq_setcur(mp, mp->lowsel);  /* update current message */
249         seq_save(mp);  /* synchronize message sequences */
250         folder_free(mp);  /* free folder/message structure */
251         context_save();  /* save the context file */
252         done(0);
253         return 1;
254 }
255
256 static void
257 make_comp(unsigned char **ap)
258 {
259         register unsigned char *cp;
260         char buffer[BUFSIZ];
261
262         if (*ap == NULL) {
263                 printf("Enter component name: ");
264                 fflush(stdout);
265
266                 if (fgets(buffer, sizeof buffer, stdin) == NULL)
267                         done(1);
268                 *ap = trimcpy(buffer);
269         }
270
271         if ((cp = *ap + strlen(*ap) - 1) > *ap && *cp == ':')
272                 *cp = 0;
273         if (strlen(*ap) == 0)
274                 adios(NULL, "null component name");
275         if (**ap == '-')
276                 adios(NULL, "invalid component name %s", *ap);
277         if (strlen(*ap) >= NAMESZ)
278                 adios(NULL, "too large component name %s", *ap);
279
280         for (cp = *ap; *cp; cp++)
281                 if (!isalnum(*cp) && *cp != '-')
282                         adios(NULL, "invalid component name %s", *ap);
283 }