static int remove_filename(struct nm_ctxdata *data, const char *path)
{
notmuch_status_t st;
notmuch_filenames_t *ls;
notmuch_message_t *msg = NULL;
notmuch_database_t *db = get_db(data, TRUE);
int trans;
dprint(2, (debugfile, "nm: remove filename '%s'\n", path));
if (!db)
return -1;
st = notmuch_database_find_message_by_filename(db, path, &msg);
if (st || !msg)
return -1;
trans = db_trans_begin(data);
if (trans < 0)
return -1;
/*
* note that unlink() is probably unnecessary here, it's already removed
* by mh_sync_mailbox_message(), but for sure...
*/
st = notmuch_database_remove_message(db, path);
switch (st) {
case NOTMUCH_STATUS_SUCCESS:
dprint(2, (debugfile, "nm: remove success, call unlink\n"));
unlink(path);
break;
case NOTMUCH_STATUS_DUPLICATE_MESSAGE_ID:
dprint(2, (debugfile, "nm: remove succes (duplicate), call unlink\n"));
unlink(path);
for (ls = notmuch_message_get_filenames(msg);
ls && notmuch_filenames_valid(ls);
notmuch_filenames_move_to_next(ls)) {
path = notmuch_filenames_get(ls);
dprint(2, (debugfile, "nm: remove duplicate: '%s'\n", path));
unlink(path);
notmuch_database_remove_message(db, path);
}
break;
default:
dprint(1, (debugfile, "nm: failed to remove '%s' [st=%d]\n", path, (int) st));
break;
}
notmuch_message_destroy(msg);
if (trans)
db_trans_end(data);
return 0;
}
/*
* call-seq: FILENAMES.each {|item| block } => FILENAMES
*
* Calls +block+ once for each element in +self+, passing that element as a
* parameter.
*/
VALUE
notmuch_rb_filenames_each(VALUE self)
{
notmuch_filenames_t *fnames;
Data_Get_Notmuch_FileNames(self, fnames);
for (; notmuch_filenames_valid(fnames); notmuch_filenames_move_to_next(fnames))
rb_yield(rb_str_new2(notmuch_filenames_get(fnames)));
return self;
}
/* Recursively remove all filenames from the database referring to
* 'path' (or to any of its children). */
static void
_remove_directory (void *ctx,
notmuch_database_t *notmuch,
const char *path,
int *renamed_files,
int *removed_files)
{
notmuch_directory_t *directory;
notmuch_filenames_t *files, *subdirs;
notmuch_status_t status;
char *absolute;
directory = notmuch_database_get_directory (notmuch, path);
for (files = notmuch_directory_get_child_files (directory);
notmuch_filenames_valid (files);
notmuch_filenames_move_to_next (files))
{
absolute = talloc_asprintf (ctx, "%s/%s", path,
notmuch_filenames_get (files));
status = notmuch_database_remove_message (notmuch, absolute);
if (status == NOTMUCH_STATUS_DUPLICATE_MESSAGE_ID)
*renamed_files = *renamed_files + 1;
else
*removed_files = *removed_files + 1;
talloc_free (absolute);
}
for (subdirs = notmuch_directory_get_child_directories (directory);
notmuch_filenames_valid (subdirs);
notmuch_filenames_move_to_next (subdirs))
{
absolute = talloc_asprintf (ctx, "%s/%s", path,
notmuch_filenames_get (subdirs));
_remove_directory (ctx, notmuch, absolute, renamed_files, removed_files);
talloc_free (absolute);
}
notmuch_directory_destroy (directory);
}
static const char *get_message_last_filename(notmuch_message_t *msg)
{
notmuch_filenames_t *ls;
const char *name = NULL;
for (ls = notmuch_message_get_filenames(msg);
ls && notmuch_filenames_valid(ls);
notmuch_filenames_move_to_next(ls)) {
name = notmuch_filenames_get(ls);
}
return name;
}
static int
msg_filenames(Tcl_Interp *interp, notmuch_message_t *msg, int argc, const char *argv[]) {
if (argc != 0) {
tcl_result_printf(interp, "msg filenames takes no arguments");
return TCL_ERROR;
}
Tcl_Obj* list = Tcl_NewListObj(0, NULL);
if (!list) {
return TCL_ERROR;
}
notmuch_filenames_t *fns = notmuch_message_get_filenames(msg);
while (notmuch_filenames_valid(fns)) {
const char *fn = notmuch_filenames_get(fns);
Tcl_ListObjAppendElement(interp, list, Tcl_NewStringObj(fn, -1));
notmuch_filenames_move_to_next(fns);
}
notmuch_filenames_destroy(fns);
Tcl_SetObjResult(interp, list);
return TCL_OK;
}
int
cmd_move_message(ClientData data, Tcl_Interp *interp, int argc, const char *argv[]) {
filter_context_t *ctx = FILTER_CONTEXT(data);
notmuch_message_t *msg = ctx->current_message;
notmuch_status_t nmrc;
if (argc != 2) {
tcl_result_printf(interp, "wrong # of args: got %d, expected move folder", argc);
return TCL_ERROR;
}
const char *folder = argv[1];
if (folder[0] == '/') {
tcl_result_printf(interp, "invalid folder '%s'", folder);
return TCL_ERROR;
}
log_debug("moving message %s to folder %s",
notmuch_message_get_message_id(msg), folder);
char *folder_path = g_strdup_printf("%s/%s",
notmuch_database_get_path(ctx->database),
folder);
int status = TCL_ERROR;
notmuch_filenames_t *fns = notmuch_message_get_filenames(msg);
int nmoved = 0;
while (notmuch_filenames_valid(fns)) {
const char *fn = notmuch_filenames_get(fns);
char *new_fn = NULL;
if (g_str_has_prefix(fn, folder_path)) {
log_debug("file %s already in folder %s", fn, folder);
} else if (ctx->dry_run) {
log_debug("moving %s to folder %s", fn, folder);
} else {
log_debug("moving %s to folder %s", fn, folder);
int rc = maildir_deliver_link(fn, folder_path, &new_fn);
if (rc) {
tcl_result_printf(interp, "delivery error: %s", strerror(errno));
goto done;
}
log_debug("delivered as %s", new_fn);
nmrc = notmuch_database_add_message(ctx->database, new_fn, NULL);
switch (nmrc) {
case NOTMUCH_STATUS_SUCCESS:
log_warning("%s: message was not in database", new_fn);
case NOTMUCH_STATUS_DUPLICATE_MESSAGE_ID:
log_debug("added new file to database");
break;
default:
tcl_result_printf(interp, "error adding %s to database: %s",
new_fn,
notmuch_status_to_string(nmrc));
goto done;
}
nmrc = notmuch_database_remove_message(ctx->database, fn);
switch (nmrc) {
case NOTMUCH_STATUS_SUCCESS:
log_warning("%s: file was only file for message", fn);
case NOTMUCH_STATUS_DUPLICATE_MESSAGE_ID:
log_debug("removed file %s from database", fn);
break;
default:
tcl_result_printf(interp, "error removing %s from database: %s",
fn, notmuch_status_to_string(nmrc));
goto done;
}
if (unlink(fn)) {
tcl_result_printf(interp, "error unlinking %s: %s", fn, strerror(errno));
goto done;
}
nmoved += 1;
}
notmuch_filenames_move_to_next(fns);
}
notmuch_filenames_destroy(fns);
fns = NULL;
if (nmoved == 0) {
Tcl_SetObjResult(interp, Tcl_NewIntObj(0));
status = TCL_OK;
goto done;
}
log_debug("syncing maildir flags");
const char *mid = notmuch_message_get_message_id(msg);
nmrc = notmuch_database_find_message(ctx->database, mid, &msg);
if (nmrc != NOTMUCH_STATUS_SUCCESS) {
tcl_result_printf(interp, "cannot re-find message %s", mid);
goto done;
}
notmuch_message_destroy(ctx->current_message);
ctx->current_message = msg;
nmrc = notmuch_message_tags_to_maildir_flags(msg);
if (nmrc != NOTMUCH_STATUS_SUCCESS) {
tcl_result_printf(interp, "error syncing tags back to flags");
goto done;
}
Tcl_SetObjResult(interp, Tcl_NewIntObj(nmoved));
status = TCL_OK;
done:
if (fns) {
notmuch_filenames_destroy(fns);
}
g_free(folder_path);
return status;
}
/* Examine 'path' recursively as follows:
*
* o Ask the filesystem for the mtime of 'path' (fs_mtime)
* o Ask the database for its timestamp of 'path' (db_mtime)
*
* o Ask the filesystem for files and directories within 'path'
* (via scandir and stored in fs_entries)
*
* o Pass 1: For each directory in fs_entries, recursively call into
* this same function.
*
* o Compare fs_mtime to db_mtime. If they are equivalent, terminate
* the algorithm at this point, (this directory has not been
* updated in the filesystem since the last database scan of PASS
* 2).
*
* o Ask the database for files and directories within 'path'
* (db_files and db_subdirs)
*
* o Pass 2: Walk fs_entries simultaneously with db_files and
* db_subdirs. Look for one of three interesting cases:
*
* 1. Regular file in fs_entries and not in db_files
* This is a new file to add_message into the database.
*
* 2. Filename in db_files not in fs_entries.
* This is a file that has been removed from the mail store.
*
* 3. Directory in db_subdirs not in fs_entries
* This is a directory that has been removed from the mail store.
*
* Note that the addition of a directory is not interesting here,
* since that will have been taken care of in pass 1. Also, we
* don't immediately act on file/directory removal since we must
* ensure that in the case of a rename that the new filename is
* added before the old filename is removed, (so that no
* information is lost from the database).
*
* o Tell the database to update its time of 'path' to 'fs_mtime'
* if fs_mtime isn't the current wall-clock time.
*/
static notmuch_status_t
add_files_recursive (notmuch_database_t *notmuch,
const char *path,
add_files_state_t *state)
{
DIR *dir = NULL;
struct dirent *entry = NULL;
char *next = NULL;
time_t fs_mtime, db_mtime;
notmuch_status_t status, ret = NOTMUCH_STATUS_SUCCESS;
notmuch_message_t *message = NULL;
struct dirent **fs_entries = NULL;
int i, num_fs_entries;
notmuch_directory_t *directory;
notmuch_filenames_t *db_files = NULL;
notmuch_filenames_t *db_subdirs = NULL;
time_t stat_time;
struct stat st;
notmuch_bool_t is_maildir, new_directory;
const char **tag;
if (stat (path, &st)) {
fprintf (stderr, "Error reading directory %s: %s\n",
path, strerror (errno));
return NOTMUCH_STATUS_FILE_ERROR;
}
stat_time = time (NULL);
/* This is not an error since we may have recursed based on a
* symlink to a regular file, not a directory, and we don't know
* that until this stat. */
if (! S_ISDIR (st.st_mode))
return NOTMUCH_STATUS_SUCCESS;
fs_mtime = st.st_mtime;
directory = notmuch_database_get_directory (notmuch, path);
db_mtime = notmuch_directory_get_mtime (directory);
new_directory = db_mtime ? FALSE : TRUE;
/* XXX This is a temporary workaround. If we don't update the
* database mtime until after processing messages in this
* directory, then a 0 mtime is *not* sufficient to indicate that
* this directory has no messages or subdirs in the database (for
* example, if an earlier run skipped the mtime update because
* fs_mtime == stat_time, or was interrupted before updating the
* mtime at the end). To address this, we record a (bogus)
* non-zero value before processing any child messages so that a
* later run won't mistake this for a new directory (and, for
* example, fail to detect removed files and subdirs).
*
* A better solution would be for notmuch_database_get_directory
* to indicate if it really created a new directory or not, either
* by a new out-argument, or by recording this information and
* providing an accessor.
*/
if (new_directory)
notmuch_directory_set_mtime (directory, -1);
/* If the database knows about this directory, then we sort based
* on strcmp to match the database sorting. Otherwise, we can do
* inode-based sorting for faster filesystem operation. */
num_fs_entries = scandir (path, &fs_entries, 0,
new_directory ?
dirent_sort_inode : dirent_sort_strcmp_name);
if (num_fs_entries == -1) {
fprintf (stderr, "Error opening directory %s: %s\n",
path, strerror (errno));
ret = NOTMUCH_STATUS_FILE_ERROR;
goto DONE;
}
/* Pass 1: Recurse into all sub-directories. */
is_maildir = _entries_resemble_maildir (fs_entries, num_fs_entries);
for (i = 0; i < num_fs_entries; i++) {
if (interrupted)
break;
entry = fs_entries[i];
/* We only want to descend into directories.
* But symlinks can be to directories too, of course.
*
* And if the filesystem doesn't tell us the file type in the
* scandir results, then it might be a directory (and if not,
* then we'll stat and return immediately in the next level of
* recursion). */
if (entry->d_type != DT_DIR &&
entry->d_type != DT_LNK &&
entry->d_type != DT_UNKNOWN)
{
continue;
}
/* Ignore special directories to avoid infinite recursion.
* Also ignore the .notmuch directory and any "tmp" directory
* that appears within a maildir.
*/
/* XXX: Eventually we'll want more sophistication to let the
* user specify files to be ignored. */
if (strcmp (entry->d_name, ".") == 0 ||
strcmp (entry->d_name, "..") == 0 ||
(is_maildir && strcmp (entry->d_name, "tmp") == 0) ||
strcmp (entry->d_name, ".notmuch") ==0)
{
continue;
}
next = talloc_asprintf (notmuch, "%s/%s", path, entry->d_name);
status = add_files_recursive (notmuch, next, state);
if (status && ret == NOTMUCH_STATUS_SUCCESS)
ret = status;
talloc_free (next);
next = NULL;
}
/* If the directory's modification time in the filesystem is the
* same as what we recorded in the database the last time we
* scanned it, then we can skip the second pass entirely.
*
* We test for strict equality here to avoid a bug that can happen
* if the system clock jumps backward, (preventing new mail from
* being discovered until the clock catches up and the directory
* is modified again).
*/
if (fs_mtime == db_mtime)
goto DONE;
/* new_directory means a directory that the database has never
* seen before. In that case, we can simply leave db_files and
* db_subdirs NULL. */
if (!new_directory) {
db_files = notmuch_directory_get_child_files (directory);
db_subdirs = notmuch_directory_get_child_directories (directory);
}
/* Pass 2: Scan for new files, removed files, and removed directories. */
for (i = 0; i < num_fs_entries; i++)
{
if (interrupted)
break;
entry = fs_entries[i];
/* Check if we've walked past any names in db_files or
* db_subdirs. If so, these have been deleted. */
while (notmuch_filenames_valid (db_files) &&
strcmp (notmuch_filenames_get (db_files), entry->d_name) < 0)
{
char *absolute = talloc_asprintf (state->removed_files,
"%s/%s", path,
notmuch_filenames_get (db_files));
_filename_list_add (state->removed_files, absolute);
notmuch_filenames_move_to_next (db_files);
}
while (notmuch_filenames_valid (db_subdirs) &&
strcmp (notmuch_filenames_get (db_subdirs), entry->d_name) <= 0)
{
const char *filename = notmuch_filenames_get (db_subdirs);
if (strcmp (filename, entry->d_name) < 0)
{
char *absolute = talloc_asprintf (state->removed_directories,
"%s/%s", path, filename);
_filename_list_add (state->removed_directories, absolute);
}
notmuch_filenames_move_to_next (db_subdirs);
}
/* If we're looking at a symlink, we only want to add it if it
* links to a regular file, (and not to a directory, say).
*
* Similarly, if the file is of unknown type (due to filesystem
* limitations), then we also need to look closer.
*
* In either case, a stat does the trick.
*/
if (entry->d_type == DT_LNK || entry->d_type == DT_UNKNOWN) {
int err;
next = talloc_asprintf (notmuch, "%s/%s", path, entry->d_name);
err = stat (next, &st);
talloc_free (next);
next = NULL;
/* Don't emit an error for a link pointing nowhere, since
* the directory-traversal pass will have already done
* that. */
if (err)
continue;
if (! S_ISREG (st.st_mode))
continue;
} else if (entry->d_type != DT_REG) {
continue;
}
/* Don't add a file that we've added before. */
if (notmuch_filenames_valid (db_files) &&
strcmp (notmuch_filenames_get (db_files), entry->d_name) == 0)
{
notmuch_filenames_move_to_next (db_files);
continue;
}
/* We're now looking at a regular file that doesn't yet exist
* in the database, so add it. */
next = talloc_asprintf (notmuch, "%s/%s", path, entry->d_name);
state->processed_files++;
if (state->verbose) {
if (state->output_is_a_tty)
printf("\r\033[K");
printf ("%i/%i: %s",
state->processed_files,
state->total_files,
next);
putchar((state->output_is_a_tty) ? '\r' : '\n');
fflush (stdout);
}
status = notmuch_database_add_message (notmuch, next, &message);
switch (status) {
/* success */
case NOTMUCH_STATUS_SUCCESS:
state->added_messages++;
notmuch_message_freeze (message);
for (tag=state->new_tags; *tag != NULL; tag++)
notmuch_message_add_tag (message, *tag);
if (state->synchronize_flags == TRUE)
notmuch_message_maildir_flags_to_tags (message);
notmuch_message_thaw (message);
break;
/* Non-fatal issues (go on to next file) */
case NOTMUCH_STATUS_DUPLICATE_MESSAGE_ID:
/* Defer sync of maildir flags until after old filenames
* are removed in the case of a rename. */
if (state->synchronize_flags == TRUE)
_filename_list_add (state->message_ids_to_sync,
notmuch_message_get_message_id (message));
break;
case NOTMUCH_STATUS_FILE_NOT_EMAIL:
fprintf (stderr, "Note: Ignoring non-mail file: %s\n",
next);
break;
/* Fatal issues. Don't process anymore. */
case NOTMUCH_STATUS_READ_ONLY_DATABASE:
case NOTMUCH_STATUS_XAPIAN_EXCEPTION:
case NOTMUCH_STATUS_OUT_OF_MEMORY:
fprintf (stderr, "Error: %s. Halting processing.\n",
notmuch_status_to_string (status));
ret = status;
goto DONE;
default:
case NOTMUCH_STATUS_FILE_ERROR:
case NOTMUCH_STATUS_NULL_POINTER:
case NOTMUCH_STATUS_TAG_TOO_LONG:
case NOTMUCH_STATUS_UNBALANCED_FREEZE_THAW:
case NOTMUCH_STATUS_LAST_STATUS:
INTERNAL_ERROR ("add_message returned unexpected value: %d", status);
goto DONE;
}
if (message) {
notmuch_message_destroy (message);
message = NULL;
}
if (do_print_progress) {
do_print_progress = 0;
generic_print_progress ("Processed", "files", state->tv_start,
state->processed_files, state->total_files);
}
talloc_free (next);
next = NULL;
}
if (interrupted)
goto DONE;
/* Now that we've walked the whole filesystem list, anything left
* over in the database lists has been deleted. */
while (notmuch_filenames_valid (db_files))
{
char *absolute = talloc_asprintf (state->removed_files,
"%s/%s", path,
notmuch_filenames_get (db_files));
_filename_list_add (state->removed_files, absolute);
notmuch_filenames_move_to_next (db_files);
}
while (notmuch_filenames_valid (db_subdirs))
{
char *absolute = talloc_asprintf (state->removed_directories,
"%s/%s", path,
notmuch_filenames_get (db_subdirs));
_filename_list_add (state->removed_directories, absolute);
notmuch_filenames_move_to_next (db_subdirs);
}
/* If the directory's mtime is the same as the wall-clock time
* when we stat'ed the directory, we skip updating the mtime in
* the database because a message could be delivered later in this
* same second. This may lead to unnecessary re-scans, but it
* avoids overlooking messages. */
if (! interrupted && fs_mtime != stat_time) {
status = notmuch_directory_set_mtime (directory, fs_mtime);
if (status && ret == NOTMUCH_STATUS_SUCCESS)
ret = status;
}
DONE:
if (next)
talloc_free (next);
if (entry)
free (entry);
if (dir)
closedir (dir);
if (fs_entries)
free (fs_entries);
if (db_subdirs)
notmuch_filenames_destroy (db_subdirs);
if (db_files)
notmuch_filenames_destroy (db_files);
if (directory)
notmuch_directory_destroy (directory);
return ret;
}
static int
do_search_messages (search_context_t *ctx)
{
notmuch_message_t *message;
notmuch_messages_t *messages;
notmuch_filenames_t *filenames;
sprinter_t *format = ctx->format;
int i;
notmuch_status_t status;
if (ctx->offset < 0) {
unsigned count;
notmuch_status_t status;
status = notmuch_query_count_messages_st (ctx->query, &count);
if (print_status_query ("notmuch search", ctx->query, status))
return 1;
ctx->offset += count;
if (ctx->offset < 0)
ctx->offset = 0;
}
status = notmuch_query_search_messages_st (ctx->query, &messages);
if (print_status_query ("notmuch search", ctx->query, status))
return 1;
format->begin_list (format);
for (i = 0;
notmuch_messages_valid (messages) && (ctx->limit < 0 || i < ctx->offset + ctx->limit);
notmuch_messages_move_to_next (messages), i++)
{
if (i < ctx->offset)
continue;
message = notmuch_messages_get (messages);
if (ctx->output == OUTPUT_FILES) {
int j;
filenames = notmuch_message_get_filenames (message);
for (j = 1;
notmuch_filenames_valid (filenames);
notmuch_filenames_move_to_next (filenames), j++)
{
if (ctx->dupe < 0 || ctx->dupe == j) {
format->string (format, notmuch_filenames_get (filenames));
format->separator (format);
}
}
notmuch_filenames_destroy( filenames );
} else if (ctx->output == OUTPUT_MESSAGES) {
/* special case 1 for speed */
if (ctx->dupe <= 1 || ctx->dupe <= _count_filenames (message)) {
format->set_prefix (format, "id");
format->string (format,
notmuch_message_get_message_id (message));
format->separator (format);
}
} else {
if (ctx->output & OUTPUT_SENDER) {
const char *addrs;
addrs = notmuch_message_get_header (message, "from");
process_address_header (ctx, addrs);
}
if (ctx->output & OUTPUT_RECIPIENTS) {
const char *hdrs[] = { "to", "cc", "bcc" };
const char *addrs;
size_t j;
for (j = 0; j < ARRAY_SIZE (hdrs); j++) {
addrs = notmuch_message_get_header (message, hdrs[j]);
process_address_header (ctx, addrs);
}
}
}
notmuch_message_destroy (message);
}
if (ctx->addresses &&
(ctx->output & OUTPUT_COUNT || ctx->dedup == DEDUP_ADDRESS))
g_hash_table_foreach (ctx->addresses, print_hash_value, ctx);
notmuch_messages_destroy (messages);
format->end (format);
return 0;
}
static int
do_search_messages (const search_format_t *format,
notmuch_query_t *query,
output_t output,
int offset,
int limit)
{
notmuch_message_t *message;
notmuch_messages_t *messages;
notmuch_filenames_t *filenames;
int first_message = 1;
int i;
if (offset < 0) {
offset += notmuch_query_count_messages (query);
if (offset < 0)
offset = 0;
}
messages = notmuch_query_search_messages (query);
if (messages == NULL)
return 1;
fputs (format->results_start, stdout);
for (i = 0;
notmuch_messages_valid (messages) && (limit < 0 || i < offset + limit);
notmuch_messages_move_to_next (messages), i++)
{
if (i < offset)
continue;
message = notmuch_messages_get (messages);
if (output == OUTPUT_FILES) {
filenames = notmuch_message_get_filenames (message);
for (;
notmuch_filenames_valid (filenames);
notmuch_filenames_move_to_next (filenames))
{
if (! first_message)
fputs (format->item_sep, stdout);
format->item_id (message, "",
notmuch_filenames_get (filenames));
first_message = 0;
}
notmuch_filenames_destroy( filenames );
} else { /* output == OUTPUT_MESSAGES */
if (! first_message)
fputs (format->item_sep, stdout);
format->item_id (message, "id:",
notmuch_message_get_message_id (message));
first_message = 0;
}
notmuch_message_destroy (message);
}
notmuch_messages_destroy (messages);
if (first_message)
fputs (format->results_null, stdout);
else
fputs (format->results_end, stdout);
return 0;
}
