1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80

use crate::cursor::CursorPosition;
use crate::update::UpdateStep;
use crate::interface::InterfaceState;
use crate::utility::{clear_rest_of_line, move_cursor_exact, render_segment};
use termion::raw::RawTerminal;
use std::io::StdoutLock;
use crate::result::{Result, TTYError};

pub struct Segment {
    pub text: String,
    // TODO: color
    // TODO: style
}

pub(crate) struct SetSegmentStep {
    pub(crate) line_index: usize,
    pub(crate) segment_index: usize,
    pub(crate) segment: Option<Segment>,
}

impl UpdateStep for SetSegmentStep {
    fn do_update(&mut self, stdout: &mut RawTerminal<StdoutLock>, state: &mut InterfaceState,
                 update_cursor: &mut CursorPosition) -> Result<()> {
        if self.line_index > state.lines.len() - 1 {
            return Err(TTYError::LineOutOfBounds);
        }

        if self.segment_index > state.lines[self.line_index].segments.len() {
            return Err(TTYError::SegmentOutOfBounds);
        }

        let diff_length = state.lines[self.line_index].segments[self.segment_index].text.len()
            != self.segment.as_ref().unwrap().text.len();

        state.lines[self.line_index].segments[self.segment_index] = self.segment.take().unwrap();

        let segment_start = state.lines[self.line_index].get_segment_start(self.segment_index);
        move_cursor_exact(stdout, update_cursor, segment_start, self.line_index as u16)?;
        render_segment(stdout, update_cursor, &state.lines[self.line_index].segments[self.segment_index])?;

        if diff_length {
            clear_rest_of_line(stdout)?;
            for segment in &state.lines[self.line_index].segments[self.segment_index+1..] {
                render_segment(stdout, update_cursor, segment)?;
            }
        }

        Ok(())
    }
}

pub(crate) struct DeleteSegmentStep {
    pub(crate) line_index: usize,
    pub(crate) segment_index: usize,
}

impl UpdateStep for DeleteSegmentStep {
    fn do_update(&mut self, stdout: &mut RawTerminal<StdoutLock>, state: &mut InterfaceState,
                 update_cursor: &mut CursorPosition) -> Result<()> {
        if self.line_index > state.lines.len() - 1 {
            return Err(TTYError::LineOutOfBounds);
        }

        if self.segment_index > state.lines[self.line_index].segments.len() - 1 {
            return Err(TTYError::SegmentOutOfBounds);
        }

        let segment_start = state.lines[self.line_index].get_segment_start(self.segment_index);
        move_cursor_exact(stdout, update_cursor, segment_start, self.line_index as u16)?;
        clear_rest_of_line(stdout)?;

        state.lines[self.line_index].segments.remove(self.segment_index);

        for segment in &state.lines[self.line_index].segments[self.segment_index..] {
            render_segment(stdout, update_cursor, segment)?;
        }

        Ok(())
    }
}