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
|
use std::{
env::current_dir,
fs::{self, create_dir_all},
path::PathBuf,
};
use cfg_file::config::ConfigFile;
use crate::{
constants::{
SERVER_FILE_README, SERVER_FILE_VAULT, SERVER_PATH_MEMBER_PUB, SERVER_PATH_MEMBERS,
SERVER_PATH_SHEETS, SERVER_PATH_VIRTUAL_FILE_ROOT,
},
current::{current_vault_path, find_vault_path},
workspace::vault::config::VaultConfig,
};
pub mod config;
pub mod member;
pub mod vitrual_file;
pub type MemberId = String;
pub struct Vault {
config: VaultConfig,
vault_path: PathBuf,
}
impl Vault {
/// Get vault path
pub fn vault_path(&self) -> &PathBuf {
&self.vault_path
}
/// Initialize vault
pub fn init(config: VaultConfig, vault_path: impl Into<PathBuf>) -> Option<Self> {
let Some(vault_path) = find_vault_path(vault_path) else {
return None;
};
Some(Self { config, vault_path })
}
/// Initialize vault
pub fn init_current_dir(config: VaultConfig) -> Option<Self> {
let Some(vault_path) = current_vault_path() else {
return None;
};
Some(Self { config, vault_path })
}
/// Setup vault
pub async fn setup_vault(vault_path: impl Into<PathBuf>) -> Result<(), std::io::Error> {
let vault_path: PathBuf = vault_path.into();
// 1. Setup main config
let config = VaultConfig::default();
VaultConfig::write_to(&config, vault_path.join(SERVER_FILE_VAULT)).await?;
// 2. Setup sheets directory
create_dir_all(vault_path.join(SERVER_PATH_SHEETS))?;
// 3. Setup key directory
create_dir_all(vault_path.join(SERVER_PATH_MEMBER_PUB))?;
// 4. Setup member directory
create_dir_all(vault_path.join(SERVER_PATH_MEMBERS))?;
// 5. Setup storage directory
create_dir_all(vault_path.join(SERVER_PATH_VIRTUAL_FILE_ROOT))?;
// Final, generate README.md
let readme_content = format!(
"\
# JustEnoughVCS Server Setup
This directory contains the server configuration and data for `JustEnoughVCS`.
## User Authentication
To allow users to connect to this server, place their public keys in the `{}` directory.
Each public key file should correspond to a registered user.
## File Storage
All version-controlled files (Virtual File) are stored in the `{}` directory.
## License
This software is distributed under the MIT License.
## Support
Repository: `https://github.com/JustEnoughVCS/VersionControl`
Please report any issues or questions on the GitHub issue tracker.
## Thanks :)
Thank you for using `JustEnoughVCS!`
",
SERVER_PATH_MEMBER_PUB, SERVER_PATH_VIRTUAL_FILE_ROOT
)
.trim()
.to_string();
fs::write(vault_path.join(SERVER_FILE_README), readme_content)?;
Ok(())
}
/// Setup vault in current directory
pub async fn setup_vault_current_dir() -> Result<(), std::io::Error> {
Self::setup_vault(current_dir()?).await?;
Ok(())
}
}
|
