sjs/agate
1
0
Fork 0
mirror of https://github.com/samsonjs/agate.git synced 2026-04-04 10:45:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Simplify percent-encoding of paths.

Browse source
This commit is contained in:
Matt Brubeck 2020-12-31 14:54:26 -08:00
parent fe62be214e
commit 879422c2cc
1 changed files with 17 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
src/main.rs
View file

@ -11,7 +11,16 @@ use rustls::{
internal::pemfile::{certs, pkcs8_private_keys},
NoClientAuth, ServerConfig,
};
use std::{error::Error, ffi::OsStr, fs::File, io::BufReader, marker::Unpin, path::Path, sync::Arc};
use std::{
borrow::Cow,
error::Error,
ffi::OsStr,
fs::File,
io::BufReader,
marker::Unpin,
path::Path,
sync::Arc,
};
use url::{Host, Url};
fn main() -> Result {
@ -229,14 +238,10 @@ async fn send_header<W: Write + Unpin>(stream: &mut W, status: u8, meta: &[&str]
}
async fn list_directory<W: Write + Unpin>(stream: &mut W, path: &Path) -> Result {
// https://url.spec.whatwg.org/#path-percent-encode-set
const ENCODE_SET: AsciiSet = CONTROLS.add(b' ')
// https://url.spec.whatwg.org/#path-percent-encode-set
.add(b'"').add(b'#').add(b'<').add(b'>')
.add(b'?').add(b'`').add(b'{').add(b'}')
// https://tools.ietf.org/html/rfc3986#section-2.2
.add(b':').add(b'/').add(b'?').add(b'#').add(b'[').add(b']').add(b'@')
.add(b'!').add(b'$').add(b'&').add(b'\'').add(b'(').add(b')')
.add(b'*').add(b'+').add(b',').add(b';').add(b'=');
.add(b'?').add(b'`').add(b'{').add(b'}');
log::info!("Listing directory {:?}", path);
send_text_gemini_header(stream).await?;
@ -251,12 +256,11 @@ async fn list_directory<W: Write + Unpin>(stream: &mut W, path: &Path) -> Result
if entry.file_type().await?.is_dir() {
name += "/";
}
if name.contains(char::is_whitespace) {
let url = percent_encode(name.as_bytes(), &ENCODE_SET);
lines.push(format!("=> {} {}\n", url, name));
} else {
lines.push(format!("=> {}\n", name));
}
let line = match percent_encode(name.as_bytes(), &ENCODE_SET).into() {
Cow::Owned(url) => format!("=> {} {}\n", url, name),
Cow::Borrowed(url) => format!("=> {}\n", url), // url and name are identical
};
lines.push(line);
}
lines.sort();
for line in lines {