// SPDX-License-Identifier: GPL-2.0
#include <asm-generic/errno-base.h>
#include <asm-generic/int-ll64.h>
#include <linux/bitops.h>
#include <linux/input.h>
#include <linux/input-event-codes.h>
#include <linux/module.h>
#include <linux/hid.h>
#include <linux/usb.h>

#include "hid-ids.h"

#define RAPOO_BTN_BACK			0x08
#define RAPOO_BTN_FORWARD		0x10

static const struct hid_device_id rapoo_devices[] = {
	{ HID_USB_DEVICE(USB_VENDOR_ID_RAPOO, USB_DEVICE_ID_RAPOO_2_4G_RECEIVER) },
	{ }
};
MODULE_DEVICE_TABLE(hid, rapoo_devices);

static int rapoo_probe(struct hid_device *hdev, const struct hid_device_id *id)
{
	int ret;
	struct input_dev *input;

	ret = hid_parse(hdev);
	if (ret) {
		hid_err(hdev, "parse failed\n");
		return ret;
	}

	ret = hid_hw_start(hdev, HID_CONNECT_DEFAULT);
	if (ret) {
		hid_err(hdev, "start failed\n");
		return ret;
	}

	if (hdev->bus == BUS_USB) {
		struct usb_interface *intf = to_usb_interface(hdev->dev.parent);

		if (intf->cur_altsetting->desc.bInterfaceNumber != 1)
			return 0;
	}

	input = devm_input_allocate_device(&hdev->dev);
	if (!input)
		return -ENOMEM;

	input->name = "Rapoo 2.4G Wireless Mouse";
	input->phys = "rapoo/input1";
	input->id.bustype = hdev->bus;
	input->id.vendor = hdev->vendor;
	input->id.product = hdev->product;
	input->id.version = hdev->version;

	__set_bit(EV_KEY, input->evbit);
	__set_bit(KEY_BACK, input->keybit);
	__set_bit(KEY_FORWARD, input->keybit);

	ret = input_register_device(input);
	if (ret)
		return ret;

	hid_set_drvdata(hdev, input);

	return ret;
}

static int rapoo_raw_event(struct hid_device *hdev, struct hid_report *report, u8 *data, int size)
{
	struct input_dev *input = hid_get_drvdata(hdev);

	if (!input)
		return 0;

	if (report->id == 1 && size >= 2) {
		u8 btn = data[1];

		input_report_key(input, KEY_BACK, btn & RAPOO_BTN_BACK);
		input_report_key(input, KEY_FORWARD, btn & RAPOO_BTN_FORWARD);
		input_sync(input);
		return 1;
	}

	return 0;
}

static struct hid_driver rapoo_driver = {
	.name = "hid-rapoo",
	.id_table = rapoo_devices,
	.probe = rapoo_probe,
	.raw_event = rapoo_raw_event,
};

module_hid_driver(rapoo_driver);

MODULE_LICENSE("GPL");
MODULE_AUTHOR("Nguyen Dinh Dang Duong <dangduong31205@gmail.com>");
MODULE_DESCRIPTION("RAPOO 2.4G Wireless Device Driver");