/*

 * 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>

#ifdef G_OS_WIN32

#include <io.h>

#endif

#ifndef STDIN_FILENO

#define STDIN_FILENO 0

#endif

#ifdef HAVE_UNISTD_H

#include <unistd.h>

#endif

#include "gio-tool.h"

static char *etag = NULL;

static gboolean backup = FALSE;

static gboolean create = FALSE;

static gboolean append = FALSE;

static gboolean priv = FALSE;

static gboolean replace_dest = FALSE;

static gboolean print_etag = FALSE;

static const GOptionEntry entries[] =

{

  { "backup", 'b', 0, G_OPTION_ARG_NONE, &backup, N_("Backup existing destination files"), NULL },

  { "create", 'c', 0, G_OPTION_ARG_NONE, &create, N_("Only create if not existing"), NULL },

  { "append", 'a', 0, G_OPTION_ARG_NONE, &append, N_("Append to end of file"), NULL },

  { "private", 'p', 0, G_OPTION_ARG_NONE, &priv, N_("When creating, restrict access to the current user"), NULL },

  { "unlink", 'u', 0, G_OPTION_ARG_NONE, &replace_dest, N_("When replacing, replace as if the destination did not exist"), NULL },

  /* Translators: The "etag" is a token allowing to verify whether a file has been modified */

  { "print-etag", 'v', 0, G_OPTION_ARG_NONE, &print_etag, N_("Print new etag at end"), NULL },

  /* Translators: The "etag" is a token allowing to verify whether a file has been modified */

  { "etag", 'e', 0, G_OPTION_ARG_STRING, &etag, N_("The etag of the file being overwritten"), N_("ETAG") },

  { NULL }

};

/* 256k minus malloc overhead */

#define STREAM_BUFFER_SIZE (1024*256 - 2*sizeof(gpointer))

static gboolean

save (GFile *file)

{

  GOutputStream *out;

  GFileCreateFlags flags;

  char *buffer;

  gssize res;

  gboolean close_res;

  GError *error;

  gboolean save_res;

  error = NULL;

  flags = priv ? G_FILE_CREATE_PRIVATE : G_FILE_CREATE_NONE;

  flags |= replace_dest ? G_FILE_CREATE_REPLACE_DESTINATION : 0;

  if (create)

    out = (GOutputStream *)g_file_create (file, flags, NULL, &error);

  else if (append)

    out = (GOutputStream *)g_file_append_to (file, flags, NULL, &error);

  else

    out = (GOutputStream *)g_file_replace (file, etag, backup, flags, NULL, &error);

  if (out == NULL)

    {

      print_file_error (file, error->message);

      g_error_free (error);

      return FALSE;

    }

  buffer = g_malloc (STREAM_BUFFER_SIZE);

  save_res = TRUE;

  while (1)

    {

      res = read (STDIN_FILENO, buffer, STREAM_BUFFER_SIZE);

      if (res > 0)

	{

          g_output_stream_write_all (out, buffer, res, NULL, NULL, &error);

          if (error != NULL)

            {

              save_res = FALSE;

              print_file_error (file, error->message);

              g_clear_error (&error);

              goto out;

            }

	}

      else if (res < 0)

	{

	  save_res = FALSE;

          print_error ("%s", _("Error reading from standard input"));

	  break;

	}

      else if (res == 0)

	break;

    }

 out:

  close_res = g_output_stream_close (out, NULL, &error);

  if (!close_res)

    {

      save_res = FALSE;

      print_file_error (file, error->message);

      g_error_free (error);

    }

  if (close_res && print_etag)

    {

      char *etag;

      etag = g_file_output_stream_get_etag (G_FILE_OUTPUT_STREAM (out));

      if (etag)

	g_print ("Etag: %s\n", etag);

      else

	/* Translators: The "etag" is a token allowing to verify whether a file has been modified */

	g_print (_("Etag not available\n"));

      g_free (etag);

    }

  g_object_unref (out);

  g_free (buffer);

  return save_res;

}

int

handle_save (int argc, char *argv[], gboolean do_help)

{

  GOptionContext *context;

  GError *error = NULL;

  GFile *file;

  gboolean res;

  g_set_prgname ("gio save");

  /* Translators: commandline placeholder */

  context = g_option_context_new (_("DESTINATION"));

  g_option_context_set_help_enabled (context, FALSE);

  g_option_context_set_summary (context,

      _("Read from standard input and save to DEST."));

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

    {

      show_help (context, _("No destination given"));

      g_option_context_free (context);

      return 1;

    }

  if (argc > 2)

    {

      show_help (context, _("Too many arguments"));

      g_option_context_free (context);

      return 1;

    }

  g_option_context_free (context);

  file = g_file_new_for_commandline_arg (argv[1]);

  res = save (file);

  g_object_unref (file);

  return res ? 0 : 2;

}