glib/gio/gio-tool-copy.c

219 lines
6.2 KiB
C
Raw Normal View History

/*
* Copyright 2015 Red Hat, Inc.
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2 of the licence, or (at your option) any later version.
*
* This library 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
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, see <http://www.gnu.org/licenses/>.
*
* Author: Matthias Clasen <mclasen@redhat.com>
*/
#include "config.h"
#include <gio/gio.h>
#include <gi18n.h>
#include <stdio.h>
#if 0
#include <locale.h>
#include <string.h>
#include <stdlib.h>
#include <errno.h>
#endif
#include "gio-tool.h"
static gboolean no_target_directory = FALSE;
static gboolean progress = FALSE;
static gboolean interactive = FALSE;
static gboolean preserve = FALSE;
static gboolean backup = FALSE;
static gboolean no_dereference = FALSE;
static const GOptionEntry entries[] = {
{ "no-target-directory", 'T', 0, G_OPTION_ARG_NONE, &no_target_directory, N_("No target directory"), NULL },
{ "progress", 'p', 0, G_OPTION_ARG_NONE, &progress, N_("Show progress"), NULL },
{ "interactive", 'i', 0, G_OPTION_ARG_NONE, &interactive, N_("Prompt before overwrite"), NULL },
{ "preserve", 'p', 0, G_OPTION_ARG_NONE, &preserve, N_("Preserve all attributes"), NULL },
{ "backup", 'b', 0, G_OPTION_ARG_NONE, &backup, N_("Backup existing destination files"), NULL },
{ "no-dereference", 'P', 0, G_OPTION_ARG_NONE, &no_dereference, N_("Never follow symbolic links"), NULL },
{ NULL }
};
static gint64 start_time;
static gint64 previous_time;
static void
show_progress (goffset current_num_bytes,
goffset total_num_bytes,
gpointer user_data)
{
gint64 tv;
char *current_size, *total_size, *rate;
tv = g_get_monotonic_time ();
if (tv - previous_time < (G_USEC_PER_SEC / 5) &&
current_num_bytes != total_num_bytes)
return;
current_size = g_format_size (current_num_bytes);
total_size = g_format_size (total_num_bytes);
rate = g_format_size (current_num_bytes /
MAX ((tv - start_time) / G_USEC_PER_SEC, 1));
g_print ("\r\033[K");
g_print (_("Transferred %s out of %s (%s/s)"), current_size, total_size, rate);
previous_time = tv;
g_free (current_size);
g_free (total_size);
g_free (rate);
}
int
handle_copy (int argc, char *argv[], gboolean do_help)
{
GOptionContext *context;
GError *error = NULL;
char *param;
GFile *source, *dest, *target;
gboolean dest_is_dir;
char *basename;
char *uri;
int i;
GFileCopyFlags flags;
int retval = 0;
g_set_prgname ("gio copy");
/* Translators: commandline placeholder */
param = g_strdup_printf ("%s... %s", _("SOURCE"), _("DESTINATION"));
context = g_option_context_new (param);
g_free (param);
g_option_context_set_help_enabled (context, FALSE);
g_option_context_set_summary (context,
_("Copy one or more files from SOURCE to DESTINATION."));
g_option_context_set_description (context,
_("gio copy is similar to the traditional cp utility, but using GIO\n"
"locations instead of local files: for example, you can use something\n"
"like smb://server/resource/file.txt as location."));
g_option_context_add_main_entries (context, entries, GETTEXT_PACKAGE);
if (do_help)
{
show_help (context, NULL);
return 0;
}
if (!g_option_context_parse (context, &argc, &argv, &error))
{
show_help (context, error->message);
g_error_free (error);
return 1;
}
if (argc < 3)
{
show_help (context, NULL);
return 1;
}
dest = g_file_new_for_commandline_arg (argv[argc - 1]);
if (no_target_directory && argc > 3)
{
show_help (context, NULL);
g_object_unref (dest);
return 1;
}
dest_is_dir = file_is_dir (dest);
if (!dest_is_dir && argc > 3)
{
char *message;
message = g_strdup_printf (_("Destination %s is not a directory"), argv[argc - 1]);
show_help (context, message);
g_free (message);
g_object_unref (dest);
return 1;
}
g_option_context_free (context);
for (i = 1; i < argc - 1; i++)
{
source = g_file_new_for_commandline_arg (argv[i]);
if (dest_is_dir && !no_target_directory)
{
basename = g_file_get_basename (source);
target = g_file_get_child (dest, basename);
g_free (basename);
}
else
target = g_object_ref (dest);
flags = 0;
if (backup)
flags |= G_FILE_COPY_BACKUP;
if (!interactive)
flags |= G_FILE_COPY_OVERWRITE;
if (no_dereference)
flags |= G_FILE_COPY_NOFOLLOW_SYMLINKS;
if (preserve)
flags |= G_FILE_COPY_ALL_METADATA;
error = NULL;
start_time = g_get_monotonic_time ();
if (!g_file_copy (source, target, flags, NULL, progress ? show_progress : NULL, NULL, &error))
{
if (interactive && g_error_matches (error, G_IO_ERROR, G_IO_ERROR_EXISTS))
{
char line[16];
g_error_free (error);
error = NULL;
uri = g_file_get_uri (target);
g_print (_("%s: overwrite “%s”? "), argv[0], uri);
g_free (uri);
if (fgets (line, sizeof (line), stdin) &&
(line[0] == 'y' || line[0] == 'Y'))
{
flags |= G_FILE_COPY_OVERWRITE;
start_time = g_get_monotonic_time ();
if (!g_file_copy (source, target, flags, NULL, progress ? show_progress : NULL, NULL, &error))
goto copy_failed;
}
}
else
{
copy_failed:
print_file_error (source, error->message);
g_error_free (error);
retval = 1;
}
}
if (progress && retval == 0)
g_print ("\n");
g_object_unref (source);
g_object_unref (target);
}
g_object_unref (dest);
return retval;
}