open-gpui-canvas 0.2.0

Extensible infinite canvas model and interaction primitives for Open GPUI.
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
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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107

use crate::layer::{
    CanvasLayerRecord, CanvasZOrderCommand, reorder_layer_z_indices, sort_layer_records,
};
use crate::record_scope::{CanvasRecordScopeOptions, collect_selection_record_scope};
use crate::{CanvasDocument, CanvasRecordId, CanvasSelection, CanvasTransaction, DocumentCommand};
use indexmap::IndexSet;

pub(crate) fn reorder_selection_transaction(
    document: &CanvasDocument,
    selection: &CanvasSelection,
    command: CanvasZOrderCommand,
) -> CanvasTransaction {
    let selection_records = z_order_selection_record_ids(document, selection);
    let mut records = z_order_records(document, &selection_records);
    let commands = reorder_layer_z_indices(&mut records, command)
        .into_iter()
        .filter_map(|(record_id, z_index)| z_order_update_command(document, &record_id, z_index))
        .collect::<Vec<_>>();

    CanvasTransaction::new(commands)
}

fn z_order_selection_record_ids(
    document: &CanvasDocument,
    selection: &CanvasSelection,
) -> IndexSet<CanvasRecordId> {
    collect_selection_record_scope(
        document,
        selection,
        CanvasRecordScopeOptions::structural_with_internal_edges(),
        |record_id| is_reorderable_record(document, record_id),
    )
}

fn is_reorderable_record(document: &CanvasDocument, record_id: &CanvasRecordId) -> bool {
    match record_id {
        CanvasRecordId::Node(id) => document.node(id).is_some_and(|node| !node.locked),
        CanvasRecordId::Edge(id) => document.edge(id).is_some_and(|edge| !edge.locked),
        CanvasRecordId::Shape(id) => document.shape(id).is_some_and(|shape| !shape.locked),
    }
}

fn z_order_records(
    document: &CanvasDocument,
    selection_records: &IndexSet<CanvasRecordId>,
) -> Vec<CanvasLayerRecord> {
    let mut ordinal = 0;
    let mut records = Vec::new();

    records.extend(document.nodes().map(|node| {
        let record = CanvasLayerRecord {
            id: CanvasRecordId::Node(node.id.clone()),
            z_index: node.z_index,
            ordinal,
            selected: selection_records.contains(&CanvasRecordId::Node(node.id.clone())),
        };
        ordinal += 1;
        record
    }));
    records.extend(document.shapes().map(|shape| {
        let record = CanvasLayerRecord {
            id: CanvasRecordId::Shape(shape.id.clone()),
            z_index: shape.z_index,
            ordinal,
            selected: selection_records.contains(&CanvasRecordId::Shape(shape.id.clone())),
        };
        ordinal += 1;
        record
    }));
    records.extend(document.edges().map(|edge| {
        let record = CanvasLayerRecord {
            id: CanvasRecordId::Edge(edge.id.clone()),
            z_index: edge.z_index,
            ordinal,
            selected: selection_records.contains(&CanvasRecordId::Edge(edge.id.clone())),
        };
        ordinal += 1;
        record
    }));

    sort_layer_records(&mut records);
    records
}

fn z_order_update_command(
    document: &CanvasDocument,
    record_id: &CanvasRecordId,
    z_index: i32,
) -> Option<DocumentCommand> {
    match record_id {
        CanvasRecordId::Node(id) => {
            let mut node = document.node(id)?.clone();
            node.z_index = z_index;
            Some(DocumentCommand::UpdateNode(node))
        }
        CanvasRecordId::Edge(id) => {
            let mut edge = document.edge(id)?.clone();
            edge.z_index = z_index;
            Some(DocumentCommand::UpdateEdge(edge))
        }
        CanvasRecordId::Shape(id) => {
            let mut shape = document.shape(id)?.clone();
            shape.z_index = z_index;
            Some(DocumentCommand::UpdateShape(shape))
        }
    }
}