Momentum-Firmware/applications/services/desktop/scenes/desktop_scene_lock_menu.c

#include <gui/scene_manager.h>
#include <applications.h>
#include <furi_hal.h>
#include <toolbox/saved_struct.h>
#include <stdbool.h>
#include <loader/loader.h>
#include <momentum/momentum.h>

#include "../desktop_i.h"
#include <desktop/desktop_settings.h>
#include "../views/desktop_view_lock_menu.h"
#include "desktop_scene.h"
#include "../helpers/pin_code.h"
#include <power/power_service/power.h>

#define TAG "DesktopSceneLock"

typedef enum {
    CheckPinNo,
    CheckPinLock,
    CheckPinLockOff,
} CheckPin;

void desktop_scene_lock_menu_callback(DesktopEvent event, void* context) {
    Desktop* desktop = (Desktop*)context;
    view_dispatcher_send_custom_event(desktop->view_dispatcher, event);
}

void desktop_scene_lock_menu_on_enter(void* context) {
    Desktop* desktop = (Desktop*)context;

    scene_manager_set_scene_state(desktop->scene_manager, DesktopSceneLockMenu, CheckPinNo);
    desktop_lock_menu_set_callback(desktop->lock_menu, desktop_scene_lock_menu_callback, desktop);
    desktop_lock_menu_set_pin_state(desktop->lock_menu, desktop_pin_code_is_set());
    desktop_lock_menu_set_stealth_mode_state(
        desktop->lock_menu, furi_hal_rtc_is_flag_set(FuriHalRtcFlagStealthMode));
    desktop_lock_menu_set_idx(desktop->lock_menu, 3);

    gui_set_hide_statusbar(desktop->gui, true);

    view_dispatcher_switch_to_view(desktop->view_dispatcher, DesktopViewIdLockMenu);
}

void desktop_scene_lock_menu_save_settings(Desktop* desktop) {
    if(desktop->lock_menu->save_notification) {
        notification_message_save_settings(desktop->lock_menu->notification);
        desktop->lock_menu->save_notification = false;
    }
    if(desktop->lock_menu->save_momentum) {
        momentum_settings_save();
        desktop->lock_menu->save_momentum = false;
    }
    if(desktop->lock_menu->save_bt) {
        bt_settings_save(&desktop->lock_menu->bt->bt_settings);
        desktop->lock_menu->save_bt = false;
    }
}

bool desktop_scene_lock_menu_on_event(void* context, SceneManagerEvent event) {
    Desktop* desktop = (Desktop*)context;
    bool consumed = false;

    if(event.type == SceneManagerEventTypeTick) {
        CheckPin check_pin_changed =
            scene_manager_get_scene_state(desktop->scene_manager, DesktopSceneLockMenu);
        if(check_pin_changed != CheckPinNo && desktop_pin_code_is_set()) {
            desktop_lock_menu_set_pin_state(desktop->lock_menu, true);
            scene_manager_set_scene_state(
                desktop->scene_manager, DesktopSceneLockMenu, CheckPinNo);
            desktop_lock(desktop, true);
            if(check_pin_changed == CheckPinLockOff) {
                Power* power = furi_record_open(RECORD_POWER);
                furi_delay_ms(500);
                power_off(power);
                furi_record_close(RECORD_POWER);
            }
        }
    } else if(event.type == SceneManagerEventTypeCustom) {
        switch(event.event) {
        case DesktopLockMenuEventSettings:
            desktop_scene_lock_menu_save_settings(desktop);
            loader_show_settings(furi_record_open(RECORD_LOADER));
            furi_record_close(RECORD_LOADER);
            consumed = true;
            break;
        case DesktopLockMenuEventLockKeypad:
            desktop_scene_lock_menu_save_settings(desktop);
            desktop_lock(desktop, false);
            consumed = true;
            break;
        case DesktopLockMenuEventLockPinCode:
            desktop_scene_lock_menu_save_settings(desktop);
            if(desktop_pin_code_is_set()) {
                desktop_lock(desktop, true);
            } else {
                loader_start_detached_with_gui_error(
                    desktop->loader, "Desktop", DESKTOP_SETTINGS_RUN_PIN_SETUP_ARG);
                scene_manager_set_scene_state(
                    desktop->scene_manager, DesktopSceneLockMenu, CheckPinLock);
            }
            consumed = true;
            break;
        case DesktopLockMenuEventLockPinOff:
            desktop_scene_lock_menu_save_settings(desktop);
            if(desktop_pin_code_is_set()) {
                desktop_lock(desktop, true);
                Power* power = furi_record_open(RECORD_POWER);
                furi_delay_ms(500);
                power_off(power);
                furi_record_close(RECORD_POWER);
            } else {
                loader_start_detached_with_gui_error(
                    desktop->loader, "Desktop", DESKTOP_SETTINGS_RUN_PIN_SETUP_ARG);
                scene_manager_set_scene_state(
                    desktop->scene_manager, DesktopSceneLockMenu, CheckPinLockOff);
            }
            consumed = true;
            break;
        case DesktopLockMenuEventMomentum:
            desktop_scene_lock_menu_save_settings(desktop);
            loader_start_detached_with_gui_error(desktop->loader, "Momentum", NULL);
            consumed = true;
            break;
        case DesktopLockMenuEventScreenSettings:
            desktop_scene_lock_menu_save_settings(desktop);
            loader_start_detached_with_gui_error(desktop->loader, "Momentum", "MiscScreen");
            consumed = true;
            break;
        case DesktopLockMenuEventStealthModeOn:
            desktop_set_stealth_mode_state(desktop, true);
            break;
        case DesktopLockMenuEventStealthModeOff:
            desktop_set_stealth_mode_state(desktop, false);
            break;
        default:
            break;
        }
    } else if(event.type == SceneManagerEventTypeBack) {
        scene_manager_set_scene_state(desktop->scene_manager, DesktopSceneLockMenu, CheckPinNo);
    }
    return consumed;
}

void desktop_scene_lock_menu_on_exit(void* context) {
    Desktop* desktop = (Desktop*)context;
    desktop_scene_lock_menu_save_settings(desktop);

    gui_set_hide_statusbar(desktop->gui, false);
}