diff options
author | Sergey Poznyakoff <gray@gnu.org.ua> | 2007-06-26 13:55:13 +0000 |
---|---|---|
committer | Sergey Poznyakoff <gray@gnu.org.ua> | 2007-06-26 13:55:13 +0000 |
commit | bb12e6174bcff633960289781d58a34503e8cb2a (patch) | |
tree | 4dac9f6ead6b976560130ef8926f430acbe333a5 /examples/header.c | |
parent | 344611f57409fc78699198b165c8acf986aceb65 (diff) | |
download | mailutils-bb12e6174bcff633960289781d58a34503e8cb2a.tar.gz mailutils-bb12e6174bcff633960289781d58a34503e8cb2a.tar.bz2 |
New program
Diffstat (limited to 'examples/header.c')
-rw-r--r-- | examples/header.c | 373 |
1 files changed, 373 insertions, 0 deletions
diff --git a/examples/header.c b/examples/header.c new file mode 100644 index 000000000..0e21afdf5 --- /dev/null +++ b/examples/header.c @@ -0,0 +1,373 @@ +/* GNU Mailutils -- a suite of utilities for electronic mail + Copyright (C) 2007 Free Software Foundation, Inc. + + GNU Mailutils is free software; you can redistribute it and/or modify + it under the terms of the GNU General Public License as published by + the Free Software Foundation; either version 2, or (at your option) + any later version. + + GNU Mailutils is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with GNU Mailutils; if not, write to the Free Software + Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, + MA 02110-1301 USA */ + +#include <stdio.h> +#include <string.h> +#include <errno.h> +#include <sys/stat.h> +#include <mailutils/mailutils.h> + +char *file; +mu_header_t header; + +char *ps[] = { "> ", ". " }; +int interactive; + +static void +prompt(int l) +{ + if (interactive) + { + printf ("%s", ps[l]); + fflush (stdout); + } +} + +static int +load_file (const char *name) +{ + struct stat st; + size_t nread; + char *buf; + FILE *fp; + int status; + + if (stat (name, &st)) + { + mu_error ("cannot stat %s: %s", name, mu_strerror (errno)); + return 1; + } + + buf = malloc (st.st_size + 2); + if (!buf) + { + mu_error ("not enough memory"); + return 1; + } + + fp = fopen (name, "r"); + if (!fp) + { + mu_error ("cannot open file %s: %s", name, mu_strerror (errno)); + free (buf); + return 1; + } + + nread = fread (buf, 1, st.st_size, fp); + fclose (fp); + if (nread != st.st_size) + { + mu_error ("short read on file %s", name); + free (buf); + return 1; + } + + buf[st.st_size] = '\n'; + buf[st.st_size+1] = 0; + status = mu_header_create (&header, buf, st.st_size + 1, NULL); + free (buf); + if (status) + { + mu_error ("cannot create header: %s", mu_strerror (status)); + return 1; + } + return 0; +} + +unsigned line_num = 0; + +static int +check_args (char const *cmdname, int argc, int amin, int amax) +{ + if (argc < amin) + { + mu_error ("%u: %s: too few arguments", + line_num, cmdname); + return 1; + } + if (amax > 0 && argc > amax) + { + mu_error ("%u: %s: too many arguments", + line_num, cmdname); + return 1; + } + return 0; +} + +void +cmd_quit (int argc, char **argv) +{ + exit (0); +} + +void +cmd_load (int argc, char **argv) +{ + if (check_args (argv[0], argc, 2, 2)) + return; + mu_header_destroy (&header, NULL); + load_file (argv[1]); +} + +void +cmd_free (int argc, char **argv) +{ + if (check_args (argv[0], argc, 1, 1)) + return; + mu_header_destroy (&header, NULL); +} + +void +cmd_print (int argc, char **argv) +{ + char *fn; + int num = 1; + int status; + const char *str; + + if (check_args (argv[0], argc, 2, 3)) + return; + fn = argv[1]; + if (argc == 3) + num = atoi (argv[2]); + + status = mu_header_sget_value_n (header, fn, num, &str); + if (status == 0) + printf ("%s: %s\n", fn, str); + else + mu_error ("%u: %s", line_num, mu_strerror (status)); +} + +void +cmd_dump (int argc, char **argv) +{ + mu_off_t off = 0; + size_t n; + mu_stream_t stream; + char buf[512]; + int status; + + if (check_args (argv[0], argc, 1, 2)) + return; + + if (argc == 2) + off = strtoul (argv[1], NULL, 0); + + status = mu_header_get_stream (header, &stream); + if (status) + { + mu_error ("%u: cannot get stream: %s", line_num, mu_strerror (status)); + return; + } + + status = mu_stream_seek (stream, off, SEEK_SET); + if (status) + { + mu_error ("%u: cannot seek: %s", line_num, mu_strerror (status)); + return; + } + + while (mu_stream_sequential_read (stream, buf, sizeof buf, &n) == 0 + && n > 0) + { + fwrite (buf, 1, n, stdout); + } +} + +void +cmd_remove (int argc, char **argv) +{ + char *fn; + int num = 1; + int status; + + if (check_args (argv[0], argc, 2, 3)) + return; + fn = argv[1]; + if (argc == 3) + num = atoi (argv[2]); + status = mu_header_remove (header, fn, num); + if (status) + mu_error ("%u: %s: %s", line_num, argv[0], mu_strerror (status)); +} + +/* insert header value [ref [num] [before|after] [replace]] */ +void +cmd_insert (int argc, char **argv) +{ + int status; + int flags = 0; + char *ref = NULL; + int num = 1; + int n; + + if (check_args (argv[0], argc, 3, 7)) + return; + + if (argc >= 4) + { + ref = argv[3]; + n = 4; + if (n < argc) + { + char *p; + int tmp; + + tmp = strtoul(argv[4], &p, 0); + if (*p == 0) + { + num = tmp; + n++; + } + + for (; n < argc; n++) + { + if (strcmp(argv[n], "before") == 0) + flags |= MU_HEADER_BEFORE; + else if (strcmp(argv[n], "after") == 0) + ; + else if (strcmp(argv[n], "replace") == 0) + flags |= MU_HEADER_REPLACE; + else + { + mu_error("%u: %s: unknown option", line_num, argv[4]); + return; + } + } + } + } + status = mu_header_insert (header, argv[1], argv[2], + ref, num, flags); + if (status) + mu_error ("%u: %s: %s", line_num, argv[0], mu_strerror (status)); +} + +void +cmd_write (int argc, char **argv) +{ + char buf[512]; + mu_stream_t str; + int status; + + if (check_args (argv[0], argc, 1, 1)) + return; + + status = mu_header_get_stream (header, &str); + if (status) + { + mu_error ("%u: cannot get stream: %s", line_num, mu_strerror (status)); + return; + } + mu_stream_seek (str, 0, SEEK_SET); + while (prompt (1), fgets(buf, sizeof buf, stdin)) + { + mu_stream_sequential_write (str, buf, strlen (buf)); + if (buf[0] == '\n') + break; + } +} + +struct cmdtab +{ + char *name; + void (*fun) (int argc, char **argv); +}; + +static struct cmdtab cmdtab[] = { + { "quit", cmd_quit }, + { "load", cmd_load }, + { "free", cmd_free }, + { "print", cmd_print }, + { "dump", cmd_dump }, + { "remove", cmd_remove }, + { "insert", cmd_insert }, + { "write", cmd_write }, + { NULL } +}; + +static struct cmdtab * +find_cmd (const char *name) +{ + struct cmdtab *p; + for (p = cmdtab; p->name; p++) + if (strcmp (p->name, name) == 0) + return p; + return NULL; +} + +int +main (int argc, char **argv) +{ + int c; + char buf[512]; + + interactive = isatty (0); + while ((c = getopt (argc, argv, "f:h")) != EOF) + { + switch (c) + { + case 'f': + file = optarg; + break; + + case 'h': + printf ("usage: header [-f file]\n"); + exit (0); + + default: + exit (1); + } + } + + if (file) + { + if (load_file (file)) + exit (1); + } + + while (prompt(0), fgets(buf, sizeof buf, stdin)) + { + int c; + char **v; + struct cmdtab *cmd; + int status; + + line_num++; + status = mu_argcv_get (buf, NULL, "#", &c, &v); + if (status) + { + mu_error ("%u: cannot parse: %s", + line_num, mu_strerror (status)); + continue; + } + + cmd = find_cmd (v[0]); + if (!cmd) + { + mu_error ("%u: unknown command %s", + line_num, v[0]); + } + else + cmd->fun (c, v); + + mu_argcv_free (c, v); + + } + exit (0); +} + |