remapper_dual_b.cc

#include <bsp/board.h>
#include <tusb.h>

#include "hardware/watchdog.h"
#include "pico/stdio.h"
#include "pico/time.h"

#include "dual.h"
#include "interval_override.h"
#include "serial.h"

bool led_state;
uint8_t buffer[SERIAL_MAX_PAYLOAD_SIZE + sizeof(device_connected_t)];
bool initialized = false;

void serial_callback(const uint8_t* data, uint16_t len) {
    switch ((DualCommand) data[0]) {
        case DualCommand::B_INIT:
            interval_override = ((b_init_t*) data)->interval_override;
            initialized = true;
            break;
        case DualCommand::RESTART:
            watchdog_reboot(0, 0, 0);
            break;
        default:
            break;
    }
}

void request_b_init() {
    request_b_init_t msg;
    serial_write((uint8_t*) &msg, sizeof(msg));
}

int main() {
    serial_init();
    board_init();

    while (!initialized) {
        request_b_init();
        serial_read(serial_callback);
    }

    tusb_init();

    while (true) {
        tuh_task();
        serial_read(serial_callback);
    }

    return 0;
}

void tuh_hid_report_received_cb(uint8_t dev_addr, uint8_t instance, uint8_t const* report, uint16_t len) {
    led_state = !led_state;
    board_led_write(led_state);

    report_received_t* msg = (report_received_t*) buffer;
    msg->command = DualCommand::REPORT_RECEIVED;
    msg->dev_addr = dev_addr;
    msg->interface = instance;
    memcpy(msg->report, report, len);
    serial_write((uint8_t*) msg, len + sizeof(report_received_t));
    tuh_hid_receive_report(dev_addr, instance);
}

void tuh_hid_mount_cb(uint8_t dev_addr, uint8_t instance, uint8_t const* desc_report, uint16_t desc_len) {
    printf("tuh_hid_mount_cb\n");
    stdio_flush();
    device_connected_t* msg = (device_connected_t*) buffer;
    msg->command = DualCommand::DEVICE_CONNECTED;
    uint16_t vid;
    uint16_t pid;
    tuh_vid_pid_get(dev_addr, &vid, &pid);
    msg->vid = vid;
    msg->pid = pid;
    msg->dev_addr = dev_addr;
    msg->interface = instance;
    memcpy(msg->report_descriptor, desc_report, desc_len);
    serial_write((uint8_t*) msg, desc_len + sizeof(device_connected_t));
    tuh_hid_receive_report(dev_addr, instance);
}

void tuh_hid_umount_cb(uint8_t dev_addr, uint8_t instance) {
    printf("tuh_hid_umount_cb %d %d\n", dev_addr, instance);
    stdio_flush();
    device_disconnected_t msg;
    msg.dev_addr = dev_addr;
    msg.interface = instance;
    serial_write((uint8_t*) &msg, sizeof(msg));
}