zed/crates/language/src/proto.rs

use std::sync::Arc;

use crate::Diagnostic;

use super::Operation;
use anyhow::{anyhow, Result};
use buffer::*;
use clock::ReplicaId;
use lsp::DiagnosticSeverity;
use rpc::proto;

pub use proto::Buffer;

pub fn serialize_operation(operation: &Operation) -> proto::Operation {
    proto::Operation {
        variant: Some(match operation {
            Operation::Buffer(buffer::Operation::Edit(edit)) => {
                proto::operation::Variant::Edit(serialize_edit_operation(edit))
            }
            Operation::Buffer(buffer::Operation::Undo {
                undo,
                lamport_timestamp,
            }) => proto::operation::Variant::Undo(proto::operation::Undo {
                replica_id: undo.id.replica_id as u32,
                local_timestamp: undo.id.value,
                lamport_timestamp: lamport_timestamp.value,
                ranges: undo
                    .ranges
                    .iter()
                    .map(|r| proto::Range {
                        start: r.start.0 as u64,
                        end: r.end.0 as u64,
                    })
                    .collect(),
                counts: undo
                    .counts
                    .iter()
                    .map(|(edit_id, count)| proto::operation::UndoCount {
                        replica_id: edit_id.replica_id as u32,
                        local_timestamp: edit_id.value,
                        count: *count,
                    })
                    .collect(),
                version: From::from(&undo.version),
            }),
            Operation::Buffer(buffer::Operation::UpdateSelections {
                set_id,
                selections,
                lamport_timestamp,
            }) => proto::operation::Variant::UpdateSelections(proto::operation::UpdateSelections {
                replica_id: set_id.replica_id as u32,
                local_timestamp: set_id.value,
                lamport_timestamp: lamport_timestamp.value,
                version: selections.version().into(),
                selections: selections
                    .full_offset_ranges()
                    .map(|(range, state)| proto::Selection {
                        id: state.id as u64,
                        start: range.start.0 as u64,
                        end: range.end.0 as u64,
                        reversed: state.reversed,
                    })
                    .collect(),
            }),
            Operation::Buffer(buffer::Operation::RemoveSelections {
                set_id,
                lamport_timestamp,
            }) => proto::operation::Variant::RemoveSelections(proto::operation::RemoveSelections {
                replica_id: set_id.replica_id as u32,
                local_timestamp: set_id.value,
                lamport_timestamp: lamport_timestamp.value,
            }),
            Operation::Buffer(buffer::Operation::SetActiveSelections {
                set_id,
                lamport_timestamp,
            }) => proto::operation::Variant::SetActiveSelections(
                proto::operation::SetActiveSelections {
                    replica_id: lamport_timestamp.replica_id as u32,
                    local_timestamp: set_id.map(|set_id| set_id.value),
                    lamport_timestamp: lamport_timestamp.value,
                },
            ),
            Operation::UpdateDiagnostics(diagnostic_set) => {
                proto::operation::Variant::UpdateDiagnostics(serialize_diagnostics(diagnostic_set))
            }
        }),
    }
}

pub fn serialize_edit_operation(operation: &EditOperation) -> proto::operation::Edit {
    let ranges = operation
        .ranges
        .iter()
        .map(|range| proto::Range {
            start: range.start.0 as u64,
            end: range.end.0 as u64,
        })
        .collect();
    proto::operation::Edit {
        replica_id: operation.timestamp.replica_id as u32,
        local_timestamp: operation.timestamp.local,
        lamport_timestamp: operation.timestamp.lamport,
        version: From::from(&operation.version),
        ranges,
        new_text: operation.new_text.clone(),
    }
}

pub fn serialize_selection_set(set: &SelectionSet) -> proto::SelectionSet {
    let version = set.selections.version();
    let entries = set.selections.full_offset_ranges();
    proto::SelectionSet {
        replica_id: set.id.replica_id as u32,
        lamport_timestamp: set.id.value as u32,
        is_active: set.active,
        version: version.into(),
        selections: entries
            .map(|(range, state)| proto::Selection {
                id: state.id as u64,
                start: range.start.0 as u64,
                end: range.end.0 as u64,
                reversed: state.reversed,
            })
            .collect(),
    }
}

pub fn serialize_diagnostics(map: &AnchorRangeMultimap<Diagnostic>) -> proto::DiagnosticSet {
    proto::DiagnosticSet {
        version: map.version().into(),
        diagnostics: map
            .full_offset_ranges()
            .map(|(range, diagnostic)| proto::Diagnostic {
                start: range.start.0 as u64,
                end: range.end.0 as u64,
                message: diagnostic.message.clone(),
                severity: match diagnostic.severity {
                    DiagnosticSeverity::ERROR => proto::diagnostic::Severity::Error,
                    DiagnosticSeverity::WARNING => proto::diagnostic::Severity::Warning,
                    DiagnosticSeverity::INFORMATION => proto::diagnostic::Severity::Information,
                    DiagnosticSeverity::HINT => proto::diagnostic::Severity::Hint,
                    _ => proto::diagnostic::Severity::None,
                } as i32,
                group_id: diagnostic.group_id as u64,
                is_primary: diagnostic.is_primary,
            })
            .collect(),
    }
}

pub fn deserialize_operation(message: proto::Operation) -> Result<Operation> {
    Ok(
        match message
            .variant
            .ok_or_else(|| anyhow!("missing operation variant"))?
        {
            proto::operation::Variant::Edit(edit) => {
                Operation::Buffer(buffer::Operation::Edit(deserialize_edit_operation(edit)))
            }
            proto::operation::Variant::Undo(undo) => Operation::Buffer(buffer::Operation::Undo {
                lamport_timestamp: clock::Lamport {
                    replica_id: undo.replica_id as ReplicaId,
                    value: undo.lamport_timestamp,
                },
                undo: UndoOperation {
                    id: clock::Local {
                        replica_id: undo.replica_id as ReplicaId,
                        value: undo.local_timestamp,
                    },
                    counts: undo
                        .counts
                        .into_iter()
                        .map(|c| {
                            (
                                clock::Local {
                                    replica_id: c.replica_id as ReplicaId,
                                    value: c.local_timestamp,
                                },
                                c.count,
                            )
                        })
                        .collect(),
                    ranges: undo
                        .ranges
                        .into_iter()
                        .map(|r| FullOffset(r.start as usize)..FullOffset(r.end as usize))
                        .collect(),
                    version: undo.version.into(),
                },
            }),
            proto::operation::Variant::UpdateSelections(message) => {
                let version = message.version.into();
                let entries = message
                    .selections
                    .iter()
                    .map(|selection| {
                        let range = (FullOffset(selection.start as usize), Bias::Left)
                            ..(FullOffset(selection.end as usize), Bias::Right);
                        let state = SelectionState {
                            id: selection.id as usize,
                            reversed: selection.reversed,
                            goal: SelectionGoal::None,
                        };
                        (range, state)
                    })
                    .collect();
                let selections = AnchorRangeMap::from_full_offset_ranges(version, entries);

                Operation::Buffer(buffer::Operation::UpdateSelections {
                    set_id: clock::Lamport {
                        replica_id: message.replica_id as ReplicaId,
                        value: message.local_timestamp,
                    },
                    lamport_timestamp: clock::Lamport {
                        replica_id: message.replica_id as ReplicaId,
                        value: message.lamport_timestamp,
                    },
                    selections: Arc::from(selections),
                })
            }
            proto::operation::Variant::RemoveSelections(message) => {
                Operation::Buffer(buffer::Operation::RemoveSelections {
                    set_id: clock::Lamport {
                        replica_id: message.replica_id as ReplicaId,
                        value: message.local_timestamp,
                    },
                    lamport_timestamp: clock::Lamport {
                        replica_id: message.replica_id as ReplicaId,
                        value: message.lamport_timestamp,
                    },
                })
            }
            proto::operation::Variant::SetActiveSelections(message) => {
                Operation::Buffer(buffer::Operation::SetActiveSelections {
                    set_id: message.local_timestamp.map(|value| clock::Lamport {
                        replica_id: message.replica_id as ReplicaId,
                        value,
                    }),
                    lamport_timestamp: clock::Lamport {
                        replica_id: message.replica_id as ReplicaId,
                        value: message.lamport_timestamp,
                    },
                })
            }
            proto::operation::Variant::UpdateDiagnostics(message) => {
                Operation::UpdateDiagnostics(deserialize_diagnostics(message))
            }
        },
    )
}

pub fn deserialize_edit_operation(edit: proto::operation::Edit) -> EditOperation {
    let ranges = edit
        .ranges
        .into_iter()
        .map(|range| FullOffset(range.start as usize)..FullOffset(range.end as usize))
        .collect();
    EditOperation {
        timestamp: InsertionTimestamp {
            replica_id: edit.replica_id as ReplicaId,
            local: edit.local_timestamp,
            lamport: edit.lamport_timestamp,
        },
        version: edit.version.into(),
        ranges,
        new_text: edit.new_text,
    }
}

pub fn deserialize_selection_set(set: proto::SelectionSet) -> SelectionSet {
    SelectionSet {
        id: clock::Lamport {
            replica_id: set.replica_id as u16,
            value: set.lamport_timestamp,
        },
        active: set.is_active,
        selections: Arc::new(AnchorRangeMap::from_full_offset_ranges(
            set.version.into(),
            set.selections
                .into_iter()
                .map(|selection| {
                    let range = (FullOffset(selection.start as usize), Bias::Left)
                        ..(FullOffset(selection.end as usize), Bias::Left);
                    let state = SelectionState {
                        id: selection.id as usize,
                        reversed: selection.reversed,
                        goal: SelectionGoal::None,
                    };
                    (range, state)
                })
                .collect(),
        )),
    }
}

pub fn deserialize_diagnostics(message: proto::DiagnosticSet) -> AnchorRangeMultimap<Diagnostic> {
    AnchorRangeMultimap::from_full_offset_ranges(
        message.version.into(),
        Bias::Left,
        Bias::Right,
        message.diagnostics.into_iter().filter_map(|diagnostic| {
            Some((
                FullOffset(diagnostic.start as usize)..FullOffset(diagnostic.end as usize),
                Diagnostic {
                    severity: match proto::diagnostic::Severity::from_i32(diagnostic.severity)? {
                        proto::diagnostic::Severity::Error => DiagnosticSeverity::ERROR,
                        proto::diagnostic::Severity::Warning => DiagnosticSeverity::WARNING,
                        proto::diagnostic::Severity::Information => DiagnosticSeverity::INFORMATION,
                        proto::diagnostic::Severity::Hint => DiagnosticSeverity::HINT,
                        proto::diagnostic::Severity::None => return None,
                    },
                    message: diagnostic.message,
                    group_id: diagnostic.group_id as usize,
                    is_primary: diagnostic.is_primary,
                },
            ))
        }),
    )
}