/*
 * gnome-keyring
 *
 * Copyright (C) 2011 Collabora Ltd.
 *
 * This program 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 program 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 program; if not, see <http://www.gnu.org/licenses/>.
 *
 * Author: Stef Walter <stefw@collabora.co.uk>
 */

#include "config.h"

#include "gcr-callback-output-stream.h"

#include <glib/gi18n-lib.h>

struct _GcrCallbackOutputStream {
	GOutputStream parent;
	GcrCallbackOutputFunc callback;
	gpointer user_data;
	GDestroyNotify destroy_func;
};

struct _GcrCallbackOutputStreamClass {
	GOutputStreamClass parent_class;
};

G_DEFINE_TYPE (GcrCallbackOutputStream, _gcr_callback_output_stream, G_TYPE_OUTPUT_STREAM);

static void
_gcr_callback_output_stream_init (GcrCallbackOutputStream *self)
{

}

static gssize
_gcr_callback_output_stream_write (GOutputStream *stream,
                                   const void *buffer,
                                   gsize count,
                                   GCancellable *cancellable,
                                   GError **error)
{
	GcrCallbackOutputStream *self = GCR_CALLBACK_OUTPUT_STREAM (stream);

	if (g_cancellable_set_error_if_cancelled (cancellable, error)) {
		return -1;
	} else if (self->callback == NULL) {
		g_set_error (error, G_IO_ERROR, G_IO_ERROR_CLOSED,
		             _("The stream was closed"));
		return -1;
	}

	return (self->callback) (buffer, count, cancellable, self->user_data, error);
}

static gboolean
_gcr_callback_output_stream_close (GOutputStream *stream,
                                   GCancellable *cancellable,
                                   GError **error)
{
	GcrCallbackOutputStream *self = GCR_CALLBACK_OUTPUT_STREAM (stream);
	if (g_cancellable_set_error_if_cancelled (cancellable, error)) {
		return FALSE;
	} else if (self->callback == NULL) {
		g_set_error (error, G_IO_ERROR, G_IO_ERROR_CLOSED,
		             _("The stream was closed"));
		return FALSE;
	}

	if (self->destroy_func != NULL)
		(self->destroy_func) (self->user_data);
	self->destroy_func = NULL;
	self->user_data = NULL;
	self->callback = NULL;

	return TRUE;
}

static void
_gcr_callback_output_stream_dispose (GObject *obj)
{
	_gcr_callback_output_stream_close (G_OUTPUT_STREAM (obj), NULL, NULL);
	G_OBJECT_CLASS (_gcr_callback_output_stream_parent_class)->dispose (obj);
}

static void
_gcr_callback_output_stream_class_init (GcrCallbackOutputStreamClass *klass)
{
	GObjectClass *gobject_class = G_OBJECT_CLASS (klass);
	GOutputStreamClass *output_class = G_OUTPUT_STREAM_CLASS (klass);

	gobject_class->dispose = _gcr_callback_output_stream_dispose;
	output_class->write_fn = _gcr_callback_output_stream_write;
	output_class->close_fn = _gcr_callback_output_stream_close;
}

/**
 * _gcr_callback_output_stream_new: (skip)
 *
 * Create a new output stream which invokes the callback.
 *
 * Returns: (transfer full) (type Gcr.CallbackOutputStream): the new stream
 */
GOutputStream *
_gcr_callback_output_stream_new (GcrCallbackOutputFunc callback,
                                 gpointer user_data,
                                 GDestroyNotify destroy_func)
{
	GcrCallbackOutputStream *self;

	g_return_val_if_fail (callback, NULL);

	self = g_object_new (GCR_TYPE_CALLBACK_OUTPUT_STREAM, NULL);
	self->callback = callback;
	self->user_data = user_data;
	self->destroy_func = destroy_func;

	return G_OUTPUT_STREAM (self);
}