rune/languageserver/
url.rs

Help
1// Implementation copied and adjusted from https://github.com/servo/rust-url
2
3use std::path::Path;
4
5use anyhow::anyhow;
6
7use crate::alloc::fmt::TryWrite;
8use crate::alloc::prelude::*;
9use crate::alloc::String;
10use crate::support::Result;
11
12use percent_encoding::{percent_encode, AsciiSet, CONTROLS};
13use url::Url;
14
15/// https://url.spec.whatwg.org/#fragment-percent-encode-set
16const FRAGMENT: &AsciiSet = &CONTROLS.add(b' ').add(b'"').add(b'<').add(b'>').add(b'`');
17/// https://url.spec.whatwg.org/#path-percent-encode-set
18const PATH: &AsciiSet = &FRAGMENT.add(b'#').add(b'?').add(b'{').add(b'}');
19const PATH_SEGMENT: &AsciiSet = &PATH.add(b'/').add(b'%');
20
21/// Convert a file path into a URL.
22pub(super) fn from_file_path<P>(path: P) -> Result<Url>
23where
24    P: AsRef<Path>,
25{
26    let mut buf = "file://".try_to_owned()?;
27    path_to_file_url_segments(path.as_ref(), &mut buf)?;
28    Ok(Url::parse(&buf)?)
29}
30
31#[cfg(any(unix, target_os = "redox", target_os = "wasi"))]
32fn path_to_file_url_segments(path: &Path, buf: &mut String) -> Result<()> {
33    #[cfg(any(unix, target_os = "redox"))]
34    use std::os::unix::prelude::OsStrExt;
35    #[cfg(target_os = "wasi")]
36    use std::os::wasi::prelude::OsStrExt;
37
38    if !path.is_absolute() {
39        return Err(anyhow!("Path must be absolute"));
40    }
41
42    let mut empty = true;
43
44    // skip the root component
45    for component in path.components().skip(1) {
46        empty = false;
47        buf.try_push('/')?;
48        buf.try_extend(percent_encode(
49            component.as_os_str().as_bytes(),
50            PATH_SEGMENT,
51        ))?;
52    }
53
54    if empty {
55        // An URL’s path must not be empty.
56        buf.try_push('/')?;
57    }
58
59    Ok(())
60}
61
62#[cfg(windows)]
63fn path_to_file_url_segments(path: &Path, buf: &mut String) -> Result<()> {
64    path_to_file_url_segments_windows(path, buf)
65}
66
67// Build this unconditionally to alleviate https://github.com/servo/rust-url/issues/102
68#[cfg_attr(not(windows), allow(dead_code))]
69fn path_to_file_url_segments_windows(path: &Path, buf: &mut String) -> Result<()> {
70    use std::path::{Component, Prefix};
71
72    if !path.is_absolute() {
73        return Err(anyhow!("Path must be absolute"));
74    }
75
76    let mut components = path.components();
77
78    match components.next() {
79        Some(Component::Prefix(ref p)) => match p.kind() {
80            Prefix::Disk(letter) | Prefix::VerbatimDisk(letter) => {
81                buf.try_push('/')?;
82                buf.try_push((letter as char).to_ascii_lowercase())?;
83                buf.try_push_str("%3A")?;
84            }
85            Prefix::UNC(server, share) | Prefix::VerbatimUNC(server, share) => {
86                let Some(server) = server.to_str() else {
87                    return Err(anyhow!("UNC server is not valid UTF-8"));
88                };
89
90                let host = url::Host::parse(server)?;
91                write!(buf, "{}", host)?;
92                buf.try_push('/')?;
93
94                let Some(share) = share.to_str() else {
95                    return Err(anyhow!("UNC share is not valid UTF-8"));
96                };
97
98                buf.try_extend(percent_encode(share.as_bytes(), PATH_SEGMENT))?;
99            }
100            _ => return Err(anyhow!("Illegal path component")),
101        },
102        _ => return Err(anyhow!("Illegal path component")),
103    }
104
105    let mut only_prefix = true;
106
107    for component in components {
108        if component == Component::RootDir {
109            continue;
110        }
111
112        only_prefix = false;
113
114        let Some(component) = component.as_os_str().to_str() else {
115            return Err(anyhow!("Path component is not valid UTF-8"));
116        };
117
118        buf.try_push('/')?;
119        buf.try_extend(percent_encode(component.as_bytes(), PATH_SEGMENT))?;
120    }
121
122    if only_prefix {
123        buf.try_push('/')?;
124    }
125
126    Ok(())
127}
rune/languageserver/url.rs

rune/languageserver/
url.rs
