oma-pm 0.63.0

APT package manager API abstraction library
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

use crate::{apt::AptConfig, dbus::change_status};
use oma_apt::progress::DynInstallProgress;
use oma_utils::zbus;
use once_cell::sync::OnceCell;
use tokio::runtime::Runtime;
use zbus::Connection;

pub use oma_apt::util::{get_apt_progress_string, terminal_height, terminal_width};

#[derive(Default, Debug)]
pub struct NoProgress {
    _lastline: usize,
    _pulse_interval: usize,
    _disable: bool,
}

pub(crate) struct InstallProgressArgs {
    pub config: AptConfig,
    pub tokio: OnceCell<Runtime>,
    pub connection: OnceCell<Connection>,
}

pub(crate) struct OmaAptInstallProgress {
    config: AptConfig,
    tokio: OnceCell<Runtime>,
    connection: OnceCell<Connection>,
    pm: Box<dyn InstallProgressManager>,
}

pub trait InstallProgressManager {
    fn status_change(&self, pkgname: &str, steps_done: u64, total_steps: u64, config: &AptConfig);
    fn no_interactive(&self) -> bool;
    fn use_pty(&self) -> bool;
}

impl OmaAptInstallProgress {
    pub fn new(args: InstallProgressArgs, pm: Box<dyn InstallProgressManager>) -> Self {
        let InstallProgressArgs {
            config,
            tokio,
            connection,
        } = args;

        if pm.no_interactive() {
            unsafe { std::env::set_var("DEBIAN_FRONTEND", "noninteractive") };
        }

        if !pm.use_pty() {
            config.set("Dpkg::Use-Pty", "false");
        }

        Self {
            config,
            tokio,
            connection,
            pm,
        }
    }
}

impl DynInstallProgress for OmaAptInstallProgress {
    fn status_changed(
        &mut self,
        pkgname: String,
        steps_done: u64,
        total_steps: u64,
        _action: String,
    ) {
        let conn = &self.connection;

        self.pm
            .status_change(&pkgname, steps_done, total_steps, &self.config);

        if let Some(tokio) = self.tokio.get()
            && let Some(conn) = conn.get()
        {
            tokio.block_on(async move {
                change_status(conn, &format!("i {pkgname}")).await.ok();
            });
        }
    }

    fn error(&mut self, _pkgname: String, _steps_done: u64, _total_steps: u64, _error: String) {}
}