// Copyright 2018-2024 the Deno authors. All rights reserved. MIT license. use super::CACHE_PERM; use crate::util::fs::atomic_write_file; use deno_cache_dir::url_to_filename; use deno_core::url::Host; use deno_core::url::Url; use std::ffi::OsStr; use std::fs; use std::path::Component; use std::path::Path; use std::path::PathBuf; use std::path::Prefix; use std::str; #[derive(Debug, Clone)] pub struct DiskCache { pub location: PathBuf, } impl DiskCache { /// `location` must be an absolute path. pub fn new(location: &Path) -> Self { assert!(location.is_absolute()); Self { location: location.to_owned(), } } fn get_cache_filename(&self, url: &Url) -> Option { let mut out = PathBuf::new(); let scheme = url.scheme(); out.push(scheme); match scheme { "wasm" => { let host = url.host_str().unwrap(); let host_port = match url.port() { // Windows doesn't support ":" in filenames, so we represent port using a // special string. Some(port) => format!("{host}_PORT{port}"), None => host.to_string(), }; out.push(host_port); for path_seg in url.path_segments().unwrap() { out.push(path_seg); } } "http" | "https" | "data" | "blob" => out = url_to_filename(url).ok()?, "file" => { let path = match url.to_file_path() { Ok(path) => path, Err(_) => return None, }; let mut path_components = path.components(); if cfg!(target_os = "windows") { if let Some(Component::Prefix(prefix_component)) = path_components.next() { // Windows doesn't support ":" in filenames, so we need to extract disk prefix // Example: file:///C:/deno/js/unit_test_runner.ts // it should produce: file\c\deno\js\unit_test_runner.ts match prefix_component.kind() { Prefix::Disk(disk_byte) | Prefix::VerbatimDisk(disk_byte) => { let disk = (disk_byte as char).to_string(); out.push(disk); } Prefix::UNC(server, share) | Prefix::VerbatimUNC(server, share) => { out.push("UNC"); let host = Host::parse(server.to_str().unwrap()).unwrap(); let host = host.to_string().replace(':', "_"); out.push(host); out.push(share); } _ => unreachable!(), } } } // Must be relative, so strip forward slash let mut remaining_components = path_components.as_path(); if let Ok(stripped) = remaining_components.strip_prefix("/") { remaining_components = stripped; }; out = out.join(remaining_components); } _ => return None, }; Some(out) } pub fn get_cache_filename_with_extension( &self, url: &Url, extension: &str, ) -> Option { let base = self.get_cache_filename(url)?; match base.extension() { None => Some(base.with_extension(extension)), Some(ext) => { let original_extension = OsStr::to_str(ext).unwrap(); let final_extension = format!("{original_extension}.{extension}"); Some(base.with_extension(final_extension)) } } } pub fn get(&self, filename: &Path) -> std::io::Result> { let path = self.location.join(filename); fs::read(path) } pub fn set(&self, filename: &Path, data: &[u8]) -> std::io::Result<()> { let path = self.location.join(filename); atomic_write_file(&path, data, CACHE_PERM) } } #[cfg(test)] mod tests { use super::*; use test_util::TempDir; #[test] fn test_set_get_cache_file() { let temp_dir = TempDir::new(); let sub_dir = temp_dir.path().join("sub_dir"); let cache = DiskCache::new(&sub_dir.to_path_buf()); let path = PathBuf::from("foo/bar.txt"); cache.set(&path, b"hello").unwrap(); assert_eq!(cache.get(&path).unwrap(), b"hello"); } #[test] fn test_get_cache_filename() { let cache_location = if cfg!(target_os = "windows") { PathBuf::from(r"C:\deno_dir\") } else { PathBuf::from("/deno_dir/") }; let cache = DiskCache::new(&cache_location); let mut test_cases = vec![ ( "http://deno.land/std/http/file_server.ts", "http/deno.land/d8300752800fe3f0beda9505dc1c3b5388beb1ee45afd1f1e2c9fc0866df15cf", ), ( "http://localhost:8000/std/http/file_server.ts", "http/localhost_PORT8000/d8300752800fe3f0beda9505dc1c3b5388beb1ee45afd1f1e2c9fc0866df15cf", ), ( "https://deno.land/std/http/file_server.ts", "https/deno.land/d8300752800fe3f0beda9505dc1c3b5388beb1ee45afd1f1e2c9fc0866df15cf", ), ("wasm://wasm/d1c677ea", "wasm/wasm/d1c677ea"), ]; if cfg!(target_os = "windows") { test_cases.push(("file:///D:/a/1/s/format.ts", "file/D/a/1/s/format.ts")); // IPv4 localhost test_cases.push(( "file://127.0.0.1/d$/a/1/s/format.ts", "file/UNC/127.0.0.1/d$/a/1/s/format.ts", )); // IPv6 localhost test_cases.push(( "file://[0:0:0:0:0:0:0:1]/d$/a/1/s/format.ts", "file/UNC/[__1]/d$/a/1/s/format.ts", )); // shared folder test_cases.push(( "file://comp/t-share/a/1/s/format.ts", "file/UNC/comp/t-share/a/1/s/format.ts", )); } else { test_cases.push(( "file:///std/http/file_server.ts", "file/std/http/file_server.ts", )); } for test_case in &test_cases { let cache_filename = cache.get_cache_filename(&Url::parse(test_case.0).unwrap()); assert_eq!(cache_filename, Some(PathBuf::from(test_case.1))); } } #[test] fn test_get_cache_filename_with_extension() { let p = if cfg!(target_os = "windows") { "C:\\foo" } else { "/foo" }; let cache = DiskCache::new(&PathBuf::from(p)); let mut test_cases = vec![ ( "http://deno.land/std/http/file_server.ts", "js", "http/deno.land/d8300752800fe3f0beda9505dc1c3b5388beb1ee45afd1f1e2c9fc0866df15cf.js", ), ( "http://deno.land/std/http/file_server.ts", "js.map", "http/deno.land/d8300752800fe3f0beda9505dc1c3b5388beb1ee45afd1f1e2c9fc0866df15cf.js.map", ), ]; if cfg!(target_os = "windows") { test_cases.push(( "file:///D:/std/http/file_server", "js", "file/D/std/http/file_server.js", )); } else { test_cases.push(( "file:///std/http/file_server", "js", "file/std/http/file_server.js", )); } for test_case in &test_cases { assert_eq!( cache.get_cache_filename_with_extension( &Url::parse(test_case.0).unwrap(), test_case.1 ), Some(PathBuf::from(test_case.2)) ) } } #[test] fn test_get_cache_filename_invalid_urls() { let cache_location = if cfg!(target_os = "windows") { PathBuf::from(r"C:\deno_dir\") } else { PathBuf::from("/deno_dir/") }; let cache = DiskCache::new(&cache_location); let mut test_cases = vec!["unknown://localhost/test.ts"]; if cfg!(target_os = "windows") { test_cases.push("file://"); test_cases.push("file:///"); } for test_case in &test_cases { let cache_filename = cache.get_cache_filename(&Url::parse(test_case).unwrap()); assert_eq!(cache_filename, None); } } }