#include <unistd.h>
#include <glib.h>
#include <dbus/dbus-glib.h>
#include <polkit-gnome/polkit-gnome.h>

#include "logout-dialog.h"
#include "ck-pk-helper.h"

static gboolean
ck_multiple_users (void)
{
	DBusGConnection * sbus = dbus_g_bus_get(DBUS_BUS_SYSTEM, NULL);
	g_return_val_if_fail(sbus != NULL, TRUE); /* worst case */
	DBusGProxy * proxy = dbus_g_proxy_new_for_name(sbus, "org.freedesktop.ConsoleKit",
	                                                     "/org/freedesktop/ConsoleKit/Manager",
	                                                     "org.freedesktop.ConsoleKit.Manager");

	if (proxy == NULL) {
		return TRUE;
	}

	gboolean result;
	GPtrArray * seats = NULL;

	result = dbus_g_proxy_call(proxy, "GetSeats", NULL, G_TYPE_INVALID,
	                           dbus_g_type_get_collection("GPtrArray", DBUS_TYPE_G_OBJECT_PATH), &seats, G_TYPE_INVALID);

	if (!result) {
		g_warning("Unable to get the seats for ConsoleKit");
		g_object_unref(proxy);
		return TRUE;
	}
	
	gchar * this_session_id = NULL;
	
	result = dbus_g_proxy_call(proxy, "GetCurrentSession", NULL, G_TYPE_INVALID,
							DBUS_TYPE_G_OBJECT_PATH, &this_session_id, G_TYPE_INVALID);
							
	g_object_unref(proxy);
							
	if (!result) {
		g_warning("Unable to get current session from ConsoleKit");
		return TRUE;
	}
	
	proxy = dbus_g_proxy_new_for_name(sbus, "org.freedesktop.ConsoleKit",
									this_session_id, "org.freedesktop.ConsoleKit.Session");

	if (proxy == NULL) {
		return TRUE;
	}
	
	guint this_session_uid;
	
	result = dbus_g_proxy_call(proxy, "GetUnixUser", NULL, G_TYPE_INVALID,
							G_TYPE_UINT, &this_session_uid, G_TYPE_INVALID);										

	if (!result) {
		g_warning("Unable to get UID from ConsoleKit");
		return TRUE;
	}
	
	guint seat;
	gboolean multiple_users = FALSE;
	for (seat = 0; seat < seats->len; seat++) {
		gchar * seat_id = g_ptr_array_index(seats, seat);
		DBusGProxy * seat_proxy = dbus_g_proxy_new_for_name(sbus, "org.freedesktop.ConsoleKit",
		                                                    seat_id, "org.freedesktop.ConsoleKit.Seat");
		g_free(seat_id);

		if (seat_proxy == NULL) {
			continue;
		}

		GPtrArray * sessions = NULL;

		gboolean result = dbus_g_proxy_call(seat_proxy,
		                                    "GetSessions", NULL, G_TYPE_INVALID,
	                                        dbus_g_type_get_collection("GPtrArray", DBUS_TYPE_G_OBJECT_PATH), &sessions, G_TYPE_INVALID);

		g_object_unref(seat_proxy);
		if (!result) {
			continue;
		}

		guint session;
		for (session = 0; session < sessions->len; session++) {
			gchar * session_id = g_ptr_array_index(sessions, session);
			if (g_strcmp0(this_session_id, session_id) == 0) {
				continue;
			}
			DBusGProxy * session_proxy = dbus_g_proxy_new_for_name(sbus, "org.freedesktop.ConsoleKit",
		                                                    	session_id, "org.freedesktop.ConsoleKit.Session");
		    g_free(session_id);
		    
		    if (session_proxy == NULL) {
		    	continue;
		    }
		    
		    guint session_uid;
		    result =  dbus_g_proxy_call(session_proxy, "GetUnixUser", NULL, G_TYPE_INVALID,
		    							G_TYPE_UINT, &session_uid, G_TYPE_INVALID);	
		    g_object_unref(session_proxy);
		    							
		    if (!result) {
		    	continue;
		    }
		    
		    if (session_uid != this_session_uid) {
		    	multiple_users = TRUE;
		    	break;
		    }
		}
		
		g_ptr_array_free(sessions, TRUE);
		
		if (multiple_users) {
			break;
		}
	}

	g_ptr_array_free(seats, TRUE);
	g_object_unref(proxy);
	g_free(this_session_id);

	return multiple_users;
}

gboolean
pk_require_auth (LogoutDialogAction action) {
	if (action == LOGOUT_DIALOG_LOGOUT) {
		return FALSE;
	}

	gchar * pk_action;
	if (ck_multiple_users()) {
		if (action == LOGOUT_DIALOG_RESTART) {
			pk_action = "org.freedesktop.consolekit.system.restart-multiple-users";
		} else {
			pk_action = "org.freedesktop.consolekit.system.stop-multiple-users";
		}
	} else {
		if (action == LOGOUT_DIALOG_RESTART) {
			pk_action = "org.freedesktop.consolekit.system.restart";
		} else {
			pk_action = "org.freedesktop.consolekit.system.stop";
		}
	}

	PolKitResult polres;
	if (pk_can_do_action(pk_action, &polres)) {
		if (polres == POLKIT_RESULT_YES) {
			return FALSE;
		}
		return TRUE;
	}
	return FALSE;
}

gboolean
pk_can_do_action (const gchar    *action_id, PolKitResult * pol_result)
{
        PolKitGnomeContext *gnome_context;
        PolKitAction *action;
        PolKitCaller *caller;
        DBusError dbus_error;
        PolKitError *error;
        PolKitResult result;

        gnome_context = polkit_gnome_context_get (NULL);

        if (gnome_context == NULL) {
                return FALSE;
        }

        if (gnome_context->pk_tracker == NULL) {
                return FALSE;
        }

        dbus_error_init (&dbus_error);
        caller = polkit_tracker_get_caller_from_pid (gnome_context->pk_tracker,
                                                     getpid (),
                                                     &dbus_error);
        dbus_error_free (&dbus_error);

        if (caller == NULL) {
                return FALSE;
        }

        action = polkit_action_new ();
        if (!polkit_action_set_action_id (action, action_id)) {
                polkit_action_unref (action);
                polkit_caller_unref (caller);
                return FALSE;
        }

        result = POLKIT_RESULT_UNKNOWN;
        error = NULL;
        result = polkit_context_is_caller_authorized (gnome_context->pk_context,
                                                      action, caller, FALSE,
                                                      &error);
        if (polkit_error_is_set (error)) {
                polkit_error_free (error);
        }
        polkit_action_unref (action);
                polkit_caller_unref (caller);

		if (pol_result != NULL) {
			*pol_result = result;
		}

        return result != POLKIT_RESULT_NO && result != POLKIT_RESULT_UNKNOWN;
}