// SPDX-License-Identifier: GPL-3.0-only

use crate::daemon::client::load_audio_themes;

use crate::state::{AudioTheme, ContextPage, DaemonStatus, RecordingStatus};

use crate::ui::icons;

use crate::ui::messages::{Message, ModelMessage, RecordingMessage};

use cosmic::prelude::*;

use cosmic::widget::nav_bar;

use std::collections::HashMap;

use super::{AppModel, DeviceState};

/// Builds the navigation bar with all Super STT pages inserted in order.

fn build_nav() -> nav_bar::Model {

    let mut nav = nav_bar::Model::default();

    // Models is the primary page (the active backend) — first in the rail and

    // active on launch. Library (manage/install backends) sits directly below.

    nav.insert()

        .text("Models")

        .data::<crate::state::Page>(crate::state::Page::Models)

        .icon(icons::phosphor(icons::BRAIN))

        .activate();

    nav.insert()

        .text("Library")

        .data::<crate::state::Page>(crate::state::Page::Library)

        .icon(icons::phosphor(icons::BOOKS));

    nav.insert()

        .text("Customization")

        .data::<crate::state::Page>(crate::state::Page::Customization)

        .icon(icons::phosphor(icons::GEAR));

    nav.insert()

        .text("Recording")

        .data::<crate::state::Page>(crate::state::Page::Recording)

        .icon(icons::phosphor(icons::MICROPHONE));

    nav.insert()

        .text("Input Simulation")

        .data::<crate::state::Page>(crate::state::Page::InputSimulation)

        .icon(icons::phosphor(icons::KEYBOARD));

    nav.insert()

        .text("Connection")

        .data::<crate::state::Page>(crate::state::Page::Connection)

        .icon(icons::phosphor(icons::PLUG));

    nav.insert()

        .text("Updates")

        .data::<crate::state::Page>(crate::state::Page::Updates)

        .icon(icons::phosphor(icons::ARROWS_CLOCKWISE));

    nav

}

/// Builds the initial batch of startup tasks (audio themes, daemon ping, data load).

fn initial_load_tasks(

    title_command: Task<cosmic::Action<Message>>,

) -> Task<cosmic::Action<Message>> {

    // Load audio themes on startup (always available)

    let load_themes = Task::perform(load_audio_themes(), |themes| {

        cosmic::Action::App(Message::Recording(RecordingMessage::AudioThemesLoaded(

            themes,

        )))

    });

    // Try to ping the daemon on startup

    let initial_ping = crate::core::app::handlers::tasks::ping_task();

    // Load initial data (models + device info) on startup

    let load_initial_data = Task::perform(

        async move {

            // Small delay to let daemon connection establish

            tokio::time::sleep(tokio::time::Duration::from_millis(500)).await;

        },

        |()| cosmic::Action::App(Message::Model(ModelMessage::LoadInitialData)),

    );

    Task::batch([title_command, load_themes, initial_ping, load_initial_data])

}

impl AppModel {

    /// Initializes the application with any given flags and startup commands.

    pub(super) fn init_model(

        core: cosmic::Core,

        _flags: (),

    ) -> (Self, Task<cosmic::Action<Message>>) {

        let nav = build_nav();

        // Construct the app model with the runtime's core.

        let mut app = AppModel {

            core,

            context_page: ContextPage::default(),

            nav,

            // Initialize Super STT state using proper socket path

            socket_path: super_stt_shared::validation::get_http_socket_path(),

            daemon_status: DaemonStatus::Disconnected,

            reconnect_retry: super_stt_shared::daemon::retry::RetryStrategy::for_initial_connection(

            ),

            recording_status: RecordingStatus::Idle,

            transcription_text: String::new(),

            preview_text: String::new(),

            audio_level: 0.0,

            is_speech_detected: false,

            audio_themes: Vec::new(),

            selected_audio_theme: AudioTheme::default(),

            last_non_silent_theme: AudioTheme::default(),

            udp_restart_counter: 0,

            last_udp_data: std::time::Instant::now(),

            // Initialize model state

            current_model: String::new(),

            current_source: String::new(),

            current_model_epoch: 0,

            model_operations: crate::state::model_operations::ModelOperations::opening(

                "Loading initial model state...".to_string(),

            ),

            // Initialize device state

            current_device: String::new(), // Empty until loaded from daemon

            gpu_info: Vec::new(),

            device_state: DeviceState::Ready,

            last_event_timestamp: None,

            // Initialize preview typing state (disabled by default as beta feature)

            preview_typing_enabled: false,

            // Replaced by the daemon's own state as soon as the settings load

            // completes; the default is the daemon's default too (off).

            post_processor: crate::daemon::client::StageState::default(),

            staged_picks: crate::state::staged_picks::StagedPicks::default(),

            device_offers: crate::state::device_offers::DeviceOffers::default(),

            stage_catalog: crate::state::stage_catalog::StageCatalog::default(),

            recording_stop_mode:

                super_stt_shared::models::recording_stop_mode::RecordingStopMode::default(),

            write_method: super_stt_shared::models::write_method::WriteMethod::default(),

            write_method_test_text: String::new(),

            resolved_write_method: None,

            write_method_test_countdown: None,

            notification_method:

                super_stt_shared::models::notification_method::NotificationMethod::default(),

            volume: 100,

            last_committed_volume: 100,

            // Custom models directory

            custom_models_dir: None,

            custom_models_dir_input: String::new(),

            // Models page UI state

            models_page: crate::state::models_page::ModelsPageState::default(),

            // Transcription language state

            language: crate::state::language::LanguageState::default(),

            // Backend catalog + per-backend configuration state

            backends: Vec::new(),

            backend_secret_inputs: HashMap::new(),

            backend_secret_configured: HashMap::new(),

            backend_option_inputs: HashMap::new(),

            // Registry state

            registry: crate::state::registry::RegistryState::default(),

            // Self-update state

            update: crate::state::update::UpdateState::default(),

            // No pending scoped action error at startup.

            action_error: None,

        };

        // Create startup commands

        let title_command = app.update_title();

        (app, initial_load_tasks(title_command))

    }

}