#include "uds.hpp"
#include "isotp.hpp"
#include "slcan_serial.hpp"

// Create SLCAN driver
slcan::SerialDriver can_driver;
can_driver.open("/dev/cu.usbserial-XXX", 500000);

// Create ISO-TP transport
isotp::Transport transport(can_driver);

// Configure addressing
uds::Address addr;
addr.type = uds::AddressType::Physical;
addr.tx_can_id = 0x7E0;  // Tester -> ECU
addr.rx_can_id = 0x7E8;  // ECU -> Tester
transport.set_address(addr);

// Create UDS client
uds::Client client(transport);

// Enter programming session
auto resp = client.diagnostic_session_control(uds::Session::ProgrammingSession);
if (resp.ok) {
    std::cout << "Programming session active\n";
}

#include "uds_comm_control.hpp"
#include "uds_dtc_control.hpp"

// Safe flash programming with automatic cleanup
{
    // Guards ensure proper restoration
    uds::dtc_control::FlashProgrammingGuard dtc_guard(client);
    uds::comm_control::CommunicationGuard comm_guard(client, &transport);
    
    // 1. Enter programming session
    client.diagnostic_session_control(uds::Session::ProgrammingSession);
    
    // 2. Disable DTC setting (CRITICAL!)
    uds::dtc_control::disable_dtc_setting(client);
    
    // 3. Disable communication
    uds::comm_control::disable_normal_communication(client, &transport);
    
    // 4. Security access
    auto seed_resp = client.security_access_request_seed(1);
    // ... compute key ...
    client.security_access_send_key(1, key);
    
    // 5. Flash programming
    client.request_download(0x00, address, size);
    for (auto& block : data_blocks) {
        client.transfer_data(block.seq, block.data);
    }
    client.request_transfer_exit();
    
    // 6. Reset ECU
    client.ecu_reset(uds::EcuResetType::HardReset);
    
} // Guards automatically restore DTC and communication settings

// Session management
client.diagnostic_session_control(Session::ProgrammingSession);
client.tester_present(suppress_response = true);

// Security
client.security_access_request_seed(level);
client.security_access_send_key(level, key);

// Data services
client.read_data_by_identifier(did);
client.write_data_by_identifier(did, data);

// Programming services
client.request_download(dfi, address, size);
client.transfer_data(block_counter, data);
client.request_transfer_exit();

// Communication control (0x28)
client.communication_control(subfunction, comm_type);

// DTC control (0x85)
client.control_dtc_setting(setting_type);

// ECU control
client.ecu_reset(EcuResetType::HardReset);
client.routine_control(action, routine_id, record);

// Communication Control helpers
uds::comm_control::disable_normal_communication(client, &transport);
uds::comm_control::enable_normal_communication(client, &transport);
uds::comm_control::disable_all_communication(client, &transport);
uds::comm_control::restore_communication(client, &transport);

// DTC Setting helpers
uds::dtc_control::disable_dtc_setting(client);
uds::dtc_control::enable_dtc_setting(client);

// RAII Guards
uds::comm_control::CommunicationGuard comm_guard(client, &transport);
uds::dtc_control::DTCSettingGuard dtc_guard(client);
uds::dtc_control::FlashProgrammingGuard flash_guard(client);