#include "../subghz_i.h"
#include "../views/subghz_transmitter.h"
#include "lib/subghz/protocols/subghz_protocol_princeton.h"

void subghz_scene_transmitter_tx(void* context) {
    SubGhz* subghz = context;
    SubGhzEncoderPrinceton* encoder = subghz_encoder_princeton_alloc();

    subghz_encoder_princeton_reset(encoder, subghz->protocol_result->code_last_found, 4);
    subghz_encoder_princeton_set_te(encoder, subghz->protocol_result);

    subghz_begin(FuriHalSubGhzPresetOokAsync);
    subghz_tx(433920000);

    furi_hal_subghz_start_async_tx(subghz_encoder_princeton_yield, encoder);

    while(!furi_hal_subghz_is_async_tx_complete()) {
        osDelay(20);
    }

    //Stop tx
    furi_hal_subghz_stop_async_tx();
    subghz_end();
    subghz_encoder_princeton_free(encoder);
}

void subghz_scene_transmitter_callback(SubghzTransmitterEvent event, void* context) {
    furi_assert(context);
    SubGhz* subghz = context;
    view_dispatcher_send_custom_event(subghz->view_dispatcher, event);
}

const void subghz_scene_transmitter_on_enter(void* context) {
    SubGhz* subghz = context;
    SubghzTransmitter* subghz_transmitter = subghz->subghz_transmitter;

    subghz_transmitter_set_callback(subghz_transmitter, subghz_scene_transmitter_callback, subghz);

    subghz_transmitter_set_protocol(subghz_transmitter, subghz->protocol_result);
    view_dispatcher_switch_to_view(subghz->view_dispatcher, SubGhzViewTransmitter);
}

const bool subghz_scene_transmitter_on_event(void* context, SceneManagerEvent event) {
    SubGhz* subghz = context;

    if(event.type == SceneManagerEventTypeCustom) {
        if(event.event == SubghzTransmitterEventSend) {
            //scene_manager_next_scene(subghz->scene_manager, SubGhzSceneSaveName);
            subghz_scene_transmitter_tx(subghz);
            return true;
        } else if(event.event == SubghzTransmitterEventBack) {
            scene_manager_search_and_switch_to_previous_scene(
                subghz->scene_manager, SubGhzSceneStart);
            return true;
        }
    }
    return false;
}

const void subghz_scene_transmitter_on_exit(void* context) {
    SubGhz* subghz = context;
    SubghzTransmitter* subghz_transmitter = subghz->subghz_transmitter;

    subghz_transmitter_set_callback(subghz_transmitter, NULL, subghz);
}