/*

 * 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 <errno.h>

#ifdef G_OS_WIN32

#include <io.h>

#endif

#ifndef STDOUT_FILENO

#define STDOUT_FILENO 1

#endif

#ifdef HAVE_UNISTD_H

#include <unistd.h>

#endif

#include "gio-tool.h"

static const GOptionEntry entries[] = {

  { NULL }

};

/* 256k minus malloc overhead */

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

static gboolean

cat (GFile *file)

{

  GInputStream *in;

  char *buffer;

  char *p;

  gssize res;

  gboolean close_res;

  GError *error;

  gboolean success;

  error = NULL;

  in = (GInputStream *) g_file_read (file, NULL, &error);

  if (in == NULL)

    {

      print_file_error (file, error->message);

      g_error_free (error);

      return FALSE;

    }

  buffer = g_malloc (STREAM_BUFFER_SIZE);

  success = TRUE;

  while (1)

    {

      res = g_input_stream_read (in, buffer, STREAM_BUFFER_SIZE, NULL, &error);

      if (res > 0)

        {

          gssize written;

          p = buffer;

          while (res > 0)

            {

              int errsv;

              written = write (STDOUT_FILENO, p, res);

              errsv = errno;

              if (written == -1 && errsv != EINTR)

                {

                  print_error ("%s", _("Error writing to stdout"));

                  success = FALSE;

                  goto out;

                }

              res -= written;

              p += written;

            }

        }

      else if (res < 0)

        {

          print_file_error (file, error->message);

          g_error_free (error);

          error = NULL;

          success = FALSE;

          break;

        }

      else if (res == 0)

        break;

    }

 out:

 close_res = g_input_stream_close (in, NULL, &error);

  if (!close_res)

    {

      print_file_error (file, error->message);

      g_error_free (error);

      success = FALSE;

    }

  g_free (buffer);

  return success;

}

int

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

{

  GOptionContext *context;

  gchar *param;

  GError *error = NULL;

  int i;

  gboolean res;

  GFile *file;

  g_set_prgname ("gio cat");

  /* Translators: commandline placeholder */

  param = g_strdup_printf ("%s...", _("LOCATION"));

  context = g_option_context_new (param);

  g_free (param);

  g_option_context_set_help_enabled (context, FALSE);

  g_option_context_set_summary (context,

      _("Concatenate files and print to standard output."));

  g_option_context_set_description (context,

      _("gio cat works just like the traditional cat 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 < 2)

    {

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

      g_option_context_free (context);

      return 1;

    }

  g_option_context_free (context);

  res = TRUE;

  for (i = 1; i < argc; i++)

    {

      file = g_file_new_for_commandline_arg (argv[i]);

      res &= cat (file);

      g_object_unref (file);

    }

  return res ? 0 : 2;

}