mirror of https://github.com/djcb/mu.git
443 lines
9.1 KiB
C
443 lines
9.1 KiB
C
/*
|
|
** Copyright (C) 2011-2016 Dirk-Jan C. Binnema <djcb@djcbsoftware.nl>
|
|
**
|
|
** This program 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 3, or (at your option) any
|
|
** later version.
|
|
**
|
|
** This program 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 this program; if not, write to the Free Software Foundation,
|
|
** Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
**
|
|
*/
|
|
|
|
#if HAVE_CONFIG_H
|
|
#include "config.h"
|
|
#endif /*HAVE_CONFIG_H*/
|
|
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <ctype.h>
|
|
|
|
#include "mu-cmd.h"
|
|
#include "mu-util.h"
|
|
#include "mu-str.h"
|
|
#include "mu-date.h"
|
|
#include "mu-contacts.h"
|
|
#include "mu-runtime.h"
|
|
|
|
/**
|
|
* guess the last name for the given name; clearly,
|
|
* this is just a rough guess for setting an initial value.
|
|
*
|
|
* @param name a name
|
|
*
|
|
* @return the last name, as a newly allocated string (free with
|
|
* g_free)
|
|
*/
|
|
static gchar*
|
|
guess_last_name (const char *name)
|
|
{
|
|
const gchar *lastsp;
|
|
|
|
if (!name)
|
|
return g_strdup ("");
|
|
|
|
lastsp = g_strrstr (name, " ");
|
|
|
|
return g_strdup (lastsp ? lastsp + 1 : "");
|
|
}
|
|
|
|
/**
|
|
* guess the first name for the given name; clearly,
|
|
* this is just a rough guess for setting an initial value.
|
|
*
|
|
* @param name a name
|
|
*
|
|
* @return the first name, as a newly allocated string (free with
|
|
* g_free)
|
|
*/
|
|
static gchar*
|
|
guess_first_name (const char *name)
|
|
{
|
|
const gchar *lastsp;
|
|
|
|
if (!name)
|
|
return g_strdup ("");
|
|
|
|
lastsp = g_strrstr (name, " ");
|
|
|
|
if (lastsp)
|
|
return g_strndup (name, lastsp - name);
|
|
else
|
|
return g_strdup (name);
|
|
}
|
|
|
|
/**
|
|
* guess some nick name for the given name; if we can determine an
|
|
* first name, last name, the nick will be first name + the first char
|
|
* of the last name. otherwise, it's just the first name. clearly,
|
|
* this is just a rough guess for setting an initial value for nicks.
|
|
*
|
|
* @param name a name
|
|
*
|
|
* @return the guessed nick, as a newly allocated string (free with g_free)
|
|
*/
|
|
static gchar*
|
|
cleanup_str (const char* str)
|
|
{
|
|
gchar *s;
|
|
const gchar *cur;
|
|
unsigned i;
|
|
|
|
if (mu_str_is_empty(str))
|
|
return g_strdup ("");
|
|
|
|
s = g_new0 (char, strlen(str) + 1);
|
|
|
|
for (cur = str, i = 0; *cur; ++cur) {
|
|
if (ispunct(*cur) || isspace(*cur))
|
|
continue;
|
|
else
|
|
s[i++] = *cur;
|
|
}
|
|
|
|
return s;
|
|
}
|
|
|
|
static char*
|
|
uniquify_nick (const char *nick, GHashTable *nicks)
|
|
{
|
|
guint u;
|
|
|
|
for (u = 2; u != 1000; ++u) {
|
|
char *cand;
|
|
cand = g_strdup_printf ("%s%u", nick, u);
|
|
if (!g_hash_table_contains (nicks, cand))
|
|
return cand;
|
|
}
|
|
|
|
return g_strdup (nick); /* if all else fails */
|
|
}
|
|
|
|
|
|
static gchar*
|
|
guess_nick (const char* name, GHashTable *nicks)
|
|
{
|
|
gchar *fname, *lname, *nick;
|
|
gchar initial[7];
|
|
|
|
fname = guess_first_name (name);
|
|
lname = guess_last_name (name);
|
|
|
|
/* if there's no last name, use first name as the nick */
|
|
if (mu_str_is_empty(fname) || mu_str_is_empty(lname)) {
|
|
g_free (lname);
|
|
nick = fname;
|
|
goto leave;
|
|
}
|
|
|
|
memset (initial, 0, sizeof(initial));
|
|
/* couldn't we get an initial for the last name? */
|
|
if (g_unichar_to_utf8 (g_utf8_get_char (lname), initial) == 0) {
|
|
g_free (lname);
|
|
nick = fname;
|
|
goto leave;
|
|
}
|
|
|
|
nick = g_strdup_printf ("%s%s", fname, initial);
|
|
g_free (fname);
|
|
g_free (lname);
|
|
|
|
leave:
|
|
{
|
|
gchar *tmp;
|
|
tmp = cleanup_str (nick);
|
|
g_free (nick);
|
|
nick = tmp;
|
|
}
|
|
|
|
if (g_hash_table_contains (nicks, nick)) {
|
|
char *tmp;
|
|
tmp = uniquify_nick (nick, nicks);
|
|
g_free (nick);
|
|
nick = tmp;
|
|
}
|
|
|
|
g_hash_table_add (nicks, g_strdup(nick));
|
|
|
|
return nick;
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
print_header (MuConfigFormat format)
|
|
{
|
|
switch (format) {
|
|
case MU_CONFIG_FORMAT_BBDB:
|
|
g_print (";; -*-coding: utf-8-emacs;-*-\n"
|
|
";;; file-version: 6\n");
|
|
break;
|
|
case MU_CONFIG_FORMAT_MUTT_AB:
|
|
g_print ("Matching addresses in the mu database:\n");
|
|
break;
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
static void
|
|
each_contact_bbdb (const char *email, const char *name, time_t tstamp)
|
|
{
|
|
char *fname, *lname, *now, *timestamp;
|
|
|
|
fname = guess_first_name (name);
|
|
lname = guess_last_name (name);
|
|
now = mu_date_str ("%Y-%m-%d", time(NULL));
|
|
timestamp = mu_date_str ("%Y-%m-%d", tstamp);
|
|
|
|
g_print ("[\"%s\" \"%s\" nil nil nil nil (\"%s\") "
|
|
"((creation-date . \"%s\") (time-stamp . \"%s\")) nil]\n",
|
|
fname, lname, email, now, timestamp);
|
|
|
|
g_free (now);
|
|
g_free (timestamp);
|
|
g_free (fname);
|
|
g_free (lname);
|
|
}
|
|
|
|
|
|
static void
|
|
each_contact_mutt_alias (const char *email, const char *name,
|
|
GHashTable *nicks)
|
|
{
|
|
gchar *nick;
|
|
|
|
if (!name)
|
|
return;
|
|
|
|
nick = guess_nick (name, nicks);
|
|
mu_util_print_encoded ("alias %s %s <%s>\n",
|
|
nick, name, email);
|
|
g_free (nick);
|
|
|
|
}
|
|
|
|
|
|
static void
|
|
each_contact_wl (const char *email, const char *name, GHashTable *nicks)
|
|
{
|
|
gchar *nick;
|
|
|
|
if (!name)
|
|
return;
|
|
|
|
nick = guess_nick (name, nicks);
|
|
mu_util_print_encoded ("%s \"%s\" \"%s\"\n",
|
|
email, nick, name);
|
|
g_free (nick);
|
|
}
|
|
|
|
|
|
static void
|
|
each_contact_org_contact (const char *email, const char *name)
|
|
{
|
|
if (name)
|
|
mu_util_print_encoded (
|
|
"* %s\n:PROPERTIES:\n:EMAIL: %s\n:END:\n\n",
|
|
name, email);
|
|
}
|
|
|
|
|
|
static void
|
|
print_csv_field (const char *str)
|
|
{
|
|
char *s;
|
|
|
|
if (!str)
|
|
return;
|
|
|
|
s = mu_str_replace (str, "\"", "\"\"");
|
|
if (strchr (s, ','))
|
|
mu_util_print_encoded ("\"%s\"", s);
|
|
else
|
|
mu_util_print_encoded ("%s", s);
|
|
|
|
g_free (s);
|
|
}
|
|
|
|
static void
|
|
each_contact_csv (const char *email, const char *name)
|
|
{
|
|
print_csv_field (name);
|
|
mu_util_print_encoded (",");
|
|
print_csv_field (email);
|
|
mu_util_print_encoded ("\n");
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
print_plain (const char *email, const char *name, gboolean color)
|
|
{
|
|
if (name) {
|
|
if (color) fputs (MU_COLOR_MAGENTA, stdout);
|
|
mu_util_fputs_encoded (name, stdout);
|
|
fputs (" ", stdout);
|
|
}
|
|
|
|
if (color)
|
|
fputs (MU_COLOR_GREEN, stdout);
|
|
|
|
mu_util_fputs_encoded (email, stdout);
|
|
|
|
if (color)
|
|
fputs (MU_COLOR_DEFAULT, stdout);
|
|
|
|
fputs ("\n", stdout);
|
|
}
|
|
|
|
typedef struct {
|
|
MuConfigFormat format;
|
|
gboolean color, personal;
|
|
time_t after;
|
|
GHashTable *nicks;
|
|
} ECData;
|
|
|
|
|
|
static void
|
|
each_contact (const char *email, const char *name, gboolean personal,
|
|
time_t tstamp, unsigned freq, ECData *ecdata)
|
|
{
|
|
if (ecdata->personal && !personal)
|
|
return;
|
|
|
|
if (tstamp < ecdata->after)
|
|
return;
|
|
|
|
switch (ecdata->format) {
|
|
case MU_CONFIG_FORMAT_MUTT_ALIAS:
|
|
each_contact_mutt_alias (email, name, ecdata->nicks);
|
|
break;
|
|
case MU_CONFIG_FORMAT_MUTT_AB:
|
|
mu_util_print_encoded ("%s\t%s\t\n",
|
|
email, name ? name : "");
|
|
break;
|
|
case MU_CONFIG_FORMAT_WL:
|
|
each_contact_wl (email, name, ecdata->nicks);
|
|
break;
|
|
case MU_CONFIG_FORMAT_ORG_CONTACT:
|
|
each_contact_org_contact (email, name);
|
|
break;
|
|
case MU_CONFIG_FORMAT_BBDB:
|
|
each_contact_bbdb (email, name, tstamp);
|
|
break;
|
|
case MU_CONFIG_FORMAT_CSV:
|
|
each_contact_csv (email, name);
|
|
break;
|
|
default:
|
|
print_plain (email, name, ecdata->color);
|
|
}
|
|
}
|
|
|
|
|
|
static MuError
|
|
run_cmd_cfind (const char* pattern,
|
|
gboolean personal, time_t after,
|
|
MuConfigFormat format,
|
|
gboolean color, GError **err)
|
|
{
|
|
gboolean rv;
|
|
MuContacts *contacts;
|
|
size_t num;
|
|
ECData ecdata;
|
|
|
|
ecdata.personal = personal;
|
|
ecdata.after = after;
|
|
ecdata.format = format;
|
|
ecdata.color = color;
|
|
ecdata.nicks = g_hash_table_new_full (g_str_hash, g_str_equal,
|
|
g_free, NULL);
|
|
|
|
contacts = mu_contacts_new
|
|
(mu_runtime_path(MU_RUNTIME_PATH_CONTACTS));
|
|
if (!contacts) {
|
|
g_set_error (err, MU_ERROR_DOMAIN,
|
|
MU_ERROR_CONTACTS_CANNOT_RETRIEVE,
|
|
"could not retrieve contacts");
|
|
return MU_ERROR_CONTACTS_CANNOT_RETRIEVE;
|
|
}
|
|
|
|
print_header (format);
|
|
rv = mu_contacts_foreach (contacts,
|
|
(MuContactsForeachFunc)each_contact,
|
|
&ecdata, pattern, &num);
|
|
g_hash_table_unref (ecdata.nicks);
|
|
mu_contacts_destroy (contacts);
|
|
|
|
|
|
if (num == 0) {
|
|
g_warning ("no matching contacts found");
|
|
return MU_ERROR_NO_MATCHES;
|
|
}
|
|
|
|
return rv ? MU_OK : MU_ERROR_CONTACTS;
|
|
}
|
|
|
|
static gboolean
|
|
cfind_params_valid (MuConfig *opts)
|
|
{
|
|
switch (opts->format) {
|
|
case MU_CONFIG_FORMAT_PLAIN:
|
|
case MU_CONFIG_FORMAT_MUTT_ALIAS:
|
|
case MU_CONFIG_FORMAT_MUTT_AB:
|
|
case MU_CONFIG_FORMAT_WL:
|
|
case MU_CONFIG_FORMAT_BBDB:
|
|
case MU_CONFIG_FORMAT_CSV:
|
|
case MU_CONFIG_FORMAT_ORG_CONTACT:
|
|
break;
|
|
default:
|
|
g_warning ("invalid output format %s",
|
|
opts->formatstr ? opts->formatstr : "<none>");
|
|
return FALSE;
|
|
}
|
|
|
|
/* only one pattern allowed */
|
|
if (opts->params[1] && opts->params[2]) {
|
|
g_warning ("usage: mu cfind [options] [<ptrn>]");
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
|
|
MuError
|
|
mu_cmd_cfind (MuConfig *opts, GError **err)
|
|
{
|
|
g_return_val_if_fail (opts, MU_ERROR_INTERNAL);
|
|
g_return_val_if_fail (opts->cmd == MU_CONFIG_CMD_CFIND,
|
|
MU_ERROR_INTERNAL);
|
|
|
|
if (!cfind_params_valid (opts)) {
|
|
g_set_error (err, MU_ERROR_DOMAIN, MU_ERROR_IN_PARAMETERS,
|
|
"invalid parameters");
|
|
return MU_ERROR_IN_PARAMETERS;
|
|
}
|
|
|
|
return run_cmd_cfind (opts->params[1],
|
|
opts->personal,
|
|
opts->after,
|
|
opts->format,
|
|
!opts->nocolor,
|
|
err);
|
|
}
|