// SPDX-FileCopyrightText: Copyright 2022 yuzu Emulator Project
// SPDX-License-Identifier: GPL-2.0-or-later

#include "common/logging/log.h"
#include "input_common/helpers/joycon_protocol/generic_functions.h"

namespace InputCommon::Joycon {

GenericProtocol::GenericProtocol(std::shared_ptr<JoyconHandle> handle)
    : JoyconCommonProtocol(handle) {}

DriverResult GenericProtocol::EnablePassiveMode() {
    SetBlocking();
    const auto result = SetReportMode(ReportMode::SIMPLE_HID_MODE);
    SetNonBlocking();
    return result;
}

DriverResult GenericProtocol::EnableActiveMode() {
    SetBlocking();
    const auto result = SetReportMode(ReportMode::STANDARD_FULL_60HZ);
    SetNonBlocking();
    return result;
}

DriverResult GenericProtocol::GetDeviceInfo(DeviceInfo& device_info) {
    std::vector<u8> output;
    SetBlocking();

    const auto result = SendSubCommand(SubCommand::REQ_DEV_INFO, {}, output);

    device_info = {};
    if (result == DriverResult::Success) {
        memcpy(&device_info, output.data(), sizeof(DeviceInfo));
    }

    SetNonBlocking();
    return result;
}

DriverResult GenericProtocol::GetControllerType(ControllerType& controller_type) {
    return GetDeviceType(controller_type);
}

DriverResult GenericProtocol::EnableImu(bool enable) {
    const std::vector<u8> buffer{static_cast<u8>(enable ? 1 : 0)};
    std::vector<u8> output;
    SetBlocking();
    const auto result = SendSubCommand(SubCommand::ENABLE_IMU, buffer, output);
    SetNonBlocking();
    return result;
}

DriverResult GenericProtocol::SetImuConfig(GyroSensitivity gsen, GyroPerformance gfrec,
                                           AccelerometerSensitivity asen,
                                           AccelerometerPerformance afrec) {
    const std::vector<u8> buffer{static_cast<u8>(gsen), static_cast<u8>(asen),
                                 static_cast<u8>(gfrec), static_cast<u8>(afrec)};
    std::vector<u8> output;
    SetBlocking();
    const auto result = SendSubCommand(SubCommand::SET_IMU_SENSITIVITY, buffer, output);
    SetNonBlocking();
    return result;
}

DriverResult GenericProtocol::GetBattery(u32& battery_level) {
    battery_level = 0;
    return DriverResult::NotSupported;
}

DriverResult GenericProtocol::GetColor(Color& color) {
    std::vector<u8> buffer;
    SetBlocking();
    const auto result = ReadSPI(CalAddr::COLOR_DATA, 12, buffer);
    SetNonBlocking();

    color = {};
    if (result == DriverResult::Success) {
        color.body = static_cast<u32>((buffer[0] << 16) | (buffer[1] << 8) | buffer[2]);
        color.buttons = static_cast<u32>((buffer[3] << 16) | (buffer[4] << 8) | buffer[5]);
        color.left_grip = static_cast<u32>((buffer[6] << 16) | (buffer[7] << 8) | buffer[8]);
        color.right_grip = static_cast<u32>((buffer[9] << 16) | (buffer[10] << 8) | buffer[11]);
    }

    return result;
}

DriverResult GenericProtocol::GetSerialNumber(SerialNumber& serial_number) {
    std::vector<u8> buffer;
    SetBlocking();
    const auto result = ReadSPI(CalAddr::SERIAL_NUMBER, 16, buffer);
    SetNonBlocking();

    serial_number = {};
    if (result == DriverResult::Success) {
        memcpy(serial_number.data(), buffer.data() + 1, sizeof(SerialNumber));
    }

    return result;
}

DriverResult GenericProtocol::GetTemperature(u32& temperature) {
    // Not all devices have temperature sensor
    temperature = 25;
    return DriverResult::NotSupported;
}

DriverResult GenericProtocol::GetVersionNumber(FirmwareVersion& version) {
    DeviceInfo device_info{};

    const auto result = GetDeviceInfo(device_info);
    version = device_info.firmware;

    return result;
}

DriverResult GenericProtocol::SetHomeLight() {
    const std::vector<u8> buffer{0x0f, 0xf0, 0x00};
    std::vector<u8> output;
    SetBlocking();

    const auto result = SendSubCommand(SubCommand::SET_HOME_LIGHT, buffer, output);

    SetNonBlocking();
    return result;
}

DriverResult GenericProtocol::SetLedBusy() {
    return DriverResult::NotSupported;
}

DriverResult GenericProtocol::SetLedPattern(u8 leds) {
    const std::vector<u8> buffer{leds};
    std::vector<u8> output;
    SetBlocking();

    const auto result = SendSubCommand(SubCommand::SET_PLAYER_LIGHTS, buffer, output);

    SetNonBlocking();
    return result;
}

DriverResult GenericProtocol::SetLedBlinkPattern(u8 leds) {
    return SetLedPattern(static_cast<u8>(leds << 4));
}

} // namespace InputCommon::Joycon