/*
   nautilus-desktop-icon-file.c: Subclass of NautilusFile to help implement the
   virtual desktop icons.

   Copyright (C) 2003 Red Hat, Inc.

   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, see <http://www.gnu.org/licenses/>.

   Author: Alexander Larsson <alexl@redhat.com>
*/

#include <config.h>
#include "nautilus-desktop-icon-file.h"
#include "nautilus-desktop-metadata.h"
#include "nautilus-desktop-directory-file.h"
#include "nautilus-desktop-directory.h"

#include <src/nautilus-directory-notify.h>
#include <src/nautilus-directory-private.h>
#include <src/nautilus-file-attributes.h>
#include <src/nautilus-file-private.h>
#include <src/nautilus-file-utilities.h>
#include <src/nautilus-file-operations.h>
#include <src/nautilus-link.h>
#include <src/nautilus-file-undo-manager.h>

#include <eel/eel-glib-extensions.h>
#include <glib/gi18n.h>
#include <string.h>
#include <gio/gio.h>

struct NautilusDesktopIconFileDetails {
	NautilusDesktopLink *link;
};

G_DEFINE_TYPE(NautilusDesktopIconFile, nautilus_desktop_icon_file, NAUTILUS_TYPE_FILE)


static void
desktop_icon_file_monitor_add (NautilusFile *file,
			       gconstpointer client,
			       NautilusFileAttributes attributes)
{
	nautilus_directory_monitor_add_internal
		(file->details->directory, file,
		 client, TRUE, attributes, NULL, NULL);
}

static void
desktop_icon_file_monitor_remove (NautilusFile *file,
				  gconstpointer client)
{
	nautilus_directory_monitor_remove_internal
		(file->details->directory, file, client);
}

static void
desktop_icon_file_call_when_ready (NautilusFile *file,
				   NautilusFileAttributes attributes,
				   NautilusFileCallback callback,
				   gpointer callback_data)
{
	nautilus_directory_call_when_ready_internal
		(file->details->directory, file,
		 attributes, FALSE, NULL, callback, callback_data);
}

static void
desktop_icon_file_cancel_call_when_ready (NautilusFile *file,
					  NautilusFileCallback callback,
					  gpointer callback_data)
{
	nautilus_directory_cancel_callback_internal
		(file->details->directory, file,
		 NULL, callback, callback_data);
}

static gboolean
desktop_icon_file_check_if_ready (NautilusFile *file,
				  NautilusFileAttributes attributes)
{
	return nautilus_directory_check_if_ready_internal
		(file->details->directory, file,
		 attributes);
}

static gboolean
desktop_icon_file_get_item_count (NautilusFile *file,
				  guint *count,
				  gboolean *count_unreadable)
{
	if (count != NULL) {
		*count = 0;
	}
	if (count_unreadable != NULL) {
		*count_unreadable = FALSE;
	}
	return TRUE;
}

static NautilusRequestStatus
desktop_icon_file_get_deep_counts (NautilusFile *file,
				   guint *directory_count,
				   guint *file_count,
				   guint *unreadable_directory_count,
				   goffset *total_size)
{
	if (directory_count != NULL) {
		*directory_count = 0;
	}
	if (file_count != NULL) {
		*file_count = 0;
	}
	if (unreadable_directory_count != NULL) {
		*unreadable_directory_count = 0;
	}
	if (total_size != NULL) {
		*total_size = 0;
	}

	return NAUTILUS_REQUEST_DONE;
}

static gboolean
desktop_icon_file_get_date (NautilusFile *file,
			    NautilusDateType date_type,
			    time_t *date)
{
	NautilusDesktopIconFile *desktop_file;

	desktop_file = NAUTILUS_DESKTOP_ICON_FILE (file);

	return nautilus_desktop_link_get_date (desktop_file->details->link,
					       date_type, date);
}

static char *
desktop_icon_file_get_where_string (NautilusFile *file)
{
	return g_strdup (_("on the desktop"));
}

static void
nautilus_desktop_icon_file_init (NautilusDesktopIconFile *desktop_file)
{
	desktop_file->details =	G_TYPE_INSTANCE_GET_PRIVATE (desktop_file,
							     NAUTILUS_TYPE_DESKTOP_ICON_FILE,
							     NautilusDesktopIconFileDetails);
}

static void
update_info_from_link (NautilusDesktopIconFile *icon_file)
{
	NautilusFile *file;
	NautilusDesktopLink *link;
	char *display_name;
	GMount *mount;

	file = NAUTILUS_FILE (icon_file);

	link = icon_file->details->link;

	if (link == NULL) {
		return;
	}

	eel_ref_str_unref (file->details->mime_type);
	file->details->mime_type = eel_ref_str_get_unique ("application/x-nautilus-link");
	file->details->type = G_FILE_TYPE_SHORTCUT;
	file->details->size = 0;
	file->details->has_permissions = FALSE;
	file->details->can_read = TRUE;
	file->details->can_write = TRUE;

	file->details->can_mount = FALSE;
	file->details->can_unmount = FALSE;
	file->details->can_eject = FALSE;
	if (file->details->mount) {
		g_object_unref (file->details->mount);
	}
	mount = nautilus_desktop_link_get_mount (link);
	file->details->mount = mount;
	if (mount) {
		file->details->can_unmount = g_mount_can_unmount (mount);
		file->details->can_eject = g_mount_can_eject (mount);
	}

	file->details->file_info_is_up_to_date = TRUE;

	display_name = nautilus_desktop_link_get_display_name (link);
	nautilus_file_set_display_name (file,
					display_name, NULL, TRUE);
	g_free (display_name);

	if (file->details->icon != NULL) {
		g_object_unref (file->details->icon);
	}
	file->details->icon = nautilus_desktop_link_get_icon (link);
	g_free (file->details->activation_uri);
	file->details->activation_uri = nautilus_desktop_link_get_activation_uri (link);
	file->details->got_link_info = TRUE;
	file->details->link_info_is_up_to_date = TRUE;

	file->details->directory_count = 0;
	file->details->got_directory_count = TRUE;
	file->details->directory_count_is_up_to_date = TRUE;
}

void
nautilus_desktop_icon_file_update (NautilusDesktopIconFile *icon_file)
{
	NautilusFile *file;

	update_info_from_link (icon_file);
	file = NAUTILUS_FILE (icon_file);
	nautilus_file_changed (file);
}

void
nautilus_desktop_icon_file_remove (NautilusDesktopIconFile *icon_file)
{
	NautilusFile *file;
	GList list;

	icon_file->details->link = NULL;

	file = NAUTILUS_FILE (icon_file);

	/* ref here because we might be removing the last ref when we
	 * mark the file gone below, but we need to keep a ref at
	 * least long enough to send the change notification.
	 */
	nautilus_file_ref (file);

	file->details->is_gone = TRUE;

	list.data = file;
	list.next = NULL;
	list.prev = NULL;

	nautilus_directory_remove_file (file->details->directory, file);
	nautilus_directory_emit_change_signals (file->details->directory, &list);

	nautilus_file_unref (file);
}

NautilusDesktopIconFile *
nautilus_desktop_icon_file_new (NautilusDesktopLink *link)
{
	NautilusFile *file;
	NautilusDirectory *directory;
	NautilusDesktopIconFile *icon_file;
	GList list;
	char *name;

	directory = nautilus_directory_get_by_uri (EEL_DESKTOP_URI);

	file = NAUTILUS_FILE (g_object_new (NAUTILUS_TYPE_DESKTOP_ICON_FILE, NULL));

#ifdef NAUTILUS_FILE_DEBUG_REF
	printf("%10p ref'd\n", file);
	eazel_dump_stack_trace ("\t", 10);
#endif

	nautilus_file_set_directory (file, directory);

	icon_file = NAUTILUS_DESKTOP_ICON_FILE (file);
	icon_file->details->link = link;

	name = nautilus_desktop_link_get_file_name (link);
	file->details->name = eel_ref_str_new (name);
	g_free (name);

	update_info_from_link (icon_file);

	nautilus_desktop_update_metadata_from_keyfile (file, file->details->name);

	nautilus_directory_add_file (directory, file);

	list.data = file;
	list.next = NULL;
	list.prev = NULL;
	nautilus_directory_emit_files_added (directory, &list);

	return icon_file;
}

/* Note: This can return NULL if the link was recently removed (i.e. unmounted) */
NautilusDesktopLink *
nautilus_desktop_icon_file_get_link (NautilusDesktopIconFile *icon_file)
{
	if (icon_file->details->link)
		return g_object_ref (icon_file->details->link);
	else
		return NULL;
}

static void
nautilus_desktop_icon_file_unmount (NautilusFile                   *file,
				    GMountOperation                *mount_op,
				    GCancellable                   *cancellable,
				    NautilusFileOperationCallback   callback,
				    gpointer                        callback_data)
{
	NautilusDesktopIconFile *desktop_file;
	GMount *mount;

	desktop_file = NAUTILUS_DESKTOP_ICON_FILE (file);
	if (desktop_file) {
		mount = nautilus_desktop_link_get_mount (desktop_file->details->link);
		if (mount != NULL) {
			nautilus_file_operations_unmount_mount (NULL, mount, FALSE, TRUE);
		}
	}

}

static void
nautilus_desktop_icon_file_eject (NautilusFile                   *file,
				  GMountOperation                *mount_op,
				  GCancellable                   *cancellable,
				  NautilusFileOperationCallback   callback,
				  gpointer                        callback_data)
{
	NautilusDesktopIconFile *desktop_file;
	GMount *mount;

	desktop_file = NAUTILUS_DESKTOP_ICON_FILE (file);
	if (desktop_file) {
		mount = nautilus_desktop_link_get_mount (desktop_file->details->link);
		if (mount != NULL) {
			nautilus_file_operations_unmount_mount (NULL, mount, TRUE, TRUE);
		}
	}
}

static char*
real_get_target_uri (NautilusFile *file)
{
	char *uri = NULL;
	GFile *location;
	NautilusDesktopLink *link;

	g_return_val_if_fail (NAUTILUS_IS_DESKTOP_ICON_FILE (file), NULL);

	link = nautilus_desktop_icon_file_get_link (NAUTILUS_DESKTOP_ICON_FILE (file));

	if (link != NULL) {
		location = nautilus_desktop_link_get_activation_location (link);
		g_object_unref (link);
		if (location != NULL) {
			uri = g_file_get_uri (location);
			g_object_unref (location);

                        return uri;
		}
	}

        return NAUTILUS_FILE_CLASS (nautilus_desktop_icon_file_parent_class)->get_target_uri (file);
}

static void
real_rename (NautilusFile                  *file,
             const char                    *new_name,
             NautilusFileOperationCallback  callback,
             gpointer                       callback_data)
{
	NautilusDesktopLink *link;
	char *old_name;
	gboolean success;
	GError *error;

	g_return_if_fail (NAUTILUS_IS_FILE (file));
	g_return_if_fail (new_name != NULL);
	g_return_if_fail (callback != NULL);

	/* Can't rename a file that's already gone.
	 * We need to check this here because there may be a new
	 * file with the same name.
	 */
        if (nautilus_file_rename_handle_file_gone (file, callback, callback_data)) {
                return;
        }

	link = nautilus_desktop_icon_file_get_link (NAUTILUS_DESKTOP_ICON_FILE (file));
	old_name = nautilus_file_get_display_name (file);

	if ((old_name != NULL && strcmp (new_name, old_name) == 0)) {
		success = TRUE;
	} else {
		success = (link != NULL && nautilus_desktop_link_rename (link, new_name));
	}

	if (success) {
		(* callback) (file, NULL, NULL, callback_data);
	} else {
		error = g_error_new (G_IO_ERROR, G_IO_ERROR_FAILED,
				     _("Unable to rename desktop icon"));
		(* callback) (file, NULL, error, callback_data);
		g_error_free (error);
	}

	g_free (old_name);
	g_object_unref (link);

	return;
}

static gboolean
real_can_rename (NautilusFile *file)
{
        NautilusDesktopLink *link;
        gboolean can_rename;

        can_rename = NAUTILUS_FILE_CLASS (nautilus_desktop_icon_file_parent_class)->can_rename (file);

        if (!can_rename)
                return FALSE;

        link = nautilus_desktop_icon_file_get_link (NAUTILUS_DESKTOP_ICON_FILE (file));

        /* Certain types of links can't be renamed */
        if (link != NULL) {
                can_rename = nautilus_desktop_link_can_rename (link);
                g_object_unref (link);
        }

        return can_rename;
}

static gboolean
real_drag_can_accept_files (NautilusFile *drop_target_item)
{
        return TRUE;
}

static void
real_invalidate_attributes_internal (NautilusFile           *file,
                                     NautilusFileAttributes  attributes)
{
        /* Desktop icon files are always up to date.
         * If we invalidate their attributes they
         * will lose data, so we just ignore them.
         */
        return;
}

static gboolean
real_opens_in_view (NautilusFile *file)
{
        return TRUE;
}

static gboolean
real_is_special_link (NautilusFile *file)
{
        return TRUE;
}

static void
nautilus_desktop_icon_file_set_metadata (NautilusFile           *file,
					 const char             *key,
					 const char             *value)
{
	nautilus_desktop_set_metadata_string (file, file->details->name, key, value);
}

static void
nautilus_desktop_icon_file_set_metadata_as_list (NautilusFile           *file,
						 const char             *key,
						 char                  **value)
{
	nautilus_desktop_set_metadata_stringv (file, file->details->name, key, (const gchar **) value);
}

static void
nautilus_desktop_icon_file_class_init (NautilusDesktopIconFileClass *klass)
{
	GObjectClass *object_class;
	NautilusFileClass *file_class;

	object_class = G_OBJECT_CLASS (klass);
	file_class = NAUTILUS_FILE_CLASS (klass);

	file_class->default_file_type = G_FILE_TYPE_DIRECTORY;

	file_class->monitor_add = desktop_icon_file_monitor_add;
	file_class->monitor_remove = desktop_icon_file_monitor_remove;
	file_class->call_when_ready = desktop_icon_file_call_when_ready;
	file_class->cancel_call_when_ready = desktop_icon_file_cancel_call_when_ready;
	file_class->check_if_ready = desktop_icon_file_check_if_ready;
	file_class->get_item_count = desktop_icon_file_get_item_count;
	file_class->get_deep_counts = desktop_icon_file_get_deep_counts;
	file_class->get_date = desktop_icon_file_get_date;
	file_class->get_where_string = desktop_icon_file_get_where_string;
	file_class->set_metadata = nautilus_desktop_icon_file_set_metadata;
	file_class->set_metadata_as_list = nautilus_desktop_icon_file_set_metadata_as_list;
	file_class->unmount = nautilus_desktop_icon_file_unmount;
	file_class->eject = nautilus_desktop_icon_file_eject;
        file_class->can_rename = real_can_rename;
        file_class->rename = real_rename;
        file_class->get_target_uri = real_get_target_uri;
        file_class->drag_can_accept_files = real_drag_can_accept_files;
        file_class->invalidate_attributes_internal = real_invalidate_attributes_internal;
        file_class->opens_in_view = real_opens_in_view;
        file_class->is_special_link = real_is_special_link;

	g_type_class_add_private (object_class, sizeof(NautilusDesktopIconFileDetails));
}