mirror of
https://github.com/sigoden/dufs.git
synced 2026-04-09 09:09:03 +03:00
Compare commits
4 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
c1c6dbc356 | ||
|
|
e29cf4c752 | ||
|
|
7f062b6705 | ||
|
|
ea8b9e9cce |
14
CHANGELOG.md
14
CHANGELOG.md
@@ -2,6 +2,20 @@
|
|||||||
|
|
||||||
All notable changes to this project will be documented in this file.
|
All notable changes to this project will be documented in this file.
|
||||||
|
|
||||||
|
## [0.21.0] - 2022-06-21
|
||||||
|
|
||||||
|
### Bug Fixes
|
||||||
|
|
||||||
|
- Escape name contains html escape code ([#65](https://github.com/sigoden/dufs/issues/65))
|
||||||
|
|
||||||
|
### Features
|
||||||
|
|
||||||
|
- Use custom logger with timestamp in rfc3339 ([#67](https://github.com/sigoden/dufs/issues/67))
|
||||||
|
|
||||||
|
### Refactor
|
||||||
|
|
||||||
|
- Split css/js from index.html ([#68](https://github.com/sigoden/dufs/issues/68))
|
||||||
|
|
||||||
## [0.20.0] - 2022-06-20
|
## [0.20.0] - 2022-06-20
|
||||||
|
|
||||||
### Bug Fixes
|
### Bug Fixes
|
||||||
|
|||||||
19
Cargo.lock
generated
19
Cargo.lock
generated
@@ -572,7 +572,7 @@ checksum = "fea41bba32d969b513997752735605054bc0dfa92b4c56bf1189f2e174be7a10"
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "dufs"
|
name = "dufs"
|
||||||
version = "0.20.0"
|
version = "0.21.0"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"assert_cmd",
|
"assert_cmd",
|
||||||
"assert_fs",
|
"assert_fs",
|
||||||
@@ -583,7 +583,6 @@ dependencies = [
|
|||||||
"chrono",
|
"chrono",
|
||||||
"clap",
|
"clap",
|
||||||
"diqwest",
|
"diqwest",
|
||||||
"env_logger",
|
|
||||||
"futures",
|
"futures",
|
||||||
"get_if_addrs",
|
"get_if_addrs",
|
||||||
"headers",
|
"headers",
|
||||||
@@ -629,16 +628,6 @@ dependencies = [
|
|||||||
"cfg-if",
|
"cfg-if",
|
||||||
]
|
]
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "env_logger"
|
|
||||||
version = "0.9.0"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "0b2cf0344971ee6c64c31be0d530793fba457d322dfec2810c453d0ef228f9c3"
|
|
||||||
dependencies = [
|
|
||||||
"humantime",
|
|
||||||
"log",
|
|
||||||
]
|
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "event-listener"
|
name = "event-listener"
|
||||||
version = "2.5.2"
|
version = "2.5.2"
|
||||||
@@ -1033,12 +1022,6 @@ version = "1.0.2"
|
|||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
checksum = "c4a1e36c821dbe04574f602848a19f742f4fb3c98d40449f11bcad18d6b17421"
|
checksum = "c4a1e36c821dbe04574f602848a19f742f4fb3c98d40449f11bcad18d6b17421"
|
||||||
|
|
||||||
[[package]]
|
|
||||||
name = "humantime"
|
|
||||||
version = "2.1.0"
|
|
||||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
|
||||||
checksum = "9a3a5bfb195931eeb336b2a7b4d761daec841b97f947d34394601737a7bba5e4"
|
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "hyper"
|
name = "hyper"
|
||||||
version = "0.14.19"
|
version = "0.14.19"
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "dufs"
|
name = "dufs"
|
||||||
version = "0.20.0"
|
version = "0.21.0"
|
||||||
edition = "2021"
|
edition = "2021"
|
||||||
authors = ["sigoden <sigoden@gmail.com>"]
|
authors = ["sigoden <sigoden@gmail.com>"]
|
||||||
description = "Dufs is a distinctive utility file server"
|
description = "Dufs is a distinctive utility file server"
|
||||||
@@ -34,7 +34,6 @@ lazy_static = "1.4"
|
|||||||
uuid = { version = "1.1", features = ["v4", "fast-rng"] }
|
uuid = { version = "1.1", features = ["v4", "fast-rng"] }
|
||||||
urlencoding = "2.1"
|
urlencoding = "2.1"
|
||||||
xml-rs = "0.8"
|
xml-rs = "0.8"
|
||||||
env_logger = { version = "0.9", default-features = false, features = ["humantime"] }
|
|
||||||
log = "0.4"
|
log = "0.4"
|
||||||
socket2 = "0.4"
|
socket2 = "0.4"
|
||||||
async-stream = "0.3"
|
async-stream = "0.3"
|
||||||
|
|||||||
@@ -64,14 +64,16 @@ class Uploader {
|
|||||||
|
|
||||||
upload() {
|
upload() {
|
||||||
const { file, idx, name } = this;
|
const { file, idx, name } = this;
|
||||||
let url = getUrl(name);
|
const url = getUrl(name);
|
||||||
|
const encodedUrl = encodedStr(url);
|
||||||
|
const encodedName = encodedStr(name);
|
||||||
$uploadersTable.insertAdjacentHTML("beforeend", `
|
$uploadersTable.insertAdjacentHTML("beforeend", `
|
||||||
<tr id="upload${idx}" class="uploader">
|
<tr id="upload${idx}" class="uploader">
|
||||||
<td class="path cell-icon">
|
<td class="path cell-icon">
|
||||||
${getSvg(file.path_type)}
|
${getSvg(file.path_type)}
|
||||||
</td>
|
</td>
|
||||||
<td class="path cell-name">
|
<td class="path cell-name">
|
||||||
<a href="${url}">${name}</a>
|
<a href="${encodedUrl}">${encodedName}</a>
|
||||||
</td>
|
</td>
|
||||||
<td class="cell-status upload-status" id="uploadStatus${idx}"></td>
|
<td class="cell-status upload-status" id="uploadStatus${idx}"></td>
|
||||||
</tr>`);
|
</tr>`);
|
||||||
@@ -141,12 +143,14 @@ function addBreadcrumb(href, uri_prefix) {
|
|||||||
}
|
}
|
||||||
path += encodeURI(name);
|
path += encodeURI(name);
|
||||||
}
|
}
|
||||||
|
const encodedPath = encodedStr(path);
|
||||||
|
const encodedName = encodedStr(name);
|
||||||
if (i === 0) {
|
if (i === 0) {
|
||||||
$breadcrumb.insertAdjacentHTML("beforeend", `<a href="${path}"><svg width="16" height="16" viewBox="0 0 16 16"><path d="M6.5 14.5v-3.505c0-.245.25-.495.5-.495h2c.25 0 .5.25.5.5v3.5a.5.5 0 0 0 .5.5h4a.5.5 0 0 0 .5-.5v-7a.5.5 0 0 0-.146-.354L13 5.793V2.5a.5.5 0 0 0-.5-.5h-1a.5.5 0 0 0-.5.5v1.293L8.354 1.146a.5.5 0 0 0-.708 0l-6 6A.5.5 0 0 0 1.5 7.5v7a.5.5 0 0 0 .5.5h4a.5.5 0 0 0 .5-.5z"/></svg></a>`);
|
$breadcrumb.insertAdjacentHTML("beforeend", `<a href="${encodedPath}"><svg width="16" height="16" viewBox="0 0 16 16"><path d="M6.5 14.5v-3.505c0-.245.25-.495.5-.495h2c.25 0 .5.25.5.5v3.5a.5.5 0 0 0 .5.5h4a.5.5 0 0 0 .5-.5v-7a.5.5 0 0 0-.146-.354L13 5.793V2.5a.5.5 0 0 0-.5-.5h-1a.5.5 0 0 0-.5.5v1.293L8.354 1.146a.5.5 0 0 0-.708 0l-6 6A.5.5 0 0 0 1.5 7.5v7a.5.5 0 0 0 .5.5h4a.5.5 0 0 0 .5-.5z"/></svg></a>`);
|
||||||
} else if (i === len - 1) {
|
} else if (i === len - 1) {
|
||||||
$breadcrumb.insertAdjacentHTML("beforeend", `<b>${name}</b>`);
|
$breadcrumb.insertAdjacentHTML("beforeend", `<b>${encodedName}</b>`);
|
||||||
} else {
|
} else {
|
||||||
$breadcrumb.insertAdjacentHTML("beforeend", `<a href="${path}">${name}</a>`);
|
$breadcrumb.insertAdjacentHTML("beforeend", `<a href="${encodedPath}">${encodedName}</a>`);
|
||||||
}
|
}
|
||||||
if (i !== len - 1) {
|
if (i !== len - 1) {
|
||||||
$breadcrumb.insertAdjacentHTML("beforeend", `<span class="separator">/</span>`);
|
$breadcrumb.insertAdjacentHTML("beforeend", `<span class="separator">/</span>`);
|
||||||
@@ -160,28 +164,31 @@ function addBreadcrumb(href, uri_prefix) {
|
|||||||
* @param {number} index
|
* @param {number} index
|
||||||
*/
|
*/
|
||||||
function addPath(file, index) {
|
function addPath(file, index) {
|
||||||
|
const encodedName = encodedStr(file.name);
|
||||||
let url = getUrl(file.name)
|
let url = getUrl(file.name)
|
||||||
|
let encodedUrl = encodedStr(url);
|
||||||
let actionDelete = "";
|
let actionDelete = "";
|
||||||
let actionDownload = "";
|
let actionDownload = "";
|
||||||
if (file.path_type.endsWith("Dir")) {
|
if (file.path_type.endsWith("Dir")) {
|
||||||
url += "/";
|
url += "/";
|
||||||
|
encodedUrl += "/";
|
||||||
actionDownload = `
|
actionDownload = `
|
||||||
<div class="action-btn">
|
<div class="action-btn">
|
||||||
<a href="${url}?zip" title="Download folder as a .zip file">
|
<a href="${encodedUrl}?zip" title="Download folder as a .zip file">
|
||||||
<svg width="16" height="16" viewBox="0 0 16 16"><path d="M.5 9.9a.5.5 0 0 1 .5.5v2.5a1 1 0 0 0 1 1h12a1 1 0 0 0 1-1v-2.5a.5.5 0 0 1 1 0v2.5a2 2 0 0 1-2 2H2a2 2 0 0 1-2-2v-2.5a.5.5 0 0 1 .5-.5z"/><path d="M7.646 11.854a.5.5 0 0 0 .708 0l3-3a.5.5 0 0 0-.708-.708L8.5 10.293V1.5a.5.5 0 0 0-1 0v8.793L5.354 8.146a.5.5 0 1 0-.708.708l3 3z"/></svg>
|
<svg width="16" height="16" viewBox="0 0 16 16"><path d="M.5 9.9a.5.5 0 0 1 .5.5v2.5a1 1 0 0 0 1 1h12a1 1 0 0 0 1-1v-2.5a.5.5 0 0 1 1 0v2.5a2 2 0 0 1-2 2H2a2 2 0 0 1-2-2v-2.5a.5.5 0 0 1 .5-.5z"/><path d="M7.646 11.854a.5.5 0 0 0 .708 0l3-3a.5.5 0 0 0-.708-.708L8.5 10.293V1.5a.5.5 0 0 0-1 0v8.793L5.354 8.146a.5.5 0 1 0-.708.708l3 3z"/></svg>
|
||||||
</a>
|
</a>
|
||||||
</div>`;
|
</div>`;
|
||||||
} else {
|
} else {
|
||||||
actionDownload = `
|
actionDownload = `
|
||||||
<div class="action-btn" >
|
<div class="action-btn" >
|
||||||
<a href="${url}" title="Download file" download>
|
<a href="${encodedUrl}" title="Download file" download>
|
||||||
<svg width="16" height="16" viewBox="0 0 16 16"><path d="M.5 9.9a.5.5 0 0 1 .5.5v2.5a1 1 0 0 0 1 1h12a1 1 0 0 0 1-1v-2.5a.5.5 0 0 1 1 0v2.5a2 2 0 0 1-2 2H2a2 2 0 0 1-2-2v-2.5a.5.5 0 0 1 .5-.5z"/><path d="M7.646 11.854a.5.5 0 0 0 .708 0l3-3a.5.5 0 0 0-.708-.708L8.5 10.293V1.5a.5.5 0 0 0-1 0v8.793L5.354 8.146a.5.5 0 1 0-.708.708l3 3z"/></svg>
|
<svg width="16" height="16" viewBox="0 0 16 16"><path d="M.5 9.9a.5.5 0 0 1 .5.5v2.5a1 1 0 0 0 1 1h12a1 1 0 0 0 1-1v-2.5a.5.5 0 0 1 1 0v2.5a2 2 0 0 1-2 2H2a2 2 0 0 1-2-2v-2.5a.5.5 0 0 1 .5-.5z"/><path d="M7.646 11.854a.5.5 0 0 0 .708 0l3-3a.5.5 0 0 0-.708-.708L8.5 10.293V1.5a.5.5 0 0 0-1 0v8.793L5.354 8.146a.5.5 0 1 0-.708.708l3 3z"/></svg>
|
||||||
</a>
|
</a>
|
||||||
</div>`;
|
</div>`;
|
||||||
}
|
}
|
||||||
if (DATA.allow_delete) {
|
if (DATA.allow_delete) {
|
||||||
actionDelete = `
|
actionDelete = `
|
||||||
<div onclick="deletePath(${index})" class="action-btn" id="deleteBtn${index}" title="Delete ${file.name}">
|
<div onclick="deletePath(${index})" class="action-btn" id="deleteBtn${index}" title="Delete ${encodedName}">
|
||||||
<svg width="16" height="16" fill="currentColor"viewBox="0 0 16 16"><path d="M6.854 7.146a.5.5 0 1 0-.708.708L7.293 9l-1.147 1.146a.5.5 0 0 0 .708.708L8 9.707l1.146 1.147a.5.5 0 0 0 .708-.708L8.707 9l1.147-1.146a.5.5 0 0 0-.708-.708L8 8.293 6.854 7.146z"/><path d="M14 14V4.5L9.5 0H4a2 2 0 0 0-2 2v12a2 2 0 0 0 2 2h8a2 2 0 0 0 2-2zM9.5 3A1.5 1.5 0 0 0 11 4.5h2V14a1 1 0 0 1-1 1H4a1 1 0 0 1-1-1V2a1 1 0 0 1 1-1h5.5v2z"/></svg>
|
<svg width="16" height="16" fill="currentColor"viewBox="0 0 16 16"><path d="M6.854 7.146a.5.5 0 1 0-.708.708L7.293 9l-1.147 1.146a.5.5 0 0 0 .708.708L8 9.707l1.146 1.147a.5.5 0 0 0 .708-.708L8.707 9l1.147-1.146a.5.5 0 0 0-.708-.708L8 8.293 6.854 7.146z"/><path d="M14 14V4.5L9.5 0H4a2 2 0 0 0-2 2v12a2 2 0 0 0 2 2h8a2 2 0 0 0 2-2zM9.5 3A1.5 1.5 0 0 0 11 4.5h2V14a1 1 0 0 1-1 1H4a1 1 0 0 1-1-1V2a1 1 0 0 1 1-1h5.5v2z"/></svg>
|
||||||
</div>`;
|
</div>`;
|
||||||
}
|
}
|
||||||
@@ -197,7 +204,7 @@ function addPath(file, index) {
|
|||||||
${getSvg(file.path_type)}
|
${getSvg(file.path_type)}
|
||||||
</td>
|
</td>
|
||||||
<td class="path cell-name">
|
<td class="path cell-name">
|
||||||
<a href="${url}" title="${file.name}">${file.name}</a>
|
<a href="${encodedUrl}">${encodedName}</a>
|
||||||
</td>
|
</td>
|
||||||
<td class="cell-mtime">${formatMtime(file.mtime)}</td>
|
<td class="cell-mtime">${formatMtime(file.mtime)}</td>
|
||||||
<td class="cell-size">${formatSize(file.size).join(" ")}</td>
|
<td class="cell-size">${formatSize(file.size).join(" ")}</td>
|
||||||
@@ -333,7 +340,14 @@ function formatPercent(precent) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
function encodedStr(rawStr) {
|
||||||
|
return rawStr.replace(/[\u00A0-\u9999<>\&]/g, function(i) {
|
||||||
|
return '&#'+i.charCodeAt(0)+';';
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
function ready() {
|
function ready() {
|
||||||
|
document.title = `Index of ${DATA.href} - Dufs`;
|
||||||
$pathsTable = document.querySelector(".paths-table")
|
$pathsTable = document.querySelector(".paths-table")
|
||||||
$pathsTableBody = document.querySelector(".paths-table tbody");
|
$pathsTableBody = document.querySelector(".paths-table tbody");
|
||||||
$uploadersTable = document.querySelector(".uploaders-table");
|
$uploadersTable = document.querySelector(".uploaders-table");
|
||||||
|
|||||||
30
src/logger.rs
Normal file
30
src/logger.rs
Normal file
@@ -0,0 +1,30 @@
|
|||||||
|
use chrono::{Local, SecondsFormat};
|
||||||
|
use log::{Level, Metadata, Record};
|
||||||
|
use log::{LevelFilter, SetLoggerError};
|
||||||
|
|
||||||
|
struct SimpleLogger;
|
||||||
|
|
||||||
|
impl log::Log for SimpleLogger {
|
||||||
|
fn enabled(&self, metadata: &Metadata) -> bool {
|
||||||
|
metadata.level() <= Level::Info
|
||||||
|
}
|
||||||
|
|
||||||
|
fn log(&self, record: &Record) {
|
||||||
|
if self.enabled(record.metadata()) {
|
||||||
|
let timestamp = Local::now().to_rfc3339_opts(SecondsFormat::Secs, true);
|
||||||
|
if record.level() < Level::Info {
|
||||||
|
eprintln!("{} {} - {}", timestamp, record.level(), record.args());
|
||||||
|
} else {
|
||||||
|
println!("{} {} - {}", timestamp, record.level(), record.args());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn flush(&self) {}
|
||||||
|
}
|
||||||
|
|
||||||
|
static LOGGER: SimpleLogger = SimpleLogger;
|
||||||
|
|
||||||
|
pub fn init() -> Result<(), SetLoggerError> {
|
||||||
|
log::set_logger(&LOGGER).map(|()| log::set_max_level(LevelFilter::Info))
|
||||||
|
}
|
||||||
15
src/main.rs
15
src/main.rs
@@ -1,5 +1,6 @@
|
|||||||
mod args;
|
mod args;
|
||||||
mod auth;
|
mod auth;
|
||||||
|
mod logger;
|
||||||
mod server;
|
mod server;
|
||||||
mod streamer;
|
mod streamer;
|
||||||
mod tls;
|
mod tls;
|
||||||
@@ -12,9 +13,8 @@ use crate::args::{matches, Args};
|
|||||||
use crate::server::{Request, Server};
|
use crate::server::{Request, Server};
|
||||||
use crate::tls::{TlsAcceptor, TlsStream};
|
use crate::tls::{TlsAcceptor, TlsStream};
|
||||||
|
|
||||||
use std::io::Write;
|
|
||||||
use std::net::{IpAddr, SocketAddr, TcpListener as StdTcpListener};
|
use std::net::{IpAddr, SocketAddr, TcpListener as StdTcpListener};
|
||||||
use std::{env, sync::Arc};
|
use std::sync::Arc;
|
||||||
|
|
||||||
use futures::future::join_all;
|
use futures::future::join_all;
|
||||||
use tokio::net::TcpListener;
|
use tokio::net::TcpListener;
|
||||||
@@ -32,16 +32,7 @@ async fn main() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
async fn run() -> BoxResult<()> {
|
async fn run() -> BoxResult<()> {
|
||||||
if env::var("RUST_LOG").is_err() {
|
logger::init().map_err(|e| format!("Failed to init logger, {}", e))?;
|
||||||
env::set_var("RUST_LOG", "info")
|
|
||||||
}
|
|
||||||
env_logger::builder()
|
|
||||||
.format(|buf, record| {
|
|
||||||
let timestamp = buf.timestamp_millis();
|
|
||||||
writeln!(buf, "[{} {}] {}", timestamp, record.level(), record.args())
|
|
||||||
})
|
|
||||||
.init();
|
|
||||||
|
|
||||||
let args = Args::parse(matches())?;
|
let args = Args::parse(matches())?;
|
||||||
let args = Arc::new(args);
|
let args = Arc::new(args);
|
||||||
let handles = serve(args.clone())?;
|
let handles = serve(args.clone())?;
|
||||||
|
|||||||
@@ -44,11 +44,16 @@ const BUF_SIZE: usize = 65536;
|
|||||||
|
|
||||||
pub struct Server {
|
pub struct Server {
|
||||||
args: Arc<Args>,
|
args: Arc<Args>,
|
||||||
|
assets_prefix: String,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Server {
|
impl Server {
|
||||||
pub fn new(args: Arc<Args>) -> Self {
|
pub fn new(args: Arc<Args>) -> Self {
|
||||||
Self { args }
|
let assets_prefix = format!("{}__dufs_v{}_", args.uri_prefix, env!("CARGO_PKG_VERSION"));
|
||||||
|
Self {
|
||||||
|
args,
|
||||||
|
assets_prefix,
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn call(
|
pub async fn call(
|
||||||
@@ -58,12 +63,15 @@ impl Server {
|
|||||||
) -> Result<Response, hyper::Error> {
|
) -> Result<Response, hyper::Error> {
|
||||||
let method = req.method().clone();
|
let method = req.method().clone();
|
||||||
let uri = req.uri().clone();
|
let uri = req.uri().clone();
|
||||||
|
let assets_prefix = self.assets_prefix.clone();
|
||||||
let enable_cors = self.args.enable_cors;
|
let enable_cors = self.args.enable_cors;
|
||||||
|
|
||||||
let mut res = match self.handle(req).await {
|
let mut res = match self.handle(req).await {
|
||||||
Ok(res) => {
|
Ok(res) => {
|
||||||
let status = res.status().as_u16();
|
let status = res.status().as_u16();
|
||||||
|
if !uri.path().starts_with(&assets_prefix) {
|
||||||
info!(r#"{} "{} {}" - {}"#, addr.ip(), method, uri, status,);
|
info!(r#"{} "{} {}" - {}"#, addr.ip(), method, uri, status,);
|
||||||
|
}
|
||||||
res
|
res
|
||||||
}
|
}
|
||||||
Err(err) => {
|
Err(err) => {
|
||||||
@@ -89,8 +97,7 @@ impl Server {
|
|||||||
let headers = req.headers();
|
let headers = req.headers();
|
||||||
let method = req.method().clone();
|
let method = req.method().clone();
|
||||||
|
|
||||||
if req_path == "/favicon.ico" && method == Method::GET {
|
if method == Method::GET && self.handle_embed_assets(req_path, &mut res).await? {
|
||||||
self.handle_send_favicon(headers, &mut res).await?;
|
|
||||||
return Ok(res);
|
return Ok(res);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -418,23 +425,38 @@ impl Server {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn handle_send_favicon(
|
async fn handle_embed_assets(&self, req_path: &str, res: &mut Response) -> BoxResult<bool> {
|
||||||
&self,
|
if let Some(name) = req_path.strip_prefix(&self.assets_prefix) {
|
||||||
headers: &HeaderMap<HeaderValue>,
|
match name {
|
||||||
res: &mut Response,
|
"index.js" => {
|
||||||
) -> BoxResult<()> {
|
*res.body_mut() = Body::from(INDEX_JS);
|
||||||
let path = self.args.path.join("favicon.ico");
|
res.headers_mut().insert(
|
||||||
let meta = fs::metadata(&path).await.ok();
|
"content-type",
|
||||||
let is_file = meta.map(|v| v.is_file()).unwrap_or_default();
|
HeaderValue::from_static("application/javascript"),
|
||||||
if is_file {
|
);
|
||||||
self.handle_send_file(path.as_path(), headers, false, res)
|
}
|
||||||
.await?;
|
"index.css" => {
|
||||||
} else {
|
*res.body_mut() = Body::from(INDEX_CSS);
|
||||||
|
res.headers_mut()
|
||||||
|
.insert("content-type", HeaderValue::from_static("text/css"));
|
||||||
|
}
|
||||||
|
"favicon.ico" => {
|
||||||
*res.body_mut() = Body::from(FAVICON_ICO);
|
*res.body_mut() = Body::from(FAVICON_ICO);
|
||||||
res.headers_mut()
|
res.headers_mut()
|
||||||
.insert("content-type", HeaderValue::from_static("image/x-icon"));
|
.insert("content-type", HeaderValue::from_static("image/x-icon"));
|
||||||
}
|
}
|
||||||
Ok(())
|
_ => {
|
||||||
|
return Ok(false);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
res.headers_mut().insert(
|
||||||
|
"cache-control",
|
||||||
|
HeaderValue::from_static("max-age=2592000, public"),
|
||||||
|
);
|
||||||
|
Ok(true)
|
||||||
|
} else {
|
||||||
|
Ok(false)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn handle_send_file(
|
async fn handle_send_file(
|
||||||
@@ -692,7 +714,7 @@ impl Server {
|
|||||||
paths.sort_unstable();
|
paths.sort_unstable();
|
||||||
let href = format!("/{}", normalize_path(path.strip_prefix(&self.args.path)?));
|
let href = format!("/{}", normalize_path(path.strip_prefix(&self.args.path)?));
|
||||||
let data = IndexData {
|
let data = IndexData {
|
||||||
href: href.clone(),
|
href,
|
||||||
uri_prefix: self.args.uri_prefix.clone(),
|
uri_prefix: self.args.uri_prefix.clone(),
|
||||||
paths,
|
paths,
|
||||||
allow_upload: self.args.allow_upload,
|
allow_upload: self.args.allow_upload,
|
||||||
@@ -701,18 +723,21 @@ impl Server {
|
|||||||
dir_exists: exist,
|
dir_exists: exist,
|
||||||
};
|
};
|
||||||
let data = serde_json::to_string(&data).unwrap();
|
let data = serde_json::to_string(&data).unwrap();
|
||||||
|
let asset_js = format!("{}index.js", self.assets_prefix);
|
||||||
|
let asset_css = format!("{}index.css", self.assets_prefix);
|
||||||
|
let asset_ico = format!("{}favicon.ico", self.assets_prefix);
|
||||||
let output = INDEX_HTML.replace(
|
let output = INDEX_HTML.replace(
|
||||||
"__SLOT__",
|
"__SLOT__",
|
||||||
&format!(
|
&format!(
|
||||||
r#"
|
r#"
|
||||||
<title>Index of {} - Dufs</title>
|
<link rel="icon" type="image/x-icon" href="{}">
|
||||||
<style>{}</style>
|
<link rel="stylesheet" href="{}">
|
||||||
<script>
|
<script>
|
||||||
const DATA =
|
DATA = {}
|
||||||
{}
|
</script>
|
||||||
{}</script>
|
<script src="{}"></script>
|
||||||
"#,
|
"#,
|
||||||
href, INDEX_CSS, data, INDEX_JS
|
asset_ico, asset_css, data, asset_js
|
||||||
),
|
),
|
||||||
);
|
);
|
||||||
res.headers_mut()
|
res.headers_mut()
|
||||||
|
|||||||
@@ -36,7 +36,6 @@ fn path_prefix_propfind(
|
|||||||
#[case("index.html")]
|
#[case("index.html")]
|
||||||
fn serve_single_file(tmpdir: TempDir, port: u16, #[case] file: &str) -> Result<(), Error> {
|
fn serve_single_file(tmpdir: TempDir, port: u16, #[case] file: &str) -> Result<(), Error> {
|
||||||
let mut child = Command::cargo_bin("dufs")?
|
let mut child = Command::cargo_bin("dufs")?
|
||||||
.env("RUST_LOG", "false")
|
|
||||||
.arg(tmpdir.path().join(file))
|
.arg(tmpdir.path().join(file))
|
||||||
.arg("-p")
|
.arg("-p")
|
||||||
.arg(port.to_string())
|
.arg(port.to_string())
|
||||||
|
|||||||
61
tests/assets.rs
Normal file
61
tests/assets.rs
Normal file
@@ -0,0 +1,61 @@
|
|||||||
|
mod fixtures;
|
||||||
|
mod utils;
|
||||||
|
|
||||||
|
use fixtures::{server, Error, TestServer};
|
||||||
|
use rstest::rstest;
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn assets(server: TestServer) -> Result<(), Error> {
|
||||||
|
let ver = env!("CARGO_PKG_VERSION");
|
||||||
|
let resp = reqwest::blocking::get(server.url())?;
|
||||||
|
let index_js = format!("/__dufs_v{}_index.js", ver);
|
||||||
|
let index_css = format!("/__dufs_v{}_index.css", ver);
|
||||||
|
let favicon_ico = format!("/__dufs_v{}_favicon.ico", ver);
|
||||||
|
let text = resp.text()?;
|
||||||
|
assert!(text.contains(&format!(r#"href="{}""#, index_css)));
|
||||||
|
assert!(text.contains(&format!(r#"href="{}""#, favicon_ico)));
|
||||||
|
assert!(text.contains(&format!(r#"src="{}""#, index_js)));
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn asset_js(server: TestServer) -> Result<(), Error> {
|
||||||
|
let url = format!(
|
||||||
|
"{}__dufs_v{}_index.js",
|
||||||
|
server.url(),
|
||||||
|
env!("CARGO_PKG_VERSION")
|
||||||
|
);
|
||||||
|
let resp = reqwest::blocking::get(url)?;
|
||||||
|
assert_eq!(resp.status(), 200);
|
||||||
|
assert_eq!(
|
||||||
|
resp.headers().get("content-type").unwrap(),
|
||||||
|
"application/javascript"
|
||||||
|
);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn asset_css(server: TestServer) -> Result<(), Error> {
|
||||||
|
let url = format!(
|
||||||
|
"{}__dufs_v{}_index.css",
|
||||||
|
server.url(),
|
||||||
|
env!("CARGO_PKG_VERSION")
|
||||||
|
);
|
||||||
|
let resp = reqwest::blocking::get(url)?;
|
||||||
|
assert_eq!(resp.status(), 200);
|
||||||
|
assert_eq!(resp.headers().get("content-type").unwrap(), "text/css");
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
#[rstest]
|
||||||
|
fn asset_ico(server: TestServer) -> Result<(), Error> {
|
||||||
|
let url = format!(
|
||||||
|
"{}__dufs_v{}_favicon.ico",
|
||||||
|
server.url(),
|
||||||
|
env!("CARGO_PKG_VERSION")
|
||||||
|
);
|
||||||
|
let resp = reqwest::blocking::get(url)?;
|
||||||
|
assert_eq!(resp.status(), 200);
|
||||||
|
assert_eq!(resp.headers().get("content-type").unwrap(), "image/x-icon");
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
@@ -6,14 +6,13 @@ use assert_cmd::prelude::*;
|
|||||||
use assert_fs::fixture::TempDir;
|
use assert_fs::fixture::TempDir;
|
||||||
use regex::Regex;
|
use regex::Regex;
|
||||||
use rstest::rstest;
|
use rstest::rstest;
|
||||||
use std::io::{BufRead, BufReader};
|
use std::io::Read;
|
||||||
use std::process::{Command, Stdio};
|
use std::process::{Command, Stdio};
|
||||||
|
|
||||||
#[rstest]
|
#[rstest]
|
||||||
#[case(&["-b", "20.205.243.166"])]
|
#[case(&["-b", "20.205.243.166"])]
|
||||||
fn bind_fails(tmpdir: TempDir, port: u16, #[case] args: &[&str]) -> Result<(), Error> {
|
fn bind_fails(tmpdir: TempDir, port: u16, #[case] args: &[&str]) -> Result<(), Error> {
|
||||||
Command::cargo_bin("dufs")?
|
Command::cargo_bin("dufs")?
|
||||||
.env("RUST_LOG", "false")
|
|
||||||
.arg(tmpdir.path())
|
.arg(tmpdir.path())
|
||||||
.arg("-p")
|
.arg("-p")
|
||||||
.arg(port.to_string())
|
.arg(port.to_string())
|
||||||
@@ -51,7 +50,6 @@ fn bind_ipv4_ipv6(
|
|||||||
#[case(&["--path-prefix", "/prefix"])]
|
#[case(&["--path-prefix", "/prefix"])]
|
||||||
fn validate_printed_urls(tmpdir: TempDir, port: u16, #[case] args: &[&str]) -> Result<(), Error> {
|
fn validate_printed_urls(tmpdir: TempDir, port: u16, #[case] args: &[&str]) -> Result<(), Error> {
|
||||||
let mut child = Command::cargo_bin("dufs")?
|
let mut child = Command::cargo_bin("dufs")?
|
||||||
.env("RUST_LOG", "false")
|
|
||||||
.arg(tmpdir.path())
|
.arg(tmpdir.path())
|
||||||
.arg("-p")
|
.arg("-p")
|
||||||
.arg(port.to_string())
|
.arg(port.to_string())
|
||||||
@@ -61,22 +59,23 @@ fn validate_printed_urls(tmpdir: TempDir, port: u16, #[case] args: &[&str]) -> R
|
|||||||
|
|
||||||
wait_for_port(port);
|
wait_for_port(port);
|
||||||
|
|
||||||
// WARN assumes urls list is terminated by an empty line
|
let stdout = child.stdout.as_mut().expect("Failed to get stdout");
|
||||||
let url_lines = BufReader::new(child.stdout.take().unwrap())
|
let mut buf = [0; 1000];
|
||||||
|
let buf_len = stdout.read(&mut buf)?;
|
||||||
|
let output = std::str::from_utf8(&buf[0..buf_len])?;
|
||||||
|
let url_lines = output
|
||||||
.lines()
|
.lines()
|
||||||
.map(|line| line.expect("Error reading stdout"))
|
|
||||||
.take_while(|line| !line.is_empty()) /* non-empty lines */
|
.take_while(|line| !line.is_empty()) /* non-empty lines */
|
||||||
.collect::<Vec<_>>();
|
.collect::<Vec<_>>()
|
||||||
let url_lines = url_lines.join("\n");
|
.join("\n");
|
||||||
|
|
||||||
let urls = Regex::new(r"http://[a-zA-Z0-9\.\[\]:/]+")
|
let urls = Regex::new(r"http://[a-zA-Z0-9\.\[\]:/]+")
|
||||||
.unwrap()
|
.unwrap()
|
||||||
.captures_iter(url_lines.as_str())
|
.captures_iter(url_lines.as_str())
|
||||||
.map(|caps| caps.get(0).unwrap().as_str())
|
.filter_map(|caps| caps.get(0).map(|v| v.as_str()))
|
||||||
.collect::<Vec<_>>();
|
.collect::<Vec<_>>();
|
||||||
|
|
||||||
assert!(!urls.is_empty());
|
assert!(!urls.is_empty());
|
||||||
|
|
||||||
for url in urls {
|
for url in urls {
|
||||||
reqwest::blocking::get(url)?.error_for_status()?;
|
reqwest::blocking::get(url)?.error_for_status()?;
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,25 +0,0 @@
|
|||||||
mod fixtures;
|
|
||||||
mod utils;
|
|
||||||
|
|
||||||
use fixtures::{server, Error, TestServer};
|
|
||||||
use rstest::rstest;
|
|
||||||
|
|
||||||
#[rstest]
|
|
||||||
fn default_favicon(server: TestServer) -> Result<(), Error> {
|
|
||||||
let resp = reqwest::blocking::get(format!("{}favicon.ico", server.url()))?;
|
|
||||||
assert_eq!(resp.status(), 200);
|
|
||||||
assert_eq!(resp.headers().get("content-type").unwrap(), "image/x-icon");
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
|
|
||||||
#[rstest]
|
|
||||||
fn exist_favicon(#[with(&["-A"])] server: TestServer) -> Result<(), Error> {
|
|
||||||
let url = format!("{}favicon.ico", server.url());
|
|
||||||
let data = b"abc";
|
|
||||||
let resp = fetch!(b"PUT", &url).body(data.to_vec()).send()?;
|
|
||||||
assert_eq!(resp.status(), 201);
|
|
||||||
let resp = reqwest::blocking::get(url)?;
|
|
||||||
assert_eq!(resp.status(), 200);
|
|
||||||
assert_eq!(resp.bytes()?, data.to_vec());
|
|
||||||
Ok(())
|
|
||||||
}
|
|
||||||
@@ -13,21 +13,7 @@ pub type Error = Box<dyn std::error::Error>;
|
|||||||
|
|
||||||
/// File names for testing purpose
|
/// File names for testing purpose
|
||||||
#[allow(dead_code)]
|
#[allow(dead_code)]
|
||||||
pub static FILES: &[&str] = &[
|
pub static FILES: &[&str] = &["test.txt", "test.html", "index.html", "😀.bin"];
|
||||||
"test.txt",
|
|
||||||
"test.html",
|
|
||||||
"index.html",
|
|
||||||
"test.mkv",
|
|
||||||
#[cfg(not(windows))]
|
|
||||||
"test \" \' & < >.csv",
|
|
||||||
"😀.data",
|
|
||||||
"⎙.mp4",
|
|
||||||
"#[]{}()@!$&'`+,;= %20.test",
|
|
||||||
#[cfg(unix)]
|
|
||||||
":?#[]{}<>()@!$&'`|*+,;= %20.test",
|
|
||||||
#[cfg(not(windows))]
|
|
||||||
"foo\\bar.test",
|
|
||||||
];
|
|
||||||
|
|
||||||
/// Directory names for testing diretory don't exist
|
/// Directory names for testing diretory don't exist
|
||||||
#[allow(dead_code)]
|
#[allow(dead_code)]
|
||||||
@@ -41,10 +27,6 @@ pub static DIR_NO_INDEX: &str = "dir-no-index/";
|
|||||||
#[allow(dead_code)]
|
#[allow(dead_code)]
|
||||||
pub static DIRECTORIES: &[&str] = &["dira/", "dirb/", "dirc/", DIR_NO_INDEX];
|
pub static DIRECTORIES: &[&str] = &["dira/", "dirb/", "dirc/", DIR_NO_INDEX];
|
||||||
|
|
||||||
/// Name of a deeply nested file
|
|
||||||
#[allow(dead_code)]
|
|
||||||
pub static DEEPLY_NESTED_FILE: &str = "very/deeply/nested/test.rs";
|
|
||||||
|
|
||||||
/// Test fixture which creates a temporary directory with a few files and directories inside.
|
/// Test fixture which creates a temporary directory with a few files and directories inside.
|
||||||
/// The directories also contain files.
|
/// The directories also contain files.
|
||||||
#[fixture]
|
#[fixture]
|
||||||
@@ -69,10 +51,6 @@ pub fn tmpdir() -> TempDir {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
tmpdir
|
|
||||||
.child(&DEEPLY_NESTED_FILE)
|
|
||||||
.write_str("File in a deeply nested directory.")
|
|
||||||
.expect("Couldn't write to file");
|
|
||||||
tmpdir
|
tmpdir
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -96,7 +74,6 @@ where
|
|||||||
let tmpdir = tmpdir();
|
let tmpdir = tmpdir();
|
||||||
let child = Command::cargo_bin("dufs")
|
let child = Command::cargo_bin("dufs")
|
||||||
.expect("Couldn't find test binary")
|
.expect("Couldn't find test binary")
|
||||||
.env("RUST_LOG", "false")
|
|
||||||
.arg(tmpdir.path())
|
.arg(tmpdir.path())
|
||||||
.arg("-p")
|
.arg("-p")
|
||||||
.arg(port.to_string())
|
.arg(port.to_string())
|
||||||
@@ -124,7 +101,6 @@ where
|
|||||||
let tmpdir = tmpdir();
|
let tmpdir = tmpdir();
|
||||||
let child = Command::cargo_bin("dufs")
|
let child = Command::cargo_bin("dufs")
|
||||||
.expect("Couldn't find test binary")
|
.expect("Couldn't find test binary")
|
||||||
.env("RUST_LOG", "false")
|
|
||||||
.arg(tmpdir.path())
|
.arg(tmpdir.path())
|
||||||
.arg("-p")
|
.arg("-p")
|
||||||
.arg(port.to_string())
|
.arg(port.to_string())
|
||||||
|
|||||||
@@ -76,12 +76,12 @@ fn get_dir_search(#[with(&["-A"])] server: TestServer) -> Result<(), Error> {
|
|||||||
|
|
||||||
#[rstest]
|
#[rstest]
|
||||||
fn get_dir_search2(#[with(&["-A"])] server: TestServer) -> Result<(), Error> {
|
fn get_dir_search2(#[with(&["-A"])] server: TestServer) -> Result<(), Error> {
|
||||||
let resp = reqwest::blocking::get(format!("{}?q={}", server.url(), "😀.data"))?;
|
let resp = reqwest::blocking::get(format!("{}?q={}", server.url(), "😀.bin"))?;
|
||||||
assert_eq!(resp.status(), 200);
|
assert_eq!(resp.status(), 200);
|
||||||
let paths = utils::retrive_index_paths(&resp.text()?);
|
let paths = utils::retrive_index_paths(&resp.text()?);
|
||||||
assert!(!paths.is_empty());
|
assert!(!paths.is_empty());
|
||||||
for p in paths {
|
for p in paths {
|
||||||
assert!(p.contains(&"😀.data"));
|
assert!(p.contains(&"😀.bin"));
|
||||||
}
|
}
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -37,12 +37,8 @@ pub fn encode_uri(v: &str) -> String {
|
|||||||
|
|
||||||
fn retrive_index_paths_impl(index: &str) -> Option<HashSet<String>> {
|
fn retrive_index_paths_impl(index: &str) -> Option<HashSet<String>> {
|
||||||
let lines: Vec<&str> = index.lines().collect();
|
let lines: Vec<&str> = index.lines().collect();
|
||||||
let (i, _) = lines
|
let line = lines.iter().find(|v| v.contains("DATA ="))?;
|
||||||
.iter()
|
let value: Value = line[7..].parse().ok()?;
|
||||||
.enumerate()
|
|
||||||
.find(|(_, v)| v.contains("const DATA"))?;
|
|
||||||
let line = lines.get(i + 1)?;
|
|
||||||
let value: Value = line.parse().ok()?;
|
|
||||||
let paths = value
|
let paths = value
|
||||||
.get("paths")?
|
.get("paths")?
|
||||||
.as_array()?
|
.as_array()?
|
||||||
|
|||||||
Reference in New Issue
Block a user