2 ** whatnow.c -- the WhatNow shell
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.
8 ** The inclusion of attachments is eased by
9 ** using the header field name mechanism added to anno and send.
10 ** The header field name for attachments is predefined.
12 ** Several commands have been added at the whatnow prompt:
14 ** cd [ directory ] This option works just like the shell's
15 ** cd command and lets the user change the
16 ** directory from which attachments are
17 ** taken so that long path names are not
18 ** needed with every file.
20 ** ls [ ls-options ] This option works just like the normal
21 ** ls command and exists to allow the user
22 ** to verify file names in the directory.
24 ** pwd This option works just like the normal
25 ** pwd command and exists to allow the user
26 ** to verify the directory.
28 ** attach files This option attaches the named files to
31 ** alist [-ln] This option lists the attachments on the
32 ** draft. -l gets long listings, -n gets
35 ** detach files This option removes attachments from the
36 ** detach -n numbers draft. This can be done by file name or
37 ** by attachment number.
46 static struct swit whatnowswitches[] = {
48 { "editor editor", 0 },
52 { "prompt string", 4 },
61 ** Options at the "whatnow" prompt
63 static struct swit aleqs[] = {
65 { "edit [<editor> <switches>]", 0 },
67 { "refile [<switches>] +folder", 0 },
69 { "display [<switches>]", 0 },
71 { "list [<switches>]", 0 },
73 { "send [<switches>]", 0 },
75 { "push [<switches>]", 0 },
81 { "cd [directory]", 0 },
86 #define ATTACHCMDSW 11
88 #define DETACHCMDSW 12
91 { "alist [-ln] ", 2 },
95 static char *myprompt = "\nWhat now? ";
100 static int editfile(char **, char **, char *, int, struct msgs *,
101 char *, char *, int);
102 static int sendfile(char **, char *, int);
103 static int removefile(char *);
104 static void writelscmd(char *, int, char **);
105 static void writesomecmd(char *buf, int bufsz, char *cmd, char *trailcmd, char **argp);
106 static FILE* popen_in_dir(const char *dir, const char *cmd, const char *type);
107 static int system_in_dir(const char *dir, const char *cmd);
111 static int copyf(char *, char *);
116 main(int argc, char **argv)
118 int nedit = 0, use = 0;
120 char *ed = NULL, *drft = NULL, *msgnam = NULL;
121 char buf[BUFSIZ], prompt[BUFSIZ];
122 char **argp, **arguments;
124 char cwd[MAXPATHLEN + 1]; /* current working directory */
125 char file[MAXPATHLEN + 1]; /* file name buffer */
126 char shell[MAXPATHLEN + 1]; /* shell response buffer */
127 FILE *f; /* read pointer for bgnd proc */
128 char *l; /* set on -l to alist command */
129 int n; /* set on -n to alist command */
132 setlocale(LC_ALL, "");
135 invo_name = mhbasename(argv[0]);
137 /* read user profile/context */
140 arguments = getarguments(invo_name, argc, argv, 1);
144 ** Get the initial current working directory.
147 if (getcwd(cwd, sizeof (cwd)) == NULL) {
148 adios("getcwd", "could not get working directory");
151 while ((cp = *argp++)) {
153 switch (smatch(++cp, whatnowswitches)) {
155 ambigsw(cp, whatnowswitches);
158 adios(NULL, "-%s unknown", cp);
161 snprintf(buf, sizeof(buf),
162 "%s [switches] [file]",
164 print_help(buf, whatnowswitches, 1);
167 print_version(invo_name);
171 if (!(ed = *argp++) || *ed == '-')
172 adios(NULL, "missing argument to %s",
181 if (!(myprompt = *argp++) || *myprompt == '-')
182 adios(NULL, "missing argument to %s",
189 adios(NULL, "only one draft at a time!");
194 if ((drft == NULL && (drft = getenv("mhdraft")) == NULL) || *drft == 0)
195 drft = getcpy(m_draft(seq_cur));
197 msgnam = (cp = getenv("mhaltmsg")) && *cp ? getcpy(cp) : NULL;
199 if ((cp = getenv("mhuse")) && *cp)
202 if (ed == NULL && ((ed = getenv("mheditor")) == NULL || *ed == 0)) {
207 /* start editing the draft, unless -noedit was given */
208 if (!nedit && editfile(&ed, NULL, drft, use, NULL, msgnam, NULL, 1)
212 snprintf(prompt, sizeof(prompt), myprompt, invo_name);
214 if (!(argp = getans(prompt, aleqs))) {
218 switch (smatch(*argp, aleqs)) {
220 /* display the msg being replied to or distributed */
222 showfile(++argp, msgnam);
224 advise(NULL, "no alternate message to display");
228 /* Call an editor on the draft file */
231 if (editfile(&ed, argp, drft, NOUSE, NULL,
232 msgnam, NULL, 1) == NOTOK)
237 /* display the draft file */
238 showfile(++argp, drft);
243 if (stat(drft, &st) != NOTOK) {
244 advise(NULL, "draft left on %s", drft);
249 /* Delete draft and exit */
254 /* Send draft in background */
255 if (sendfile(++argp, drft, 1))
261 sendfile(++argp, drft, 0);
265 /* Refile the draft */
266 if (refile(++argp, drft) == 0)
272 ** Change the working directory for attachments
274 ** Run the directory through the user's shell
275 ** so that we can take advantage of any syntax
276 ** that the user is accustomed to. Read back
277 ** the absolute path.
280 if (*(argp+1) == NULL) {
281 sprintf(buf, "$SHELL -c \"cd;pwd\"");
283 writesomecmd(buf, BUFSIZ, "cd", "pwd", argp);
285 if ((f = popen_in_dir(cwd, buf, "r")) != (FILE *)0) {
286 fgets(cwd, sizeof (cwd), f);
288 if (strchr(cwd, '\n') != NULL)
289 *strchr(cwd, '\n') = '\0';
293 advise("popen", "could not get directory");
299 /* Print the working directory for attachments */
305 ** List files in the current attachment working
308 ** Use the user's shell so that we can take
309 ** advantage of any syntax that the user is
312 writelscmd(buf, sizeof(buf), argp);
313 system_in_dir(cwd, buf);
318 ** List attachments on current draft. Options are:
320 ** -l long listing (full path names)
321 ** -n numbers listing
327 while (*++argp != NULL) {
328 if (strcmp(*argp, "-l") == 0)
331 else if (strcmp(*argp, "-n") == 0)
334 else if (strcmp(*argp, "-ln") == 0 ||
335 strcmp(*argp, "-nl") == 0) {
345 advise(NULL, "usage is alist [-ln].");
348 annolist(drft, attach_hdr, l, n);
354 ** Attach files to current draft.
357 if (*(argp+1) == NULL) {
358 advise(NULL, "attach command requires file argument(s).");
363 ** Build a command line that causes the user's
364 ** shell to list the file name arguments.
365 ** This handles and wildcard expansion, tilde
368 writelscmd(buf, sizeof(buf), argp);
371 ** Read back the response from the shell,
372 ** which contains a number of lines with one
373 ** file name per line. Remove off the newline.
374 ** Determine whether we have an absolute or
375 ** relative path name. Prepend the current
376 ** working directory to relative path names.
377 ** Add the attachment annotation to the draft.
379 if ((f = popen_in_dir(cwd, buf, "r")) != (FILE *)0) {
380 while (fgets(shell, sizeof (shell), f)
382 *(strchr(shell, '\n')) = '\0';
385 annotate(drft, attach_hdr,
389 sprintf(file, "%s/%s", cwd,
391 annotate(drft, attach_hdr,
399 advise("popen", "could not get file from shell");
406 ** Detach files from current draft.
408 ** Scan the arguments for a -n. Mixed file
409 ** names and numbers aren't allowed, so this
410 ** catches a -n anywhere in the argument list.
413 for (n = 0, arguments = argp + 1;
416 if (strcmp(*arguments, "-n") == 0) {
423 ** A -n was found so interpret the arguments as
424 ** attachment numbers. Decrement any remaining
425 ** argument number that is greater than the one
426 ** just processed after processing each one so
427 ** that the numbering stays correct.
431 for (arguments = argp + 1;
434 if (strcmp(*arguments, "-n") == 0)
437 if (**arguments != '\0') {
438 n = atoi(*arguments);
439 annotate(drft, attach_hdr,
443 for (argp = arguments + 1; *argp != NULL; argp++) {
444 if (atoi(*argp) > n) {
445 if (atoi(*argp) == 1)
448 sprintf(*argp, "%d", atoi(*argp) - 1);
456 ** The arguments are interpreted as file names.
457 ** Run them through the user's shell for wildcard
458 ** expansion and other goodies. Do this from
459 ** the current working directory if the argument
460 ** is not an absolute path name (does not begin
463 ** We feed all the file names to the shell at
464 ** once, otherwise you can't provide a file name
465 ** with a space in it.
467 writelscmd(buf, sizeof(buf), argp);
468 if ((f = popen_in_dir(cwd, buf, "r")) != (FILE *)0) {
469 while (fgets(shell, sizeof (shell), f)
471 *(strchr(shell, '\n')) = '\0';
472 annotate(drft, attach_hdr, shell,
477 advise("popen", "could not get file from shell");
483 /* Unknown command */
484 advise(NULL, "say what?");
494 ** Build a command line of the form $SHELL -c "cd 'cwd'; cmd argp ... ;
498 writesomecmd(char *buf, int bufsz, char *cmd, char *trailcmd, char **argp)
502 ** Note that we do not quote -- the argp from the user
503 ** is assumed to be quoted as they desire. (We can't treat
504 ** it as pure literal as that would prevent them using ~,
505 ** wildcards, etc.) The buffer produced by this function
506 ** should be given to popen_in_dir() or system_in_dir() so
507 ** that the current working directory is set correctly.
509 int ln = snprintf(buf, bufsz, "$SHELL -c \"%s", cmd);
511 ** NB that some snprintf() return -1 on overflow rather than the
512 ** new C99 mandated 'number of chars that would have been written'
515 ** length checks here and inside the loop allow for the
516 ** trailing ';', trailcmd, '"' and NUL
518 int trailln = strlen(trailcmd) + 3;
519 if (ln < 0 || ln + trailln > bufsz)
520 adios(NULL, "arguments too long");
524 while (*++argp != NULL) {
526 /* +1 for leading space */
527 if (ln + trailln + 1 > bufsz - (cp-buf))
528 adios(NULL, "arguments too long");
530 memcpy(cp, *argp, ln+1);
535 strcpy(cp, trailcmd);
543 ** Build a command line that causes the user's shell to list the file name
544 ** arguments. This handles and wildcard expansion, tilde expansion, etc.
547 writelscmd(char *buf, int bufsz, char **argp)
549 writesomecmd(buf, bufsz, "ls", "", argp);
553 ** Like system(), but run the command in directory dir.
554 ** This assumes the program is single-threaded!
557 system_in_dir(const char *dir, const char *cmd)
561 if (getcwd(olddir, sizeof(olddir)) == 0)
562 adios("getcwd", "could not get working directory");
564 adios("chdir", "could not change working directory");
566 if (chdir(olddir) != 0)
567 adios("chdir", "could not change working directory");
571 /* ditto for popen() */
573 popen_in_dir(const char *dir, const char *cmd, const char *type)
577 if (getcwd(olddir, sizeof(olddir)) == 0)
578 adios("getcwd", "could not get working directory");
580 adios("chdir", "could not change working directory");
581 f = popen(cmd, type);
582 if (chdir(olddir) != 0)
583 adios("chdir", "could not change working directory");
592 static int reedit = 0; /* have we been here before? */
593 static char *edsave = NULL; /* the editor we used previously */
597 editfile(char **ed, char **arg, char *file, int use, struct msgs *mp,
598 char *altmsg, char *cwd, int save_editor)
600 int pid, status, vecp;
601 char altpath[BUFSIZ], linkpath[BUFSIZ];
602 char *cp, *vec[MAXARGS];
607 #endif /* HAVE_LSTAT */
609 /* Was there a previous edit session? */
611 if (!*ed) { /* no explicit editor */
612 *ed = edsave; /* so use the previous one */
613 if ((cp = mhbasename(*ed)) == NULL)
616 /* unless we've specified it via "editor-next" */
617 cp = concat(cp, "-next", NULL);
618 if ((cp = context_find(cp)) != NULL)
622 /* set initial editor */
623 if (*ed == NULL && (*ed = context_find("editor")) == NULL)
628 if (mp == NULL || *altmsg == '/' || cwd == NULL)
629 strncpy(altpath, altmsg, sizeof(altpath));
631 snprintf(altpath, sizeof(altpath), "%s/%s",
632 mp->foldpath, altmsg);
634 strncpy(linkpath, altmsglink, sizeof(linkpath));
636 snprintf(linkpath, sizeof(linkpath), "%s/%s",
643 if (link(altpath, linkpath) == NOTOK) {
644 symlink(altpath, linkpath);
649 #else /* not HAVE_LSTAT */
650 link(altpath, linkpath);
651 #endif /* not HAVE_LSTAT */
654 context_save(); /* save the context file */
657 switch (pid = fork()) {
659 advise("fork", "unable to");
668 m_putenv("mhfolder", mp->foldpath);
669 m_putenv("editalt", altpath);
673 vec[vecp++] = mhbasename(*ed);
676 vec[vecp++] = *arg++;
681 fprintf(stderr, "unable to exec ");
686 if ((status = pidwait(pid, NOTOK))) {
687 if (((status & 0xff00) != 0xff00)
688 && (!reedit || (status & 0x00ff))) {
689 if (!use && (status & 0xff00) && (rename(file, cp = m_backup (file)) != NOTOK)) {
690 advise(NULL, "problems with edit--draft left in %s", cp);
692 advise(NULL, "problems with edit--%s preserved", file);
695 status = -2; /* maybe "reedit ? -2 : -1"? */
701 if (altmsg && mp && !is_readonly(mp) && (slinked ?
702 lstat (linkpath, &st) != NOTOK &&
703 S_ISREG(st.st_mode) &&
704 copyf(linkpath, altpath) == NOTOK :
705 stat(linkpath, &st) != NOTOK &&
707 (unlink(altpath) == NOTOK ||
708 link(linkpath, altpath) == NOTOK)))
709 advise(linkpath, "unable to update %s from", altmsg);
710 #else /* HAVE_LSTAT */
711 if (altmsg && mp && !is_readonly(mp) &&
712 stat(linkpath, &st) != NOTOK &&
714 (unlink(altpath) == NOTOK ||
715 link(linkpath, altpath) == NOTOK))
716 advise(linkpath, "unable to update %s from", altmsg);
717 #endif /* HAVE_LSTAT */
720 /* normally, we remember which editor we used */
722 edsave = getcpy(*ed);
734 copyf(char *ifile, char *ofile)
739 if ((in = open(ifile, O_RDONLY)) == NOTOK)
741 if ((out = open(ofile, O_WRONLY | O_TRUNC)) == NOTOK) {
742 admonish(ofile, "unable to open and truncate");
747 while ((i = read(in, buffer, sizeof(buffer))) > OK)
748 if (write(out, buffer, i) != i) {
749 advise(ofile, "may have damaged");
758 #endif /* HAVE_LSTAT */
766 sendfile(char **arg, char *file, int pushsw)
772 context_save(); /* save the context file */
775 for (i = 0; (child_id = fork()) == NOTOK && i < 5; i++)
779 advise(NULL, "unable to fork, so sending directly...");
783 vec[vecp++] = invo_name;
785 vec[vecp++] = "-push";
788 vec[vecp++] = *arg++;
792 execvp(sendproc, vec);
793 fprintf(stderr, "unable to exec ");
798 if (pidwait(child_id, OK) == 0)
806 ** Remove the draft file
810 removefile(char *drft)
812 if (unlink(drft) == NOTOK)
813 adios(drft, "unable to unlink");