/*
 * Copyright © 2011 Intel Corporation
 *
 * Permission to use, copy, modify, distribute, and sell this software and
 * its documentation for any purpose is hereby granted without fee, provided
 * that the above copyright notice appear in all copies and that both that
 * copyright notice and this permission notice appear in supporting
 * documentation, and that the name of the copyright holders not be used in
 * advertising or publicity pertaining to distribution of the software
 * without specific, written prior permission.  The copyright holders make
 * no representations about the suitability of this software for any
 * purpose.  It is provided "as is" without express or implied warranty.
 *
 * THE COPYRIGHT HOLDERS DISCLAIM ALL WARRANTIES WITH REGARD TO THIS
 * SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND
 * FITNESS, IN NO EVENT SHALL THE COPYRIGHT HOLDERS BE LIABLE FOR ANY
 * SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER
 * RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF
 * CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
 * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <sys/wait.h>
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <fcntl.h>
#include <linux/input.h>

#include "compositor.h"
#include "meego-tablet-server-protocol.h"

/*
 * TODO: Don't fade back from black until we've received a lockscreen
 * attachment.
 */

struct meego_tablet_shell {
	struct wl_object object;

	struct wlsc_shell shell;

	struct wlsc_compositor *compositor;
	struct wl_event_source *sigchld_source;
	pid_t pid;
	struct wlsc_input_device *device;
	struct wl_client *client;
	struct wlsc_animation animation;
	struct wlsc_tweener scale;
	int width, height;

	struct wlsc_surface *surface;

	struct wlsc_surface *lockscreen_surface;
	struct wl_listener lockscreen_listener;

	struct wlsc_surface *home_surface;

	struct wlsc_surface *switcher_surface;
	struct wl_listener switcher_listener;

	/* FIXME: Need a state machine here. */
	int starting, locked;
	int home_active;
	int task_switcher_active;
	int long_press_active;
	struct wl_event_source *long_press_source;
};

static int
sigchld_handler(int signal_number, void *data)
{
	struct meego_tablet_shell *shell = data;
	int status;

	wait(&status);
	shell->pid = 0;

	fprintf(stderr, "meego-ux-daemon crashed, exit code %d\n", status);

	return 1;
}

static void
tablet_shell_frame(struct wlsc_animation *animation,
		   struct wlsc_output *output, uint32_t msecs)
{
	struct meego_tablet_shell *shell =
		container_of(animation,
			     struct meego_tablet_shell, animation);
	struct wlsc_surface *es = shell->surface;
	float scale;

	scale = shell->scale.current;
	wlsc_tweener_update(&shell->scale, msecs);

	if (wlsc_tweener_done(&shell->scale)) {
		wl_list_remove(&animation->link);
		shell->surface = NULL;
		fprintf(stderr, "animation done\n");
	}

	es->width = scale * shell->width;
	es->height = scale * shell->height;
	es->x = (output->width - es->width) / 2;
	es->y = (output->height - es->height) / 2;
	wlsc_surface_update_matrix(es);
	wlsc_surface_damage(es);
}

static void
meego_tablet_shell_attach(struct wlsc_shell *base,
			  struct wlsc_surface *surface)
{
	struct meego_tablet_shell *shell =
		container_of(base, struct meego_tablet_shell, shell);
	struct wlsc_compositor *ec = shell->compositor;
	const char *visual;

	if (surface == shell->lockscreen_surface) {
		if (shell->starting) {
			wlsc_compositor_fade(shell->compositor, 0.0);
			wlsc_compositor_wake(shell->compositor);
			wlsc_surface_damage(surface);
			shell->starting = 0;
		}
	} else if (surface == shell->switcher_surface) {

		if (surface->visual == &ec->compositor.argb_visual) {
			visual = "argb";
		} else if (surface->visual == &ec->compositor.premultiplied_argb_visual) {
			visual = "premultiplied argb";
		} else {
			visual = "rgb";
		}

		fprintf(stderr, "switcher attach, size %dx%d, visual %s\n",
			shell->switcher_surface->width,
			shell->switcher_surface->height, visual);
	} else if (surface == shell->home_surface) {
		surface->x = 0;
		surface->y = 0;
		wlsc_surface_update_matrix(surface);
	} else {
		/* FIXME: Fullscreen everything else. */
		fprintf(stderr, "attach from surface %p, client %p\n",
			surface, surface->surface.client);
		if (shell->surface == NULL) {
			fprintf(stderr, "starting animation for surface %p\n",
				surface);

			shell->width = surface->width;
			shell->height = surface->height;
			wlsc_tweener_init(&shell->scale, 3.0, 0.3, 1.0);
			shell->animation.frame = tablet_shell_frame;
			wl_list_insert(ec->animation_list.prev,
				       &shell->animation.link);
			shell->surface = surface;
		}
	}
}

static void
handle_lockscreen_surface_destroy(struct wl_listener *listener,
				struct wl_surface *surface, uint32_t time)
{
	struct meego_tablet_shell *shell =
		container_of(listener,
			     struct meego_tablet_shell, lockscreen_listener);

	shell->lockscreen_surface = NULL;
	shell->locked = 0;
}

static void
tablet_shell_set_lockscreen(struct wl_client *client,
			    struct meego_tablet_shell *shell,
			    struct wl_surface *surface)
{
	struct wlsc_surface *es = (struct wlsc_surface *) surface;
	struct wlsc_input_device *device =
		(struct wlsc_input_device *) shell->compositor->input_device;

	fprintf(stderr, "set lock screen\n");
	es->x = 0;
	es->y = 0;
	wlsc_surface_update_matrix(es);
	wlsc_surface_activate(es, device, wlsc_compositor_get_time());
	shell->lockscreen_surface = es;

	shell->lockscreen_listener.func = handle_lockscreen_surface_destroy;
	wl_list_insert(es->surface.destroy_listener_list.prev,
		       &shell->lockscreen_listener.link);
}

static void
handle_switcher_surface_destroy(struct wl_listener *listener,
				struct wl_surface *surface, uint32_t time)
{
	struct meego_tablet_shell *shell =
		container_of(listener,
			     struct meego_tablet_shell, switcher_listener);

	shell->switcher_surface = NULL;
	shell->task_switcher_active = 0;
}

static void
tablet_shell_set_switcher(struct wl_client *client,
			  struct meego_tablet_shell *shell,
			  struct wl_surface *surface)
{
	struct wlsc_input_device *device =
		(struct wlsc_input_device *) shell->compositor->input_device;
	struct wlsc_surface *es = (struct wlsc_surface *) surface;

	/* FIXME: Switcher should be centered and the compositor
	 * should do the tinting of the background.  With the cache
	 * layer idea, we should be able to hit the framerate on the
	 * fade/zoom in. */
	shell->switcher_surface = es;
	shell->switcher_surface->x = 0;
	shell->switcher_surface->y = 0;
	wlsc_surface_update_matrix(es);

	wlsc_surface_activate(es, device, wlsc_compositor_get_time());

	fprintf(stderr, "set switcher %p, size %dx%d\n",
		es, es->width, es->height);

	shell->switcher_listener.func = handle_switcher_surface_destroy;
	wl_list_insert(es->surface.destroy_listener_list.prev,
		       &shell->switcher_listener.link);
}

static void
tablet_shell_set_homescreen(struct wl_client *client,
			    struct meego_tablet_shell *shell,
			    struct wl_surface *surface)
{
	struct wlsc_input_device *device =
		(struct wlsc_input_device *) shell->compositor->input_device;

	shell->home_surface = (struct wlsc_surface *) surface;
	shell->home_surface->x = 0;
	shell->home_surface->y = 0;
	wlsc_surface_update_matrix(shell->home_surface);

	wlsc_surface_activate(shell->home_surface, device,
			      wlsc_compositor_get_time());
	fprintf(stderr, "set home screen\n");
}

struct meego_tablet_client {
	struct wl_resource resource;
	struct meego_tablet_shell *shell;
	struct wl_client *client;
	char *name;
};

static void
destroy_tablet_client(struct wl_resource *resource, struct wl_client *client)
{
	struct meego_tablet_client *tablet_client =
		container_of(resource, struct meego_tablet_client, resource);

	free(tablet_client->name);
	free(tablet_client);
}

static void
tablet_client_poke(struct wl_client *client,
		   struct meego_tablet_client *tablet_client)
{
	struct wlsc_compositor *compositor = tablet_client->shell->compositor;
	struct wlsc_surface *surface;
	struct wlsc_input_device *device =
		(struct wlsc_input_device *) compositor->input_device;

	wl_list_for_each(surface, &compositor->surface_list, link) {
		if (surface->surface.client == tablet_client->client) {
			wlsc_surface_activate(surface, device,
					      wlsc_compositor_get_time());
			break;
		}
	}

	fprintf(stderr, "poke client %s\n", tablet_client->name);
}

static const struct meego_tablet_client_interface tablet_client_interface = {
	tablet_client_poke
};

static void
tablet_shell_create_client(struct wl_client *client,
			   struct meego_tablet_shell *shell,
			   uint32_t id, const char *name, int fd)
{
	struct wlsc_compositor *compositor = shell->compositor;
	struct meego_tablet_client *tablet_client;

	tablet_client = malloc(sizeof *tablet_client);
	if (tablet_client == NULL) {
		wl_client_post_no_memory(client);
		return;
	}

	tablet_client->client = wl_client_create(compositor->wl_display, fd);
	tablet_client->shell = shell;
	tablet_client->name = strdup(name);

	tablet_client->resource.destroy = destroy_tablet_client;
	tablet_client->resource.object.id = id;
	tablet_client->resource.object.interface =
		&meego_tablet_client_interface;
	tablet_client->resource.object.implementation =
		(void (**)(void)) &tablet_client_interface;

	wl_client_add_resource(client, &tablet_client->resource);

	fprintf(stderr, "created client %p, id %d, name %s, fd %d\n",
		tablet_client->client, id, name, fd);
}

static const struct meego_tablet_shell_interface tablet_shell_interface = {
	tablet_shell_set_lockscreen,
	tablet_shell_set_switcher,
	tablet_shell_set_homescreen,
	tablet_shell_create_client
};

static void
launch_switcher(struct meego_tablet_shell *shell)
{
	struct wlsc_compositor *compositor = shell->compositor;
	char s[32];
	int sv[2], flags;

	if (socketpair(AF_UNIX, SOCK_STREAM | SOCK_CLOEXEC, 0, sv) < 0) {
		fprintf(stderr, "socketpair failed\n");
		return;
	}

	/* SOCK_CLOEXEC closes both ends, so we need to unset the flag
	 * on the client fd. */
	flags = fcntl(sv[1], F_GETFD);
	if (flags != -1)
		fcntl(sv[1], F_SETFD, flags & ~FD_CLOEXEC);

	shell->pid = fork();
	switch (shell->pid) {
	case 0:
		snprintf(s, sizeof s, "%d", sv[1]);
		setenv("WAYLAND_SOCKET", s, 1);
		setenv("EGL_PLATFORM", "wayland", 1);
		setenv("QT_QPA_PLATFORM", "waylandgl", 1);
		if (execl("/usr/libexec/meego-ux-daemon",
			  "/usr/libexec/meego-ux-daemon", NULL) < 0)
			fprintf(stderr, "exec failed: %m\n");
		exit(-1);

	default:
		close(sv[1]);
		shell->client =
			wl_client_create(compositor->wl_display, sv[0]);
		break;

	case -1:
		fprintf(stderr, "failed to fork\n");
		break;
	}
}

static void
toggle_switcher(struct meego_tablet_shell *shell)
{
	struct wlsc_surface *surface = shell->switcher_surface;

	if (surface) {
		wl_client_post_event(shell->client, &shell->object,
				     MEEGO_TABLET_SHELL_HIDE_SWITCHER);
		printf("hide switcher\n");
		/* cleanupTaskSwitcher(); */
	} else {
		shell->task_switcher_active = 1;
		wl_client_post_event(shell->client, &shell->object,
				     MEEGO_TABLET_SHELL_SHOW_SWITCHER);
	}
}

static void
meego_tablet_shell_lock(struct wlsc_shell *base)
{
	struct meego_tablet_shell *shell =
		container_of(base, struct meego_tablet_shell, shell);

	if (!shell->locked)
		wl_client_post_event(shell->client, &shell->object,
				     MEEGO_TABLET_SHELL_SHOW_LOCKSCREEN);
	shell->locked = 1;
}

static void
go_home(struct meego_tablet_shell *shell)
{
	struct wlsc_input_device *device =
		(struct wlsc_input_device *) shell->compositor->input_device;

	if (shell->task_switcher_active)
		wl_client_post_event(shell->client, &shell->object,
				     MEEGO_TABLET_SHELL_HIDE_SWITCHER);

	wlsc_surface_activate(shell->home_surface, device,
			      wlsc_compositor_get_time());
	printf("go home\n");
}

static int
long_press_handler(void *data)
{
	struct meego_tablet_shell *shell = data;

	shell->long_press_active = 0;
	toggle_switcher(shell);

	return 1;
}

static void
menu_key_binding(struct wl_input_device *device, uint32_t time,
		 uint32_t key, uint32_t button, uint32_t state, void *data)
{
	struct meego_tablet_shell *shell = data;

	if (shell->locked)
		return;

	if (state)
		toggle_switcher(shell);
}

static void
home_key_binding(struct wl_input_device *device, uint32_t time,
		 uint32_t key, uint32_t button, uint32_t state, void *data)
{
	struct meego_tablet_shell *shell = data;

	if (shell->locked)
		return;

	shell->device = (struct wlsc_input_device *) device;

	if (state) {
		wl_event_source_timer_update(shell->long_press_source, 500);
		shell->long_press_active = 1;
	} else if (shell->long_press_active) {
		wl_event_source_timer_update(shell->long_press_source, 0);
		shell->long_press_active = 0;

		if (shell->home_active || shell->task_switcher_active)
			toggle_switcher(shell);
		else
			go_home(shell);
	}
}

static void
tablet_shell_bind(struct wl_client *client,
		  struct wl_object *object, uint32_t version)
{
	struct meego_tablet_shell *shell =
		(struct meego_tablet_shell *) object;

	wl_client_post_event(shell->client, &shell->object,
			     MEEGO_TABLET_SHELL_SHOW_LOCKSCREEN);
}

void
shell_init(struct wlsc_compositor *compositor);

WL_EXPORT void
shell_init(struct wlsc_compositor *compositor)
{
	struct meego_tablet_shell *shell;
	struct wl_event_loop *loop;

	shell = malloc(sizeof *shell);
	if (shell == NULL)
		return;

	memset(shell, 0, sizeof *shell);
	shell->compositor = compositor;
	shell->locked = 1;

	shell->object.interface = &meego_tablet_shell_interface;
	shell->object.implementation =
		(void (**)(void)) &tablet_shell_interface;
	wl_display_add_object(compositor->wl_display, &shell->object);

	/* FIXME: This will make the object available to all clients. */
	wl_display_add_global(compositor->wl_display,
			      &shell->object, tablet_shell_bind);

	loop = wl_display_get_event_loop(compositor->wl_display);
	shell->sigchld_source =
		wl_event_loop_add_signal(loop, SIGCHLD,
					 sigchld_handler, shell);

	shell->long_press_source =
		wl_event_loop_add_timer(loop, long_press_handler, shell);

	wlsc_compositor_add_binding(compositor, KEY_SPACE, 0, MODIFIER_CTRL,
				    home_key_binding, shell);
	wlsc_compositor_add_binding(compositor, KEY_ENTER, 0, MODIFIER_CTRL,
				    menu_key_binding, shell);

	compositor->shell = &shell->shell;

	shell->shell.lock = meego_tablet_shell_lock;
	shell->shell.attach = meego_tablet_shell_attach;

	launch_switcher(shell);

	wlsc_tweener_init(&compositor->fade.tweener, 40.0, 1.0, 1.0);
	shell->starting = 1;
}