blob: bf6ec5df7ec090644cea8ab970c0d5f29ec589a8 (
plain) (
blame)
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
|
#![feature(async_closure)]
use config::Config;
use db::Database;
use error::Error;
use tokio::{fs::File, io::AsyncWriteExt};
mod artist;
mod artwork;
mod comment;
pub mod config;
mod db;
mod error;
mod file;
pub mod routes;
mod ructe_poem;
pub type Result<T> = std::result::Result<T, error::Error>;
#[derive(Clone)]
pub struct Critch {
db: Database,
config: Config,
}
impl Critch {
pub async fn new(config: Config) -> Self {
let db = Database::new(config.database_connection()).await;
Self { db, config }
}
pub async fn save_file(&self, file_name: &str, file_data: Vec<u8>) -> Result<()> {
let upload_dir = self.config.files_dir();
if upload_dir.is_dir() {
let file_handle = upload_dir.join(file_name);
let mut file = File::create(file_handle).await?;
file.write_all(&file_data).await?;
return Ok(());
} else {
return Err(Error::UploadDirectory(
self.config.files_dir().to_string_lossy().to_string(),
));
}
}
}
include!(concat!(env!("OUT_DIR"), "/templates.rs"));
|
