zap_rs/
manager.rs

1use std::{
2    io::{self, Write},
3    path::PathBuf,
4};
5use tokio::fs;
6
7use crate::{
8    AppImage, Downloader, Error, Index, Result, SymlinkManager, desktops_dir,
9    get_github_release_url, icons_dir, index_dir,
10};
11
12#[derive(Debug, Default)]
13pub struct PackageManager {
14    pub downloader: Downloader,
15    pub index: Index,
16    pub symlink_manager: SymlinkManager,
17}
18
19impl PackageManager {
20    pub fn new() -> Self {
21        Self {
22            downloader: Downloader::new(),
23            index: Index::new(),
24            symlink_manager: SymlinkManager::new(),
25        }
26    }
27    pub async fn install(&self, appimage: &mut AppImage, appname: &str) -> Result<()> {
28        if self.index.exists(&appimage.executable)? {
29            println!("{} is already installed.", appimage.executable);
30            return Ok(());
31        }
32
33        appimage.file_path = self
34            .downloader
35            .prepare_path(&appimage.source.meta.url, &appimage.executable)?;
36
37        if appimage.source.identifier != "git.github" {
38            self.downloader
39                .download_with_progress(&appimage.source.meta.url, &appimage.file_path)
40                .await?;
41        } else {
42            self.downloader
43                .download_with_progress(
44                    &get_github_release_url(appimage).await?,
45                    &appimage.file_path,
46                )
47                .await?;
48        }
49
50        self.index.add(appimage, appname).await?;
51        self.symlink_manager.create(appimage).await?;
52
53        print!("Do you want to integrate this appimage? (y/N) ");
54        io::stdout().flush()?;
55
56        let mut input = String::new();
57        io::stdin().read_line(&mut input)?;
58
59        if input.to_lowercase().trim() == "y" || input.to_lowercase().trim() == "yes" {
60            appimage.integrate_desktop().await?;
61        }
62
63        Ok(())
64    }
65    pub async fn remove(&self, appname: &str) -> Result<()> {
66        let appimage = self.index.get(appname).await?;
67
68        fs::remove_file(&appimage.file_path).await?;
69        self.symlink_manager.remove(&appimage.executable).await?;
70        self.index.remove(appname).await?;
71
72        if fs::try_exists(desktops_dir()?.join(format!("{}.desktop", appimage.executable))).await? {
73            fs::remove_file(desktops_dir()?.join(format!("{}.desktop", appimage.executable)))
74                .await?;
75        }
76        if fs::try_exists(PathBuf::from(std::env::var("HOME")?).join(format!(
77            ".local/share/applications/{}.desktop",
78            appimage.executable
79        )))
80        .await?
81        {
82            fs::remove_file(PathBuf::from(std::env::var("HOME")?).join(format!(
83                ".local/share/applications/{}.desktop",
84                appimage.executable
85            )))
86            .await?;
87        }
88        if fs::try_exists(icons_dir()?.join(format!("{}.png", appimage.executable))).await? {
89            fs::remove_file(icons_dir()?.join(format!("{}.png", appimage.executable))).await?;
90        }
91
92        Ok(())
93    }
94    pub async fn list(&self) -> Result<()> {
95        let mut appimages = fs::read_dir(index_dir()?).await?;
96
97        while let Some(appimage) = appimages.next_entry().await? {
98            if let Some(stem) = appimage.path().file_stem().and_then(|s| s.to_str()) {
99                println!("- {stem}");
100            }
101        }
102
103        Ok(())
104    }
105    pub async fn update(&self, appimage: &mut AppImage) -> Result<()> {
106        if appimage.source.identifier != "git.github".to_string() {
107            return Err(Error::CantUpdatePkg);
108        }
109
110        self.downloader
111            .download_with_progress(
112                &get_github_release_url(appimage).await?,
113                &appimage.file_path,
114            )
115            .await?;
116
117        Ok(())
118    }
119}
zap_rs/manager.rs

zap_rs/
manager.rs
