/*
 * Copyright (c) 2019 Derek Stevens, 2005 Russ Cox, 1994-1996 David Hogan
 * see README for license details
 */

#include <stdio.h>
#include <stdlib.h>
#include <signal.h>
#include <X11/X.h>
#include <X11/Xos.h>
#include <X11/Xlib.h>
#include <X11/Xutil.h>
#include <X11/Xatom.h>
#include <X11/keysym.h>
#include <X11/extensions/shape.h>
#include "config.h"
#include "dat.h"
#include "fns.h"
#include "patchlevel.h"

enum
{
	GrabAltTab,
	GrabAltAny
};

/*static int tabcode = 0x17; */
/*static int altcode = 0x40; */
/*static int pgupcode = 0x63; */
/*static int pgdowncode = 0x69; */

static void alttab(int shift);
void quickreshape(Client*c, int x, int y, int dx, int dy);

void
keysetup(void)
{
	int i;
	int tabcode = XKeysymToKeycode(dpy, XK_Tab);
	int dcode = XKeysymToKeycode(dpy, DESTROY_KEY);
	int icode = XKeysymToKeycode(dpy, ICON_KEY);
	int ucode = XKeysymToKeycode(dpy, UNHIDE_KEY);
	int mcode = XKeysymToKeycode(dpy, MAX_KEY);
	int vcode = XKeysymToKeycode(dpy, MOVE_KEY);
	int rcode = XKeysymToKeycode(dpy, RESIZE_KEY);
	int hcode = XKeysymToKeycode(dpy, SNAPLEFT_KEY);
	int lcode = XKeysymToKeycode(dpy, SNAPRIGHT_KEY);
	int jcode = XKeysymToKeycode(dpy, SNAPBOTTOM_KEY);
	int kcode = XKeysymToKeycode(dpy, SNAPTOP_KEY);
	int qcode = XKeysymToKeycode(dpy, SNAPTOPLEFT_KEY);
	int wcode = XKeysymToKeycode(dpy, SNAPBOTTOMLEFT_KEY);
	int ocode = XKeysymToKeycode(dpy, SNAPBOTTOMRIGHT_KEY);
	int pcode = XKeysymToKeycode(dpy, SNAPTOPRIGHT_KEY);
	int ccode = XKeysymToKeycode(dpy, SNAPCENTER_KEY);
	int slcode = XKeysymToKeycode(dpy, LAUNCH_KEY);
	int rightcode = XKeysymToKeycode(dpy, NEXTVIRT_KEY);
	int leftcode = XKeysymToKeycode(dpy, PREVVIRT_KEY);

	for(i=0; i<num_screens; i++){
		XGrabKey(dpy, tabcode, Mod1Mask, screens[i].root, 0, GrabModeSync, GrabModeAsync);
		XGrabKey(dpy, dcode, SHORTCUTMOD, screens[i].root, 0, GrabModeSync, GrabModeAsync);
		XGrabKey(dpy, icode, SHORTCUTMOD, screens[i].root, 0, GrabModeSync, GrabModeAsync);
		XGrabKey(dpy, ucode, SHORTCUTMOD, screens[i].root, 0, GrabModeSync, GrabModeAsync);
		XGrabKey(dpy, rcode, SHORTCUTMOD, screens[i].root, 0, GrabModeSync, GrabModeAsync);
		XGrabKey(dpy, vcode, SHORTCUTMOD, screens[i].root, 0, GrabModeSync, GrabModeAsync);
		XGrabKey(dpy, mcode, SHORTCUTMOD, screens[i].root, 0, GrabModeSync, GrabModeAsync);
		XGrabKey(dpy, hcode, SHORTCUTMOD, screens[i].root, 0, GrabModeSync, GrabModeAsync);
		XGrabKey(dpy, lcode, SHORTCUTMOD, screens[i].root, 0, GrabModeSync, GrabModeAsync);
		XGrabKey(dpy, jcode, SHORTCUTMOD, screens[i].root, 0, GrabModeSync, GrabModeAsync);
		XGrabKey(dpy, kcode, SHORTCUTMOD, screens[i].root, 0, GrabModeSync, GrabModeAsync);
		XGrabKey(dpy, qcode, SHORTCUTMOD, screens[i].root, 0, GrabModeSync, GrabModeAsync);
		XGrabKey(dpy, wcode, SHORTCUTMOD, screens[i].root, 0, GrabModeSync, GrabModeAsync);
		XGrabKey(dpy, ocode, SHORTCUTMOD, screens[i].root, 0, GrabModeSync, GrabModeAsync);
		XGrabKey(dpy, pcode, SHORTCUTMOD, screens[i].root, 0, GrabModeSync, GrabModeAsync);
		XGrabKey(dpy, ccode, SHORTCUTMOD, screens[i].root, 0, GrabModeSync, GrabModeAsync);
		XGrabKey(dpy, leftcode, SHORTCUTMOD, screens[i].root, 0, GrabModeSync, GrabModeAsync);
		XGrabKey(dpy, rightcode, SHORTCUTMOD, screens[i].root, 0, GrabModeSync, GrabModeAsync);
		XGrabKey(dpy, slcode, SHORTCUTMOD, screens[i].root, 0, GrabModeSync, GrabModeAsync);
		XGrabKey(dpy, tabcode, Mod1Mask|ShiftMask, screens[i].root, 0, GrabModeSync, GrabModeAsync);
	/*	XGrabKey(dpy, pgupcode, Mod1Mask, screens[i].root, 0, GrabModeSync, GrabModeAsync); */
	/*	XGrabKey(dpy, pgdowncode, Mod1Mask, screens[i].root, 0, GrabModeSync, GrabModeAsync); */
	/*	XGrabKey(dpy, altcode, 0, screens[i].root, 0, GrabModeSync, GrabModeAsync); */
	}
}

void
keypress(XKeyEvent *e)
{
	/*
	 * process key press here
	 */
	int tabcode = XKeysymToKeycode(dpy, XK_Tab);
	int dcode = XKeysymToKeycode(dpy, DESTROY_KEY);
	int icode = XKeysymToKeycode(dpy, ICON_KEY);
	int ucode = XKeysymToKeycode(dpy, UNHIDE_KEY);
	int mcode = XKeysymToKeycode(dpy, MAX_KEY);
	int vcode = XKeysymToKeycode(dpy, MOVE_KEY);
	int rcode = XKeysymToKeycode(dpy, RESIZE_KEY);
	int slcode = XKeysymToKeycode(dpy, LAUNCH_KEY);
	int hcode = XKeysymToKeycode(dpy, SNAPLEFT_KEY);
	int lcode = XKeysymToKeycode(dpy, SNAPRIGHT_KEY);
	int jcode = XKeysymToKeycode(dpy, SNAPBOTTOM_KEY);
	int kcode = XKeysymToKeycode(dpy, SNAPTOP_KEY);
	int qcode = XKeysymToKeycode(dpy, SNAPTOPLEFT_KEY);
	int wcode = XKeysymToKeycode(dpy, SNAPBOTTOMLEFT_KEY);
	int ocode = XKeysymToKeycode(dpy, SNAPBOTTOMRIGHT_KEY);
	int pcode = XKeysymToKeycode(dpy, SNAPTOPRIGHT_KEY);
	int ccode = XKeysymToKeycode(dpy, SNAPCENTER_KEY);
	int rightcode = XKeysymToKeycode(dpy, NEXTVIRT_KEY);
	int leftcode = XKeysymToKeycode(dpy, PREVVIRT_KEY);

	static XWindowAttributes ra;
	XGetWindowAttributes(dpy, DefaultRootWindow(dpy), &ra);


/* basic wm functionality */
  if(e->keycode == tabcode && (e->state&Mod1Mask) == (1<<3))
    alttab(e->state&ShiftMask);
	if(e->keycode == dcode && (e->state&SHORTCUTMOD) == (MODBITS))
		delete(current, 0);
	if (e->keycode == icode && (e->state&SHORTCUTMOD) == (MODBITS))
		hide(current);
	if (e->keycode == ucode && (e->state&SHORTCUTMOD) == (MODBITS))
		unhide(0, 1);
	if (e->keycode == vcode && (e->state&SHORTCUTMOD) == (MODBITS))
		move(current, Button3);
	if (e->keycode == rcode && (e->state&SHORTCUTMOD) == (MODBITS))
		reshape(current, Button3, sweep, 0);
	if (e->keycode == mcode && (e->state&SHORTCUTMOD) == (MODBITS))
		quickreshape(current, -BORDER, -BORDER, ra.width + 2*BORDER, ra.height + 2*BORDER);

/* half snap */
	if (e->keycode == hcode && (e->state&SHORTCUTMOD) == (MODBITS))
		quickreshape(current, 0, 0, ra.width/2, ra.height);
	if (e->keycode == lcode && (e->state&SHORTCUTMOD) == (MODBITS))
		quickreshape(current, ra.width/2, 0, ra.width/2, ra.height);
	if (e->keycode == jcode && (e->state&SHORTCUTMOD) == (MODBITS))
		quickreshape(current, 0, ra.height/2, ra.width, ra.height/2);
	if (e->keycode == kcode && (e->state&SHORTCUTMOD) == (MODBITS))
		quickreshape(current, 0, 0, ra.width, ra.height/2);

/* quarter snap */
	if (e->keycode == qcode && (e->state&SHORTCUTMOD) == (MODBITS))
		quickreshape(current, 0, 0, ra.width/2, ra.height/2);
	if (e->keycode == wcode && (e->state&SHORTCUTMOD) == (MODBITS))
		quickreshape(current, 0, ra.height/2, ra.width/2, ra.height/2);
	if (e->keycode == ocode && (e->state&SHORTCUTMOD) == (MODBITS))
		quickreshape(current, ra.width/2, ra.height/2, ra.width/2, ra.height/2);
	if (e->keycode == pcode && (e->state&SHORTCUTMOD) == (MODBITS))
		quickreshape(current, ra.width/2, 0, ra.width/2, ra.height/2);

/* center snap */
	if (e->keycode == ccode && (e->state&SHORTCUTMOD) == (MODBITS))
		quickreshape(current, ra.width/5, ra.height/5, 3*ra.width/5, 3*ra.height/5);


/* launch */
	if (e->keycode == slcode && (e->state&SHORTCUTMOD) == (MODBITS)){
		kbLaunch = 1;
		if(fork() == 0){
			close(ConnectionNumber(dpy));
			if(dpy != '\0')
				putenv(dpy);
			signal(SIGINT, SIG_DFL);
			signal(SIGTERM, SIG_DFL);
			signal(SIGHUP, SIG_DFL);
			if(termprog != NULL){
				execl(shell, shell, "-c", termprog, (char*)0);
				fprintf(stderr, "ryudo: exec %s", shell);
				perror(" failed");
			}
			execlp("urxvt", "urxvt", (char*)0);
			execlp("9term", "9term", scrolling ? "-ws" : "-w", (char*)0);
			execlp("xterm", "xterm", "-ut", (char*)0);
			perror("ryudo: exec urxvt/9term/xterm failed");
	 	}
  }

/* switch virts */
  if (e->keycode == rightcode && (e->state&SHORTCUTMOD) == (MODBITS))
  	if (numvirtuals > 1 && virt < numvirtuals - 1)
	   	switch_to(virt + 1);
  if (e->keycode == leftcode && (e->state&SHORTCUTMOD) == (MODBITS))
  	if (numvirtuals >1 && virt > 0)
	   	switch_to(virt - 1);

	XAllowEvents(dpy, SyncKeyboard, e->time);
}

void
keyrelease(XKeyEvent *e)
{
	XAllowEvents(dpy, SyncKeyboard, e->time);
}

void
quickreshape(Client *c, int x, int y, int dx, int dy)
{
	if (c == 0)
	  return;
	XMoveResizeWindow(dpy, c->parent, x,y, dx, dy);
	c->x = x + BORDER;
	c->y = y + BORDER;
	c->dx = dx-2*BORDER;
	c->dy = dy-2*BORDER;
	XMoveResizeWindow(dpy, c->window, BORDER, BORDER, c->dx, c->dy);
	sendconfig(c);
}

static void
alttab(int shift)
{
	shuffle(shift);
/*	fprintf(stderr, "%sTab\n", shift ? "Back" : ""); */
}