4 * by JH <jheinonen@users.sourceforge.net>
6 * Copyright (C) Jaakko Heinonen
19 #if defined(HAVE_LOCALE_H) && defined(HAVE_SETLOCALE)
34 static void init_abook();
35 static void quit_abook_sig(int i);
36 static void set_filenames();
37 static void parse_command_line(int argc, char **argv);
38 static void show_usage();
39 static void mutt_query(char *str);
40 static void init_mutt_query();
41 static void convert(char *srcformat, char *srcfile,
42 char *dstformat, char *dstfile);
43 static void add_email(int);
45 char *datafile = NULL;
48 bool alternative_datafile = FALSE;
49 bool alternative_rcfile = FALSE;
56 assert(datafile != NULL);
58 if( (f = fopen(datafile, "a")) == NULL)
67 check_abook_directory()
72 assert(!is_ui_initialized());
74 if(alternative_datafile)
77 dir = strconcat(getenv("HOME"), "/" DIR_IN_HOME, NULL);
80 if(stat(dir, &s) == -1) {
86 if(mkdir(dir, 0700) == -1) {
87 printf("Cannot create directory %s\n", dir);
92 } else if(!S_ISDIR(s.st_mode)) {
93 printf("%s is not a directory\n", dir);
105 check_abook_directory();
107 if(load_opts(rcfile) > 0) {
108 printf("Press enter to continue...\n");
112 signal(SIGKILL, quit_abook_sig);
113 signal(SIGTERM, quit_abook_sig);
118 umask(DEFAULT_UMASK);
120 if(!datafile_writeable()) {
121 char *s = mkstr("File %s is not writeable", datafile);
125 if(load_database(datafile) || !statusline_ask_boolean(
126 "If you continue all changes will "
127 "be lost. Do you want to continue?", FALSE)) {
129 /*close_database();*/
134 load_database(datafile);
140 quit_abook(int save_db)
143 if(opt_get_bool(BOOL_AUTOSAVE))
145 else if(statusline_ask_boolean("Save database", TRUE))
147 } else if(!statusline_ask_boolean("Quit without saving", FALSE))
159 quit_abook_sig(int i)
161 quit_abook(QUIT_SAVE);
165 main(int argc, char **argv)
167 #if defined(HAVE_SETLOCALE) && defined(HAVE_LOCALE_H)
168 setlocale(LC_ALL, "");
171 parse_command_line(argc, argv);
177 quit_abook(QUIT_SAVE);
195 if( (stat(getenv("HOME"), &s)) == -1 || ! S_ISDIR(s.st_mode) ) {
196 fprintf(stderr,"%s is not a valid HOME directory\n", getenv("HOME") );
201 datafile = strconcat(getenv("HOME"), "/" DIR_IN_HOME "/"
205 rcfile = strconcat(getenv("HOME"), "/" DIR_IN_HOME "/"
208 atexit(free_filenames);
218 MODE_ADD_EMAIL_QUIET,
224 change_mode(int *current, int mode)
226 if(*current != MODE_CONT) {
227 fprintf(stderr, "Cannot combine options --mutt-query, "
229 "--add-email or --add-email-quiet\n");
237 set_filename(char **var, char *path)
242 assert(*var == NULL); /* or else we probably leak memory */
243 assert(path != NULL);
252 *var = strconcat(cwd, "/", path, NULL);
257 #define set_convert_var(X) do { if(mode != MODE_CONVERT) {\
258 fprintf(stderr, "please use option --%s after --convert option\n",\
259 long_options[option_index].name);\
266 parse_command_line(int argc, char **argv)
268 int mode = MODE_CONT;
269 char *query_string = NULL;
270 char *informat = "abook",
277 int option_index = 0;
289 static struct option long_options[] = {
290 { "help", 0, 0, 'h' },
291 { "add-email", 0, 0, OPT_ADD_EMAIL },
292 { "add-email-quiet", 0, 0, OPT_ADD_EMAIL_QUIET },
293 { "datafile", 1, 0, 'f' },
294 { "mutt-query", 1, 0, OPT_MUTT_QUERY },
295 { "config", 1, 0, 'C' },
296 { "convert", 0, 0, OPT_CONVERT },
297 { "informat", 1, 0, OPT_INFORMAT },
298 { "outformat", 1, 0, OPT_OUTFORMAT },
299 { "infile", 1, 0, OPT_INFILE },
300 { "outfile", 1, 0, OPT_OUTFILE },
301 { "formats", 0, 0, OPT_FORMATS },
305 c = getopt_long(argc, argv, "hC:",
306 long_options, &option_index);
316 change_mode(&mode, MODE_ADD_EMAIL);
318 case OPT_ADD_EMAIL_QUIET:
319 change_mode(&mode, MODE_ADD_EMAIL_QUIET);
322 set_filename(&datafile, optarg);
323 alternative_datafile = TRUE;
326 query_string = optarg;
327 change_mode(&mode, MODE_QUERY);
330 set_filename(&rcfile, optarg);
331 alternative_rcfile = TRUE;
334 change_mode(&mode, MODE_CONVERT);
337 set_convert_var(informat);
340 set_convert_var(outformat);
343 set_convert_var(infile);
346 set_convert_var(outfile);
357 fprintf(stderr, "%s: unrecognized arguments on command line\n",
365 case MODE_ADD_EMAIL_QUIET:
368 mutt_query(query_string);
370 convert(informat, infile, outformat, outfile);
378 puts (PACKAGE " v " VERSION "\n");
379 puts (" -h --help show usage");
380 puts (" -C --config <file> use an alternative configuration file");
381 puts (" --datafile <file> use an alternative addressbook file");
382 puts (" --mutt-query <string> make a query for mutt");
383 puts (" --add-email "
384 "read an e-mail message from stdin and\n"
386 "add the sender to the addressbook");
387 puts (" --add-email-quiet "
388 "same as --add-email but doesn't\n"
391 puts (" --convert convert address book files");
392 puts (" options to use with --convert:");
393 puts (" --informat <format> format for input file");
394 puts (" (default: abook)");
395 puts (" --infile <file> source file");
396 puts (" (default: stdin)");
397 puts (" --outformat <format> format for output file");
398 puts (" (default: text)");
399 puts (" --outfile <file> destination file");
400 puts (" (default: stdout)");
401 puts (" --formats list available formats");
408 extern list_item *database;
411 quit_mutt_query(int status)
420 muttq_print_item(FILE *file, int item)
422 char emails[MAX_EMAILS][MAX_EMAIL_LEN];
425 split_emailstr(item, emails);
427 for(i = 0; i < (opt_get_bool(BOOL_MUTT_RETURN_ALL_EMAILS) ?
428 MAX_EMAILS : 1) ; i++)
430 fprintf(file, "%s\t%s\t%s\n", emails[i],
431 database[item][NAME],
432 database[item][NOTES] == NULL ? " " :
433 database[item][NOTES]
438 mutt_query(char *str)
442 if( str == NULL || !strcasecmp(str, "all") ) {
443 struct db_enumerator e = init_db_enumerator(ENUM_ALL);
444 printf("All items\n");
445 db_enumerate_items(e)
446 muttq_print_item(stdout, e.item);
448 int search_fields[] = {NAME, EMAIL, NICK, -1};
450 if( (i = find_item(str, 0, search_fields)) < 0 ) {
451 printf("Not found\n");
456 muttq_print_item(stdout, i);
457 i = find_item(str, i + 1, search_fields);
471 if( load_database(datafile) ) {
472 printf("Cannot open database\n");
480 make_mailstr(int item)
482 char email[MAX_EMAIL_LEN];
484 char *name = mkstr("\"%s\"", database[item][NAME]);
486 get_first_email(email, item);
488 ret = *database[item][EMAIL] ?
489 mkstr("%s <%s>", name, email) :
498 print_stderr(int item)
500 fprintf (stderr, "%c", '\n');
502 if( is_valid_item(item) )
503 muttq_print_item(stderr, item);
505 struct db_enumerator e = init_db_enumerator(ENUM_SELECTED);
506 db_enumerate_items(e) {
507 muttq_print_item(stderr, e.item);
514 launch_mutt(int item)
516 char *cmd = NULL, *mailstr = NULL;
517 char *mutt_command = opt_get_str(STR_MUTT_COMMAND);
519 if(mutt_command == NULL || !*mutt_command)
522 if( is_valid_item(item) )
523 mailstr = make_mailstr(item);
525 struct db_enumerator e = init_db_enumerator(ENUM_SELECTED);
527 db_enumerate_items(e) {
530 strconcat(tmp, ",", make_mailstr(e.item), NULL):
531 strconcat(make_mailstr(e.item), NULL);
536 cmd = strconcat(mutt_command, " \'", mailstr, "\'", NULL);
539 fprintf(stderr, "cmd: %s\n", cmd);
545 * we need to make sure that curses settings are correct
551 launch_wwwbrowser(int item)
555 if( !is_valid_item(item) )
558 if( database[item][URL] )
559 cmd = mkstr("%s '%s'",
560 opt_get_str(STR_WWW_COMMAND),
561 safe_str(database[item][URL]));
571 * we need to make sure that curses settings are correct
577 abook_malloc(size_t size)
581 if ( (ptr = malloc(size)) == NULL ) {
582 if( is_ui_initialized() )
583 quit_abook(QUIT_SAVE);
584 perror("malloc() failed");
592 abook_realloc(void *ptr, size_t size)
594 ptr = realloc(ptr, size);
600 if(is_ui_initialized())
601 quit_abook(QUIT_SAVE);
602 perror("realloc() failed");
610 abook_fopen (const char *path, const char *mode)
615 stat_ok = (stat(path, &s) != -1);
617 if(strchr(mode, 'r'))
618 return (stat_ok && S_ISREG(s.st_mode)) ?
619 fopen(path, mode) : NULL;
621 return (stat_ok && S_ISDIR(s.st_mode)) ?
622 NULL : fopen(path, mode);
626 convert(char *srcformat, char *srcfile, char *dstformat, char *dstfile)
630 if( !srcformat || !srcfile || !dstformat || !dstfile ) {
631 fprintf(stderr, "too few argumets to make conversion\n");
632 fprintf(stderr, "try --help\n");
636 if( !strcasecmp(srcformat, dstformat) ) {
637 printf( "input and output formats are the same\n"
647 switch(import_file(srcformat, srcfile)) {
650 "input format %s not supported\n", srcformat);
654 fprintf(stderr, "cannot read file %s\n", srcfile);
660 switch(export_file(dstformat, dstfile)) {
663 "output format %s not supported\n",
669 "cannot write file %s\n", dstfile);
680 * --add-email handling
683 static int add_email_count = 0;
688 if(add_email_count > 0) {
689 if(save_database() < 0) {
690 fprintf(stderr, "cannot open %s\n", datafile);
693 printf("%d item(s) added to %s\n", add_email_count, datafile);
695 puts("Valid sender address not found");
702 quit_add_email_sig(int signal)
711 atexit(free_filenames);
712 check_abook_directory();
718 * we don't actually care if loading fails or not
720 load_database(datafile);
722 atexit(close_database);
724 signal(SIGINT, quit_add_email_sig);
728 add_email_add_item(int quiet, char *name, char *email)
732 if(opt_get_bool(BOOL_ADD_EMAIL_PREVENT_DUPLICATES)) {
733 int search_fields[] = { EMAIL, -1 };
734 if(find_item(email, 0, search_fields) >= 0) {
736 printf("Address %s already in addressbook\n",
743 FILE *in = fopen("/dev/tty", "r");
746 fprintf(stderr, "cannot open /dev/tty\n"
747 "you may want to use --add-email-quiet\n");
750 printf("Add ``%s <%s>'' to %s ? (y/n)\n",
757 if(c == 'n' || c == 'N') {
761 } while(c != 'y' && c != 'Y');
765 memset(item, 0, sizeof(item));
766 item[NAME] = strdup(name);
767 item[EMAIL] = strdup(email);
768 add_item2database(item);
777 char *name = NULL, *email = NULL;
780 if( (fstat(fileno(stdin), &s)) == -1 || S_ISDIR(s.st_mode) ) {
781 fprintf(stderr, "stdin is a directory or cannot stat stdin\n");
788 line = getaline(stdin);
789 if(line && !strncasecmp("From:", line, 5) ) {
790 getname(line, &name, &email);
791 add_email_count += add_email_add_item(quiet,
797 } while( !feof(stdin) );
803 * end of --add-email handling