From de00a4de664eef61fbe6628d8559539f3bfb80c0 Mon Sep 17 00:00:00 2001
From: Condorra <kvox990h@anonaddy.me>
Date: Fri, 22 Nov 2024 23:52:49 +1100
Subject: [PATCH] Implement HTML view type.

---
 src/frame_view.rs        |  5 +++
 src/html_view.rs         | 12 +++++++
 src/lua_engine.rs        |  1 +
 src/lua_engine/frames.rs | 54 ++++++++++++++++++++++++++++++-
 src/main.rs              |  1 +
 src/term_split.rs        | 68 +++++++++++++++++++++++++++++++++++++++-
 styles.css               |  7 +++++
 7 files changed, 146 insertions(+), 2 deletions(-)
 create mode 100644 src/html_view.rs

diff --git a/src/frame_view.rs b/src/frame_view.rs
index f38c761..4e319d5 100644
--- a/src/frame_view.rs
+++ b/src/frame_view.rs
@@ -8,6 +8,7 @@ use yew::prelude::*;
 use crate::{
     command_handler::command_handler,
     editor_view::CodeEditorView,
+    html_view::HtmlView,
     lineengine::line::{Readline, ReadlineEvent},
     term_split::TermSplit,
     timer_host::TimerHost,
@@ -76,6 +77,7 @@ pub struct FrameId(pub u64);
 pub enum FrameViewType {
     Terminal,
     Editor,
+    Html { inner: String },
 }
 
 #[derive(Properties)]
@@ -239,6 +241,9 @@ pub fn term_view(props: &TermViewProps) -> Html {
                 global_memo={props.global_memo.clone()}
                 global_layout={props.global_layout.clone()}/>
         },
+        FrameViewType::Html { inner } => html! {
+            <HtmlView inner={inner.clone()}/>
+        },
     }
 }
 
diff --git a/src/html_view.rs b/src/html_view.rs
new file mode 100644
index 0000000..74575e1
--- /dev/null
+++ b/src/html_view.rs
@@ -0,0 +1,12 @@
+use yew::prelude::*;
+use yew::{function_component, Properties};
+
+#[derive(Properties, PartialEq)]
+pub struct HtmlViewProps {
+    pub inner: String,
+}
+
+#[function_component(HtmlView)]
+pub fn html_view(props: &HtmlViewProps) -> Html {
+    html! { <iframe class="htmlview" srcdoc={props.inner.clone()} sandbox=""/> }
+}
diff --git a/src/lua_engine.rs b/src/lua_engine.rs
index 314c49f..7adef36 100644
--- a/src/lua_engine.rs
+++ b/src/lua_engine.rs
@@ -221,6 +221,7 @@ pub fn install_lua_globals(
             register_command!(echo_frame_raw);
             register_command!(editor);
             register_stateless_command!(cmd_help, "help");
+            register_command!(html);
             register_command!(hsplit);
             register_stateless_command!(cmd_include, "include");
             register_command!(cmd_list_logs, "listlogs");
diff --git a/src/lua_engine/frames.rs b/src/lua_engine/frames.rs
index 181680e..e473d73 100644
--- a/src/lua_engine/frames.rs
+++ b/src/lua_engine/frames.rs
@@ -12,6 +12,8 @@ use piccolo::{
     SequenceReturn, StashedTable, StashedUserData, StashedValue, Table, UserData, Value, Variadic,
 };
 use std::{rc::Rc, str};
+use wasm_bindgen::JsValue;
+use web_sys::console;
 use yew::UseStateSetter;
 
 use super::call_checking_metatable;
@@ -774,7 +776,57 @@ pub(super) fn editor<'gc>(
                 new_gl
                     .frame_views
                     .insert(cur_frame_id, FrameViewType::Editor);
-                global_layout.set(new_gl.into());
+                let new_gl: Rc<GlobalLayoutState> = new_gl.into();
+                global_layout.set(new_gl.clone());
+                (*global_memo.layout.borrow_mut()) = new_gl;
+            }
+        };
+        Ok(piccolo::CallbackReturn::Return)
+    })
+}
+
+pub fn html<'gc>(
+    ctx: Context<'gc>,
+    global_memo: &GlobalMemoCell,
+    global_layout: &UseStateSetter<GlobalLayoutCell>,
+) -> Callback<'gc> {
+    let global_memo = global_memo.clone();
+    let global_layout = global_layout.clone();
+    Callback::from_fn(&ctx, move |ctx, _ex, mut stack| {
+        let (frame, inner): (u64, String) = stack.consume(ctx)?;
+        let target_frame_id: FrameId = FrameId(frame);
+        let will_have_remaining_terminal = {
+            let layout = global_memo.layout.borrow();
+            layout.term_splits.iter().any(|f| {
+                (*f != target_frame_id)
+                    && matches!(
+                        layout
+                            .frame_views
+                            .get(f)
+                            .unwrap_or(&FrameViewType::Terminal),
+                        FrameViewType::Terminal
+                    )
+            })
+        };
+        if !will_have_remaining_terminal {
+            Err(anyhow::Error::msg(
+                "html command that would leave user with no remaining terminal is invalid",
+            ))?
+        }
+        match global_memo
+            .frame_registry
+            .borrow_mut()
+            .get_mut(&target_frame_id)
+        {
+            None => Err(anyhow::Error::msg("Frame no longer exists"))?,
+            Some(_frame_dat) => {
+                let mut new_gl: GlobalLayoutState = (*global_memo.layout.borrow().as_ref()).clone();
+                new_gl
+                    .frame_views
+                    .insert(target_frame_id, FrameViewType::Html { inner });
+                let new_gl: Rc<GlobalLayoutState> = new_gl.into();
+                global_layout.set(new_gl.clone());
+                (*global_memo.layout.borrow_mut()) = new_gl;
             }
         };
         Ok(piccolo::CallbackReturn::Return)
diff --git a/src/main.rs b/src/main.rs
index 3d1683a..a28501d 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -11,6 +11,7 @@ use yew::prelude::*;
 pub mod command_handler;
 pub mod editor_view;
 pub mod frame_view;
+pub mod html_view;
 pub mod id_intern;
 pub mod lineengine;
 pub mod logging;
diff --git a/src/term_split.rs b/src/term_split.rs
index 3d1827d..55b2e74 100644
--- a/src/term_split.rs
+++ b/src/term_split.rs
@@ -1,5 +1,8 @@
 use itertools::Itertools;
-use std::{collections::BTreeMap, rc::Rc};
+use std::{
+    collections::{BTreeMap, VecDeque},
+    rc::Rc,
+};
 
 use crate::FrameId;
 
@@ -19,6 +22,12 @@ pub enum TermSplit {
 }
 
 impl TermSplit {
+    pub fn iter<'t>(&'t self) -> AccessibleSplitIter<'t> {
+        AccessibleSplitIter {
+            queue: vec![self].into(),
+        }
+    }
+
     fn collect_term_frames(&self, into: &mut BTreeMap<FrameId, usize>) {
         match self {
             TermSplit::Term { frame } => {
@@ -136,6 +145,31 @@ impl TermSplit {
     }
 }
 
+pub struct AccessibleSplitIter<'t> {
+    queue: VecDeque<&'t TermSplit>,
+}
+
+impl<'t> Iterator for AccessibleSplitIter<'t> {
+    type Item = &'t FrameId;
+
+    fn next(&mut self) -> Option<Self::Item> {
+        loop {
+            match self.queue.pop_back() {
+                Some(TermSplit::Horizontal { left, right }) => {
+                    self.queue.push_front(left);
+                    self.queue.push_front(right);
+                }
+                Some(TermSplit::Vertical { top, bottom }) => {
+                    self.queue.push_front(top);
+                    self.queue.push_front(bottom);
+                }
+                Some(TermSplit::Term { frame }) => break Some(frame),
+                None => break None,
+            }
+        }
+    }
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;
@@ -237,4 +271,36 @@ mod tests {
             })
         );
     }
+
+    #[test]
+    fn iterates_termframes() {
+        use TermSplit::*;
+        let t = Vertical {
+            top: Horizontal {
+                left: Horizontal {
+                    left: Term { frame: FrameId(42) }.into(),
+                    right: Term { frame: FrameId(64) }.into(),
+                }
+                .into(),
+                right: Term { frame: FrameId(42) }.into(),
+            }
+            .into(),
+            bottom: Vertical {
+                top: Term { frame: FrameId(43) }.into(),
+                bottom: Term { frame: FrameId(44) }.into(),
+            }
+            .into(),
+        };
+        let frames: Vec<_> = t.iter().collect();
+        assert_eq!(
+            frames,
+            vec![
+                &FrameId(42),
+                &FrameId(43),
+                &FrameId(44),
+                &FrameId(42),
+                &FrameId(64),
+            ]
+        );
+    }
 }
diff --git a/styles.css b/styles.css
index 1f4447d..2ef6893 100644
--- a/styles.css
+++ b/styles.css
@@ -85,3 +85,10 @@ body {
     min-width: 1px;
     min-height: 1px;
 }
+.htmlview {
+    margin: 10px;
+    min-width: 1px;
+    min-height: 1px;
+    width: 100%;
+    height: 100%;
+}