/**************************************************************************/
/*!
    @file     hid_device.c
    @author   hathach (tinyusb.org)

    @section LICENSE

    Software License Agreement (BSD License)

    Copyright (c) 2013, hathach (tinyusb.org)
    All rights reserved.

    Redistribution and use in source and binary forms, with or without
    modification, are permitted provided that the following conditions are met:
    1. Redistributions of source code must retain the above copyright
    notice, this list of conditions and the following disclaimer.
    2. 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.
    3. Neither the name of the copyright holders nor the
    names of its contributors may be used to endorse or promote products
    derived from this software without specific prior written permission.

    THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS ''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 COPYRIGHT HOLDER 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.

    This file is part of the tinyusb stack.
*/
/**************************************************************************/

#include "tusb_option.h"

#if (TUSB_OPT_DEVICE_ENABLED && TUD_OPT_HID_ENABLED)

#define _TINY_USB_SOURCE_FILE_
//--------------------------------------------------------------------+
// INCLUDE
//--------------------------------------------------------------------+
#include "common/tusb_common.h"
#include "hid_device.h"
#include "device/usbd_pvt.h"

//--------------------------------------------------------------------+
// MACRO CONSTANT TYPEDEF
//--------------------------------------------------------------------+

// Max report len is keyboard's one with 8 byte + 1 byte report id
#define REPORT_BUFSIZE      9

typedef struct {
  uint8_t itf_num;
  uint8_t ep_in;
  uint8_t idle_rate;

  uint8_t  report_id;
  uint16_t report_len;
  uint8_t const * report_desc;

  // class specific control request
  uint16_t (*get_report_cb) (hid_report_type_t type, uint8_t* buffer, uint16_t reqlen);
  void     (*set_report_cb) (hid_report_type_t type, uint8_t const* buffer, uint16_t bufsize);

  CFG_TUSB_MEM_ALIGN uint8_t report_buf[REPORT_BUFSIZE];
}hidd_interface_t;

#if CFG_TUD_HID_BOOT_PROTOCOL

CFG_TUSB_ATTR_USBRAM static hidd_interface_t _kbd_itf;
CFG_TUSB_ATTR_USBRAM static hidd_interface_t _mse_itf;

#else

CFG_TUSB_ATTR_USBRAM static hidd_interface_t _composite_itf;

#endif

//--------------------------------------------------------------------+
// KEYBOARD APPLICATION API
//--------------------------------------------------------------------+
#if CFG_TUD_HID_KEYBOARD
bool tud_hid_keyboard_ready(void)
{
  VERIFY( _kbd_itf.ep_in != 0 );
  return !dcd_edpt_busy(TUD_OPT_RHPORT, _kbd_itf.ep_in);
}

static bool hidd_kbd_report(hid_keyboard_report_t const *p_report)
{
  VERIFY( tud_hid_keyboard_ready() );

  hidd_interface_t * p_hid = &_kbd_itf;

  if ( p_report )
  {
    memcpy(p_hid->report_buf, p_report, sizeof(hid_keyboard_report_t));
  }else
  {
    // only send empty report if previous report is not empty
    // TODO idle rate
    if ( mem_all_zero(p_hid->report_buf, sizeof(hid_keyboard_report_t)) ) return true;

    arrclr_(p_hid->report_buf);
  }

  return dcd_edpt_xfer(TUD_OPT_RHPORT, p_hid->ep_in, p_hid->report_buf, sizeof(hid_keyboard_report_t));
}

bool tud_hid_keyboard_keycode(uint8_t modifier, uint8_t keycode[6])
{
  hid_keyboard_report_t report = { .modifier = modifier };

  if ( keycode )
  {
    memcpy(report.keycode, keycode, 6);
  }else
  {
    memclr_(report.keycode, 6);
  }

  return hidd_kbd_report(&report);
}

#if CFG_TUD_HID_ASCII_TO_KEYCODE_LOOKUP

bool tud_hid_keyboard_key_press(char ch)
{
  uint8_t keycode[6] = { 0 };
  uint8_t modifier   = 0;

  if ( HID_ASCII_TO_KEYCODE[(uint8_t)ch].shift ) modifier = KEYBOARD_MODIFIER_LEFTSHIFT;
  keycode[0] = HID_ASCII_TO_KEYCODE[(uint8_t)ch].keycode;

  return tud_hid_keyboard_keycode(modifier, keycode);
}

bool tud_hid_keyboard_key_sequence(const char* str, uint32_t interval_ms)
{
  // Send each key in string
  char ch;
  while( (ch = *str++) != 0 )
  {
    char lookahead = *str;

    tud_hid_keyboard_key_press(ch);

    // Blocking delay
    tu_timeout_wait(interval_ms);

    /* Only need to empty report if the next character is NULL or the same with
     * the current one, else no need to send */
    if ( lookahead == ch || lookahead == 0 )
    {
      tud_hid_keyboard_key_release();
      tu_timeout_wait(interval_ms);
    }
  }

}

#endif // CFG_TUD_HID_ASCII_TO_KEYCODE_LOOKUP

#endif

//--------------------------------------------------------------------+
// MOUSE APPLICATION API
//--------------------------------------------------------------------+
#if CFG_TUD_HID_MOUSE
bool tud_hid_mouse_ready(void)
{
  VERIFY( _mse_itf.ep_in != 0 );
  return !dcd_edpt_busy(TUD_OPT_RHPORT, _mse_itf.ep_in);
}

static bool hidd_mouse_report(hid_mouse_report_t const *p_report)
{
  VERIFY( tud_hid_mouse_ready() );

  hidd_interface_t * p_hid = &_mse_itf;
  memcpy(p_hid->report_buf, p_report, sizeof(hid_mouse_report_t));

  return dcd_edpt_xfer(TUD_OPT_RHPORT, p_hid->ep_in, p_hid->report_buf, sizeof(hid_mouse_report_t));
}

bool tud_hid_mouse_data(uint8_t buttons, int8_t x, int8_t y, int8_t scroll, int8_t pan)
{
  hid_mouse_report_t report =
  {
      .buttons = buttons,
      .x       = x,
      .y       = y,
      .wheel   = scroll,
//      .pan     = pan
  };

  return hidd_mouse_report( &report );
}

bool tud_hid_mouse_move(int8_t x, int8_t y)
{
  hidd_interface_t * p_hid = &_mse_itf;
  uint8_t prev_buttons = p_hid->report_buf[0];

  return tud_hid_mouse_data(prev_buttons, x, y, 0, 0);
}

bool tud_hid_mouse_scroll(int8_t vertical, int8_t horizontal)
{
  hidd_interface_t * p_hid = &_mse_itf;
  uint8_t prev_buttons = p_hid->report_buf[0];

  return tud_hid_mouse_data(prev_buttons, 0, 0, vertical, horizontal);
}

#endif

static inline hidd_interface_t* get_interface_by_edpt(uint8_t ep_addr)
{
  return ( ep_addr == _kbd_itf.ep_in ) ? &_kbd_itf :
         ( ep_addr == _mse_itf.ep_in ) ? &_mse_itf : NULL;
}

static inline hidd_interface_t* get_interface_by_number(uint8_t itf_num)
{
  return ( itf_num == _kbd_itf.itf_num ) ? &_kbd_itf :
         ( itf_num == _mse_itf.itf_num ) ? &_mse_itf : NULL;
}

//--------------------------------------------------------------------+
// USBD-CLASS API
//--------------------------------------------------------------------+
void hidd_init(void)
{
  hidd_reset(TUD_OPT_RHPORT);
}

void hidd_reset(uint8_t rhport)
{
  #if CFG_TUD_HID_MOUSE
  varclr_(&_mse_itf);
  #endif

  #if CFG_TUD_HID_KEYBOARD
  varclr_(&_kbd_itf);
  #endif
}

tusb_error_t hidd_open(uint8_t rhport, tusb_desc_interface_t const * desc_itf, uint16_t *p_length)
{
  uint8_t const *p_desc = (uint8_t const *) desc_itf;

  //------------- HID descriptor -------------//
  p_desc += p_desc[DESC_OFFSET_LEN];
  tusb_hid_descriptor_hid_t const *desc_hid = (tusb_hid_descriptor_hid_t const *) p_desc;
  TU_ASSERT(HID_DESC_TYPE_HID == desc_hid->bDescriptorType, TUSB_ERROR_HIDD_DESCRIPTOR_INTERFACE);

  //------------- Endpoint Descriptor -------------//
  p_desc += p_desc[DESC_OFFSET_LEN];
  tusb_desc_endpoint_t const *desc_edpt = (tusb_desc_endpoint_t const *) p_desc;
  TU_ASSERT(TUSB_DESC_ENDPOINT == desc_edpt->bDescriptorType, TUSB_ERROR_HIDD_DESCRIPTOR_INTERFACE);

  if (desc_itf->bInterfaceSubClass == HID_SUBCLASS_BOOT)
  {
#if CFG_TUD_HID_BOOT_PROTOCOL
    if ( (desc_itf->bInterfaceProtocol != HID_PROTOCOL_KEYBOARD) && (desc_itf->bInterfaceProtocol != HID_PROTOCOL_MOUSE) )
    {
      // unknown, unsupported protocol
      return TUSB_ERROR_HIDD_DESCRIPTOR_INTERFACE;
    }else
    {
      hidd_interface_t * p_hid = NULL;

      #if CFG_TUD_HID_KEYBOARD
      if (desc_itf->bInterfaceProtocol == HID_PROTOCOL_KEYBOARD)
      {
        p_hid = &_kbd_itf;
        p_hid->report_desc   = tud_desc_set.hid_report.boot_keyboard;
        p_hid->get_report_cb = tud_hid_keyboard_get_report_cb;
        p_hid->set_report_cb = tud_hid_keyboard_set_report_cb;
      }
      #endif

      #if CFG_TUD_HID_MOUSE
      if (desc_itf->bInterfaceProtocol == HID_PROTOCOL_MOUSE)
      {
        p_hid = &_mse_itf;
        p_hid->report_desc   = tud_desc_set.hid_report.boot_mouse;
        p_hid->get_report_cb = tud_hid_mouse_get_report_cb;
        p_hid->set_report_cb = tud_hid_mouse_set_report_cb;
      }
      #endif

      TU_ASSERT(p_hid, TUSB_ERROR_HIDD_DESCRIPTOR_INTERFACE);
      VERIFY(p_hid->report_desc, TUSB_ERROR_DESCRIPTOR_CORRUPTED);

      TU_ASSERT( dcd_edpt_open(rhport, desc_edpt), TUSB_ERROR_DCD_FAILED );

      p_hid->report_len = desc_hid->wReportLength;
      p_hid->itf_num    = desc_itf->bInterfaceNumber;
      p_hid->ep_in      = desc_edpt->bEndpointAddress;
      p_hid->report_id  = 0;

      *p_length = sizeof(tusb_desc_interface_t) + sizeof(tusb_hid_descriptor_hid_t) + sizeof(tusb_desc_endpoint_t);
    }
#else
    return TUSB_ERROR_HIDD_DESCRIPTOR_INTERFACE;
#endif
  }
  else
  {
    // TODO HID generic
    // TODO parse report ID for keyboard, mouse
    *p_length = 0;
    return TUSB_ERROR_HIDD_DESCRIPTOR_INTERFACE;
  }

  return TUSB_ERROR_NONE;
}

tusb_error_t hidd_control_request_st(uint8_t rhport, tusb_control_request_t const * p_request)
{
  hidd_interface_t* p_hid = get_interface_by_number( (uint8_t) p_request->wIndex );
  TU_ASSERT(p_hid, TUSB_ERROR_FAILED);

  OSAL_SUBTASK_BEGIN

  //------------- STD Request -------------//
  if (p_request->bmRequestType_bit.type == TUSB_REQ_TYPE_STANDARD)
  {
    uint8_t const desc_type  = u16_high_u8(p_request->wValue);
    uint8_t const desc_index = u16_low_u8 (p_request->wValue);

    (void) desc_index;

    if (p_request->bRequest == TUSB_REQ_GET_DESCRIPTOR && desc_type == HID_DESC_TYPE_REPORT)
    {
      STASK_ASSERT ( p_hid->report_len <= CFG_TUD_CTRL_BUFSIZE );

      // use device control buffer (in USB SRAM)
      memcpy(_usbd_ctrl_buf, p_hid->report_desc, p_hid->report_len);

      usbd_control_xfer_st(rhport, p_request->bmRequestType_bit.direction, _usbd_ctrl_buf, p_hid->report_len);
    }else
    {
      dcd_control_stall(rhport);
    }
  }
  //------------- Class Specific Request -------------//
  else if (p_request->bmRequestType_bit.type == TUSB_REQ_TYPE_CLASS)
  {
    if( HID_REQ_CONTROL_GET_REPORT == p_request->bRequest )
    {
      // wValue = Report Type | Report ID
      uint8_t const report_type = u16_high_u8(p_request->wValue);
      uint8_t const report_id   = u16_low_u8(p_request->wValue);

      // Composite interface need to determine it is Keyboard, Mouse or Gamepad
      if ( report_id > 0 )
      {

      }

      uint16_t xferlen;
      if ( p_hid->get_report_cb )
      {
        xferlen = p_hid->get_report_cb((hid_report_type_t) report_type, p_hid->report_buf, p_request->wLength);
      }else
      {
        xferlen = p_request->wLength;
        // re-use report_buf -> report has no change
      }

      STASK_ASSERT( xferlen > 0 );
      usbd_control_xfer_st(rhport, p_request->bmRequestType_bit.direction, p_hid->report_buf, xferlen);
    }
    else if ( HID_REQ_CONTROL_SET_REPORT == p_request->bRequest )
    {
      // wValue = Report Type | Report ID
      usbd_control_xfer_st(rhport, p_request->bmRequestType_bit.direction, _usbd_ctrl_buf, p_request->wLength);

      if ( p_hid->set_report_cb )
      {
        p_hid->set_report_cb(u16_high_u8(p_request->wValue), _usbd_ctrl_buf, p_request->wLength);
      }
    }
    else if (HID_REQ_CONTROL_SET_IDLE == p_request->bRequest)
    {
      p_hid->idle_rate = u16_high_u8(p_request->wValue);
      dcd_control_status(rhport, p_request->bmRequestType_bit.direction);
    }
    else if (HID_REQ_CONTROL_GET_IDLE == p_request->bRequest)
    {
      _usbd_ctrl_buf[0] = p_hid->idle_rate;
      usbd_control_xfer_st(rhport, p_request->bmRequestType_bit.direction, _usbd_ctrl_buf, 1);
    }
    else if (HID_REQ_CONTROL_GET_PROTOCOL == p_request->bRequest )
    {
      _usbd_ctrl_buf[0] = 1 - CFG_TUD_HID_BOOT_PROTOCOL; // 0 is Boot, 1 is Report protocol
      usbd_control_xfer_st(rhport, p_request->bmRequestType_bit.direction, _usbd_ctrl_buf, 1);
    }
    else
    {
//      HID_REQ_CONTROL_SET_PROTOCOL:
      dcd_control_stall(rhport);
    }
  }else
  {
    dcd_control_stall(rhport);
  }

  OSAL_SUBTASK_END
}

tusb_error_t hidd_xfer_cb(uint8_t rhport, uint8_t edpt_addr, tusb_event_t event, uint32_t xferred_bytes)
{
  // nothing to do
  return TUSB_ERROR_NONE;
}


/*------------------------------------------------------------------*/
/* Ascii to Keycode
 *------------------------------------------------------------------*/
const hid_ascii_to_keycode_entry_t HID_ASCII_TO_KEYCODE[128] =
{
    {0, 0                     }, // 0x00 Null
    {0, 0                     }, // 0x01
    {0, 0                     }, // 0x02
    {0, 0                     }, // 0x03
    {0, 0                     }, // 0x04
    {0, 0                     }, // 0x05
    {0, 0                     }, // 0x06
    {0, 0                     }, // 0x07
    {0, HID_KEY_BACKSPACE     }, // 0x08 Backspace
    {0, HID_KEY_TAB           }, // 0x09 Horizontal Tab
    {0, HID_KEY_RETURN        }, // 0x0A Line Feed
    {0, 0                     }, // 0x0B
    {0, 0                     }, // 0x0C
    {0, HID_KEY_RETURN        }, // 0x0D Carriage return
    {0, 0                     }, // 0x0E
    {0, 0                     }, // 0x0F
    {0, 0                     }, // 0x10
    {0, 0                     }, // 0x11
    {0, 0                     }, // 0x12
    {0, 0                     }, // 0x13
    {0, 0                     }, // 0x14
    {0, 0                     }, // 0x15
    {0, 0                     }, // 0x16
    {0, 0                     }, // 0x17
    {0, 0                     }, // 0x18
    {0, 0                     }, // 0x19
    {0, 0                     }, // 0x1A
    {0, HID_KEY_ESCAPE        }, // 0x1B Escape
    {0, 0                     }, // 0x1C
    {0, 0                     }, // 0x1D
    {0, 0                     }, // 0x1E
    {0, 0                     }, // 0x1F

    {0, HID_KEY_SPACE         }, // 0x20
    {1, HID_KEY_1             }, // 0x21 !
    {1, HID_KEY_APOSTROPHE    }, // 0x22 "
    {1, HID_KEY_3             }, // 0x23 #
    {1, HID_KEY_4             }, // 0x24 $
    {1, HID_KEY_5             }, // 0x25 %
    {1, HID_KEY_7             }, // 0x26 &
    {0, HID_KEY_APOSTROPHE    }, // 0x27 '
    {1, HID_KEY_9             }, // 0x28 (
    {1, HID_KEY_0             }, // 0x29 )
    {1, HID_KEY_8             }, // 0x2A *
    {1, HID_KEY_EQUAL         }, // 0x2B +
    {0, HID_KEY_COMMA         }, // 0x2C ,
    {0, HID_KEY_MINUS         }, // 0x2D -
    {0, HID_KEY_PERIOD        }, // 0x2E .
    {0, HID_KEY_SLASH         }, // 0x2F /
    {0, HID_KEY_0             }, // 0x30 0
    {0, HID_KEY_1             }, // 0x31 1
    {0, HID_KEY_2             }, // 0x32 2
    {0, HID_KEY_3             }, // 0x33 3
    {0, HID_KEY_4             }, // 0x34 4
    {0, HID_KEY_5             }, // 0x35 5
    {0, HID_KEY_6             }, // 0x36 6
    {0, HID_KEY_7             }, // 0x37 7
    {0, HID_KEY_8             }, // 0x38 8
    {0, HID_KEY_9             }, // 0x39 9
    {1, HID_KEY_SEMICOLON     }, // 0x3A :
    {0, HID_KEY_SEMICOLON     }, // 0x3B ;
    {1, HID_KEY_COMMA         }, // 0x3C <
    {0, HID_KEY_EQUAL         }, // 0x3D =
    {1, HID_KEY_PERIOD        }, // 0x3E >
    {1, HID_KEY_SLASH         }, // 0x3F ?

    {1, HID_KEY_2             }, // 0x40 @
    {1, HID_KEY_A             }, // 0x41 A
    {1, HID_KEY_B             }, // 0x42 B
    {1, HID_KEY_C             }, // 0x43 C
    {1, HID_KEY_D             }, // 0x44 D
    {1, HID_KEY_E             }, // 0x45 E
    {1, HID_KEY_F             }, // 0x46 F
    {1, HID_KEY_G             }, // 0x47 G
    {1, HID_KEY_H             }, // 0x48 H
    {1, HID_KEY_I             }, // 0x49 I
    {1, HID_KEY_J             }, // 0x4A J
    {1, HID_KEY_K             }, // 0x4B K
    {1, HID_KEY_L             }, // 0x4C L
    {1, HID_KEY_M             }, // 0x4D M
    {1, HID_KEY_N             }, // 0x4E N
    {1, HID_KEY_O             }, // 0x4F O
    {1, HID_KEY_P             }, // 0x50 P
    {1, HID_KEY_Q             }, // 0x51 Q
    {1, HID_KEY_R             }, // 0x52 R
    {1, HID_KEY_S             }, // 0x53 S
    {1, HID_KEY_T             }, // 0x55 T
    {1, HID_KEY_U             }, // 0x55 U
    {1, HID_KEY_V             }, // 0x56 V
    {1, HID_KEY_W             }, // 0x57 W
    {1, HID_KEY_X             }, // 0x58 X
    {1, HID_KEY_Y             }, // 0x59 Y
    {1, HID_KEY_Z             }, // 0x5A Z
    {0, HID_KEY_BRACKET_LEFT  }, // 0x5B [
    {0, HID_KEY_BACKSLASH     }, // 0x5C '\'
    {0, HID_KEY_BRACKET_RIGHT }, // 0x5D ]
    {1, HID_KEY_6             }, // 0x5E ^
    {1, HID_KEY_MINUS         }, // 0x5F _

    {0, HID_KEY_GRAVE         }, // 0x60 `
    {0, HID_KEY_A             }, // 0x61 a
    {0, HID_KEY_B             }, // 0x62 b
    {0, HID_KEY_C             }, // 0x63 c
    {0, HID_KEY_D             }, // 0x66 d
    {0, HID_KEY_E             }, // 0x65 e
    {0, HID_KEY_F             }, // 0x66 f
    {0, HID_KEY_G             }, // 0x67 g
    {0, HID_KEY_H             }, // 0x68 h
    {0, HID_KEY_I             }, // 0x69 i
    {0, HID_KEY_J             }, // 0x6A j
    {0, HID_KEY_K             }, // 0x6B k
    {0, HID_KEY_L             }, // 0x6C l
    {0, HID_KEY_M             }, // 0x6D m
    {0, HID_KEY_N             }, // 0x6E n
    {0, HID_KEY_O             }, // 0x6F o
    {0, HID_KEY_P             }, // 0x70 p
    {0, HID_KEY_Q             }, // 0x71 q
    {0, HID_KEY_R             }, // 0x72 r
    {0, HID_KEY_S             }, // 0x73 s
    {0, HID_KEY_T             }, // 0x75 t
    {0, HID_KEY_U             }, // 0x75 u
    {0, HID_KEY_V             }, // 0x76 v
    {0, HID_KEY_W             }, // 0x77 w
    {0, HID_KEY_X             }, // 0x78 x
    {0, HID_KEY_Y             }, // 0x79 y
    {0, HID_KEY_Z             }, // 0x7A z
    {1, HID_KEY_BRACKET_LEFT  }, // 0x7B {
    {1, HID_KEY_BACKSLASH     }, // 0x7C |
    {1, HID_KEY_BRACKET_RIGHT }, // 0x7D }
    {1, HID_KEY_GRAVE         }, // 0x7E ~
    {0, HID_KEY_DELETE        }  // 0x7F Delete
};

#endif