Hexor/mus-fuse
1
0
Fork 0
mirror of https://github.com/house-of-vanity/mus-fuse.git synced 2025-07-06 13:14:08 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add arg parser.

Browse Source
This commit is contained in:
Alexandr Bogomyakov
2020-04-25 22:20:32 +03:00
parent 824ccbb366
commit 3446e7f3d5
2 changed files with 43 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
Cargo.toml
View File

@ -1,6 +1,6 @@
[package]
name = "musfuse"
version = "0.5.0"
version = "0.6.0"
authors = ["AB <ultradesu@hexor.ru>"]
edition = "2018"
@ -9,6 +9,7 @@ edition = "2018"
reqwest = { version = "0.10", features = ["json", "blocking"] }
tokio = { version = "0.2", features = ["full"] }
serde = { version = "1.0", features = ["derive"] }
clap = {version = "2.33", features = ["yaml"]}
serde_json = "1.0"
percent-encoding = "2.1.0"
fuse = "0.3.1"

75
src/main.rs
View File

@ -1,4 +1,5 @@
extern crate base64;
extern crate clap;
extern crate fuse;
extern crate libc;
extern crate time;
@ -6,12 +7,15 @@ extern crate time;
extern crate log;
extern crate chrono;
use time::Timespec;
use percent_encoding::percent_decode_str;
use serde::Deserialize;
use clap::{App, Arg};
use env_logger::Env;
use fuse::{
FileAttr, FileType, Filesystem, ReplyAttr, ReplyData, ReplyDirectory, ReplyEntry, Request,
};
use libc::{EIO, ENOENT};
use percent_encoding::percent_decode_str;
use reqwest::{blocking::Client, header::CONTENT_LENGTH};
use serde::Deserialize;
use size_format::SizeFormatterBinary;
use std::{
collections::{BTreeMap, HashMap, HashSet},
@ -23,10 +27,11 @@ use std::{
thread::sleep,
time::Duration,
};
use fuse::{
FileAttr, FileType, Filesystem, ReplyAttr, ReplyData, ReplyDirectory, ReplyEntry, Request,
};
use time::Timespec;
const CACHE_HEAD: i64 = 768 * 1024; // bytes from beginning of each file cached
const MAX_CACHE_SIZE: i64 = 10; // Count of files cached
static mut HTTP_AUTH: String = String::new();
struct Metrics {
http_requests: u64,
@ -80,10 +85,6 @@ pub struct Track {
pub size: Option<i64>,
}
const CACHE_HEAD: i64 = 768 * 1024;
const MAX_CACHE_SIZE: i64 = 10; // Count
static mut HTTP_AUTH: String = String::new();
fn get_basename(path: Option<&String>) -> Option<String> {
let base = match percent_decode_str(path.unwrap().as_str()).decode_utf8() {
Ok(path) => {
@ -200,7 +201,6 @@ impl JsonFilesystem {
};
attrs.insert(metrics_attr.ino, metrics_attr);
inodes.insert("METRICS.TXT".to_string(), metrics_attr.ino);
warn!("Len: attrs: {}, ino: {}", attrs.len(), inodes.len());
info!(
"Filesystem initialized. Size: {} files, {}B in total.",
inodes.len(),
@ -484,28 +484,35 @@ impl Filesystem for JsonFilesystem {
}
fn main() {
env_logger::init();
env_logger::from_env(Env::default().default_filter_or("info")).init();
info!("Logger initialized. Set RUST_LOG=[debug,error,info,warn,trace] Default: info");
let mountpoint = match env::args().nth(1) {
Some(path) => path,
None => {
println!(
"Usage: {} <MOUNTPOINT> <SERVER>",
env::args().nth(0).unwrap()
);
return;
}
};
let server = match env::args().nth(2) {
Some(server) => server,
None => {
println!(
"Usage: {} <MOUNTPOINT> <SERVER>",
env::args().nth(0).unwrap()
);
return;
}
};
let cli_args = App::new("mus-fuse")
.version(env!("CARGO_PKG_VERSION"))
.author(env!("CARGO_PKG_AUTHORS"))
.about("Mount FUSE filesystem with your own remote library.")
.arg(
Arg::with_name("server")
.short("s")
.long("server")
.value_name("SERVER")
.help("Sets a server hosting your library")
.required(true)
.takes_value(true),
)
.arg(
Arg::with_name("mountpoint")
.short("m")
.long("mountpoint")
.value_name("MOUNT POINT")
.help("Mount point for library.")
.required(true)
.takes_value(true),
)
.get_matches();
let server = cli_args.value_of("server").unwrap().to_string();
let mountpoint = cli_args.value_of("mountpoint").unwrap().to_string();
unsafe {
METRICS.server_addr = server.clone();
}
@ -588,6 +595,6 @@ fn main() {
})
.expect("Error setting Ctrl-C handler");
loop {
sleep(Duration::from_millis(2));
sleep(Duration::from_millis(300));
}
}