/*
 * Copyright (c) 2011 Lubos Slovak
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * - Redistributions in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in the
 *   documentation and/or other materials provided with the distribution.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/** @addtogroup drvusbhid
 * @{
 */
/**
 * @file
 * USB HID driver API.
 */

#include <usb/debug.h>
#include <usb/classes/classes.h>
#include <usb/classes/hid.h>
#include <usb/classes/hidparser.h>
#include <usb/classes/hidreport.h>
#include <usb/classes/hidreq.h>
#include <errno.h>

#include "usbhid.h"

#include "kbd/kbddev.h"
#include "generic/hiddev.h"
#include "mouse/mousedev.h"

/*----------------------------------------------------------------------------*/

/* Array of endpoints expected on the device, NULL terminated. */
usb_endpoint_description_t *usb_hid_endpoints[USB_HID_POLL_EP_COUNT + 1] = {
	&usb_hid_kbd_poll_endpoint_description,
	&usb_hid_mouse_poll_endpoint_description,
	&usb_hid_generic_poll_endpoint_description,
	NULL
};

/*----------------------------------------------------------------------------*/

usb_hid_dev_t *usb_hid_new(void)
{
	usb_hid_dev_t *hid_dev = (usb_hid_dev_t *)calloc(1,
	    sizeof(usb_hid_dev_t));
	
	if (hid_dev == NULL) {
		usb_log_fatal("No memory!\n");
		return NULL;
	}
	
	hid_dev->parser = (usb_hid_report_parser_t *)(malloc(sizeof(
	    usb_hid_report_parser_t)));
	if (hid_dev->parser == NULL) {
		usb_log_fatal("No memory!\n");
		free(hid_dev);
		return NULL;
	}
	
	return hid_dev;
}

/*----------------------------------------------------------------------------*/

static int usb_hid_check_pipes(usb_hid_dev_t *hid_dev, usb_device_t *dev)
{
	if (dev->pipes[USB_HID_KBD_POLL_EP_NO].present) {
		usb_log_debug("Found keyboard endpoint.\n");
		
		// save the pipe index and device type
		hid_dev->poll_pipe_index = USB_HID_KBD_POLL_EP_NO;
		hid_dev->device_type = USB_HID_PROTOCOL_KEYBOARD;
		
		// set the polling callback
		hid_dev->poll_callback = usb_kbd_polling_callback;

	} else if (dev->pipes[USB_HID_MOUSE_POLL_EP_NO].present) {
		usb_log_debug("Found mouse endpoint.\n");
		
		// save the pipe index and device type
		hid_dev->poll_pipe_index = USB_HID_MOUSE_POLL_EP_NO;
		hid_dev->device_type = USB_HID_PROTOCOL_MOUSE;
		
		// set the polling callback
		hid_dev->poll_callback = usb_mouse_polling_callback;
		
	} else if (dev->pipes[USB_HID_GENERIC_POLL_EP_NO].present) {
		usb_log_debug("Found generic HID endpoint.\n");
		
		// save the pipe index and device type
		hid_dev->poll_pipe_index = USB_HID_GENERIC_POLL_EP_NO;
		hid_dev->device_type = USB_HID_PROTOCOL_NONE;
		
		// set the polling callback
		hid_dev->poll_callback = usb_hid_polling_callback;
		
	} else {
		usb_log_warning("None of supported endpoints found - probably"
		    " not a supported device.\n");
		return ENOTSUP;
	}
	
	return EOK;
}

/*----------------------------------------------------------------------------*/

static int usb_hid_init_parser(usb_hid_dev_t *hid_dev)
{
	/* Initialize the report parser. */
	int rc = usb_hid_parser_init(hid_dev->parser);
	if (rc != EOK) {
		usb_log_error("Failed to initialize report parser.\n");
		return rc;
	}
	
	/* Get the report descriptor and parse it. */
	rc = usb_hid_process_report_descriptor(hid_dev->usb_dev, 
	    hid_dev->parser);
	
	if (rc != EOK || hid_dev->device_type == USB_HID_PROTOCOL_MOUSE) {
		usb_log_warning("Could not process report descriptor.\n");
		
		if (hid_dev->device_type == USB_HID_PROTOCOL_KEYBOARD) {
			usb_log_warning("Falling back to boot protocol.\n");
			rc = usb_kbd_set_boot_protocol(hid_dev);
		} else if (hid_dev->device_type == USB_HID_PROTOCOL_MOUSE) {
			usb_log_warning("Falling back to boot protocol.\n");
			rc = usb_mouse_set_boot_protocol(hid_dev);
		}
	}
	
	return rc;
}

/*----------------------------------------------------------------------------*/

int usb_hid_init(usb_hid_dev_t *hid_dev, usb_device_t *dev)
{
	int rc;
	
	usb_log_debug("Initializing HID structure...\n");
	
	if (hid_dev == NULL) {
		usb_log_error("Failed to init HID structure: no structure given"
		    ".\n");
		return EINVAL;
	}
	
	if (dev == NULL) {
		usb_log_error("Failed to init HID structure: no USB device"
		    " given.\n");
		return EINVAL;
	}
	
	/* The USB device should already be initialized, save it in structure */
	hid_dev->usb_dev = dev;
	
	rc = usb_hid_check_pipes(hid_dev, dev);
	if (rc != EOK) {
		return rc;
	}
	
	rc = usb_hid_init_parser(hid_dev);
	if (rc != EOK) {
		usb_log_error("Failed to initialize HID parser.\n");
		return rc;
	}
	
	switch (hid_dev->device_type) {
	case USB_HID_PROTOCOL_KEYBOARD:
		// initialize the keyboard structure
		rc = usb_kbd_init(hid_dev);
		if (rc != EOK) {
			usb_log_warning("Failed to initialize KBD structure."
			    "\n");
		}
		break;
	case USB_HID_PROTOCOL_MOUSE:
		rc = usb_mouse_init(hid_dev);
		if (rc != EOK) {
			usb_log_warning("Failed to initialize Mouse structure."
			    "\n");
		}
		break;
	default:
		break;
	}
	
	return rc;
}

/*----------------------------------------------------------------------------*/

void usb_hid_polling_ended_callback(usb_device_t *dev, bool reason, 
     void *arg)
{
	if (dev == NULL || arg == NULL) {
		return;
	}
	
	usb_hid_dev_t *hid_dev = (usb_hid_dev_t *)arg;
	
	usb_hid_free(&hid_dev);
}

/*----------------------------------------------------------------------------*/

const char *usb_hid_get_function_name(usb_hid_iface_protocol_t device_type)
{
	switch (device_type) {
	case USB_HID_PROTOCOL_KEYBOARD:
		return HID_KBD_FUN_NAME;
		break;
	case USB_HID_PROTOCOL_MOUSE:
		return HID_MOUSE_FUN_NAME;
		break;
	default:
		return HID_GENERIC_FUN_NAME;
	}
}

/*----------------------------------------------------------------------------*/

const char *usb_hid_get_class_name(usb_hid_iface_protocol_t device_type)
{
	switch (device_type) {
	case USB_HID_PROTOCOL_KEYBOARD:
		return HID_KBD_CLASS_NAME;
		break;
	case USB_HID_PROTOCOL_MOUSE:
		return HID_MOUSE_CLASS_NAME;
		break;
	default:
		return HID_GENERIC_CLASS_NAME;
	}
}

/*----------------------------------------------------------------------------*/

void usb_hid_free(usb_hid_dev_t **hid_dev)
{
	if (hid_dev == NULL || *hid_dev == NULL) {
		return;
	}
	
	switch ((*hid_dev)->device_type) {
	case USB_HID_PROTOCOL_KEYBOARD:
		usb_kbd_deinit(*hid_dev);
		break;
	case USB_HID_PROTOCOL_MOUSE:
		usb_mouse_deinit(*hid_dev);
		break;
	default:
		break;
	}

	// destroy the parser
	if ((*hid_dev)->parser != NULL) {
		usb_hid_free_report_parser((*hid_dev)->parser);
	}

	if ((*hid_dev)->report_desc != NULL) {
		free((*hid_dev)->report_desc);
	}

	free(*hid_dev);
	*hid_dev = NULL;
}

/**
 * @}
 */