/*

 * 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.1 of the License, 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);

      g_option_context_free (context);

      return 0;

    }

  if (!g_option_context_parse (context, &argc, &argv, &error))

    {

      show_help (context, error->message);

      g_error_free (error);

      g_option_context_free (context);

      return 1;

    }

  if (argc < 3)

    {

      show_help (context, NULL);

      g_option_context_free (context);

      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);

      g_option_context_free (context);

      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);

      g_option_context_free (context);

      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;

}