#include <stdio.h>

#include "usb.h"
#include "max3421e.h"
#include "timer.h"
#include "../user_io.h"

static unsigned char kbd_led_state = 0;  // default: all leds off

static void hexdump(void *data, int size) {
  int i,n = 0, b2c;
  char *ptr = data;

  if(!size) return;

  while(size>0) {
    iprintf("%04x: ", n);

    b2c = (size>16)?16:size;

    for(i=0;i<b2c;i++)
      iprintf("%02x ", 0xff&ptr[i]);

    iprintf("  ");

    for(i=0;i<(16-b2c);i++)
      iprintf("   ");

    for(i=0;i<b2c;i++)
      iprintf("%c", isprint(ptr[i])?ptr[i]:'.');

    iprintf("\n");

    ptr  += b2c;
    size -= b2c;
    n    += b2c;
  }
}

//get HID report descriptor 
static uint8_t hid_get_report_descr(usb_device_t *dev, uint8_t iface, uint16_t size)  {
  iprintf("%s(%x, if=%d, size=%d)\n", __FUNCTION__, dev->bAddress, iface, size);

  uint8_t buf[size];
  uint8_t rcode = usb_ctrl_req( dev, HID_REQ_HIDREPORT, USB_REQUEST_GET_DESCRIPTOR, 0x00, 
			      HID_DESCRIPTOR_REPORT, iface, size, size, buf, NULL);
  
  if(!rcode)
    hexdump(buf, size);
  
  return rcode;
}

static uint8_t hid_set_idle(usb_device_t *dev, uint8_t iface, uint8_t reportID, uint8_t duration ) {
  iprintf("%s(%x, if=%d id=%d, dur=%d)\n", __FUNCTION__, dev->bAddress, iface, reportID, duration);

  return( usb_ctrl_req( dev, HID_REQ_HIDOUT, HID_REQUEST_SET_IDLE, reportID, 
		       duration, iface, 0x0000, 0x0000, NULL, NULL ));
}

static uint8_t hid_set_protocol(usb_device_t *dev, uint8_t iface, uint8_t protocol) {
  iprintf("%s(%x, if=%d proto=%d)\n", __FUNCTION__, dev->bAddress, iface, protocol);

  return( usb_ctrl_req( dev, HID_REQ_HIDOUT, HID_REQUEST_SET_PROTOCOL, protocol, 
		       0x00, iface, 0x0000, 0x0000, NULL, NULL ));
}

static uint8_t hid_set_report(usb_device_t *dev, uint8_t iface, uint8_t report_type, uint8_t report_id, 
			      uint16_t nbytes, uint8_t* dataptr ) {
  //  iprintf("%s(%x, if=%d data=%x)\n", __FUNCTION__, dev->bAddress, iface, dataptr[0]);

  return( usb_ctrl_req(dev, HID_REQ_HIDOUT, HID_REQUEST_SET_REPORT, report_id, 
		       report_type, iface, nbytes, nbytes, dataptr, NULL ));
}

/* todo: handle parsing in chunks */
static uint8_t usb_hid_parse_conf(usb_device_t *dev, uint8_t conf, uint16_t len) {
  usb_hid_info_t *info = &(dev->hid_info);
  uint8_t rcode;
  bool isGoodInterface = false;

  union buf_u {
    usb_configuration_descriptor_t conf_desc;
    usb_interface_descriptor_t iface_desc;
    usb_endpoint_descriptor_t ep_desc;
    usb_hid_descriptor_t hid_desc;
    uint8_t raw[len];
  } buf, *p;

  // usb_interface_descriptor

  if(rcode = usb_get_conf_descr(dev, len, conf, &buf.conf_desc)) 
    return rcode;

  /* scan through all descriptors */
  p = &buf;
  while(len > 0) {

    switch(p->conf_desc.bDescriptorType) {
    case USB_DESCRIPTOR_CONFIGURATION:
      iprintf("conf descriptor size %d\n", p->conf_desc.bLength);
      // we already had this, so we simply ignore it
      break;

    case USB_DESCRIPTOR_INTERFACE:
      isGoodInterface = false;
      iprintf("iface descriptor size %d\n", p->iface_desc.bLength);

      /* check the interface descriptors for supported class */

      // only HID interfaces are supported
      if(p->iface_desc.bInterfaceClass == USB_CLASS_HID) {
	puts("iface is HID");

	if(info->bNumIfaces < MAX_IFACES) {
	  // ok, let's use this interface
	  isGoodInterface = true;

	  info->iface_info[info->bNumIfaces].iface_idx = p->iface_desc.bInterfaceNumber;
	  info->iface_info[info->bNumIfaces].has_boot_mode = false;
	  info->iface_info[info->bNumIfaces].device_type = HID_DEVICE_UNKNOWN;

	  if(p->iface_desc.bInterfaceSubClass == HID_BOOT_INTF_SUBCLASS) {
	    iprintf("Iface %d is Boot sub class\n", info->bNumIfaces);
	    info->iface_info[info->bNumIfaces].has_boot_mode = true;
	  }
	  
	  switch(p->iface_desc.bInterfaceProtocol) {
	  case HID_PROTOCOL_NONE:
	    iprintf("HID protocol is NONE\n");
	    break;
	    
	  case HID_PROTOCOL_KEYBOARD:
	    iprintf("HID protocol is KEYBOARD\n");
	    info->iface_info[info->bNumIfaces].device_type = HID_DEVICE_KEYBOARD;
	    //	    hid_set_report(dev, info->iface_info[info->bNumIfaces].iface_idx, 2, 0, 1, &kbd_led_state);
	    break;
	    
	  case HID_PROTOCOL_MOUSE:
	    iprintf("HID protocol is MOUSE\n");
	    info->iface_info[info->bNumIfaces].device_type = HID_DEVICE_MOUSE;
	    break;
	    
	  default:
	    iprintf("HID protocol is %d\n", p->iface_desc.bInterfaceProtocol);
	    break;
	  }
	}
      }
      break;

    case USB_DESCRIPTOR_ENDPOINT:
      iprintf("endpoint descriptor size %d\n", p->ep_desc.bLength);

      if(isGoodInterface) {

	// only interrupt in endpoints are supported
	if ((p->ep_desc.bmAttributes & 0x03) == 3 && (p->ep_desc.bEndpointAddress & 0x80) == 0x80) {
	  iprintf("endpint %d, interval = %dms\n", 
		  p->ep_desc.bEndpointAddress & 0x0F, p->ep_desc.bInterval);

	  // Fill in the endpoint info structure
	  uint8_t epidx = info->bNumIfaces;
	  info->ep[epidx].epAddr	 = (p->ep_desc.bEndpointAddress & 0x0F);
	  info->ep[epidx].maxPktSize = p->ep_desc.wMaxPacketSize[0];
	  info->ep[epidx].epAttribs	 = 0;
	  info->ep[epidx].bmNakPower = USB_NAK_NOWAIT;
	  info->bNumIfaces++;
	}
      }
      break;

    case HID_DESCRIPTOR_HID:
      iprintf("hid descriptor size %d\n", p->ep_desc.bLength);

      if(isGoodInterface) {
	// we need a report descriptor
	if(p->hid_desc.bDescrType == HID_DESCRIPTOR_REPORT) {
	  uint16_t len = p->hid_desc.wDescriptorLength[0] + 
	    256 * p->hid_desc.wDescriptorLength[1];
	  iprintf(" -> report descriptor size = %d\n", len);
	  
	  info->iface_info[info->bNumIfaces].report_size = len;
	}
      }
      break;

    default:
      iprintf("unsupported descriptor type %d size %d\n", p->raw[1], p->raw[0]);
    }

    // advance to next descriptor
    len -= p->conf_desc.bLength;
    p = (union buf_u*)(p->raw + p->conf_desc.bLength);
  }
  
  if(len != 0) {
    iprintf("URGS, underrun: %d\n", len);
    return USB_ERROR_CONFIGURAION_SIZE_MISMATCH;
  }

  iprintf("done\n");

  return 0;
}

static uint8_t usb_hid_init(usb_device_t *dev) {
  iprintf("%s()\n", __FUNCTION__);

  iprintf("init with address %x\n", dev->bAddress);

  uint8_t rcode;
  uint8_t i;

  usb_hid_info_t *info = &(dev->hid_info);
  
  union {
    usb_device_descriptor_t dev_desc;
    usb_configuration_descriptor_t conf_desc;
  } buf;

  // reset status
  info->qNextPollTime = 0;
  info->bPollEnable = false;
  info->bNumIfaces = 0;

  for(i=0;i<MAX_IFACES;i++) {
    info->ep[i].epAddr	= i;
    info->ep[i].maxPktSize	= 8;
    info->ep[i].epAttribs	= 0;
    info->ep[i].bmNakPower	= USB_NAK_MAX_POWER;
  }

  // try to re-read full device descriptor from newly assigned address
  if(rcode = usb_get_dev_descr( dev, sizeof(usb_device_descriptor_t), &buf.dev_desc )) {
    puts("failed to get device descriptor");
    return rcode;
  }

  uint8_t num_of_conf = buf.dev_desc.bNumConfigurations;
  iprintf("number of configurations: %d\n", num_of_conf);

  for(i=0; i<num_of_conf; i++) {
    if(rcode = usb_get_conf_descr(dev, sizeof(usb_configuration_descriptor_t), i, &buf.conf_desc)) 
      return rcode;
    
    iprintf("conf descriptor %d has total size %d\n", i, buf.conf_desc.wTotalLength);

    // extract number of interfaces
    iprintf("number of interfaces: %d\n", buf.conf_desc.bNumInterfaces);
    
    // parse directly if it already fitted completely into the buffer
    usb_hid_parse_conf(dev, i, buf.conf_desc.wTotalLength);
  }

  // check if we found valid hid interfaces
  if(!info->bNumIfaces) {
    puts("no hid interfaces found");
    return USB_DEV_CONFIG_ERROR_DEVICE_NOT_SUPPORTED;
  }

  // Set Configuration Value
  iprintf("conf value = %d\n", buf.conf_desc.bConfigurationValue);
  rcode = usb_set_conf(dev, buf.conf_desc.bConfigurationValue);

  // process all supported interfaces
  for(i=0; i<info->bNumIfaces; i++) {
    rcode = hid_get_report_descr(dev, 
	       info->iface_info[i].iface_idx, info->iface_info[i].report_size);
    if (rcode)
      return rcode;

    rcode = hid_set_idle(dev, info->iface_info[i].iface_idx, 0, 0);
    if (rcode && rcode != hrSTALL)
      return rcode;

    // enable boot mode
    if(info->iface_info[i].has_boot_mode)
      hid_set_protocol(dev, info->iface_info[i].iface_idx, HID_BOOT_PROTOCOL);
  }
  
  puts("HID configured");

  // update leds
  for(i=0;i<MAX_IFACES;i++)
    if(dev->hid_info.iface_info[i].device_type == HID_DEVICE_KEYBOARD)
      hid_set_report(dev, dev->hid_info.iface_info[i].iface_idx, 2, 0, 1, &kbd_led_state);

  info->bPollEnable = true;
  return 0;
}

static uint8_t usb_hid_release(usb_device_t *dev) {
  puts(__FUNCTION__);
  return 0;
}

static uint8_t usb_hid_poll(usb_device_t *dev) {
  usb_hid_info_t *info = &(dev->hid_info);

  if (!info->bPollEnable)
    return 0;
  
  if (info->qNextPollTime <= timer_get_msec()) {
    int8_t i;
    for(i=0;i<info->bNumIfaces;i++) {
      //      iprintf("poll %d...\n", info->ep[i].epAddr);

      uint16_t read = info->ep[i].maxPktSize;
      uint8_t buf[32];
      uint8_t rcode = 
	usb_in_transfer(dev, &(info->ep[i]), &read, buf);

      if (rcode) {
	if (rcode != hrNAK)
	  iprintf("%s() error: %d %d\n", __FUNCTION__, rcode, timer_get_msec());
	//	else
	//	  puts("nak");
      } else {

	//	iprintf("interface %d: received %d bytes\n", i, read);

	// successfully received some bytes
	if(info->iface_info[i].has_boot_mode) {
	  if(info->iface_info[i].device_type == HID_DEVICE_MOUSE) {
	    // boot mouse needs at least three bytes
	    if(read >= 3) {
	      // forward all three bytes to the user_io layer
	      user_io_mouse(buf[0], buf[1], buf[2]);
	    }
	  }

	  if(info->iface_info[i].device_type == HID_DEVICE_KEYBOARD) {
	    // boot kbd needs at least eight bytes
	    if(read >= 8) {
	      user_io_kbd(buf[0], buf+2);
	    }
	  }
	}
      }
    }

    info->qNextPollTime = timer_get_msec() + 20;   // poll 50 times a second
  }

  return 0;
}

void hid_set_kbd_led(unsigned char led, bool on) {
  // check if led state has changed
  if( (on && !(kbd_led_state&led)) || (!on && (kbd_led_state&led))) {
    if(on) kbd_led_state |=  led;
    else   kbd_led_state &= ~led;

    // search for all keyboard interfaces on all hid devices
    usb_device_t *dev = usb_get_devices();
    int i;
    for(i=0;i<USB_NUMDEVICES;i++) {
      if(dev[i].bAddress && (dev[i].class == &usb_hid_class)) {
	// search for keyboard interfaces
	int j;
	for(j=0;j<MAX_IFACES;j++)
	  if(dev[i].hid_info.iface_info[j].device_type == HID_DEVICE_KEYBOARD)
	    hid_set_report(dev+i, dev[i].hid_info.iface_info[j].iface_idx, 2, 0, 1, &kbd_led_state);
      }
    }
  }
}

const usb_device_class_config_t usb_hid_class = {
  usb_hid_init, usb_hid_release, usb_hid_poll };