3 * new.c -- as new, list all folders with unseen messages
4 * -- as fnext, move to next folder with unseen messages
5 * -- as fprev, move to previous folder with unseen messages
6 * -- as unseen, scan all unseen messages
7 * This code is Copyright (c) 2008, by the authors of nmh. See the
8 * COPYRIGHT file in the root directory of the nmh distribution for
9 * complete copyright information.
11 * Inspired by Luke Mewburn's new: http://www.mewburn.net/luke/src/new
14 #include <sys/types.h>
21 #include <h/crawl_folders.h>
24 static struct swit switches[] = {
36 static enum { NEW, FNEXT, FPREV, UNSEEN } run_mode = NEW;
38 /* check_folders uses this to maintain state with both .folders list of
39 * folders and with crawl_folders. */
41 struct node **first, **cur_node;
48 /* Return the number of messages in a string list of message numbers. */
50 count_messages(char *field)
56 field = getcpy(field);
58 /* copied from seq_read.c:seq_init */
59 for (ap = brkstring (field, " ", "\n"); *ap; ap++) {
60 if ((cp = strchr(*ap, '-')))
62 if ((j = m_atoi (*ap)) > 0) {
63 k = cp ? m_atoi (cp) : j;
74 /* Return TRUE if the sequence 'name' is in 'sequences'. */
76 seq_in_list(char *name, char *sequences[])
80 for (i = 0; sequences[i] != NULL; i++) {
81 if (strcmp(name, sequences[i]) == 0) {
89 /* Return the string list of message numbers from the sequences file, or NULL
92 get_msgnums(char *folder, char *sequences[])
94 char *seqfile = concat(m_maildir(folder), "/", mh_seq, (void *)NULL);
95 FILE *fp = fopen(seqfile, "r");
97 char name[NAMESZ], field[BUFSIZ];
99 char *msgnums = NULL, *this_msgnums, *old_msgnums;
101 /* no sequences file -> no messages */
106 /* copied from seq_read.c:seq_public */
107 for (state = FLD;;) {
108 switch (state = m_getfld (state, name, field, sizeof(field), fp)) {
112 if (state == FLDPLUS) {
114 while (state == FLDPLUS) {
115 state = m_getfld (state, name, field,
117 cp = add (field, cp);
120 /* Here's where we differ from seq_public: if it's in a
121 * sequence we want, save the list of messages. */
122 if (seq_in_list(name, sequences)) {
123 this_msgnums = trimcpy(cp);
124 if (msgnums == NULL) {
125 msgnums = this_msgnums;
127 old_msgnums = msgnums;
128 msgnums = concat(old_msgnums, " ",
129 this_msgnums, (void *)NULL);
137 if (seq_in_list(name, sequences)) {
138 this_msgnums = trimcpy(field);
139 if (msgnums == NULL) {
140 msgnums = this_msgnums;
142 old_msgnums = msgnums;
143 msgnums = concat(old_msgnums, " ",
144 this_msgnums, (void *)NULL);
157 adios (NULL, "no blank lines are permitted in %s", seqfile);
164 adios (NULL, "%s is poorly formatted", seqfile);
166 break; /* break from for loop */
174 /* Check `folder' (of length `len') for interesting messages, filling in the
177 check_folder(char *folder, size_t len, struct list_state *b)
179 char *msgnums = get_msgnums(folder, b->sequences);
180 int is_cur = strcmp(folder, b->cur) == 0;
182 if (is_cur || msgnums != NULL) {
183 if (*b->first == NULL) {
184 *b->first = b->node = mh_xmalloc(sizeof(*b->node));
186 b->node->n_next = mh_xmalloc(sizeof(*b->node));
187 b->node = b->node->n_next;
189 b->node->n_name = folder;
190 b->node->n_field = msgnums;
192 if (*b->maxlen < len) {
197 /* Save the node for the current folder, so we can fall back to it. */
199 *b->cur_node = b->node;
204 crawl_callback(char *folder, void *baton)
206 check_folder(folder, strlen(folder), baton);
210 /* Scan folders, returning:
211 * first -- list of nodes for all folders which have desired messages;
212 * if the current folder is listed in .folders, it is also in
213 * the list regardless of whether it has any desired messages
214 * last -- last node in list
215 * cur_node -- node of current folder, if listed in .folders
216 * maxlen -- length of longest folder name
218 * `cur' points to the name of the current folder, `folders' points to the
219 * name of a .folder (if NULL, crawl all folders), and `sequences' points to
220 * the array of sequences for which to look.
222 * An empty list is returned as first=last=NULL.
225 check_folders(struct node **first, struct node **last,
226 struct node **cur_node, size_t *maxlen,
227 char *cur, char *folders, char *sequences[])
234 *first = *last = *cur_node = NULL;
238 b.cur_node = cur_node;
241 b.sequences = sequences;
243 if (folders == NULL) {
244 chdir(m_maildir(""));
245 crawl_folders(".", crawl_callback, &b);
247 fp = fopen(folders, "r");
249 adios(NULL, "failed to read %s", folders);
251 while (vfgets(fp, &line) == OK) {
252 len = strlen(line) - 1;
254 check_folder(getcpy(line), len, &b);
259 if (*first != NULL) {
260 b.node->n_next = NULL;
265 /* Return a single string of the `sequences' joined by a space (' '). */
267 join_sequences(char *sequences[])
273 for (i = 0; sequences[i] != NULL; i++) {
274 len += strlen(sequences[i]) + 1;
276 result = mh_xmalloc(len + 1);
278 for (i = 0, cp = result; sequences[i] != NULL; i++, cp += len + 1) {
279 len = strlen(sequences[i]);
280 memcpy(cp, sequences[i], len);
283 /* -1 to overwrite the last delimiter */
289 /* Return a struct node for the folder to change to. This is the next
290 * (previous, if FPREV mode) folder with desired messages, or the current
291 * folder if no folders have desired. If NEW or UNSEEN mode, print the
292 * output but don't change folders.
294 * n_name is the folder to change to, and n_field is the string list of
295 * desired message numbers.
298 doit(char *cur, char *folders, char *sequences[])
300 struct node *first, *cur_node, *node, *last, *prev;
302 int count, total = 0;
303 char *command = NULL, *sequences_s = NULL;
305 if (cur == NULL || cur[0] == '\0') {
309 check_folders(&first, &last, &cur_node, &folder_len, cur,
312 if (run_mode == FNEXT || run_mode == FPREV) {
314 /* No folders at all... */
316 } else if (first->n_next == NULL) {
317 /* We have only one node; any desired messages in it? */
318 if (first->n_field == NULL) {
323 } else if (cur_node == NULL) {
324 /* Current folder is not listed in .folders, return first. */
327 } else if (run_mode == UNSEEN) {
328 sequences_s = join_sequences(sequences);
331 for (node = first, prev = NULL;
333 prev = node, node = node->n_next) {
334 if (run_mode == FNEXT) {
335 /* If we have a previous node and it is the current
336 * folder, return this node. */
337 if (prev != NULL && strcmp(prev->n_name, cur) == 0) {
340 } else if (run_mode == FPREV) {
341 if (strcmp(node->n_name, cur) == 0) {
342 /* Found current folder in fprev mode; if we have a
343 * previous node in the list, return it; else return
350 } else if (run_mode == UNSEEN) {
351 if (node->n_field == NULL) {
355 printf("\n%d %s messages in %s",
356 count_messages(node->n_field),
359 if (strcmp(node->n_name, cur) == 0) {
360 puts(" (*: current folder)");
366 /* TODO: Split enough of scan.c out so that we can call it here. */
367 command = concat("scan +", node->n_name, " ", sequences_s,
372 if (node->n_field == NULL) {
376 count = count_messages(node->n_field);
379 printf("%-*s %6d.%c %s\n",
380 (int) folder_len, node->n_name,
382 (strcmp(node->n_name, cur) == 0 ? '*' : ' '),
387 /* If we're fnext, we haven't checked the last node yet. If it's the
388 * current folder, return the first node. */
389 if (run_mode == FNEXT && strcmp(last->n_name, cur) == 0) {
393 if (run_mode == NEW) {
394 printf("%-*s %6d.\n", (int) folder_len, " total", total);
401 main(int argc, char **argv)
403 char **ap, *cp, **argp, **arguments;
405 char *folders = NULL;
406 char *sequences[NUMATTRS + 1];
415 setlocale(LC_ALL, "");
417 invo_name = r1bindex(argv[0], '/');
419 /* read user profile/context */
422 arguments = getarguments (invo_name, argc, argv, 1);
428 while ((cp = *argp++)) {
430 switch (smatch (++cp, switches)) {
432 ambigsw (cp, switches);
435 adios (NULL, "-%s unknown", cp);
438 snprintf (help, sizeof(help), "%s [switches] [sequences]",
440 print_help (help, switches, 1);
443 print_version(invo_name);
447 if (!(folders = *argp++) || *folders == '-')
448 adios(NULL, "missing argument to %s", argp[-2]);
451 if (!(invo_name = *argp++) || *invo_name == '-')
452 adios(NULL, "missing argument to %s", argp[-2]);
453 invo_name = r1bindex(invo_name, '/');
457 /* have a sequence argument */
458 if (!seq_in_list(cp, sequences)) {
464 if (strcmp(invo_name, "fnext") == 0) {
466 } else if (strcmp(invo_name, "fprev") == 0) {
468 } else if (strcmp(invo_name, "unseen") == 0) {
472 if (folders == NULL) {
475 if (folders[0] != '/') {
476 folders = m_maildir(folders);
481 /* no sequence arguments; use unseen */
482 unseen = context_find(usequence);
483 if (unseen == NULL || unseen[0] == '\0') {
484 adios(NULL, "must specify sequences or set %s", usequence);
486 for (ap = brkstring(unseen, " ", "\n"); *ap; ap++) {
487 sequences[i++] = *ap;
492 folder = doit(context_find(pfolder), folders, sequences);
493 if (folder == NULL) {
498 if (run_mode == UNSEEN) {
499 /* All the scan(1)s it runs change the current folder, so we
500 * need to put it back. Unfortunately, context_replace lamely
501 * ignores the new value you give it if it is the same one it
502 * has in memory. So, we'll be lame, too. I'm not sure if i
503 * should just change context_replace... */
504 context_replace(pfolder, "defeat_context_replace_optimization");
507 /* update current folder */
508 context_replace(pfolder, folder->n_name);
510 if (run_mode == FNEXT || run_mode == FPREV) {
511 printf("%s %s\n", folder->n_name, folder->n_field);