about summary refs log tree commit diff
path: root/src/history.rs
blob: 25a9555df425ddc6e8439ba64bb2a83f01a784c7 (plain)
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
81
82
83
84
85
86
87
88
89
use std::{fs::File, io::{self, Read, Seek, Write}};

use serde::{Deserialize, Serialize};

use crate::text_diff::TextChange;

#[derive(Serialize, Deserialize, Debug)]
pub struct FileHistory {
    cursor: usize,
    changes: Vec<FileChange>,
}

impl FileHistory {
    pub fn from_file(file: &mut File) -> Result<FileHistory, ()> {
        let mut buffer = Vec::new();
        file.read_to_end(&mut buffer)
            .expect("Could not read file history,");

        let file_history = serde_json::from_slice::<FileHistory>(&buffer);
        Ok(file_history.expect("Corrupted file history."))
    }

    pub fn cursor(&self) -> usize {
        self.cursor
    }

    pub fn file_is_deleted(&self) -> bool {
        match self.changes.last() {
            Some(change) => match change {
                FileChange::Deleted => true,
                FileChange::Updated(_) => false,
            },
            None => false,
        }
    }

    pub fn get_content(&self) -> String {
        let mut buffer = String::new();
        for file_change in self.changes.iter().take(self.cursor) {
            if let FileChange::Updated(ref updated) = file_change {
                for change in updated.iter() {
                    change.apply(&mut buffer)
                }
            } else {
                buffer = String::new();
            }
        }
        buffer
    }

    pub fn write_to_file(&self, file: &mut File) -> io::Result<()> {
        let encoded: Vec<u8> = serde_json::to_vec(self).unwrap();
        file.rewind()?;
        file.set_len(0)?;
        file.write_all(encoded.as_ref())?;
        Ok(())
    }

    pub fn set_cursor(&mut self, new_cursor: usize) {
        if new_cursor < self.changes.len() {
            self.cursor = new_cursor;
        } else {
            panic!(
                "Out-of-bounds cursor for file history: {}, can be at most {}",
                new_cursor,
                self.changes.len()
            );
        }
    }

    pub fn add_change(&mut self, change: FileChange) {
        self.changes.push(change);
    }
}

impl Default for FileHistory {
    fn default() -> Self {
        Self {
            cursor: 0,
            changes: Vec::new(),
        }
    }
}

#[derive(Serialize, Deserialize, Debug)]
pub enum FileChange {
    Updated(Vec<TextChange>),
    Deleted,
}