1use std::env;
10use std::fs;
11use std::path::{Path, PathBuf};
12
13use crate::Result;
14use crate::common::env_vars::FRM_DIR;
15use crate::errors::Error;
16use crate::version::Version;
17
18#[derive(Debug, Clone)]
19pub struct Paths {
20 base_dir: PathBuf,
21}
22
23impl Paths {
24 pub fn new() -> Result<Self> {
25 let base_dir = Self::detect_base_dir()?;
26 Ok(Self { base_dir })
27 }
28
29 pub fn with_base_dir(base_dir: PathBuf) -> Self {
30 Self { base_dir }
31 }
32
33 fn detect_base_dir() -> Result<PathBuf> {
34 if let Ok(dir) = env::var(FRM_DIR) {
35 return Ok(PathBuf::from(dir));
36 }
37
38 let home =
39 dirs::home_dir().ok_or_else(|| Error::Config("cannot find home directory".into()))?;
40
41 Ok(home.join(".local").join("frm"))
42 }
43
44 pub fn base_dir(&self) -> &Path {
45 &self.base_dir
46 }
47
48 pub fn versions_dir(&self) -> PathBuf {
49 self.base_dir.join("versions")
50 }
51
52 pub fn version_dir(&self, version: &Version) -> PathBuf {
53 self.versions_dir().join(version.dir_name())
54 }
55
56 pub fn version_sbin_dir(&self, version: &Version) -> PathBuf {
57 self.version_dir(version).join("sbin")
58 }
59
60 pub fn version_etc_dir(&self, version: &Version) -> PathBuf {
61 self.version_dir(version).join("etc").join("rabbitmq")
62 }
63
64 pub fn version_var_log_dir(&self, version: &Version) -> PathBuf {
65 self.version_dir(version)
66 .join("var")
67 .join("log")
68 .join("rabbitmq")
69 }
70
71 pub fn etc_dir(&self) -> PathBuf {
72 self.base_dir.join("etc").join("rabbitmq")
73 }
74
75 pub fn downloads_dir(&self) -> PathBuf {
76 self.base_dir.join("downloads")
77 }
78
79 pub fn config_file(&self) -> PathBuf {
80 self.base_dir.join("config.toml")
81 }
82
83 pub fn default_file(&self) -> PathBuf {
84 self.base_dir.join("default")
85 }
86
87 pub fn timestamps_file(&self) -> PathBuf {
88 self.base_dir.join("version_timestamps.json")
89 }
90
91 pub fn ensure_dirs(&self) -> Result<()> {
92 fs::create_dir_all(self.versions_dir())?;
93 fs::create_dir_all(self.downloads_dir())?;
94 fs::create_dir_all(self.etc_dir())?;
95 Ok(())
96 }
97
98 pub fn version_installed(&self, version: &Version) -> bool {
99 self.version_dir(version).exists()
100 }
101
102 pub fn installed_versions(&self) -> Result<Vec<Version>> {
103 let versions_dir = self.versions_dir();
104 if !versions_dir.exists() {
105 return Ok(Vec::new());
106 }
107
108 let mut versions = Vec::new();
109 for entry in fs::read_dir(versions_dir)? {
110 let entry = entry?;
111 if entry.file_type()?.is_dir()
112 && let Some(name) = entry.file_name().to_str()
113 && let Ok(version) = name.parse::<Version>()
114 {
115 versions.push(version);
116 }
117 }
118
119 versions.sort();
120 Ok(versions)
121 }
122
123 pub fn installed_alpha_versions(&self) -> Result<Vec<Version>> {
124 let versions = self.installed_versions()?;
125 Ok(versions
126 .into_iter()
127 .filter(|v| v.is_distributed_via_server_packages_repository())
128 .collect())
129 }
130
131 pub fn latest_ga_version(&self) -> Result<Option<Version>> {
132 let versions = self.installed_versions()?;
133 Ok(versions.into_iter().rev().find(|v| v.is_ga()))
134 }
135
136 pub fn latest_alpha_version(&self) -> Result<Option<Version>> {
137 let versions = self.installed_versions()?;
138 Ok(versions
139 .into_iter()
140 .rev()
141 .find(|v| v.is_distributed_via_server_packages_repository()))
142 }
143}
144
145impl Default for Paths {
146 fn default() -> Self {
147 Self::new().expect("failed to initialize paths")
148 }
149}