#include "pc_monitor.h"

static void render_callback(Canvas* canvas, void* ctx) {
    furi_assert(ctx);
    PcMonitorApp* app = ctx;

    if(app->bt_state == BtStateRecieving) {
        canvas_clear(canvas);
        canvas_set_color(canvas, ColorBlack);
        canvas_set_font(canvas, FontKeyboard);

        uint8_t line = 0;
        char str[32];

		if (app->data.cpu_usage <= 100) {
			canvas_draw_str(canvas, 0, 10, "CPU");
			snprintf(str, 32, "%d%%", app->data.cpu_usage);
			elements_progress_bar_with_text(canvas, BAR_X, 1, BAR_WIDTH, app->data.cpu_usage / 100.0f, str);
			
			line++;
		}

		if (app->data.ram_usage <= 100) {
			canvas_draw_str(canvas, 0, 10 + BAR_MARGIN * line, "RAM");
			snprintf(
				str,
				32,
				"%.1f/%.1f %s",
				(double)(app->data.ram_max * 0.1f * app->data.ram_usage * 0.01f),
				(double)(app->data.ram_max * 0.1f),
				app->data.ram_unit);
			elements_progress_bar_with_text(
				canvas, BAR_X, 1 + BAR_MARGIN * line, BAR_WIDTH, app->data.ram_usage * 0.01f, str);
			
			line++;
		}

		if (app->data.gpu_usage <= 100) {
			canvas_draw_str(canvas, 0, 10 + BAR_MARGIN * line, "GPU");
			snprintf(str, 32, "%d%%", app->data.gpu_usage);
			elements_progress_bar_with_text(
				canvas, BAR_X, 1 + BAR_MARGIN * line, BAR_WIDTH, app->data.gpu_usage / 100.0f, str);
			
			line++;
		}

		if (app->data.vram_usage <= 100) {
			canvas_draw_str(canvas, 0, 10 + BAR_MARGIN * line, "VRAM");
			snprintf(
				str,
				32,
				"%.1f/%.1f %s",
				(double)(app->data.vram_max * 0.1f * app->data.vram_usage * 0.01f),
				(double)(app->data.vram_max * 0.1f),
				app->data.vram_unit); 
			elements_progress_bar_with_text(
				canvas, BAR_X, 1 + BAR_MARGIN * line, BAR_WIDTH, app->data.vram_usage * 0.01f, str);
			
			line++;
		}
    } else {
        canvas_draw_str_aligned(
            canvas,
            64,
            32,
            AlignCenter,
            AlignCenter,
            app->bt_state == BtStateChecking ? "Checking BLE..." :
            app->bt_state == BtStateInactive ? "BLE inactive!" :
            app->bt_state == BtStateWaiting  ? "Waiting for data..." :
                                               "Connection lost!");
    }
}

static void input_callback(InputEvent* input_event, void* ctx) {
    furi_assert(ctx);
    FuriMessageQueue* event_queue = ctx;
    furi_message_queue_put(event_queue, input_event, FuriWaitForever);
}

static uint16_t bt_serial_callback(SerialServiceEvent event, void* ctx) {
    furi_assert(ctx);
    PcMonitorApp* app = ctx;

    if(event.event == SerialServiceEventTypeDataReceived) {
        FURI_LOG_D(
            TAG,
            "SerialServiceEventTypeDataReceived. Size: %u/%u. Data: %s",
            event.data.size,
            sizeof(DataStruct),
            (char*)event.data.buffer);

        if(event.data.size == sizeof(DataStruct)) {
            memcpy(&app->data, event.data.buffer, sizeof(DataStruct));
            app->bt_state = BtStateRecieving;
			app->last_packet = furi_hal_rtc_get_timestamp();
        }
    }

    return 0;
}

static PcMonitorApp* pc_monitor_alloc() {
    PcMonitorApp* app = malloc(sizeof(PcMonitorApp));
    app->view_port = view_port_alloc();
    app->event_queue = furi_message_queue_alloc(8, sizeof(InputEvent));
    app->notification = furi_record_open(RECORD_NOTIFICATION);
    app->gui = furi_record_open(RECORD_GUI);
    app->bt = furi_record_open(RECORD_BT);

	app->data.cpu_usage = UINT8_MAX;
	app->data.ram_usage = UINT8_MAX;
	app->data.gpu_usage = UINT8_MAX;
	app->data.vram_usage = UINT8_MAX;

    gui_add_view_port(app->gui, app->view_port, GuiLayerFullscreen);
    view_port_draw_callback_set(app->view_port, render_callback, app);
    view_port_input_callback_set(app->view_port, input_callback, app->event_queue);
    return app;
}

static void pc_monitor_free(PcMonitorApp* app) {
    gui_remove_view_port(app->gui, app->view_port);
    view_port_free(app->view_port);
    furi_message_queue_free(app->event_queue);
    furi_record_close(RECORD_NOTIFICATION);
    furi_record_close(RECORD_GUI);
    furi_record_close(RECORD_BT);
    free(app);
}

int32_t pc_monitor_app(void* p) {
    UNUSED(p);
    PcMonitorApp* app = pc_monitor_alloc();

    if(furi_hal_bt_is_active()) {
        furi_hal_bt_serial_set_event_callback(BT_SERIAL_BUFFER_SIZE, bt_serial_callback, app);
        furi_hal_bt_start_advertising();

        app->bt_state = BtStateWaiting;
        FURI_LOG_D(TAG, "Bluetooth is active!");
    } else {
        app->bt_state = BtStateInactive;
        FURI_LOG_W(TAG, "Please, enable the Bluetooth and restart the app");
    }

	// Main loop
    InputEvent event;
    while(true) {
		if (furi_message_queue_get(app->event_queue, &event, 1) == FuriStatusOk) {
        	if(event.type == InputTypeShort && event.key == InputKeyBack) break;
		}
		
		if(app->bt_state == BtStateRecieving && (furi_hal_rtc_get_timestamp() - app->last_packet > 5)) app->bt_state = BtStateLost; 
    }

    furi_hal_bt_serial_set_event_callback(0, NULL, NULL);

    pc_monitor_free(app);

    return 0;
}