#include "karabiner_virtual_hid_device_methods.hpp"
#include <CoreFoundation/CoreFoundation.h>
#include <IOKit/IOKitLib.h>
#include <IOKit/hid/IOHIDDevice.h>
#include <IOKit/hid/IOHIDElement.h>
#include <IOKit/hid/IOHIDManager.h>
#include <IOKit/hid/IOHIDQueue.h>
#include <IOKit/hid/IOHIDValue.h>
#include <IOKit/hidsystem/IOHIDShared.h>
#include <IOKit/hidsystem/ev_keymap.h>
#include <cmath>
#include <iostream>
#include <thread>

int main(int argc, const char* argv[]) {
  if (getuid() != 0) {
    std::cerr << "virtual_pointing_example requires root privilege." << std::endl;
  }

  kern_return_t kr;
  io_connect_t connect = IO_OBJECT_NULL;
  auto service = IOServiceGetMatchingService(kIOMasterPortDefault, IOServiceNameMatching(pqrs::karabiner_virtual_hid_device::get_virtual_hid_root_name()));
  if (!service) {
    std::cerr << "IOServiceGetMatchingService error" << std::endl;
    goto finish;
  }

  kr = IOServiceOpen(service, mach_task_self(), kIOHIDServerConnectType, &connect);
  if (kr != KERN_SUCCESS) {
    std::cerr << "IOServiceOpen error" << std::endl;
    goto finish;
  }

  kr = pqrs::karabiner_virtual_hid_device_methods::initialize_virtual_hid_pointing(connect);
  if (kr != KERN_SUCCESS) {
    std::cerr << "initialize_virtual_hid_pointing error" << std::endl;
  }

  {
    std::this_thread::sleep_for(std::chrono::milliseconds(100));

    pqrs::karabiner_virtual_hid_device::hid_report::pointing_input report;
    report.buttons.insert(2);

    kr = pqrs::karabiner_virtual_hid_device_methods::post_pointing_input_report(connect, report);
    if (kr != KERN_SUCCESS) {
      std::cerr << "post_pointing_input_report error" << std::endl;
    }

    std::this_thread::sleep_for(std::chrono::milliseconds(100));

    report.buttons.clear();

    kr = pqrs::karabiner_virtual_hid_device_methods::post_pointing_input_report(connect, report);
    if (kr != KERN_SUCCESS) {
      std::cerr << "post_pointing_input_report error" << std::endl;
    }

    std::this_thread::sleep_for(std::chrono::milliseconds(100));
  }

  for (int i = 0; i < 400; ++i) {
    pqrs::karabiner_virtual_hid_device::hid_report::pointing_input report;
    report.x = static_cast<uint8_t>(cos(0.1 * i) * 20);
    report.y = static_cast<uint8_t>(sin(0.1 * i) * 20);

    kr = pqrs::karabiner_virtual_hid_device_methods::post_pointing_input_report(connect, report);
    if (kr != KERN_SUCCESS) {
      std::cerr << "post_pointing_input_report error" << std::endl;
    }

    std::this_thread::sleep_for(std::chrono::milliseconds(5));
  }

  kr = pqrs::karabiner_virtual_hid_device_methods::reset_virtual_hid_pointing(connect);
  if (kr != KERN_SUCCESS) {
    std::cerr << "reset_virtual_hid_pointing error" << std::endl;
  }

finish:
  if (connect) {
    IOServiceClose(connect);
  }
  if (service) {
    IOObjectRelease(service);
  }

  return 0;
}