/*
 * usb_dev.c
 *
 * Copyright (C) 2019-2022  Sylvain Munaut <tnt@246tNt.com>
 * SPDX-License-Identifier: GPL-3.0-or-later
 */

#include <stdint.h>
#include <string.h>

#include <no2usb/usb.h>
#include <no2usb/usb_proto.h>

#include "console.h"
#include "gpsdo.h"
#include "misc.h"

#include "ice1usb_proto.h"


const char *fw_build_str = BUILD_INFO;


static void
_get_gpsdo_status(struct usb_ctrl_req *req, struct usb_xfer *xfer)
{
	struct e1usb_gpsdo_status status;

	gpsdo_get_status(&status);

	memcpy(xfer->data, &status, sizeof(struct e1usb_gpsdo_status));
	xfer->len = sizeof(struct e1usb_gpsdo_status);
}

static void
_get_gpsdo_mode(struct usb_ctrl_req *req, struct usb_xfer *xfer)
{
	xfer->data[0] = gpsdo_enabled() ? ICE1USB_GPSDO_MODE_DISABLED : ICE1USB_GPSDO_MODE_AUTO;
	xfer->len = 1;
}

static void
_set_gpsdo_mode(struct usb_ctrl_req *req, struct usb_xfer *xfer)
{
	gpsdo_enable(req->wValue != ICE1USB_GPSDO_MODE_DISABLED);
}

static void
_get_gpsdo_tune(struct usb_ctrl_req *req, struct usb_xfer *xfer)
{
	uint16_t coarse, fine;
	struct e1usb_gpsdo_tune tune;

	gpsdo_get_tune(&coarse, &fine);
	tune.coarse = coarse;
	tune.fine = fine;

	memcpy(xfer->data, &tune, sizeof(struct e1usb_gpsdo_tune));
	xfer->len = sizeof(struct e1usb_gpsdo_tune);
}

static bool
_set_gpsdo_tune_done(struct usb_xfer *xfer)
{
	const struct e1usb_gpsdo_tune *tune = (const void *) xfer->data;
	gpsdo_set_tune(tune->coarse, tune->fine);
	return true;
}

static void
_set_gpsdo_tune(struct usb_ctrl_req *req, struct usb_xfer *xfer)
{
	xfer->cb_done = _set_gpsdo_tune_done;
	xfer->cb_ctx  = req;
	xfer->len     = sizeof(struct e1usb_gpsdo_tune);
}


static enum usb_fnd_resp
_usb_dev_ctrl_req(struct usb_ctrl_req *req, struct usb_xfer *xfer)
{
	/* Check it's a device-wide vendor request */
	if (USB_REQ_TYPE_RCPT(req) != (USB_REQ_TYPE_VENDOR | USB_REQ_RCPT_DEV))
		return USB_FND_CONTINUE;

	/* Dispatch / Handle */
	switch (req->bRequest) {
	case ICE1USB_DEV_GET_CAPABILITIES:
		xfer->data[0] = (1 << ICE1USB_DEV_CAP_GPSDO);
		xfer->len = 1;
		break;
	case ICE1USB_DEV_GET_FW_BUILD:
		xfer->data = (void*) fw_build_str;
		xfer->len  = strlen(fw_build_str);
		break;
	case ICE1USB_DEV_GET_GPSDO_STATUS:
		_get_gpsdo_status(req, xfer);
		break;
	case ICE1USB_DEV_GET_GPSDO_MODE:
		_get_gpsdo_mode(req, xfer);
		break;
	case ICE1USB_DEV_SET_GPSDO_MODE:
		_set_gpsdo_mode(req, xfer);
		break;
	case ICE1USB_DEV_GET_GPSDO_TUNE:
		_get_gpsdo_tune(req, xfer);
		break;
	case ICE1USB_DEV_SET_GPSDO_TUNE:
		_set_gpsdo_tune(req, xfer);
		break;
	default:
		return USB_FND_ERROR;
	}

	return USB_FND_SUCCESS;
}


static struct usb_fn_drv _dev_drv = {
	.ctrl_req = _usb_dev_ctrl_req,
};

void
usb_dev_init(void)
{
	usb_register_function_driver(&_dev_drv);
}