duat_term/area/

mod.rs

mod iter;
mod print_info;

use std::{io::Write, iter::Peekable, sync::Arc};

use crossterm::{
    cursor, queue,
    style::{Attribute, Attributes},
};
use duat_core::{
    context::{
        self,
        cache::{Decode, Encode},
    },
    form::{CONTROL_CHAR_ID, Painter},
    mode::{TwoPointsPlace, VPoint},
    opts::PrintOpts,
    text::{Point, Text, TextPart, TextPlace, TwoPoints, txt},
    ui::{
        self, Columns, DynSpawnSpecs, PrintedLine, PushSpecs, SpawnId,
        traits::{RawArea, UiPass},
    },
};
use iter::{PrintedPlace, print_iter, rev_print_iter};

pub use self::print_info::PrintInfo;
use crate::{
    AreaId, Mutex,
    layout::{Frame, Layouts},
    print_style,
    printer::Lines,
};

#[derive(Default, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Encode, Decode)]
#[bincode(crate = "duat_core::context::cache::bincode")]
pub struct Coord {
    pub x: u32,
    pub y: u32,
}

impl std::fmt::Debug for Coord {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.write_fmt(format_args!("x: {}, y: {}", self.x, self.y))
    }
}

impl Coord {
    pub fn new(x: u32, y: u32) -> Coord {
        Coord { x, y }
    }
}

#[derive(Default, Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Encode, Decode)]
#[bincode(crate = "duat_core::context::cache::bincode")]
pub struct Coords {
    pub tl: Coord,
    pub br: Coord,
}

impl Coords {
    pub fn new(tl: Coord, br: Coord) -> Self {
        Coords { tl, br }
    }

    pub fn width(&self) -> u32 {
        self.br.x - self.tl.x
    }

    pub fn height(&self) -> u32 {
        self.br.y - self.tl.y
    }

    pub fn area(&self) -> u32 {
        self.width() * self.height()
    }

    pub fn intersects(&self, other: Self) -> bool {
        self.tl.x < other.br.x
            && self.br.x > other.tl.x
            && self.tl.y < other.br.y
            && self.br.y > other.tl.y
    }

    pub fn x_range(&self) -> std::ops::Range<u32> {
        self.tl.x..self.br.x
    }

    pub fn y_range(&self) -> std::ops::Range<u32> {
        self.tl.y..self.br.y
    }
}

#[derive(Clone)]
pub struct Area {
    prev_print_info: Arc<Mutex<PrintInfo>>,
    layouts: Layouts,
    id: AreaId,
}

impl PartialEq for Area {
    fn eq(&self, other: &Self) -> bool {
        self.id == other.id
    }
}

impl Area {
    /// Returns a new `Area` from raw parts
    pub(crate) fn new(id: AreaId, layouts: Layouts) -> Self {
        Self {
            prev_print_info: Arc::new(Mutex::default()),
            layouts,
            id,
        }
    }

    /// Adds a frame to this [`Area`]
    ///
    /// This function will fail if the `Area` was either deleted or is
    /// not a spawned `Area`.
    #[track_caller]
    pub fn set_frame(&mut self, frame: Frame) {
        if !self.layouts.set_frame(self.id, frame) {
            context::warn!("This Area was already deleted");
        }
    }

    /// Prints the `Text`
    fn print(&self, text: &Text, opts: PrintOpts, mut painter: Painter) {
        let Some(coords) = self.layouts.coords_of(self.id, true) else {
            context::warn!("This Area was already deleted");
            return;
        };

        let (has_edge_ahead, max) = self
            .layouts
            .inspect(self.id, |_, layout| {
                let master_id = layout.get_cluster_master(self.id).unwrap();
                let master_coords = self.layouts.coords_of(master_id, true).unwrap();
                let master_has_edge = layout.get(master_id).unwrap().edge().is_some();

                (
                    master_has_edge && master_coords.br.x == coords.br.x,
                    layout.max_value(),
                )
            })
            .unwrap();

        if coords.width() == 0 || coords.height() == 0 {
            return;
        }

        let (s_points, x_shift) = {
            let mut info = self.layouts.get_info_of(self.id).unwrap();
            let s_points = info.start_points(coords, text, opts);
            *self.prev_print_info.lock().unwrap() = info;
            self.layouts.set_info_of(self.id, info);
            (s_points, info.x_shift())
        };

        let is_active = self.id == self.layouts.get_active_id();

        let Some((lines, observed_spawns)) = print_text(
            (text, opts, &mut painter),
            (coords, max, has_edge_ahead),
            (is_active, s_points, x_shift),
            |lines, len| lines.write_all(&SPACES[..len as usize]).unwrap(),
            |lines, len, max_x| {
                if lines.coords().br.x == max_x {
                    lines.write_all(b"\x1b[0K").unwrap();
                } else {
                    lines
                        .write_all(&SPACES[..(lines.coords().width() - len) as usize])
                        .unwrap();
                }
            },
            |lines, spacer_len| lines.write_all(&SPACES[..spacer_len as usize]).unwrap(),
        ) else {
            return;
        };

        let spawns = text.get_spawned_ids();

        // To prevent deadlocks while sending spawned lines.
        drop(painter);

        self.layouts
            .send_lines(self.id, lines, spawns, &observed_spawns);
    }
}

impl RawArea for Area {
    type Cache = PrintInfo;
    type PrintInfo = PrintInfo;

    /////////// Modification

    fn push(
        &self,
        _: UiPass,
        specs: PushSpecs,
        on_buffers: bool,
        cache: PrintInfo,
    ) -> Option<(Area, Option<Area>)> {
        let (child, parent) = self.layouts.push(self.id, specs, on_buffers, cache)?;

        Some((
            Self::new(child, self.layouts.clone()),
            parent.map(|parent| Self::new(parent, self.layouts.clone())),
        ))
    }

    fn delete(&self, _: UiPass) -> (bool, Vec<Self>) {
        let (do_rm_window, rm_areas) = self.layouts.delete(self.id);
        (
            do_rm_window,
            rm_areas
                .into_iter()
                .map(|id| Self::new(id, self.layouts.clone()))
                .collect(),
        )
    }

    fn swap(&self, _: UiPass, rhs: &Self) -> bool {
        self.layouts.swap(self.id, rhs.id)
    }

    fn spawn(
        &self,
        _: UiPass,
        spawn_id: SpawnId,
        specs: DynSpawnSpecs,
        cache: Self::Cache,
    ) -> Option<Self> {
        Some(Self::new(
            self.layouts
                .spawn_on_widget(self.id, spawn_id, specs, cache)?,
            self.layouts.clone(),
        ))
    }

    fn set_width(&self, _: UiPass, width: f32) -> Result<(), Text> {
        if self
            .layouts
            .set_constraints(self.id, Some(width), None, None)
        {
            Ok(())
        } else {
            Err(txt!("This Area was already deleted"))
        }
    }

    fn set_height(&self, _: UiPass, height: f32) -> Result<(), Text> {
        if self
            .layouts
            .set_constraints(self.id, None, Some(height), None)
        {
            Ok(())
        } else {
            Err(txt!("This Area was already deleted"))
        }
    }

    fn hide(&self, _: UiPass) -> Result<(), Text> {
        if self
            .layouts
            .set_constraints(self.id, None, None, Some(true))
        {
            Ok(())
        } else {
            Err(txt!("This Area was already deleted"))
        }
    }

    fn reveal(&self, _: UiPass) -> Result<(), Text> {
        if self
            .layouts
            .set_constraints(self.id, None, None, Some(false))
        {
            Ok(())
        } else {
            Err(txt!("This Area was already deleted"))
        }
    }

    fn size_of_text(&self, _: UiPass, opts: PrintOpts, text: &Text) -> Result<ui::Coord, Text> {
        let max = self
            .layouts
            .inspect(self.id, |_, layout| layout.max_value())
            .ok_or_else(|| txt!("This Area was already deleted"))?;

        let iter = iter::print_iter(text, TwoPoints::default(), max.x, opts);

        let mut max_x = 0;
        let mut max_y = 0;
        let mut width = 0;
        let mut is_first = true;

        for (place, item) in iter {
            if place.wrap && !is_first {
                if max_y == max.y {
                    break;
                }
                max_x = width.max(max_x);
                max_y += 1;
                width = 0;
            } else {
                is_first = false;
            }
            if item.part.is_char() {
                width += place.len;
            }
        }

        Ok(ui::Coord::new(max_x.max(width) as f32, max_y as f32))
    }

    fn scroll_ver(&self, _: UiPass, text: &Text, by: i32, opts: PrintOpts) {
        if by == 0 {
            return;
        }

        let Some(coords) = self.layouts.coords_of(self.id, false) else {
            context::warn!("This Area was already deleted");
            return;
        };

        if coords.width() == 0 || coords.height() == 0 {
            return;
        }

        let mut info = self.layouts.get_info_of(self.id).unwrap();
        info.scroll_ver(by, coords, text, opts);
        self.layouts.set_info_of(self.id, info);
    }

    ////////// Printing

    fn scroll_around_points(&self, _: UiPass, text: &Text, points: TwoPoints, opts: PrintOpts) {
        let Some(coords) = self.layouts.coords_of(self.id, false) else {
            context::warn!("This Area was already deleted");
            return;
        };

        if coords.width() == 0 || coords.height() == 0 {
            return;
        }

        let mut info = self.layouts.get_info_of(self.id).unwrap();
        info.scroll_around(points.real, coords, text, opts);
        self.layouts.set_info_of(self.id, info);
    }

    fn scroll_to_points(&self, _: UiPass, text: &Text, points: TwoPoints, opts: PrintOpts) {
        let Some(coords) = self.layouts.coords_of(self.id, false) else {
            context::warn!("This Area was already deleted");
            return;
        };

        if coords.width() == 0 || coords.height() == 0 {
            return;
        }

        let mut info = self.layouts.get_info_of(self.id).unwrap();
        info.scroll_to_points(points, coords, text, opts);
        self.layouts.set_info_of(self.id, info);
    }

    fn set_as_active(&self, _: UiPass) {
        self.layouts.set_active_id(self.id);
    }

    fn print(&self, _: UiPass, text: &Text, opts: PrintOpts, painter: Painter) {
        self.print(text, opts, painter)
    }

    ////////// Queries

    fn get_print_info(&self, _: UiPass) -> Self::PrintInfo {
        self.layouts.get_info_of(self.id).unwrap_or_default()
    }

    fn set_print_info(&self, _: UiPass, info: Self::PrintInfo) {
        self.layouts.set_info_of(self.id, info);
    }

    fn get_printed_lines(
        &self,
        pa: UiPass,
        text: &Text,
        opts: PrintOpts,
    ) -> Option<Vec<ui::PrintedLine>> {
        let coords = self.layouts.coords_of(self.id, true)?;
        let points = self.start_points(pa, text, opts);

        if coords.height() == 0 || coords.width() == 0 {
            return Some(Vec::new());
        }

        let mut prev_point = rev_print_iter(text, points, coords.width(), opts)
            .find_map(|(place, item)| place.wrap.then_some(item.points()))
            .map(|points| points.real.line());

        let mut printed_lines = Vec::<PrintedLine>::new();
        let mut y = coords.tl.y;
        let mut is_ghost = true;

        for (place, item) in print_iter(text, points, coords.width(), opts) {
            if y == coords.br.y || item.line() == text.end_point().line() {
                break;
            }
            y += place.wrap as u32;

            if place.wrap {
                if let Some(last) = printed_lines.last_mut() {
                    last.is_ghost = is_ghost;
                }
                is_ghost = true;

                let number = item.line();
                let is_wrapped = prev_point.is_some_and(|ll| ll == number);
                prev_point = Some(number);
                printed_lines.push(PrintedLine { number, is_wrapped, is_ghost: true });
            }
            
            if item.as_real_char().is_some() {
                is_ghost = false;
            }
        }

        if let Some(last) = printed_lines.last_mut() {
            last.is_ghost = is_ghost;
        }

        Some(printed_lines)
    }

    fn move_ver(
        &self,
        _: UiPass,
        by: i32,
        text: &Text,
        point: Point,
        desired_col: Option<usize>,
        opts: PrintOpts,
    ) -> VPoint {
        let Some(coords) = self.layouts.coords_of(self.id, true) else {
            panic!("Tried to move vertically on a deleted area");
        };

        let cap = coords.width();

        if by == 0 {
            return calculate_vpoint(text, point, cap, opts);
        }

        let desired_col = match desired_col {
            Some(desired_col) => desired_col as u16,
            None => calculate_vpoint(text, point, cap, opts).desired_visual_col() as u16,
        };

        let line_start = {
            let target = point.line().saturating_add_signed(by as isize);
            text.point_at_coords(target.min(text.last_point().line()), 0)
        };

        let mut wraps = 0;
        let mut vcol = 0;

        let points = line_start.to_two_points_before();
        let (wcol, point) = print_iter(text, points, cap, opts)
            .find_map(|(PrintedPlace { len, x, wrap }, item)| {
                wraps += wrap as usize;

                if let Some((p, char)) = item.as_real_char()
                    && (vcol + len as u16 > desired_col || char == '\n')
                {
                    return Some((x as u16, p));
                }

                vcol += len as u16;
                None
            })
            .unwrap_or((0, text.last_point()));

        let ccol = (point.char() - line_start.char()) as u16;

        VPoint::new(point, ccol, vcol, desired_col, wcol, wcol)
    }

    fn move_ver_wrapped(
        &self,
        _: UiPass,
        by: i32,
        text: &Text,
        point: Point,
        desired_col: Option<usize>,
        opts: PrintOpts,
    ) -> VPoint {
        let mut wraps = 0;

        let Some(coords) = self.layouts.coords_of(self.id, true) else {
            panic!("Tried to move vertically on a deleted area");
        };

        let cap = coords.width();

        if by == 0 {
            return calculate_vpoint(text, point, cap, opts);
        }

        let desired_col = match desired_col {
            Some(desired_col) => desired_col as u16,
            None => calculate_vpoint(text, point, cap, opts).desired_wrapped_col() as u16,
        };

        if by > 0 {
            let line_start = text.point_at_coords(point.line(), 0);
            let points = line_start.to_two_points_after();

            let mut vcol = 0;
            let mut last = (0, 0, line_start);
            let mut last_valid = last;

            let (vcol, wcol, point) = print_iter(text, points, cap, opts)
                .find_map(|(PrintedPlace { x, len, wrap }, item)| {
                    wraps += (wrap && item.char() > point.char()) as i32;
                    if let Some((p, char)) = item.as_real_char() {
                        if (x..x + len).contains(&(desired_col as u32))
                            || (char == '\n' && x <= desired_col as u32)
                        {
                            last_valid = (vcol, x as u16, p);
                            if wraps == by {
                                return Some((vcol, x as u16, p));
                            }
                        } else if wraps > by {
                            return Some(last);
                        }
                        last = (vcol, x as u16, p);
                    }
                    vcol += len as u16;
                    None
                })
                .unwrap_or(last_valid);

            let ccol = (point.char() - line_start.char()) as u16;

            VPoint::new(point, ccol, vcol, vcol, wcol, desired_col)
        } else {
            let end_points = text.points_after(point.to_two_points_after()).unwrap();
            let mut just_wrapped = false;
            let mut last_valid = None;

            let mut iter = rev_print_iter(text, end_points, cap, opts);
            let wcol_and_p = iter.find_map(|(PrintedPlace { x, len, wrap }, item)| {
                if let Some((p, _)) = item.as_real_char() {
                    // max(1) because it could be a '\n'
                    if (x..x + len.max(1)).contains(&(desired_col as u32))
                        || (just_wrapped && x + len < desired_col as u32)
                    {
                        last_valid = Some((x as u16, p));
                        if wraps == by {
                            return Some((x as u16, p));
                        }
                    }
                    just_wrapped = false;
                }
                wraps -= wrap as i32;
                just_wrapped |= wrap;
                None
            });

            if let Some((wcol, point)) = wcol_and_p {
                let (ccol, vcol) = iter
                    .take_while(|(_, item)| item.as_real_char().is_none_or(|(_, c)| c != '\n'))
                    .fold((0, 0), |(ccol, vcol), (place, item)| {
                        (ccol + item.is_real() as u16, vcol + place.len as u16)
                    });

                VPoint::new(point, ccol, vcol, vcol, wcol, desired_col)
            } else if let Some((wcol, point)) = last_valid {
                let points = point.to_two_points_before();
                let (ccol, vcol) = rev_print_iter(text, points, cap, opts)
                    .take_while(|(_, item)| item.as_real_char().is_none_or(|(_, c)| c != '\n'))
                    .fold((0, 0), |(ccol, vcol), (place, item)| {
                        (ccol + item.is_real() as u16, vcol + place.len as u16)
                    });

                VPoint::new(point, ccol, vcol, vcol, wcol, desired_col)
            } else {
                VPoint::default()
            }
        }
    }

    fn has_changed(&self, _: UiPass) -> bool {
        self.layouts
            .inspect(self.id, |rect, layout| {
                rect.has_changed(layout)
                    || rect
                        .print_info()
                        .is_some_and(|info| *info != *self.prev_print_info.lock().unwrap())
            })
            .unwrap_or(false)
    }

    fn is_master_of(&self, _: UiPass, other: &Self) -> bool {
        if other.id == self.id {
            return true;
        }

        let mut parent_id = other.id;

        self.layouts.inspect(self.id, |_, layout| {
            while let Some((_, parent)) = layout.get_parent(parent_id) {
                parent_id = parent.id();
                if parent.id() == self.id {
                    break;
                }
            }
        });

        parent_id == self.id
    }

    fn get_cluster_master(&self, _: UiPass) -> Option<Self> {
        let id = self
            .layouts
            .inspect(self.id, |_, layout| layout.get_cluster_master(self.id))??;

        Some(Self {
            prev_print_info: Arc::default(),
            layouts: self.layouts.clone(),
            id,
        })
    }

    fn cache(&self, _: UiPass) -> Option<Self::Cache> {
        let info = self.layouts.get_info_of(self.id)?.for_caching();
        Some(info)
    }

    fn top_left(&self, _: UiPass) -> ui::Coord {
        self.layouts.update(self.id);
        self.layouts
            .coords_of(self.id, false)
            .map(|coords| ui::Coord {
                x: coords.tl.x as f32,
                y: coords.tl.y as f32,
            })
            .unwrap_or_default()
    }

    fn bottom_right(&self, _: UiPass) -> ui::Coord {
        self.layouts.update(self.id);
        self.layouts
            .coords_of(self.id, false)
            .map(|coords| ui::Coord {
                x: coords.br.x as f32,
                y: coords.br.y as f32,
            })
            .unwrap_or_default()
    }

    #[track_caller]
    fn points_at_coord(
        &self,
        _: UiPass,
        text: &Text,
        coord: ui::Coord,
        opts: PrintOpts,
    ) -> Option<TwoPointsPlace> {
        self.layouts.update(self.id);
        let Some(coords) = self.layouts.coords_of(self.id, false) else {
            context::warn!("This Area was already deleted");
            return None;
        };

        if coords.width() == 0 || coords.height() == 0 {
            return None;
        } else if !(coords.tl.x..coords.br.x).contains(&(coord.x as u32))
            || !(coords.tl.y..coords.br.y).contains(&(coord.y as u32))
        {
            context::warn!("Coordinate not contained in area");
            return None;
        }

        let (s_points, x_shift) = {
            let mut info = self.layouts.get_info_of(self.id).unwrap();
            let s_points = info.start_points(coords, text, opts);
            self.layouts.set_info_of(self.id, info);
            (s_points, info.x_shift())
        };

        let mut is_first = true;
        let mut row = coords.tl.y;
        let mut backup = None;
        for (place, item) in print_iter(text, s_points, coords.width(), opts) {
            if is_first {
                is_first = false;
            } else {
                row += place.wrap as u32;
            }

            if row > coord.y as u32 {
                return backup;
            } else if row == coord.y as u32
                && place.len > 0
                && let Some(col) = place.x.checked_sub(x_shift)
            {
                if (coords.tl.x + col..coords.tl.x + col + place.len).contains(&(coord.x as u32)) {
                    return Some(TwoPointsPlace::Within(item.points()));
                } else if coords.tl.x + col >= coord.x as u32 {
                    break;
                }
            }

            backup = Some(TwoPointsPlace::AheadOf(item.points()));
        }

        None
    }

    fn coord_at_points(
        &self,
        _: UiPass,
        text: &Text,
        points: TwoPoints,
        opts: PrintOpts,
    ) -> Option<ui::Coord> {
        self.layouts.update(self.id);
        let Some(coords) = self.layouts.coords_of(self.id, false) else {
            context::warn!("This Area was already deleted");
            return None;
        };

        if coords.width() == 0 || coords.height() == 0 {
            return None;
        }

        let (s_points, x_shift) = {
            let mut info = self.layouts.get_info_of(self.id).unwrap();
            let s_points = info.start_points(coords, text, opts);
            self.layouts.set_info_of(self.id, info);
            (s_points, info.x_shift())
        };

        let mut row = coords.tl.y;
        for (place, item) in print_iter(text, s_points, coords.width(), opts) {
            row += place.wrap as u32;

            if row > coords.br.y {
                break;
            }

            if item.points() == points && item.part.is_char() {
                if place.x >= x_shift && place.x <= x_shift + coords.width() {
                    return Some(ui::Coord {
                        x: (coords.tl.x + place.x - x_shift) as f32,
                        y: (row - 1) as f32,
                    });
                } else {
                    break;
                }
            }
        }

        None
    }

    fn columns_at(
        &self,
        _: UiPass,
        text: &Text,
        points: TwoPoints,
        opts: PrintOpts,
    ) -> Option<Columns> {
        self.layouts.update(self.id);
        let Some(coords) = self.layouts.coords_of(self.id, false) else {
            context::warn!("This Area was already deleted");
            return None;
        };

        if coords.width() == 0 {
            return None;
        }

        let mut line = 0;

        for (place, item) in print_iter(text, points, coords.width(), opts) {
            if item.points() == points && item.part.is_char() {
                return Some(Columns {
                    line,
                    wrapped: place.x as usize,
                    len: place.len as usize,
                });
            } else if let Some('\n') = item.part.as_char() {
                break;
            }

            line += place.len as usize;
        }

        None
    }

    fn start_points(&self, _: UiPass, text: &Text, opts: PrintOpts) -> TwoPoints {
        self.layouts.update(self.id);
        let Some(coords) = self.layouts.coords_of(self.id, false) else {
            context::warn!("This Area was already deleted");
            return Default::default();
        };

        let mut info = self.layouts.get_info_of(self.id).unwrap();
        let start_points = info.start_points(coords, text, opts);
        self.layouts.set_info_of(self.id, info);

        start_points
    }

    fn end_points(&self, _: UiPass, text: &Text, opts: PrintOpts) -> TwoPoints {
        self.layouts.update(self.id);
        let Some(coords) = self.layouts.coords_of(self.id, false) else {
            context::warn!("This Area was already deleted");
            return Default::default();
        };

        let mut info = self.layouts.get_info_of(self.id).unwrap();
        let end_points = info.end_points(coords, text, opts);
        self.layouts.set_info_of(self.id, info);

        end_points
    }

    fn is_active(&self, _: UiPass) -> bool {
        self.layouts.get_active_id() == self.id
    }
}

/// The [`Text`] printing function
#[allow(clippy::type_complexity)]
pub fn print_text(
    (text, opts, painter): (&Text, PrintOpts, &mut Painter),
    (coords, max, has_edge_ahead): (Coords, Coord, bool),
    (is_active, s_points, x_shift): (bool, TwoPoints, u32),
    start_line: impl Fn(&mut Lines, u32),
    end_line: impl Fn(&mut Lines, u32, u32),
    print_spacer: impl Fn(&mut Lines, u32),
) -> Option<(Lines, Vec<(SpawnId, Coord, u32)>)> {
    if coords.width() == 0 || coords.height() == 0 {
        return None;
    }

    let (mut lines, iter, mut next_cursor_end) = {
        let lines = Lines::new(coords, has_edge_ahead);
        let width = opts.wrap_width(coords.width()).unwrap_or(coords.width());
        let iter = print_iter(text, s_points, width, opts);
        let mut selections = iter_selections(text, s_points.real).peekable();

        let next_c = move |lines: &mut Lines, painter: &mut Painter, real, ghost: Option<_>| {
            let mut last_found = None;
            while let Some(sel) = selections
                .next_if(|parts| (parts.point == real && ghost.is_none()) || parts.point < real)
            {
                last_found = Some(sel)
            }

            match last_found {
                Some(CursorParts { is_main: true, is_caret, is_start, .. }) => {
                    if let Some(shape) = painter.main_cursor()
                        && is_active
                    {
                        lines.show_real_cursor();
                        queue!(lines, shape, cursor::SavePosition).unwrap();
                    }

                    lines.hide_real_cursor();
                    painter.apply_main_selection(is_caret, is_start == Some(true));
                    Some(Cursor::Main(is_caret, is_start == Some(false)))
                }
                Some(CursorParts { is_main: false, is_caret, is_start, .. }) => {
                    painter.apply_extra_selection(is_caret, is_start == Some(true));
                    Some(Cursor::Extra(is_caret, is_start == Some(false)))
                }
                None => None,
            }
        };

        (lines, iter, next_c)
    };

    // The y here represents the bottom of the current row of cells.
    let tl_y = lines.coords().tl.y;
    let mut y = tl_y;
    let mut last_x = 0;

    // For spawns.
    let mut spawns_for_next: Vec<SpawnId> = Vec::new();
    let mut spawns = Vec::new();
    // For Forms and Masks.
    let mut style_was_set = false;
    // For Overlays.
    let mut overlays = Vec::new();
    let mut minimum_x;

    let endl = |lines: &mut _, painter: &mut Painter, x, overlays: &mut Vec<_>, spawns: &mut _| {
        let mut default = painter.get_default();
        default.style.foreground_color = None;
        default.style.underline_color = None;
        default.style.attributes = Attributes::from(Attribute::Reset);
        print_style(lines, default.style);
        end_line(lines, x, max.x);

        if !overlays.is_empty() {
            let coords = lines.coords();
            painter.reset_prev_style();
            continue_overlays(
                (lines, overlays),
                (coords.br.x, x_shift, u32::MAX, &mut true),
                (painter, spawns),
            );
            write!(lines, "\x1b[{}G", lines.coords().br.x + 1).unwrap();
            overlays.clear();
        }

        lines.flush().unwrap();
    };

    for (place, item) in iter {
        let lines = &mut lines;

        let PrintedPlace { x, len, wrap } = place;
        let TextPlace { part, real, ghost } = item;

        if wrap {
            if y == lines.coords().br.y {
                break;
            }
            if y > lines.coords().tl.y {
                endl(lines, painter, last_x, &mut overlays, &mut spawns);
                overlays.clear();
            }

            let initial_space = x.saturating_sub(x_shift).min(lines.coords().width());
            if initial_space > 0 {
                let mut default_style = painter.get_default().style;
                default_style.attributes.set(Attribute::Reset);
                print_style(lines, default_style);
                start_line(lines, initial_space);
            }
            y += 1;

            // Resetting painter to prevent erroneous printing.
            painter.reset_prev_style();
            style_was_set = true;
            last_x = initial_space;
        }

        let is_contained = x + len > x_shift && x < x_shift + lines.coords().width();

        match part {
            TextPart::Char(char) if is_contained => {
                let cursor_style = next_cursor_end(lines, painter, real, ghost);
                style_was_set |= cursor_style.is_some();

                minimum_x = continue_overlays(
                    (lines, &mut overlays),
                    (x, x_shift, x + len, &mut style_was_set),
                    (painter, &mut spawns),
                );

                if let Some(str) = get_control_str(char) {
                    painter.apply(CONTROL_CHAR_ID, 100);
                    if let Some(style) = painter.relative_style() {
                        print_style(lines, style);
                    }
                    lines.write_all(str.as_bytes()).unwrap();
                    painter.remove(CONTROL_CHAR_ID)
                } else {
                    if style_was_set && let Some(style) = painter.relative_style() {
                        print_style(lines, style);
                    }

                    let mut bytes = [0; 4];
                    match char {
                        '\t' => {
                            let truncated_start = x_shift.saturating_sub(x);
                            let truncated_end =
                                (x + len).saturating_sub(lines.coords().width() + x_shift);
                            let tab_len = (len - (truncated_start + truncated_end))
                                .saturating_sub(minimum_x.saturating_sub(x));
                            lines.write_all(&SPACES[..tab_len as usize]).unwrap()
                        }
                        _ if minimum_x >= x + len => {}
                        _ if minimum_x > x => {
                            let range = minimum_x as usize..(x + len) as usize;
                            lines.write_all(&SPACES[range]).unwrap()
                        }
                        '\n' if len == 1 => lines.write_all(b" ").unwrap(),
                        '\n' | '\r' => {}
                        char => {
                            char.encode_utf8(&mut bytes);
                            lines.write_all(&bytes[..char.len_utf8()]).unwrap();
                        }
                    }
                }

                if let Some(cursor) = cursor_style {
                    match cursor {
                        Cursor::Main(is_caret, end_range) => {
                            painter.remove_main_selection(is_caret, end_range)
                        }
                        Cursor::Extra(is_caret, end_range) => {
                            painter.remove_extra_selection(is_caret, end_range)
                        }
                    }
                    if let Some(style) = painter.relative_style() {
                        print_style(lines, style)
                    }
                }

                for id in spawns_for_next.drain(..) {
                    spawns.push((
                        id,
                        Coord::new(lines.coords().tl.x + x - x_shift, y - 1),
                        len,
                    ));
                }

                last_x = x + len - x_shift;
                style_was_set = false;
            }
            TextPart::Char(_) => {
                let cursor_style = next_cursor_end(lines, painter, real, ghost);
                style_was_set |= cursor_style.is_some();

                match cursor_style {
                    Some(Cursor::Main(is_caret, end_range)) => {
                        painter.remove_main_selection(is_caret, end_range)
                    }
                    Some(Cursor::Extra(is_caret, end_range)) => {
                        painter.remove_extra_selection(is_caret, end_range)
                    }
                    None => {}
                }
                spawns_for_next.clear();
            }
            TextPart::PushForm(id, prio) => {
                painter.apply(id, prio);
                style_was_set = true;
            }
            TextPart::PopForm(id) => {
                painter.remove(id);
                style_was_set = true;
            }
            TextPart::Spacer => {
                if style_was_set && let Some(style) = painter.relative_style() {
                    print_style(lines, style);
                }
                style_was_set = false;
                let truncated_start = x_shift.saturating_sub(x).min(len);
                let truncated_end = (x + len)
                    .saturating_sub(lines.coords().width().saturating_sub(x_shift))
                    .min(len);
                let spacer_len = len - (truncated_start + truncated_end);
                print_spacer(lines, spacer_len);
                last_x = (x + len)
                    .saturating_sub(x_shift)
                    .min(lines.coords().width());
            }
            TextPart::ResetState => print_style(lines, painter.reset()),
            TextPart::SpawnedWidget(id) => spawns_for_next.push(id),
            TextPart::PushMask(id) => {
                painter.apply_mask(id);
                style_was_set = true;
            }
            TextPart::PopMask(id) => {
                painter.remove_mask(id);
                style_was_set = true;
            }
            TextPart::Overlay(overlay) => {
                let opts = PrintOpts { print_new_line: false, ..opts };
                let overlay_coords =
                    Coords::new(Coord::new(coords.tl.x + x, 0), Coord::new(coords.br.x, 1));

                overlays.push(
                    print_iter(overlay, TwoPoints::default(), overlay_coords.width(), opts)
                        .map(move |(mut place, item)| {
                            place.x += x;
                            (place, item)
                        })
                        .peekable(),
                );
            }
        }
    }

    endl(&mut lines, painter, last_x, &mut overlays, &mut spawns);

    for _ in 0..lines.coords().br.y - y {
        endl(&mut lines, painter, 0, &mut Vec::new(), &mut spawns);
    }

    Some((lines, spawns))
}

#[inline(always)]
fn continue_overlays<'t, Iter: Iterator<Item = (PrintedPlace, TextPlace<'t>)>>(
    (lines, overlays): (&mut Lines, &mut Vec<Peekable<Iter>>),
    (original_x, x_shift, until_x, style_was_set): (u32, u32, u32, &mut bool),
    (painter, spawns): (&mut Painter, &mut Vec<(SpawnId, Coord, u32)>),
) -> u32 {
    if overlays.is_empty() {
        return original_x;
    }

    // For specific Tags
    let mut spawns_for_next: Vec<SpawnId> = Vec::new();
    let mut current_x = original_x;

    while let Some((place, item)) = {
        let mut min = None;
        for (i, overlay) in overlays.iter_mut().enumerate() {
            if let Some((place, _)) = overlay.peek()
                && min.is_none_or(|(x, _)| place.x < x)
            {
                min = Some((place.x, i));
            }
        }

        min.and_then(|(_, i)| overlays[i].next_if(|(place, _)| place.x < until_x))
    } {
        let PrintedPlace { x, len, wrap } = place;
        let TextPlace { part, real, .. } = item;

        if wrap && real != Point::default() {
            break;
        }

        let is_contained = x + len > x_shift && x < x_shift + lines.coords().width();

        match part {
            TextPart::Char(char) if is_contained => {
                if x != original_x {
                    write!(lines, "\x1b[{}G", lines.coords().tl.x + x + 1).unwrap();
                }

                if let Some(str) = get_control_str(char) {
                    painter.apply(CONTROL_CHAR_ID, 100);
                    if let Some(style) = painter.relative_style() {
                        print_style(lines, style);
                    }
                    lines.write_all(str.as_bytes()).unwrap();
                    painter.remove(CONTROL_CHAR_ID)
                } else {
                    if *style_was_set && let Some(style) = painter.relative_style() {
                        print_style(lines, style);
                    }
                    let mut bytes = [0; 4];

                    match char {
                        '\t' => {
                            let truncated_start = x_shift.saturating_sub(x);
                            let truncated_end =
                                (x + len).saturating_sub(lines.coords().width() + x_shift);
                            let tab_len = len - (truncated_start + truncated_end);
                            lines.write_all(&SPACES[..tab_len as usize]).unwrap()
                        }
                        '\n' if len == 1 => lines.write_all(b" ").unwrap(),
                        '\n' | '\r' => {}
                        char => {
                            char.encode_utf8(&mut bytes);
                            lines.write_all(&bytes[..char.len_utf8()]).unwrap();
                        }
                    }
                }

                for id in spawns_for_next.drain(..) {
                    spawns.push((
                        id,
                        Coord::new(lines.coords().tl.x + x - x_shift, lines.coords().tl.y),
                        len,
                    ));
                }

                current_x = x + len;
                *style_was_set = false;
            }
            TextPart::Char(_) => spawns_for_next.clear(),
            TextPart::PushForm(id, prio) => {
                painter.apply(id, prio);
                *style_was_set = true;
            }
            TextPart::PopForm(id) => {
                painter.remove(id);
                *style_was_set = true;
            }
            TextPart::Spacer => {}
            TextPart::ResetState => print_style(lines, painter.reset()),
            TextPart::SpawnedWidget(id) => spawns_for_next.push(id),
            TextPart::PushMask(id) => {
                painter.apply_mask(id);
                *style_was_set = true;
            }
            TextPart::PopMask(id) => {
                painter.remove_mask(id);
                *style_was_set = true;
            }
            TextPart::Overlay(_) => unreachable!(),
        }
    }

    if current_x < original_x
        && let Some(column) = (lines.coords().tl.x + original_x).checked_sub(x_shift)
        && column < lines.coords().br.x
    {
        write!(lines, "\x1b[{}G", column + 1).unwrap();
    }
    overlays.retain_mut(|iter| iter.peek().is_some());

    current_x
}

fn calculate_vpoint(text: &Text, point: Point, cap: u32, opts: PrintOpts) -> VPoint {
    let start = text.point_at_coords(point.line(), 0);

    let mut vcol = 0;

    let points = start.to_two_points_before();
    let wcol = print_iter(text, points, cap, opts)
        .find_map(|(place, item)| {
            if let Some((lhs, _)) = item.as_real_char()
                && lhs == point
            {
                return Some(place.x as u16);
            }
            vcol += place.len as u16;
            None
        })
        .unwrap_or(0);

    VPoint::new(
        point,
        (point.char() - start.char()) as u16,
        vcol,
        vcol,
        wcol,
        wcol,
    )
}

const fn get_control_str(char: char) -> Option<&'static str> {
    match char {
        '\0' => Some("^@"),
        '\u{01}' => Some("^A"),
        '\u{02}' => Some("^B"),
        '\u{03}' => Some("^C"),
        '\u{04}' => Some("^D"),
        '\u{05}' => Some("^E"),
        '\u{06}' => Some("^F"),
        '\u{07}' => Some("^G"),
        '\u{08}' => Some("^H"),
        '\u{0b}' => Some("^K"),
        '\u{0c}' => Some("^L"),
        '\u{0e}' => Some("^N"),
        '\u{0f}' => Some("^O"),
        '\u{10}' => Some("^P"),
        '\u{11}' => Some("^Q"),
        '\u{12}' => Some("^R"),
        '\u{13}' => Some("^S"),
        '\u{14}' => Some("^T"),
        '\u{15}' => Some("^U"),
        '\u{16}' => Some("^V"),
        '\u{17}' => Some("^W"),
        '\u{18}' => Some("^X"),
        '\u{19}' => Some("^Y"),
        '\u{1a}' => Some("^Z"),
        '\u{1b}' => Some("^["),
        '\u{1c}' => Some("^\\"),
        '\u{1d}' => Some("^]"),
        '\u{1e}' => Some("^^"),
        '\u{1f}' => Some("^_"),
        '\u{80}' => Some("<80>"),
        '\u{81}' => Some("<81>"),
        '\u{82}' => Some("<82>"),
        '\u{83}' => Some("<83>"),
        '\u{84}' => Some("<84>"),
        '\u{85}' => Some("<85>"),
        '\u{86}' => Some("<86>"),
        '\u{87}' => Some("<87>"),
        '\u{88}' => Some("<88>"),
        '\u{89}' => Some("<89>"),
        '\u{8a}' => Some("<8a>"),
        '\u{8b}' => Some("<8b>"),
        '\u{8c}' => Some("<8c>"),
        '\u{8d}' => Some("<8d>"),
        '\u{8e}' => Some("<8e>"),
        '\u{8f}' => Some("<8f>"),
        '\u{90}' => Some("<90>"),
        '\u{91}' => Some("<91>"),
        '\u{92}' => Some("<92>"),
        '\u{93}' => Some("<93>"),
        '\u{94}' => Some("<94>"),
        '\u{95}' => Some("<95>"),
        '\u{96}' => Some("<96>"),
        '\u{97}' => Some("<97>"),
        '\u{98}' => Some("<98>"),
        '\u{99}' => Some("<99>"),
        '\u{9a}' => Some("<9a>"),
        '\u{9b}' => Some("<9b>"),
        '\u{9c}' => Some("<9c>"),
        '\u{9d}' => Some("<9d>"),
        '\u{9e}' => Some("<9e>"),
        '\u{9f}' => Some("<9f>"),
        _ => None,
    }
}

const SPACES: &[u8] = &[b' '; 3000];

fn iter_selections(text: &Text, from: Point) -> impl Iterator<Item = CursorParts> {
    text.selections()
        .iter_within(from..)
        .flat_map(|(_, sel, is_main)| {
            if let Some(anchor) = sel.anchor()
                && anchor != sel.caret()
            {
                let caret_is_start = sel.caret() < anchor;
                let start = sel.caret().min(anchor);
                let end = sel.caret().max(anchor);
                [
                    Some(CursorParts::new(start, is_main, caret_is_start, Some(true))),
                    Some(CursorParts::new(end, is_main, !caret_is_start, Some(false))),
                ]
            } else {
                [
                    Some(CursorParts::new(sel.caret(), is_main, true, None)),
                    None,
                ]
            }
        })
        .flatten()
}

pub struct CursorParts {
    point: Point,
    is_main: bool,
    is_caret: bool,
    is_start: Option<bool>,
}

impl CursorParts {
    fn new(point: Point, is_main: bool, is_caret: bool, is_start: Option<bool>) -> Self {
        Self { point, is_main, is_caret, is_start }
    }
}

#[derive(Clone, Copy)]
enum Cursor {
    Main(bool, bool),
    Extra(bool, bool),
}