gio/gio-tool-move.c

   1 /*
   2  * Copyright 2015 Red Hat, Inc.
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, see <http://www.gnu.org/licenses/>.
  16  *
  17  * Author: Matthias Clasen <mclasen@redhat.com>
  18  */
  19
  20 #include "config.h"
  21
  22 #include <gio/gio.h>
  23 #include <gi18n.h>
  24 #include <stdio.h>
  25
  26 #include "gio-tool.h"
  27
  28
  29 /* statics {{{1 */
  30
  31 static gboolean no_target_directory = FALSE;
  32 static gboolean progress = FALSE;
  33 static gboolean interactive = FALSE;
  34 static gboolean backup = FALSE;
  35 static gboolean no_copy_fallback = FALSE;
  36
  37 static const GOptionEntry entries[] = {
  38   { "no-target-directory", 'T', 0, G_OPTION_ARG_NONE, &no_target_directory, N_("No target directory"), NULL },
  39   { "progress", 'p', 0, G_OPTION_ARG_NONE, &progress, N_("Show progress"), NULL },
  40   { "interactive", 'i', 0, G_OPTION_ARG_NONE, &interactive, N_("Prompt before overwrite"), NULL },
  41   { "backup", 'b', 0, G_OPTION_ARG_NONE, &backup, N_("Backup existing destination files"), NULL },
  42   { "no-copy-fallback", 'C', 0, G_OPTION_ARG_NONE, &no_copy_fallback, N_("Don’t use copy and delete fallback"), NULL },
  43   { NULL }
  44 };
  45
  46 static gint64 start_time;
  47 static gint64 previous_time;
  48
  49 static void
  50 show_progress (goffset current_num_bytes,
  51                goffset total_num_bytes,
  52                gpointer user_data)
  53 {
  54   gint64 tv;
  55   char *current_size, *total_size, *rate;
  56
  57   tv = g_get_monotonic_time ();
  58   if (tv - previous_time < (G_USEC_PER_SEC / 5) &&
  59       current_num_bytes != total_num_bytes)
  60     return;
  61
  62   current_size = g_format_size (current_num_bytes);
  63   total_size = g_format_size (total_num_bytes);
  64   rate = g_format_size (current_num_bytes /
  65                         MAX ((tv - start_time) / G_USEC_PER_SEC, 1));
  66   g_print ("\r\033[K");
  67   g_print (_("Transferred %s out of %s (%s/s)"),
  68            current_size, total_size, rate);
  69
  70   previous_time = tv;
  71
  72   g_free (current_size);
  73   g_free (total_size);
  74   g_free (rate);
  75 }
  76
  77 int
  78 handle_move (int argc, char *argv[], gboolean do_help)
  79 {
  80   GOptionContext *context;
  81   gchar *param;
  82   GError *error = NULL;
  83   GFile *source, *dest, *target;
  84   gboolean dest_is_dir;
  85   char *basename;
  86   char *uri;
  87   int i;
  88   GFileCopyFlags flags;
  89   int retval = 0;
  90
  91   g_set_prgname ("gio move");
  92
  93   /* Translators: commandline placeholder */
  94   param = g_strdup_printf ("%s... %s", _("SOURCE"), _("DESTINATION"));
  95   context = g_option_context_new (param);
  96   g_free (param);
  97   g_option_context_set_help_enabled (context, FALSE);
  98   g_option_context_set_summary (context,
  99       _("Move one or more files from SOURCE to DEST."));
 100   g_option_context_set_description (context,
 101       _("gio move is similar to the traditional mv utility, but using GIO\n"
 102         "locations instead of local files: for example, you can use something\n"
 103         "like smb://server/resource/file.txt as location"));
 104   g_option_context_add_main_entries (context, entries, GETTEXT_PACKAGE);
 105
 106   if (do_help)
 107     {
 108       show_help (context, NULL);
 109       return 0;
 110     }
 111
 112   if (!g_option_context_parse (context, &argc, &argv, &error))
 113     {
 114       show_help (context, error->message);
 115       g_error_free (error);
 116       g_option_context_free (context);
 117       return 1;
 118     }
 119
 120   if (argc < 3)
 121     {
 122       show_help (context, NULL);
 123       g_option_context_free (context);
 124       return 1;
 125     }
 126
 127   dest = g_file_new_for_commandline_arg (argv[argc - 1]);
 128
 129   if (no_target_directory && argc > 3)
 130     {
 131       show_help (context, NULL);
 132       g_object_unref (dest);
 133       g_option_context_free (context);
 134       return 1;
 135     }
 136
 137   dest_is_dir = file_is_dir (dest);
 138
 139   if (!dest_is_dir && argc > 3)
 140     {
 141       char *message;
 142       message = g_strdup_printf (_("Target %s is not a directory"), argv[argc - 1]);
 143       show_help (context, message);
 144       g_free (message);
 145       g_object_unref (dest);
 146       g_option_context_free (context);
 147       return 1;
 148     }
 149
 150   g_option_context_free (context);
 151
 152   for (i = 1; i < argc - 1; i++)
 153     {
 154       source = g_file_new_for_commandline_arg (argv[i]);
 155
 156       if (dest_is_dir && !no_target_directory)
 157         {
 158           basename = g_file_get_basename (source);
 159           target = g_file_get_child (dest, basename);
 160           g_free (basename);
 161         }
 162       else
 163         target = g_object_ref (dest);
 164
 165       flags = 0;
 166       if (backup)
 167         flags |= G_FILE_COPY_BACKUP;
 168       if (!interactive)
 169         flags |= G_FILE_COPY_OVERWRITE;
 170       if (no_copy_fallback)
 171         flags |= G_FILE_COPY_NO_FALLBACK_FOR_MOVE;
 172
 173       error = NULL;
 174       start_time = g_get_monotonic_time ();
 175       if (!g_file_move (source, target, flags, NULL, progress ? show_progress : NULL, NULL, &error))
 176         {
 177           if (interactive && g_error_matches (error, G_IO_ERROR, G_IO_ERROR_EXISTS))
 178             {
 179               char line[16];
 180
 181               g_error_free (error);
 182               error = NULL;
 183
 184               uri = g_file_get_uri (target);
 185               g_print (_("%s: overwrite “%s”? "), argv[0], uri);
 186               g_free (uri);
 187               if (fgets (line, sizeof (line), stdin) &&
 188                   (line[0] == 'y' || line[0] == 'Y'))
 189                 {
 190                   flags |= G_FILE_COPY_OVERWRITE;
 191                   start_time = g_get_monotonic_time ();
 192                   if (!g_file_move (source, target, flags, NULL, progress ? show_progress : NULL, NULL, &error))
 193                     goto move_failed;
 194                 }
 195             }
 196           else
 197             {
 198             move_failed:
 199               print_file_error (source, error->message);
 200               g_error_free (error);
 201               retval = 1;
 202             }
 203         }
 204
 205       if (progress && retval == 0)
 206         g_print("\n");
 207
 208       g_object_unref (source);
 209       g_object_unref (target);
 210     }
 211
 212   g_object_unref (dest);
 213
 214   return retval;
 215 }