summaryrefslogtreecommitdiff
path: root/src/manager.rs
blob: bdb152a9ce4c720fa19b2d16e9198810f93156a3 (plain)
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
use std::{
    io::{self, Write},
    path::PathBuf,
};
use tokio::fs;

use crate::{
    AppImage, Downloader, Index, Result, SymlinkManager, desktops_dir, get_github_release_url,
    icons_dir, index_dir,
};

#[derive(Debug, Default)]
pub struct PackageManager {
    pub downloader: Downloader,
    pub index: Index,
    pub symlink_manager: SymlinkManager,
}

impl PackageManager {
    pub fn new() -> Self {
        Self {
            downloader: Downloader::new(),
            index: Index::new(),
            symlink_manager: SymlinkManager::new(),
        }
    }
    pub async fn install(&self, appimage: &mut AppImage, appname: &str) -> Result<()> {
        if self.index.exists(&appimage.executable)? {
            println!("{} is already installed.", appimage.executable);
            return Ok(());
        }

        appimage.file_path = self
            .downloader
            .prepare_path(&appimage.source.meta.url, &appimage.executable)?;

        if appimage.source.identifier != "git.github" {
            self.downloader
                .download_with_progress(&appimage.source.meta.url, &appimage.file_path)
                .await?;
        } else {
            self.downloader
                .download_with_progress(
                    &get_github_release_url(appimage).await?,
                    &appimage.file_path,
                )
                .await?;
        }

        self.index.add(appimage, appname).await?;
        self.symlink_manager.create(appimage).await?;

        print!("Do you want to integrate this appimage? (y/N) ");
        io::stdout().flush()?;

        let mut input = String::new();
        io::stdin().read_line(&mut input)?;

        if input.to_lowercase().trim() == "y" || input.to_lowercase().trim() == "yes" {
            appimage.integrate_desktop().await?;
        }

        Ok(())
    }
    pub async fn remove(&self, appname: &str) -> Result<()> {
        let appimage = self.index.get(appname).await?;

        fs::remove_file(&appimage.file_path).await?;
        self.symlink_manager.remove(&appimage.executable).await?;
        self.index.remove(appname).await?;

        if fs::try_exists(desktops_dir()?.join(format!("{}.desktop", appimage.executable))).await? {
            fs::remove_file(desktops_dir()?.join(format!("{}.desktop", appimage.executable)))
                .await?;
        }
        if fs::try_exists(PathBuf::from(std::env::var("HOME")?).join(format!(
            ".local/share/applications/{}.desktop",
            appimage.executable
        )))
        .await?
        {
            fs::remove_file(PathBuf::from(std::env::var("HOME")?).join(format!(
                ".local/share/applications/{}.desktop",
                appimage.executable
            )))
            .await?;
        }
        if fs::try_exists(icons_dir()?.join(format!("{}.png", appimage.executable))).await? {
            fs::remove_file(icons_dir()?.join(format!("{}.png", appimage.executable))).await?;
        }

        Ok(())
    }
    pub async fn list(&self) -> Result<()> {
        let mut appimages = fs::read_dir(index_dir()?).await?;

        while let Some(appimage) = appimages.next_entry().await? {
            if let Some(stem) = appimage.path().file_stem().and_then(|s| s.to_str()) {
                println!("- {stem}");
            }
        }

        Ok(())
    }
}