clankerdiff-protocol 0.3.3

Versioned subprocess and live session protocols for Clankerdiff
Documentation 
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

use crate::{
    server::{ServerEvent, ServerMessage},
    shared::{DiffSnapshot, DocumentUpdate, FileEntry},
};
use clankerdiff_core::{FileDiff, RepoPath};
use std::{collections::BTreeMap, sync::Arc};

#[derive(Debug, Default)]
pub struct SentDocument {
    files: BTreeMap<RepoPath, Arc<FileDiff>>,
}

impl SentDocument {
    pub fn encode_event(&mut self, event: ServerEvent) -> ServerMessage {
        event.map(|snapshot| self.encode(&snapshot))
    }

    pub fn encode(&mut self, snapshot: &DiffSnapshot) -> DocumentUpdate {
        let mut files = Vec::with_capacity(snapshot.document.files.len());
        let mut sent = BTreeMap::new();
        for file in &snapshot.document.files {
            let entry = match self.files.get(&file.path) {
                Some(held) if held.as_ref() == file => {
                    sent.insert(file.path.clone(), Arc::clone(held));
                    FileEntry::Unchanged(file.path.clone())
                }
                _ => {
                    let file = Arc::new(file.clone());
                    sent.insert(file.path.clone(), Arc::clone(&file));
                    FileEntry::Changed(file)
                }
            };
            files.push(entry);
        }
        self.files = sent;
        DocumentUpdate {
            scope: snapshot.scope,
            repo_root: snapshot.document.repo_root.clone(),
            files,
        }
    }
}