/**************************************************************************
*
* Tint2 panel
*
* Copyright (C) 2007 Pål Staurland (staura@gmail.com)
* Modified (C) 2008 thierry lorthiois (lorthiois@bbsoft.fr)
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License version 2
* as published by the Free Software Foundation.
*
* 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 <X11/extensions/Xrender.h>
#include <X11/extensions/Xrandr.h>


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

#include "server.h"
#include "config.h"
#include "task.h"
#include "window.h"

void server_catch_error (Display *d, XErrorEvent *ev){}

void server_init_atoms ()
{
	server.atom._XROOTPMAP_ID = XInternAtom (server.dsp, "_XROOTPMAP_ID", False);
	server.atom._XROOTMAP_ID = XInternAtom (server.dsp, "_XROOTMAP_ID", False);
	server.atom._NET_CURRENT_DESKTOP = XInternAtom (server.dsp, "_NET_CURRENT_DESKTOP", False);
	server.atom._NET_NUMBER_OF_DESKTOPS = XInternAtom (server.dsp, "_NET_NUMBER_OF_DESKTOPS", False);
	server.atom._NET_DESKTOP_NAMES = XInternAtom (server.dsp, "_NET_DESKTOP_NAMES", False);
	server.atom._NET_DESKTOP_GEOMETRY = XInternAtom (server.dsp, "_NET_DESKTOP_GEOMETRY", False);
	server.atom._NET_DESKTOP_VIEWPORT = XInternAtom (server.dsp, "_NET_DESKTOP_VIEWPORT", False);
	server.atom._NET_ACTIVE_WINDOW = XInternAtom (server.dsp, "_NET_ACTIVE_WINDOW", False);
	server.atom._NET_WM_WINDOW_TYPE = XInternAtom (server.dsp, "_NET_WM_WINDOW_TYPE", False);
	server.atom._NET_WM_STATE_SKIP_PAGER = XInternAtom (server.dsp, "_NET_WM_STATE_SKIP_PAGER", False);
	server.atom._NET_WM_STATE_SKIP_TASKBAR = XInternAtom (server.dsp, "_NET_WM_STATE_SKIP_TASKBAR", False);
	server.atom._NET_WM_STATE_STICKY = XInternAtom (server.dsp, "_NET_WM_STATE_STICKY", False);
	server.atom._NET_WM_STATE_DEMANDS_ATTENTION = XInternAtom (server.dsp, "_NET_WM_STATE_DEMANDS_ATTENTION", False);
	server.atom._NET_WM_WINDOW_TYPE_DOCK = XInternAtom (server.dsp, "_NET_WM_WINDOW_TYPE_DOCK", False);
	server.atom._NET_WM_WINDOW_TYPE_DESKTOP = XInternAtom (server.dsp, "_NET_WM_WINDOW_TYPE_DESKTOP", False);
	server.atom._NET_WM_WINDOW_TYPE_TOOLBAR = XInternAtom (server.dsp, "_NET_WM_WINDOW_TYPE_TOOLBAR", False);
	server.atom._NET_WM_WINDOW_TYPE_MENU = XInternAtom (server.dsp, "_NET_WM_WINDOW_TYPE_MENU", False);
	server.atom._NET_WM_WINDOW_TYPE_SPLASH = XInternAtom (server.dsp, "_NET_WM_WINDOW_TYPE_SPLASH", False);
	server.atom._NET_WM_WINDOW_TYPE_DIALOG = XInternAtom (server.dsp, "_NET_WM_WINDOW_TYPE_DIALOG", False);
	server.atom._NET_WM_WINDOW_TYPE_NORMAL = XInternAtom (server.dsp, "_NET_WM_WINDOW_TYPE_NORMAL", False);
	server.atom._NET_WM_DESKTOP = XInternAtom (server.dsp, "_NET_WM_DESKTOP", False);
	server.atom.WM_STATE = XInternAtom (server.dsp, "WM_STATE", False);
	server.atom._NET_WM_STATE = XInternAtom (server.dsp, "_NET_WM_STATE", False);
	server.atom._NET_WM_STATE_MAXIMIZED_VERT = XInternAtom (server.dsp, "_NET_WM_STATE_MAXIMIZED_VERT", False);
	server.atom._NET_WM_STATE_MAXIMIZED_HORZ = XInternAtom (server.dsp, "_NET_WM_STATE_MAXIMIZED_HORZ", False);
	server.atom._NET_WM_STATE_SHADED = XInternAtom (server.dsp, "_NET_WM_STATE_SHADED", False);
	server.atom._NET_WM_STATE_HIDDEN = XInternAtom (server.dsp, "_NET_WM_STATE_HIDDEN", False);
	server.atom._NET_WM_STATE_BELOW = XInternAtom (server.dsp, "_NET_WM_STATE_BELOW", False);
	server.atom._NET_WM_STATE_ABOVE = XInternAtom (server.dsp, "_NET_WM_STATE_ABOVE", False);
	server.atom._NET_WM_STATE_MODAL = XInternAtom (server.dsp, "_NET_WM_STATE_MODAL", False);
	server.atom._NET_CLIENT_LIST = XInternAtom (server.dsp, "_NET_CLIENT_LIST", False);
	server.atom._NET_WM_VISIBLE_NAME = XInternAtom (server.dsp, "_NET_WM_VISIBLE_NAME", False);
	server.atom._NET_WM_NAME = XInternAtom (server.dsp, "_NET_WM_NAME", False);
	server.atom._NET_WM_STRUT = XInternAtom (server.dsp, "_NET_WM_STRUT", False);
	server.atom._NET_WM_ICON = XInternAtom (server.dsp, "_NET_WM_ICON", False);
	server.atom._NET_WM_ICON_GEOMETRY = XInternAtom(server.dsp, "_NET_WM_ICON_GEOMETRY", False );
	server.atom._NET_CLOSE_WINDOW = XInternAtom (server.dsp, "_NET_CLOSE_WINDOW", False);
	server.atom.UTF8_STRING = XInternAtom (server.dsp, "UTF8_STRING", False);
	server.atom._NET_SUPPORTING_WM_CHECK = XInternAtom (server.dsp, "_NET_SUPPORTING_WM_CHECK", False);
	server.atom._NET_WM_CM_S0 = XInternAtom (server.dsp, "_NET_WM_CM_S0", False);
	server.atom._NET_SUPPORTING_WM_CHECK = XInternAtom (server.dsp, "_NET_WM_NAME", False);
	server.atom._NET_WM_STRUT_PARTIAL = XInternAtom (server.dsp, "_NET_WM_STRUT_PARTIAL", False);
	server.atom.WM_NAME = XInternAtom(server.dsp, "WM_NAME", False);
	server.atom.__SWM_VROOT = XInternAtom(server.dsp, "__SWM_VROOT", False);
	server.atom._MOTIF_WM_HINTS = XInternAtom(server.dsp, "_MOTIF_WM_HINTS", False);
	server.atom.WM_HINTS = XInternAtom(server.dsp, "WM_HINTS", False);
	char *name = g_strdup_printf("_XSETTINGS_S%d", DefaultScreen(server.dsp));
	server.atom._XSETTINGS_SCREEN = XInternAtom(server.dsp, name, False);
	g_free(name);
	server.atom._XSETTINGS_SETTINGS = XInternAtom(server.dsp, "_XSETTINGS_SETTINGS", False);

	// systray protocol
	name = g_strdup_printf("_NET_SYSTEM_TRAY_S%d", DefaultScreen(server.dsp));
	server.atom._NET_SYSTEM_TRAY_SCREEN = XInternAtom(server.dsp, name, False);
	g_free(name);
	server.atom._NET_SYSTEM_TRAY_OPCODE = XInternAtom(server.dsp, "_NET_SYSTEM_TRAY_OPCODE", False);
	server.atom.MANAGER = XInternAtom(server.dsp, "MANAGER", False);
	server.atom._NET_SYSTEM_TRAY_MESSAGE_DATA = XInternAtom(server.dsp, "_NET_SYSTEM_TRAY_MESSAGE_DATA", False);
	server.atom._NET_SYSTEM_TRAY_ORIENTATION = XInternAtom(server.dsp, "_NET_SYSTEM_TRAY_ORIENTATION", False);
	server.atom._XEMBED = XInternAtom(server.dsp, "_XEMBED", False);
	server.atom._XEMBED_INFO = XInternAtom(server.dsp, "_XEMBED_INFO", False);

	// drag 'n' drop
	server.atom.XdndAware = XInternAtom(server.dsp, "XdndAware", False);
	server.atom.XdndEnter = XInternAtom(server.dsp, "XdndEnter", False);
	server.atom.XdndPosition = XInternAtom(server.dsp, "XdndPosition", False);
	server.atom.XdndStatus = XInternAtom(server.dsp, "XdndStatus", False);
	server.atom.XdndDrop = XInternAtom(server.dsp, "XdndDrop", False);
	server.atom.XdndLeave = XInternAtom(server.dsp, "XdndLeave", False);
	server.atom.XdndSelection = XInternAtom(server.dsp, "XdndSelection", False);
	server.atom.XdndTypeList = XInternAtom(server.dsp, "XdndTypeList", False);
	server.atom.XdndActionCopy = XInternAtom(server.dsp, "XdndActionCopy", False);
	server.atom.XdndFinished = XInternAtom(server.dsp, "XdndFinished", False);
	server.atom.TARGETS = XInternAtom(server.dsp, "TARGETS", False);
}


void cleanup_server()
{
	if (server.colormap) XFreeColormap(server.dsp, server.colormap);
	if (server.colormap32) XFreeColormap(server.dsp, server.colormap32);
	if (server.monitor) {
		int i;
		for (i=0; i<server.nb_monitor; ++i)
			if (server.monitor[i].names)
				g_strfreev(server.monitor[i].names);
		free(server.monitor);
	}
	if (server.gc) XFreeGC(server.dsp, server.gc);
}


void send_event32 (Window win, Atom at, long data1, long data2, long data3)
{
	XEvent event;

	event.xclient.type = ClientMessage;
	event.xclient.serial = 0;
	event.xclient.send_event = True;
	event.xclient.display = server.dsp;
	event.xclient.window = win;
	event.xclient.message_type = at;

	event.xclient.format = 32;
	event.xclient.data.l[0] = data1;
	event.xclient.data.l[1] = data2;
	event.xclient.data.l[2] = data3;
	event.xclient.data.l[3] = 0;
	event.xclient.data.l[4] = 0;

	XSendEvent(server.dsp, server.root_win, False, SubstructureRedirectMask|SubstructureNotifyMask, &event);
}


int get_property32 (Window win, Atom at, Atom type)
{
	Atom type_ret;
	int format_ret = 0, data = 0;
	unsigned long nitems_ret = 0;
	unsigned long bafter_ret = 0;
	unsigned char *prop_value = 0;
	int result;

	if (!win) return 0;

	result = XGetWindowProperty(server.dsp, win, at, 0, 0x7fffffff, False, type, &type_ret, &format_ret, &nitems_ret, &bafter_ret, &prop_value);

	if (result == Success && prop_value) {
		data = ((gulong*)prop_value)[0];
		XFree (prop_value);
	}
	return data;
}


void *server_get_property (Window win, Atom at, Atom type, int *num_results)
{
	Atom type_ret;
	int format_ret = 0;
	unsigned long nitems_ret = 0;
	unsigned long bafter_ret = 0;
	unsigned char *prop_value;
	int result;

	if (!win) return 0;

	result = XGetWindowProperty(server.dsp, win, at, 0, 0x7fffffff, False, type, &type_ret, &format_ret, &nitems_ret, &bafter_ret, &prop_value);

	// Send back resultcount
	if (num_results) *num_results = (int)nitems_ret;

	if (result == Success && prop_value) return prop_value;
	else return 0;
}


void get_root_pixmap()
{
	Pixmap ret = None;

	unsigned long *res;
	Atom pixmap_atoms[] = { server.atom._XROOTPMAP_ID, server.atom._XROOTMAP_ID };
	int i;

	for (i=0; i<sizeof(pixmap_atoms)/sizeof(Atom); ++i) {
		res = server_get_property (server.root_win, pixmap_atoms[i], XA_PIXMAP, 0);
		if (res) {
			ret = *((Pixmap*)res);
			XFree(res);
			break;
		}
	}
	server.root_pmap = ret;

	if (server.root_pmap == None)
		fprintf(stderr, "tint2 : pixmap background detection failed\n");
	else {
		XGCValues  gcv;
		gcv.ts_x_origin = 0;
		gcv.ts_y_origin = 0;
		gcv.fill_style = FillTiled;
		uint mask = GCTileStipXOrigin | GCTileStipYOrigin | GCFillStyle | GCTile;

		gcv.tile = server.root_pmap;
		XChangeGC(server.dsp, server.gc, mask, &gcv);
	}
}


int compareMonitorPos(const void *monitor1, const void *monitor2)
{
	Monitor *m1 = (Monitor*)monitor1;
	Monitor *m2 = (Monitor*)monitor2;

	if (m1->x < m2->x) {
		return -1;
	}
	else if (m1->x > m2->x) {
		return 1;
	}
	else if (m1->y < m2->y) {
		return -1;
	}
	else if (m1->y > m2->y) {
		return 1;
	}
	else {
		return 0;
	}
}


int compareMonitorIncluded(const void *monitor1, const void *monitor2)
{
	Monitor *m1 = (Monitor*)monitor1;
	Monitor *m2 = (Monitor*)monitor2;

	if (m1->x >= m2->x && m1->y >= m2->y && (m1->x+m1->width) <= (m2->x+m2->width) && (m1->y+m1->height) <= (m2->y+m2->height)) {
		// m1 included inside m2
		return 1;
	}
	else {
		return -1;
	}
}


void get_monitors()
{
	int i, j, nbmonitor;
	if (XineramaIsActive(server.dsp)) {
		XineramaScreenInfo *info = XineramaQueryScreens(server.dsp, &nbmonitor);
		XRRScreenResources *res = XRRGetScreenResourcesCurrent(server.dsp, server.root_win);

		if (res && res->ncrtc >= nbmonitor) {
			// use xrandr to identify monitors (does not work with proprietery nvidia drivers)

			// Workaround for issue https://code.google.com/p/tint2/issues/detail?id=353
			// on some recent configs, XRRGetScreenResourcesCurrent returns a fantom monitor at last position
			{
				int i = res->ncrtc - 1;
				XRRCrtcInfo* crtc_info = XRRGetCrtcInfo(server.dsp, res, res->crtcs[i]);
				if (!(crtc_info->x || crtc_info->y || crtc_info->width || crtc_info->height)) {
					res->ncrtc -= 1;
				}
				XRRFreeCrtcInfo(crtc_info);
			}

			printf("xRandr: Found crtc's: %d\n", res->ncrtc );
			server.monitor = malloc(res->ncrtc * sizeof(Monitor));
			for (i=0; i<res->ncrtc; ++i) {
				XRRCrtcInfo* crtc_info = XRRGetCrtcInfo(server.dsp, res, res->crtcs[i]);
				server.monitor[i].x = crtc_info->x;
				server.monitor[i].y = crtc_info->y;
				server.monitor[i].width = crtc_info->width;
				server.monitor[i].height = crtc_info->height;
				server.monitor[i].names = malloc((crtc_info->noutput+1) * sizeof(char*));
				for (j=0; j<crtc_info->noutput; ++j) {
					XRROutputInfo* output_info = XRRGetOutputInfo(server.dsp, res, crtc_info->outputs[j]);
					printf("xRandr: Linking output %s with crtc %d\n", output_info->name, i);
					server.monitor[i].names[j] = g_strdup(output_info->name);
					XRRFreeOutputInfo(output_info);
				}
				server.monitor[i].names[j] = 0;
				XRRFreeCrtcInfo(crtc_info);
			}
			nbmonitor = res->ncrtc;
		}
		else if (info && nbmonitor > 0) {
			server.monitor = malloc(nbmonitor * sizeof(Monitor));
			for (i=0 ; i < nbmonitor ; i++) {
				server.monitor[i].x = info[i].x_org;
				server.monitor[i].y = info[i].y_org;
				server.monitor[i].width = info[i].width;
				server.monitor[i].height = info[i].height;
				server.monitor[i].names = 0;
			}
		}

		// ordered monitor
		qsort(server.monitor, nbmonitor, sizeof(Monitor), compareMonitorIncluded);

		// remove monitor included into another one
		i = 0;
		while (i < nbmonitor) {
			for (j=0; j < i ; j++) {
				if (compareMonitorIncluded(&server.monitor[i], &server.monitor[j]) > 0) {
					goto next;
				}
			}
			i++;
		}
next:
		for (j=i; j<nbmonitor; ++j)
			if (server.monitor[j].names)
				g_strfreev(server.monitor[j].names);
		server.nb_monitor = i;
		server.monitor = realloc(server.monitor, server.nb_monitor * sizeof(Monitor));
		qsort(server.monitor, server.nb_monitor, sizeof(Monitor), compareMonitorPos);

		if (res)
			XRRFreeScreenResources(res);
		XFree(info);
	}

	if (!server.nb_monitor) {
		server.nb_monitor = 1;
		server.monitor = malloc(sizeof(Monitor));
		server.monitor[0].x = server.monitor[0].y = 0;
		server.monitor[0].width = DisplayWidth (server.dsp, server.screen);
		server.monitor[0].height = DisplayHeight (server.dsp, server.screen);
		server.monitor[0].names = 0;
	}
}


void get_desktops()
{
	int i;

	// detect number of desktops
	// wait 15s to leave some time for window manager startup
	for (i=0 ; i < 15 ; i++) {
		server.nb_desktop = server_get_number_of_desktop ();
		if (server.nb_desktop > 0) break;
		sleep(1);
	}
	if (server.nb_desktop == 0) {
		server.nb_desktop = 1;
		fprintf(stderr, "warning : WM doesn't respect NETWM specs. tint2 default to 1 desktop.\n");
	}
}


void server_init_visual()
{
	// inspired by freedesktops fdclock ;)
	XVisualInfo *xvi;
	XVisualInfo templ = { .screen=server.screen, .depth=32, .class=TrueColor };
	int nvi;
	xvi = XGetVisualInfo(server.dsp, VisualScreenMask|VisualDepthMask|VisualClassMask, &templ, &nvi);

	Visual *visual = 0;
	if (xvi) {
		int i;
		XRenderPictFormat *format;
		for (i = 0; i < nvi; i++) {
			format = XRenderFindVisualFormat(server.dsp, xvi[i].visual);
			if (format->type == PictTypeDirect && format->direct.alphaMask) {
				visual = xvi[i].visual;
				break;
			}
		}
	}
	XFree (xvi);

	// check composite manager
	server.composite_manager = XGetSelectionOwner(server.dsp, server.atom._NET_WM_CM_S0);
	if (server.colormap)
		XFreeColormap(server.dsp, server.colormap);
	if (server.colormap32)
		XFreeColormap(server.dsp, server.colormap32);

	if (visual) {
		server.visual32 = visual;
		server.colormap32 = XCreateColormap(server.dsp, server.root_win, visual, AllocNone);
	}

	if (visual && server.composite_manager != None && snapshot_path == 0) {
		XSetWindowAttributes attrs;
		attrs.event_mask = StructureNotifyMask;
		XChangeWindowAttributes (server.dsp, server.composite_manager, CWEventMask, &attrs);

		server.real_transparency = 1;
		server.depth = 32;
		printf("real transparency on... depth: %d\n", server.depth);
		server.colormap = XCreateColormap(server.dsp, server.root_win, visual, AllocNone);
		server.visual = visual;
	}
	else {
		// no composite manager or snapshot mode => fake transparency
		server.real_transparency = 0;
		server.depth = DefaultDepth(server.dsp, server.screen);
		printf("real transparency off.... depth: %d\n", server.depth);
		server.colormap = DefaultColormap(server.dsp, server.screen);
		server.visual = DefaultVisual(server.dsp, server.screen);
	}
}