/* GNU Mailutils -- a suite of utilities for electronic mail
   Copyright (C) 1999-2021 Free Software Foundation, Inc.

   GNU Mailutils is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; either version 3, or (at your option)
   any later version.

   GNU Mailutils 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 General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with GNU Mailutils.  If not, see <http://www.gnu.org/licenses/>. */

#include "mail.h"

/*
 * w[rite] [file] -- GNU extension
 * w[rite] [msglist] file
 * W[rite] [msglist] -- GNU extension
 */

int
mail_write (int argc, char **argv)
{
  int rc;
  mu_stream_t output;
  char *filename = NULL;
  char *namebuf = NULL;
  msgset_t *msglist = NULL, *mp;
  size_t total_size = 0, total_lines = 0;

  if (mu_isupper (argv[0][0]))
    {
      if (msgset_parse (argc, argv, MSG_NODELETED|MSG_SILENT, &msglist))
	return 1;
      filename = namebuf = util_get_sender (msgset_msgno (msglist), 1);
    }
  else
    {
      if (argc >= 2)
	{
	  filename = argv[--argc];
	}
      else
	{
	  size_t n = get_cursor ();
	  if (n == 0)
	    {
	      mu_error (_("No applicable message"));
	      return 1;
	    }
	  mu_asprintf (&namebuf, "%lu", (unsigned long) n);
	  filename = namebuf;
	}
      if (msgset_parse (argc, argv, MSG_NODELETED|MSG_SILENT, &msglist))
	return 1;
    }
  
  rc = mu_mailbox_expand_name (filename, &filename);
  free (namebuf);
  if (rc)
    {
      mu_diag_funcall (MU_DIAG_ERROR, "mu_mailbox_expand_name", NULL, rc);
      msgset_free (msglist);
      return 1;
    }

  rc = mu_file_stream_create (&output, filename,
			      MU_STREAM_APPEND|MU_STREAM_CREAT);
  if (rc)
    {
      mu_error (_("can't open %s: %s"), filename, mu_strerror (rc));
      free (filename);
      msgset_free (msglist);
      return 1;
    }

  for (mp = msglist; mp; mp = mp->next)
    {
      mu_message_t msg;
      mu_attribute_t attr;
      size_t stat[2];
      
      if (util_get_message_part (mbox, mp, &msg))
        continue;

      rc = print_message_body (msg, output, stat);
      if (rc == 0)
	{
	  total_size += stat[0];
	  total_lines += stat[1];
	  
	  /* mark as saved. */
	  if (msgset_length (mp) > 1)
	    util_get_message (mbox, msgset_msgno (mp), &msg);
	  mu_message_get_attribute (msg, &attr);
	  mu_attribute_set_userflag (attr, MAIL_ATTRIBUTE_SAVED);
	}
      else
	mu_error (_("cannot save %lu: %s"),
		  (unsigned long) msgset_msgno (mp), mu_strerror (rc));
    }

  mu_stream_close (output);
  mu_stream_destroy (&output);
  
  mu_printf ("\"%s\" %3lu/%-5lu\n", filename,
		    (unsigned long) total_lines, (unsigned long) total_size);

  free (filename);
  msgset_free (msglist);
  return 0;
}