gpui-component/crates/story/src/webview_story.rs

use gpui::{
    div, ClickEvent, FocusHandle, FocusableView, IntoElement, ParentElement as _, Render,
    Styled as _, View, ViewContext, VisualContext as _, WindowContext,
};
use ui::{
    h_flex,
    input::{InputEvent, TextInput},
    v_flex,
    webview::WebView,
    ActiveTheme,
};

pub struct WebViewStory {
    focus_handle: FocusHandle,
    webview: View<WebView>,
    address_input: View<TextInput>,
}

impl super::Story for WebViewStory {
    fn title() -> &'static str {
        "WebView"
    }

    fn new_view(cx: &mut WindowContext) -> View<impl gpui::FocusableView> {
        Self::view(cx)
    }
}

impl WebViewStory {
    pub fn view(cx: &mut WindowContext) -> View<Self> {
        let focus_handle = cx.focus_handle();

        let webview = cx.new_view(|cx| {
            let webview = ui::wry::WebViewBuilder::new()
                .build_as_child(&cx.raw_window_handle())
                .unwrap();
            WebView::new(cx, webview)
        });

        let address_input = cx.new_view(|cx| {
            let mut input = TextInput::new(cx);
            input.set_text("https://google.com", cx);
            input
        });

        let url = address_input.read(cx).text();
        webview.update(cx, |view, _| {
            view.load_url(&url);
        });

        cx.new_view(|cx| {
            let this = WebViewStory {
                focus_handle,
                webview,
                address_input: address_input.clone(),
            };

            cx.subscribe(
                &address_input,
                |this: &mut Self, input, event: &InputEvent, cx| match event {
                    InputEvent::PressEnter => {
                        let url = input.read(cx).text();
                        this.webview.update(cx, |view, _| {
                            view.load_url(&url);
                        });
                    }
                    _ => {}
                },
            )
            .detach();

            this
        })
    }

    pub fn hide(&self, cx: &mut WindowContext) {
        self.webview.update(cx, |webview, _| webview.hide())
    }

    #[allow(unused)]
    fn go_back(&mut self, _: &ClickEvent, cx: &mut ViewContext<Self>) {
        self.webview.update(cx, |webview, _| {
            webview.back().unwrap();
        });
    }
}

impl FocusableView for WebViewStory {
    fn focus_handle(&self, _cx: &gpui::AppContext) -> FocusHandle {
        self.focus_handle.clone()
    }
}

impl Render for WebViewStory {
    fn render(&mut self, cx: &mut ViewContext<Self>) -> impl IntoElement {
        let webview = self.webview.clone();
        let address_input = self.address_input.clone();

        v_flex()
            .p_2()
            .gap_3()
            .size_full()
            .child(h_flex().gap_2().items_center().child(address_input.clone()))
            .child(
                div()
                    .flex_1()
                    .border_1()
                    .h(gpui::px(400.))
                    .border_color(cx.theme().border)
                    .child(webview.clone()),
            )
    }
}