/*

  * Copyright (C) 2010  J. Ali Harlow <ali@juiblex.co.uk>

  *

  * 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 02110-1301 USA.

  */

 #include "config.h"

 #include <stdlib.h>

 #include <string.h>

 #ifdef WIN32

 #include <windows.h>

 #endif

 #include <glib.h>

 #include <gio/gio.h>

 #include <gtk/gtk.h>

 #include <plover-gtk/packagestore.h>

 #include "app-manager.h"

 #include "localmedia.h"

 G_DEFINE_TYPE(PloverLocalMediaStore,plover_local_media_store,

   PLOVER_TYPE_PACKAGE_STORE);

 static void plover_local_media_store_dispose(GObject *obj)

 {

     PloverLocalMediaStore *store=PLOVER_LOCAL_MEDIA_STORE(obj);

     if (store->monitor)

     {

 	g_object_unref(store->monitor);

 	store->monitor=NULL;

     }

     if (G_OBJECT_CLASS(plover_local_media_store_parent_class)->dispose)

 	G_OBJECT_CLASS(plover_local_media_store_parent_class)->dispose(obj);

 }

 static void

   plover_local_media_store_class_init(PloverLocalMediaStoreClass *klass)

 {

     GObjectClass *oclass=G_OBJECT_CLASS(klass);

     oclass->dispose=plover_local_media_store_dispose;

 }

 GtkTreeModel *plover_local_media_store_new(void)

 {

     return g_object_new(PLOVER_TYPE_LOCAL_MEDIA_STORE,NULL);

 }

 static void local_media_scan_mount(PloverLocalMediaStore *store,GMount *mount)

 {

     GFile *root;

     gchar *path;

 #ifdef WIN32

     gunichar2 *path2;

     UINT type;

 #endif

     PloverPackageSet *set;

     root=g_mount_get_root(mount);

     path=g_file_get_path(root);

 #ifdef WIN32

     if (path)

     {

 	path2=g_utf8_to_utf16(path,-1,NULL,NULL,NULL);

 	if (path2)

 	{

 	    type=GetDriveTypeW(path2);

 	    g_free(path2);

 	}

 	else

 	    type=DRIVE_UNKNOWN;

 	if (type!=DRIVE_REMOVABLE && type!=DRIVE_CDROM)

 	{

 	    gchar *name=g_mount_get_name(mount);

 	    g_debug("Skipping non-local mount \"%s\"",name);

 	    g_free(name);

 	    g_free(path);

 	    path=NULL;

 	}

     }

 #endif

     if (path)

     {

 	set=plover_package_set_new_from_yum(path,relocations,NULL);

 	if (set)

 	{

 	    g_object_set_data(G_OBJECT(mount),"plover-local-media-set",set);

 	    plover_package_store_add_set(PLOVER_PACKAGE_STORE(store),set);

 	    g_object_ref(mount);

 	}

 	g_free(path);

     }

     g_object_unref(root);

 }

 static void local_media_mounted(GObject *source,GAsyncResult *res,gpointer data)

 {

     GVolume *volume=G_VOLUME(source);

     GMount *mount;

     PloverLocalMediaStore *store=PLOVER_LOCAL_MEDIA_STORE(data);

     if (g_volume_mount_finish(volume,res,NULL))

     {

 	mount=g_volume_get_mount(volume);

 	if (mount)

 	{

 	    local_media_scan_mount(store,mount);

 	    g_object_unref(mount);

 	}

     }

     g_object_unref(volume);

 }

 static void local_media_scan_drive(PloverLocalMediaStore *store,GDrive *drive)

 {

     GVolume *volume;

     GMount *mount;

     GList *volumes,*mounts,*link;

     if (!drive)

 	volumes=g_volume_monitor_get_volumes(store->monitor);

     else if (g_drive_has_media(drive))

 	volumes=g_drive_get_volumes(drive);

     else

 	volumes=NULL;

     for(link=volumes;link;link=link->next)

     {

 	volume=G_VOLUME(link->data);

 	mount=g_volume_get_mount(volume);

 	if (mount)

 	{

 	    local_media_scan_mount(store,mount);

 	    g_object_unref(mount);

 	}

 	else if (!store->implicit_scan && g_volume_can_mount(volume))

 	    g_volume_mount(volume,G_MOUNT_MOUNT_NONE,NULL,NULL,

 	      local_media_mounted,store);

 	g_object_unref(volume);

     }

     g_list_free(volumes);

     if (!drive)

     {

 	mounts=g_volume_monitor_get_mounts(store->monitor);

 	for(link=mounts;link;link=link->next)

 	{

 	    mount=G_MOUNT(link->data);

 	    local_media_scan_mount(store,mount);

 	    g_object_unref(mount);

 	}

 	g_list_free(mounts);

     }

 }

 static void local_media_polled(GObject *source,GAsyncResult *res,gpointer data)

 {

     GDrive *drive=G_DRIVE(source);

     PloverLocalMediaStore *store=PLOVER_LOCAL_MEDIA_STORE(data);

     if (g_drive_poll_for_media_finish(drive,res,NULL))

 	local_media_scan_drive(store,drive);

     g_object_unref(drive);

 }

 void plover_local_media_store_scan(PloverLocalMediaStore *store)

 {

     GSList *sets,*slink;

     GList *drives,*link;

     GDrive *drive;

     g_return_if_fail(PLOVER_IS_LOCAL_MEDIA_STORE(store));

     sets=

       g_slist_copy(plover_package_store_get_sets(PLOVER_PACKAGE_STORE(store)));

     for(slink=sets;slink;slink=slink->next)

 	plover_package_store_remove_set(PLOVER_PACKAGE_STORE(store),

 	  PLOVER_PACKAGE_SET(slink->data));

     g_slist_free(sets);

     drives=g_volume_monitor_get_connected_drives(store->monitor);

     for(link=drives;link;link=link->next)

     {

 	drive=G_DRIVE(link->data);

 	if (g_drive_is_media_removable(drive))

 	{

 	    if (!g_drive_is_media_check_automatic(drive) &&

 	      g_drive_can_poll_for_media(drive) ||

 	      !g_drive_has_media(drive) && !store->implicit_scan)

 	    {

 		g_object_ref(drive);

 		g_drive_poll_for_media(drive,NULL,local_media_polled,store);

 	    }

 	    else

 		local_media_scan_drive(store,drive);

 	}

 	g_object_unref(drive);

     }

     g_list_free(drives);

     local_media_scan_drive(store,NULL);

 }

 static void local_media_mount_added(GVolumeMonitor *volume_monitor,

   GMount *mount,PloverLocalMediaStore *store)

 {

     local_media_scan_mount(store,mount);

 }

 static void local_media_mount_removed(GVolumeMonitor *volume_monitor,

   GMount *mount,PloverLocalMediaStore *store)

 {

     PloverPackageSet *set=

       g_object_get_data(G_OBJECT(mount),"plover-local-media-set");

     if (set)

     {

 	plover_package_store_remove_set(PLOVER_PACKAGE_STORE(store),set);

 	g_object_set_data(G_OBJECT(mount),"plover-local-media-set",NULL);

 	g_object_unref(set);

 	g_object_unref(mount);

     }

 }

 static void local_media_mount_changed(GVolumeMonitor *volume_monitor,

   GMount *mount,PloverLocalMediaStore *store)

 {

     local_media_mount_removed(volume_monitor,mount,store);

     local_media_mount_added(volume_monitor,mount,store);

 }

 static void plover_local_media_store_init(PloverLocalMediaStore *store)

 {

     store->monitor=g_volume_monitor_get();

     store->implicit_scan=TRUE;

     plover_local_media_store_scan(store);

     store->implicit_scan=FALSE;

     g_signal_connect(store->monitor,"mount-added",

       G_CALLBACK(local_media_mount_added),store);

     g_signal_connect(store->monitor,"mount-changed",

       G_CALLBACK(local_media_mount_changed),store);

     g_signal_connect(store->monitor,"mount-removed",

       G_CALLBACK(local_media_mount_removed),store);

 }