/*
 *  Window Maker window manager
 * 
 *  Copyright (c) 1997-2002 Alfredo K. Kojima
 * 
 *  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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, 
 *  USA.
 */

#include "wconfig.h"

#include <stdlib.h>
#include <signal.h>
#include <unistd.h>

#include <X11/Xlib.h>
#include <X11/Xutil.h>
#include <X11/Intrinsic.h>

#include "WindowMaker.h"
#include "window.h"
#include "client.h"
#include "funcs.h"
#include "properties.h"
#include "session.h"
#include "winspector.h"
#ifdef KWM_HINTS
# include "kwm.h"
#endif
#ifdef OLWM_HINTS
# include "openlook.h"
#endif

extern Atom _XA_WM_DELETE_WINDOW;
extern Time LastTimestamp;
extern int wScreenCount;


static void wipeDesktop(WScreen *scr);


/*
 *----------------------------------------------------------------------
 * Shutdown-
 * 	Exits the window manager cleanly. If mode is WSLogoutMode,
 * the whole X session will be closed, by killing all clients if
 * no session manager is running or by asking a shutdown to
 * it if its present.
 * 
 *----------------------------------------------------------------------
 */
void
Shutdown(WShutdownMode mode)
{
    int i;

    switch (mode) {
     case WSLogoutMode:
#ifdef XSMP_ENABLED
	wSessionRequestShutdown();
	break;
#else
	/* fall through */
#endif
     case WSKillMode:
     case WSExitMode:
	/* if there is no session manager, send SAVE_YOURSELF to
	 * the clients */
#if 0
#ifdef XSMP_ENABLED
	if (!wSessionIsManaged())
#endif
	    for (i = 0; i < wScreenCount; i++) {
		WScreen *scr;

		scr = wScreenWithNumber(i);
		if (scr) {
		    wSessionSendSaveYourself(scr);
		}
	    }
#endif
	for (i = 0; i < wScreenCount; i++) {
	    WScreen *scr;

	    scr = wScreenWithNumber(i);
	    if (scr) {
		if (scr->helper_pid)
		    kill(scr->helper_pid, SIGKILL);

		/* if the session is not being managed, save restart info */
#ifdef XSMP_ENABLED
		if (!wSessionIsManaged())
#endif
		    wSessionSaveClients(scr);

#ifdef KWM_HINTS
		wKWMShutdown(scr, True);
#endif
		wScreenSaveState(scr);

		if (mode == WSKillMode)
		    wipeDesktop(scr);
		else
		    RestoreDesktop(scr);
	    }
	}
	ExecExitScript();
	Exit(0);
	break;

     case WSRestartPreparationMode:
	for (i=0; i<wScreenCount; i++) {
	    WScreen *scr;

	    scr = wScreenWithNumber(i);
	    if (scr) {
		if (scr->helper_pid)
		    kill(scr->helper_pid, SIGKILL);
#ifdef KWM_HINTS
		wKWMShutdown(scr, False);
#endif
#ifdef OLWM_HINTS
		wOLWMShutdown(scr);
#endif
		wScreenSaveState(scr);

		RestoreDesktop(scr);
	    }
	}
	break;
    }
}


static void
restoreWindows(WMBag *bag, WMBagIterator iter)
{
    WCoreWindow *next;
    WCoreWindow *core;
    WWindow *wwin;


    if (iter == NULL) {
	core = WMBagFirst(bag, &iter);
    } else {
	core = WMBagNext(bag, &iter);
    }

    if (core == NULL)
	return;

    restoreWindows(bag, iter);

    /* go to the end of the list */
    while (core->stacking->under)
	core = core->stacking->under;

    while (core) {
	next = core->stacking->above;

	if (core->descriptor.parent_type==WCLASS_WINDOW) {
	    Window window;

	    wwin = core->descriptor.parent;
	    window = wwin->client_win;
	    wUnmanageWindow(wwin, !wwin->flags.internal_window, False);
	    XMapWindow(dpy, window);
	}
	core = next;
    }
}


/*
 *----------------------------------------------------------------------
 * RestoreDesktop--
 * 	Puts the desktop in a usable state when exiting.
 *
 * Side effects:
 * 	All frame windows are removed and windows are reparented
 * back to root. Windows that are outside the screen are 
 * brought to a viable place. 
 * 
 *---------------------------------------------------------------------- 
 */
void
RestoreDesktop(WScreen *scr)
{
    if (scr->helper_pid > 0) {
	kill(scr->helper_pid, SIGTERM);
	scr->helper_pid = 0;
    }

    XGrabServer(dpy);
    wDestroyInspectorPanels();

    /* reparent windows back to the root window, keeping the stacking order */
    restoreWindows(scr->stacking_list, NULL);

    XUngrabServer(dpy);
    XSetInputFocus(dpy, PointerRoot, RevertToParent, CurrentTime);
    wColormapInstallForWindow(scr, NULL);
    PropCleanUp(scr->root_win);
    XSync(dpy, 0);
}


/*
 *----------------------------------------------------------------------
 * wipeDesktop--
 * 	Kills all windows in a screen. Send DeleteWindow to all windows
 * that support it and KillClient on all windows that don't.
 * 
 * Side effects:
 * 	All managed windows are closed.
 * 
 * TODO: change to XQueryTree()
 *---------------------------------------------------------------------- 
 */
static void
wipeDesktop(WScreen *scr)
{
    WWindow *wwin;

    wwin = scr->focused_window;
    while (wwin) {
	if (wwin->protocols.DELETE_WINDOW)
	    wClientSendProtocol(wwin, _XA_WM_DELETE_WINDOW, LastTimestamp);
	else
	    wClientKill(wwin);
	wwin = wwin->prev;
    }
    XSync(dpy, False);
}