/* purple
 *
 * Purple is the legal property of its developers, whose names are too numerous
 * to list here.  Please refer to the COPYRIGHT file distributed with this
 * source distribution.
 *
 * This program 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 2 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02111-1301  USA
 */

#include "debug.h"
#include "image.h"
#include "util.h"

typedef struct {
	gchar *path;

	GBytes *contents;

	const gchar *extension;
	const gchar *mime;
	gchar *gen_filename;
	gchar *friendly_filename;
} PurpleImagePrivate;

enum {
	PROP_0,
	PROP_PATH,
	PROP_CONTENTS,
	PROP_SIZE,
	PROP_LAST
};

static GParamSpec *properties[PROP_LAST];

G_DEFINE_TYPE_WITH_PRIVATE(PurpleImage, purple_image, G_TYPE_OBJECT);

/******************************************************************************
 * Helpers
 ******************************************************************************/
static void
_purple_image_set_path(PurpleImage *image, const gchar *path) {
	PurpleImagePrivate *priv = purple_image_get_instance_private(image);

	g_free(priv->path);

	priv->path = g_strdup(path);
}

static void
_purple_image_set_contents(PurpleImage *image, GBytes *bytes) {
	PurpleImagePrivate *priv = purple_image_get_instance_private(image);

	if(priv->contents)
		g_bytes_unref(priv->contents);

	priv->contents = (bytes) ? g_bytes_ref(bytes) : NULL;
}

/******************************************************************************
 * Object stuff
 ******************************************************************************/
static void
purple_image_init(G_GNUC_UNUSED PurpleImage *image) {
}

static void
purple_image_finalize(GObject *obj) {
	PurpleImage *image = PURPLE_IMAGE(obj);
	PurpleImagePrivate *priv = purple_image_get_instance_private(image);

	if(priv->contents)
		g_bytes_unref(priv->contents);

	g_free(priv->path);
	g_free(priv->gen_filename);
	g_free(priv->friendly_filename);

	G_OBJECT_CLASS(purple_image_parent_class)->finalize(obj);
}

static void
purple_image_set_property(GObject *obj, guint param_id,
                          const GValue *value, GParamSpec *pspec)
{
	PurpleImage *image = PURPLE_IMAGE(obj);

	switch (param_id) {
		case PROP_PATH:
			_purple_image_set_path(image, g_value_get_string(value));
			break;
		case PROP_CONTENTS:
			_purple_image_set_contents(image, g_value_get_boxed(value));
			break;
		default:
			G_OBJECT_WARN_INVALID_PROPERTY_ID(obj, param_id, pspec);
			break;
	}
}

static void
purple_image_get_property(GObject *obj, guint param_id, GValue *value,
                          GParamSpec *pspec)
{
	PurpleImage *image = PURPLE_IMAGE(obj);

	switch (param_id) {
		case PROP_PATH:
			g_value_set_string(value, purple_image_get_path(image));
			break;
		case PROP_CONTENTS:
			g_value_set_boxed(value, purple_image_get_contents(image));
			break;
		case PROP_SIZE:
			g_value_set_uint64(value, purple_image_get_data_size(image));
			break;
		default:
			G_OBJECT_WARN_INVALID_PROPERTY_ID(obj, param_id, pspec);
			break;
	}
}

static void
purple_image_class_init(PurpleImageClass *klass) {
	GObjectClass *gobj_class = G_OBJECT_CLASS(klass);

	gobj_class->finalize = purple_image_finalize;
	gobj_class->get_property = purple_image_get_property;
	gobj_class->set_property = purple_image_set_property;

	properties[PROP_PATH] = g_param_spec_string(
		"path",
		"path",
		"The filepath for the image if one was provided",
		NULL,
		G_PARAM_READWRITE | G_PARAM_CONSTRUCT_ONLY | G_PARAM_STATIC_STRINGS
	);

	properties[PROP_CONTENTS] = g_param_spec_boxed(
		"contents",
		"contents",
		"The contents of the image stored in a GBytes",
		G_TYPE_BYTES,
		G_PARAM_READWRITE | G_PARAM_CONSTRUCT_ONLY | G_PARAM_STATIC_STRINGS
	);

	properties[PROP_SIZE] = g_param_spec_uint64(
		"size",
		"size",
		"The size of the image in bytes",
		0,
		G_MAXUINT64,
		0,
		G_PARAM_READABLE | G_PARAM_STATIC_STRINGS
	);

	g_object_class_install_properties(gobj_class, PROP_LAST, properties);
}

/******************************************************************************
 * API
 ******************************************************************************/
PurpleImage *
purple_image_new_from_bytes(GBytes *bytes) {
	return g_object_new(
		PURPLE_TYPE_IMAGE,
		"contents", bytes,
		NULL
	);
}

PurpleImage *
purple_image_new_from_file(const gchar *path, GError **error) {
	PurpleImage *image = NULL;
	GBytes *bytes = NULL;
	gchar *contents = NULL;
	gsize length = 0;

	if(!g_file_get_contents(path, &contents, &length, error)) {
		return NULL;
	}

	bytes = g_bytes_new_take(contents, length);

	image = g_object_new(
		PURPLE_TYPE_IMAGE,
		"contents", bytes,
		"path", path,
		NULL
	);

	g_bytes_unref(bytes);

	return image;
}

PurpleImage *
purple_image_new_from_data(const guint8 *data, gsize length) {
	PurpleImage *image;
	GBytes *bytes = NULL;

	bytes = g_bytes_new(data, length);

	image = purple_image_new_from_bytes(bytes);

	g_bytes_unref(bytes);

	return image;
}

PurpleImage *
purple_image_new_take_data(guint8 *data, gsize length) {
	PurpleImage *image;
	GBytes *bytes = NULL;

	bytes = g_bytes_new_take(data, length);

	image = purple_image_new_from_bytes(bytes);

	g_bytes_unref(bytes);

	return image;
}

gboolean
purple_image_save(PurpleImage *image, const gchar *path) {
	PurpleImagePrivate *priv = NULL;
	gconstpointer data;
	gsize len;
	gboolean succ;

	g_return_val_if_fail(PURPLE_IS_IMAGE(image), FALSE);
	g_return_val_if_fail(path != NULL, FALSE);
	g_return_val_if_fail(path[0] != '\0', FALSE);

	priv = purple_image_get_instance_private(image);
	data = purple_image_get_data(image);
	len = purple_image_get_data_size(image);

	g_return_val_if_fail(data != NULL, FALSE);
	g_return_val_if_fail(len > 0, FALSE);

	succ = g_file_set_contents(path, data, len, NULL);
	if (succ && priv->path == NULL)
		priv->path = g_strdup(path);

	return succ;
}

GBytes *
purple_image_get_contents(PurpleImage *image)
{
	PurpleImagePrivate *priv = NULL;

	g_return_val_if_fail(PURPLE_IS_IMAGE(image), NULL);

	priv = purple_image_get_instance_private(image);

	if(priv->contents)
		return g_bytes_ref(priv->contents);

	return NULL;
}

const gchar *
purple_image_get_path(PurpleImage *image) {
	PurpleImagePrivate *priv = NULL;

	g_return_val_if_fail(PURPLE_IS_IMAGE(image), NULL);

	priv = purple_image_get_instance_private(image);

	return priv->path ? priv->path : purple_image_generate_filename(image);
}

gsize
purple_image_get_data_size(PurpleImage *image) {
	PurpleImagePrivate *priv;

	g_return_val_if_fail(PURPLE_IS_IMAGE(image), 0);

	priv = purple_image_get_instance_private(image);

	if(priv->contents)
		return g_bytes_get_size(priv->contents);

	return 0;
}

gconstpointer
purple_image_get_data(PurpleImage *image) {
	PurpleImagePrivate *priv = NULL;

	g_return_val_if_fail(PURPLE_IS_IMAGE(image), NULL);

	priv = purple_image_get_instance_private(image);

	if(priv->contents)
		return g_bytes_get_data(priv->contents, NULL);

	return NULL;
}

const gchar *
purple_image_get_extension(PurpleImage *image) {
	PurpleImagePrivate *priv = NULL;
	gconstpointer data;

	g_return_val_if_fail(PURPLE_IS_IMAGE(image), NULL);

	priv = purple_image_get_instance_private(image);

	if (priv->extension)
		return priv->extension;

	if (purple_image_get_data_size(image) < 4)
		return NULL;

	data = purple_image_get_data(image);
	g_assert(data != NULL);

	if (memcmp(data, "GIF8", 4) == 0)
		return priv->extension = "gif";
	if (memcmp(data, "\xff\xd8\xff", 3) == 0) /* 4th may be e0 through ef */
		return priv->extension = "jpg";
	if (memcmp(data, "\x89PNG", 4) == 0)
		return priv->extension = "png";
	if (memcmp(data, "MM", 2) == 0)
		return priv->extension = "tif";
	if (memcmp(data, "II", 2) == 0)
		return priv->extension = "tif";
	if (memcmp(data, "BM", 2) == 0)
		return priv->extension = "bmp";
	if (memcmp(data, "\x00\x00\x01\x00", 4) == 0)
		return priv->extension = "ico";

	return NULL;
}

const gchar *
purple_image_get_mimetype(PurpleImage *image) {
	PurpleImagePrivate *priv = NULL;
	const gchar *ext = purple_image_get_extension(image);

	g_return_val_if_fail(PURPLE_IS_IMAGE(image), NULL);

	priv = purple_image_get_instance_private(image);

	if (priv->mime)
		return priv->mime;

	g_return_val_if_fail(ext != NULL, NULL);

	if (g_strcmp0(ext, "gif") == 0)
		return priv->mime = "image/gif";
	if (g_strcmp0(ext, "jpg") == 0)
		return priv->mime = "image/jpeg";
	if (g_strcmp0(ext, "png") == 0)
		return priv->mime = "image/png";
	if (g_strcmp0(ext, "tif") == 0)
		return priv->mime = "image/tiff";
	if (g_strcmp0(ext, "bmp") == 0)
		return priv->mime = "image/bmp";
	if (g_strcmp0(ext, "ico") == 0)
		return priv->mime = "image/vnd.microsoft.icon";

	return NULL;
}

const gchar *
purple_image_generate_filename(PurpleImage *image) {
	PurpleImagePrivate *priv = NULL;
	gconstpointer data;
	gsize len;
	const gchar *ext = NULL;
	gchar *checksum;

	g_return_val_if_fail(PURPLE_IS_IMAGE(image), NULL);

	priv = purple_image_get_instance_private(image);

	if (priv->gen_filename)
		return priv->gen_filename;

	/* grab the image's data and size of that data */
	data = purple_image_get_data(image);
	len = purple_image_get_data_size(image);

	/* create a checksum of it and use it as the start of our filename */
	checksum = g_compute_checksum_for_data(G_CHECKSUM_SHA1, data, len);

	/* if the image has a known format, set the extension appropriately */
	ext = purple_image_get_extension(image);
	if(ext != NULL) {
		priv->gen_filename = g_strdup_printf("%s.%s", checksum, ext);
		g_free(checksum);
	} else {
		priv->gen_filename = checksum;
	}

	return priv->gen_filename;
}

void
purple_image_set_friendly_filename(PurpleImage *image, const gchar *filename) {
	PurpleImagePrivate *priv = NULL;
	gchar *newname;
	const gchar *escaped;

	g_return_if_fail(PURPLE_IS_IMAGE(image));

	priv = purple_image_get_instance_private(image);

	newname = g_path_get_basename(filename);
	escaped = purple_escape_filename(newname);
	g_free(newname);
	newname = NULL;

	if (g_strcmp0(escaped, "") == 0 || g_strcmp0(escaped, ".") == 0 ||
		g_strcmp0(escaped, G_DIR_SEPARATOR_S) == 0 ||
		g_strcmp0(escaped, "/") == 0 || g_strcmp0(escaped, "\\") == 0)
	{
		escaped = NULL;
	}

	g_free(priv->friendly_filename);
	priv->friendly_filename = g_strdup(escaped);
}

const gchar *
purple_image_get_friendly_filename(PurpleImage *image) {
	PurpleImagePrivate *priv = NULL;

	g_return_val_if_fail(PURPLE_IS_IMAGE(image), NULL);

	priv = purple_image_get_instance_private(image);

	if(priv->friendly_filename) {
		return priv->friendly_filename;
	}

	return purple_image_generate_filename(image);
}