Yoda/src/app/browser/window/tab/item/page/navigation/request.rs

mod database;
mod identity;
mod primary_icon;
mod search;
mod suggestion;

use super::{ItemAction, Profile};
use adw::{prelude::AdwDialogExt, AlertDialog};
use anyhow::Result;
use gtk::{
    glib::{gformat, GString, Uri, UriFlags},
    prelude::{EditableExt, EntryExt, WidgetExt},
    Entry, EntryIconPosition, StateFlags,
};
use primary_icon::PrimaryIcon;
use sqlite::Transaction;
use std::{cell::Cell, rc::Rc};
use suggestion::Suggestion;

const PREFIX_DOWNLOAD: &str = "download:";
const PREFIX_SOURCE: &str = "source:";

pub trait Request {
    // Constructors

    fn request(item_action: &Rc<ItemAction>, profile: &Rc<Profile>) -> Self;

    // Actions

    fn clean(
        &self,
        transaction: &Transaction,
        app_browser_window_tab_item_page_navigation_id: &i64,
    ) -> Result<()>;

    fn restore(
        &self,
        transaction: &Transaction,
        app_browser_window_tab_item_page_navigation_id: &i64,
    ) -> Result<()>;

    fn save(
        &self,
        transaction: &Transaction,
        app_browser_window_tab_item_page_navigation_id: &i64,
    ) -> Result<()>;

    fn update_primary_icon(&self, profile: &Profile);
    fn update_secondary_icon(&self);

    fn show_identity_dialog(&self, profile: &Rc<Profile>);
    fn show_search_dialog(&self, profile: &Rc<Profile>);

    // Setters

    fn to_download(&self);
    fn to_source(&self);

    // Getters

    fn prefix_less(&self) -> GString;
    fn download(&self) -> GString;
    fn source(&self) -> GString;
    fn uri(&self) -> Option<Uri>;
    fn home(&self) -> Option<Uri>;
    fn is_file(&self) -> bool;
}

impl Request for Entry {
    // Constructors

    /// Build new `Self`
    fn request(item_action: &Rc<ItemAction>, profile: &Rc<Profile>) -> Self {
        // Init main widget
        let entry = Entry::builder()
            .placeholder_text("URL or search term...")
            .secondary_icon_tooltip_text("Go to the location")
            .hexpand(true)
            .build();

        // Detect primary icon on construct
        entry.update_primary_icon(profile);

        // Init additional features
        let suggestion = Rc::new(Suggestion::build(profile, &entry));

        // Connect events
        entry.connect_icon_release({
            let profile = profile.clone();
            move |this, position| match position {
                EntryIconPosition::Primary => {
                    if matches!(primary_icon::from(&this.text()), PrimaryIcon::Search { .. }) {
                        this.show_search_dialog(&profile)
                    } else {
                        this.show_identity_dialog(&profile)
                    }
                }
                EntryIconPosition::Secondary => this.emit_activate(),
                _ => todo!(), // unexpected
            }
        });

        entry.connect_has_focus_notify(|this| this.update_secondary_icon());

        suggestion
            .signal_handler_id
            .borrow_mut()
            .replace(entry.connect_changed({
                let profile = profile.clone();
                let item_action = item_action.clone();
                let suggestion = suggestion.clone();
                move |this| {
                    // Update actions
                    item_action.reload.set_enabled(!this.text().is_empty());
                    item_action.home.set_enabled(this.home().is_some());

                    // Update icons
                    this.update_primary_icon(&profile);
                    this.update_secondary_icon();

                    // Show search suggestions
                    if this.focus_child().is_some() {
                        suggestion.update(None);
                    }
                }
            })); // `suggestion` wants `signal_handler_id` to block this event on autocomplete navigation

        entry.connect_activate({
            let item_action = item_action.clone();
            let suggestion = suggestion.clone();
            move |_| {
                use gtk::prelude::ActionExt;
                item_action.reload.activate(None);
                suggestion.hide();
            }
        });

        entry.connect_state_flags_changed({
            // Define last focus state container
            let has_focus = Cell::new(false);
            move |this, state| {
                // Select entire text on first click (release)
                // this behavior implemented in most web-browsers,
                // to simply overwrite current request with new value
                // Note:
                // * Custom GestureClick is not an option here, as GTK Entry has default controller
                // * This is experimental feature does not follow native GTK behavior @TODO make optional
                if !has_focus.take()
                    && state.contains(StateFlags::ACTIVE | StateFlags::FOCUS_WITHIN)
                    && this.selection_bounds().is_none()
                {
                    this.select_region(0, -1);
                }
                // Update last focus state
                has_focus.replace(state.contains(StateFlags::FOCUS_WITHIN));
            }
        });

        entry
    }

    // Actions
    fn clean(
        &self,
        transaction: &Transaction,
        app_browser_window_tab_item_page_navigation_id: &i64,
    ) -> Result<()> {
        for record in database::select(transaction, app_browser_window_tab_item_page_navigation_id)?
        {
            database::delete(transaction, &record.id)?;
            // Delegate clean action to the item childs
            // nothing yet..
        }
        Ok(())
    }

    fn restore(
        &self,
        transaction: &Transaction,
        app_browser_window_tab_item_page_navigation_id: &i64,
    ) -> Result<()> {
        for record in database::select(transaction, app_browser_window_tab_item_page_navigation_id)?
        {
            if let Some(text) = record.text {
                self.set_text(&text);
            }
            // Delegate restore action to the item childs
            // nothing yet..
        }
        Ok(())
    }

    fn save(
        &self,
        transaction: &Transaction,
        app_browser_window_tab_item_page_navigation_id: &i64,
    ) -> Result<()> {
        // Keep value in memory until operation complete
        let text = self.text();
        let _id = database::insert(
            transaction,
            app_browser_window_tab_item_page_navigation_id,
            match text.is_empty() {
                true => None,
                false => Some(text.as_str()),
            },
        )?;
        // Delegate save action to childs
        // nothing yet..
        Ok(())
    }

    fn update_primary_icon(&self, profile: &Profile) {
        self.first_child().unwrap().remove_css_class("success"); // @TODO handle

        match primary_icon::from(&self.text()) {
            PrimaryIcon::Download { name, tooltip } | PrimaryIcon::File { name, tooltip } => {
                self.set_primary_icon_activatable(false);
                self.set_primary_icon_sensitive(false);
                self.set_primary_icon_name(Some(name));
                self.set_primary_icon_tooltip_text(Some(tooltip));
            }
            PrimaryIcon::Gemini { name, tooltip } | PrimaryIcon::Titan { name, tooltip } => {
                self.set_primary_icon_activatable(true);
                self.set_primary_icon_sensitive(true);
                self.set_primary_icon_name(Some(name));
                if profile.identity.get(&self.prefix_less()).is_some() {
                    self.first_child().unwrap().add_css_class("success"); // @TODO handle
                    self.set_primary_icon_tooltip_text(Some(tooltip.1));
                } else {
                    self.set_primary_icon_tooltip_text(Some(tooltip.0));
                }
            }
            PrimaryIcon::Search { name, tooltip } => {
                self.set_primary_icon_activatable(true);
                self.set_primary_icon_sensitive(true);
                self.set_primary_icon_name(Some(name));
                self.set_primary_icon_tooltip_text(Some(tooltip));
            }
            PrimaryIcon::Source { name, tooltip } => {
                self.set_primary_icon_activatable(false);
                self.set_primary_icon_sensitive(false);
                self.set_primary_icon_name(Some(name));
                self.set_primary_icon_tooltip_text(Some(tooltip));
            }
        }
    }

    fn update_secondary_icon(&self) {
        if !self.text().is_empty() && self.focus_child().is_some_and(|text| text.has_focus()) {
            self.set_secondary_icon_name(Some("pan-end-symbolic"));
        } else {
            self.set_secondary_icon_name(None);
            self.select_region(0, 0);
        }
    }

    /// Present Identity [AlertDialog](https://gnome.pages.gitlab.gnome.org/libadwaita/doc/main/class.AlertDialog.html) for `Self`
    fn show_identity_dialog(&self, profile: &Rc<Profile>) {
        // connect identity traits
        use identity::{Common, Unsupported};
        if let Some(uri) = self.uri() {
            if ["gemini", "titan"].contains(&uri.scheme().as_str()) {
                return AlertDialog::common(
                    profile,
                    &uri,
                    &Rc::new({
                        let profile = profile.clone();
                        let this = self.clone();
                        move |is_reload| {
                            this.update_primary_icon(&profile);
                            if is_reload {
                                this.emit_activate();
                            }
                        }
                    }),
                )
                .present(Some(self));
            }
        }
        AlertDialog::unsupported().present(Some(self));
    }

    /// Present Search providers [AlertDialog](https://gnome.pages.gitlab.gnome.org/libadwaita/doc/main/class.AlertDialog.html) for `Self`
    fn show_search_dialog(&self, profile: &Rc<Profile>) {
        use search::Search;
        AlertDialog::search(profile).present(Some(self))
    }

    // Setters

    fn to_download(&self) {
        self.set_text(&self.download());
    }

    fn to_source(&self) {
        self.set_text(&self.source());
    }

    // Getters

    /// Get current request value without system prefix
    /// * the `prefix` is not `scheme`
    fn prefix_less(&self) -> GString {
        let mut request = self.text();

        if let Some(postfix) = request.strip_prefix(PREFIX_SOURCE) {
            request = postfix.into()
        }
        if let Some(postfix) = request.strip_prefix(PREFIX_DOWNLOAD) {
            request = postfix.into()
        }
        request
    }

    /// Get request value with formatted `download` prefix
    fn download(&self) -> GString {
        gformat!("{PREFIX_DOWNLOAD}{}", self.prefix_less())
    }

    /// Get request value with formatted `source` prefix
    fn source(&self) -> GString {
        gformat!("{PREFIX_SOURCE}{}", self.prefix_less())
    }

    /// Try get current request value as [Uri](https://docs.gtk.org/glib/struct.Uri.html)
    /// * `strip_prefix` on parse
    fn uri(&self) -> Option<Uri> {
        match Uri::parse(&self.prefix_less(), UriFlags::NONE) {
            Ok(uri) => Some(uri),
            _ => None,
        }
    }

    /// Try build home [Uri](https://docs.gtk.org/glib/struct.Uri.html) for `Self`
    /// * return `None` if current request already match home or Uri not parsable
    fn home(&self) -> Option<Uri> {
        let uri = self.uri()?;
        if uri.path().len() > 1 || uri.query().is_some() || uri.fragment().is_some() {
            Some(Uri::build(
                UriFlags::NONE,
                &if uri.scheme() == "titan" {
                    GString::from("gemini")
                } else {
                    uri.scheme()
                },
                uri.userinfo().as_deref(),
                uri.host().as_deref(),
                uri.port(),
                "/",
                None,
                None,
            ))
        } else {
            None
        }
    }

    fn is_file(&self) -> bool {
        self.text().starts_with("file://")
    }
}

// Tools

pub fn migrate(tx: &Transaction) -> Result<()> {
    // Migrate self components
    database::init(tx)?;

    // Delegate migration to childs
    // nothing yet..

    // Success
    Ok(())
}