apeirogond 0.17.0

Daemon for Apeirogon, a cross-platform software KVM
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
108
109
110
111
112
113
114

// SPDX-License-Identifier: GPL-3.0-or-later

use a7n_core::layout::{Layout, Merge, Placement};
use a7n_ipc::{LayoutInfo, LayoutPlacement, PlacedMachine, UnplacedMachine};
use a7n_proto::{
    MachineId,
    v1::{self, Envelope, envelope::Body},
};

use super::{Daemon, DaemonError};

impl Daemon {
    pub(super) fn layout_info(&self) -> LayoutInfo {
        let placed = self
            .layout
            .state
            .placements()
            .iter()
            .map(|p| PlacedMachine {
                machine: p.id,
                name: self.layout_name(p.id),
                screen_mm: self.layout_screen_mm(p.id),
                x: p.x,
                y: p.y,
                width: p.width,
                height: p.height,
                blocked: p.blocked,
            })
            .collect();
        let unplaced = std::iter::once(self.identity.machine_id)
            .chain(
                self.pins
                    .machine_ids()
                    .filter(|id| *id != self.identity.machine_id),
            )
            .filter(|id| !self.layout.state.placements().iter().any(|p| p.id == *id))
            .map(|machine| UnplacedMachine {
                machine,
                name: self.layout_name(machine),
                screen_mm: self.layout_screen_mm(machine),
            })
            .collect();
        LayoutInfo {
            placed,
            unplaced,
            lost_edit: self.layout.lost_edit,
        }
    }

    fn layout_screen_mm(&self, id: MachineId) -> Option<(u32, u32)> {
        if id == self.identity.machine_id {
            return self
                .info
                .screen_mm
                .as_ref()
                .map(|size| (size.width, size.height));
        }
        self.pins.get(&id).and_then(|entry| entry.screen_mm)
    }

    fn layout_name(&self, id: MachineId) -> String {
        if id == self.identity.machine_id {
            return self.info.name.clone();
        }
        self.peer_name(id).unwrap_or("unknown").to_owned()
    }

    pub(super) fn set_layout(&mut self, placed: Vec<LayoutPlacement>) -> Result<(), DaemonError> {
        let placements = placed
            .into_iter()
            .map(|p| Placement {
                id: p.machine,
                x: p.x,
                y: p.y,
                width: p.width,
                height: p.height,
                blocked: p.blocked,
            })
            .collect();
        let edited = self
            .layout
            .state
            .edit(placements, self.identity.machine_id)?;
        self.layout.persist(edited)?;
        self.layout.lost_edit = false;
        self.broadcast(&Envelope {
            body: Some(Body::Layout(self.layout.state.to_proto())),
        });
        Ok(())
    }

    pub(super) fn receive_layout(&mut self, wire: &v1::Layout) -> Result<(), DaemonError> {
        let Ok(incoming) = Layout::try_from(wire) else {
            crate::log::event!("kind=layout_rejected");
            return Ok(());
        };
        let mut staged = self.layout.state.clone();
        if let Merge::Replaced { lost_own_edit } = staged.merge(incoming, self.identity.machine_id)
        {
            self.layout.persist(staged)?;
            if lost_own_edit {
                crate::log::event!("kind=layout_edit_lost");
                self.layout.lost_edit = true;
            }
            self.broadcast(&Envelope {
                body: Some(Body::Layout(self.layout.state.to_proto())),
            });
        }
        Ok(())
    }
}

#[cfg(test)]
mod tests;