/* Reduced variant of the Flipper Zero SubGhz Class */

#include "subghz_i.h"
#include "../../helpers/xremote_custom_event.h"
#include "../../helpers/xremote_led.h"

SubGhz* subghz_alloc() {
    SubGhz* subghz = malloc(sizeof(SubGhz));

    subghz->file_path = furi_string_alloc();
    subghz->txrx = subghz_txrx_alloc();
    subghz_txrx_set_need_save_callback(subghz->txrx, subghz_save_to_file, subghz);
    subghz->dialogs = furi_record_open(RECORD_DIALOGS);

    return subghz;
}

void subghz_free(SubGhz* subghz) {
    subghz_txrx_free(subghz->txrx);
    furi_string_free(subghz->file_path);
    furi_record_close(RECORD_DIALOGS);

    // The rest
    free(subghz);
}

void subghz_scene_transmit_callback_end_tx(void* context) {
    furi_assert(context);
    FURI_LOG_D(TAG, "callback end");
    XRemote* app = context;
    xremote_cross_remote_set_transmitting(app->cross_remote, XRemoteTransmittingStopSubghz);
}

void subghz_send(void* context, const char* path) {
    XRemote* app = context;

    if(!subghz_load_protocol_from_file(app->subghz, path)) {
        xremote_cross_remote_set_transmitting(app->cross_remote, XRemoteTransmittingStop);
        app->transmitting = false;
        return;
    }

    if(subghz_get_load_type_file(app->subghz) == SubGhzLoadTypeFileRaw) {
        FURI_LOG_D(TAG, "Starting Transmission");
        subghz_txrx_tx_start(
            app->subghz->txrx,
            subghz_txrx_get_fff_data(app->subghz->txrx)); //Seems like it must be done this way

        FURI_LOG_D(TAG, "setting sugbhz raw file encoder worker callback");
        subghz_txrx_set_raw_file_encoder_worker_callback_end(
            app->subghz->txrx, subghz_scene_transmit_callback_end_tx, app);
        app->state_notifications = SubGhzNotificationStateTx;

        FURI_LOG_D(TAG, "Finished Transmitting");
    } else {
        subghz_tx_start(app->subghz, subghz_txrx_get_fff_data(app->subghz->txrx));
        app->state_notifications = SubGhzNotificationStateTx;
        furi_thread_flags_wait(0, FuriFlagWaitAny, app->sg_timing);
        app->state_notifications = SubGhzNotificationStateIDLE;
        subghz_txrx_stop(app->subghz->txrx);

        xremote_cross_remote_set_transmitting(app->cross_remote, XRemoteTransmittingStop);
        app->transmitting = false;
    }
}