yazi/yazi-shared/src/url/traits.rs

use std::{hash::BuildHasher, path::{Path, PathBuf}};

use hashbrown::{HashMap, hash_map::EntryRef};
use indexmap::{IndexMap, map::RawEntryApiV1};

use crate::{loc::Loc, url::{Url, UrlBuf, UrlBufCov, UrlCov, UrlCow}};

// --- AsUrl
pub trait AsUrl {
	fn as_url(&self) -> Url<'_>;
}

impl AsUrl for Path {
	#[inline]
	fn as_url(&self) -> Url<'_> { Url::Regular(Loc::bare(self)) }
}

impl AsUrl for &Path {
	#[inline]
	fn as_url(&self) -> Url<'_> { (*self).as_url() }
}

impl AsUrl for PathBuf {
	#[inline]
	fn as_url(&self) -> Url<'_> { self.as_path().as_url() }
}

impl AsUrl for &PathBuf {
	#[inline]
	fn as_url(&self) -> Url<'_> { (*self).as_path().as_url() }
}

impl AsUrl for Url<'_> {
	#[inline]
	fn as_url(&self) -> Url<'_> { *self }
}

impl AsUrl for UrlBuf {
	#[inline]
	fn as_url(&self) -> Url<'_> {
		match self {
			Self::Regular(loc) => Url::Regular(loc.as_loc()),
			Self::Search { loc, auth } => Url::Search { loc: loc.as_loc(), auth },
			Self::Mount { loc, auth } => Url::Mount { loc: loc.as_loc(), auth },
			Self::Hub { loc, auth } => Url::Hub { loc: loc.as_loc(), auth },
			Self::Scope { loc, auth } => Url::Scope { loc: loc.as_loc(), auth },
			Self::Sftp { loc, auth } => Url::Sftp { loc: loc.as_loc(), auth },
		}
	}
}

impl AsUrl for &UrlBuf {
	#[inline]
	fn as_url(&self) -> Url<'_> { (**self).as_url() }
}

impl AsUrl for &mut UrlBuf {
	#[inline]
	fn as_url(&self) -> Url<'_> { (**self).as_url() }
}

impl AsUrl for UrlCow<'_> {
	fn as_url(&self) -> Url<'_> {
		match self {
			Self::Regular(loc) => Url::Regular(loc.as_loc()),
			Self::Search { loc, auth } => Url::Search { loc: loc.as_loc(), auth },
			Self::Mount { loc, auth } => Url::Mount { loc: loc.as_loc(), auth },
			Self::Hub { loc, auth } => Url::Hub { loc: loc.as_loc(), auth },
			Self::Scope { loc, auth } => Url::Scope { loc: loc.as_loc(), auth },
			Self::Sftp { loc, auth } => Url::Sftp { loc: loc.as_loc(), auth },
		}
	}
}

impl AsUrl for &UrlCow<'_> {
	fn as_url(&self) -> Url<'_> { (**self).as_url() }
}

impl AsUrl for super::Components<'_> {
	fn as_url(&self) -> Url<'_> { self.url() }
}

impl<'a, T> From<&'a T> for Url<'a>
where
	T: AsUrl + ?Sized,
{
	fn from(value: &'a T) -> Self { value.as_url() }
}

impl<'a, T> From<&'a mut T> for Url<'a>
where
	T: AsUrl + ?Sized,
{
	fn from(value: &'a mut T) -> Self { value.as_url() }
}

// --- UrlMapExt
pub trait UrlMapExt<V> {
	fn get_or_insert_default<U>(&mut self, url: U) -> &mut V
	where
		U: AsUrl,
		V: Default;

	fn get_or_insert_with<U, F>(&mut self, url: U, default: F) -> &mut V
	where
		U: AsUrl,
		F: FnOnce(Url<'_>) -> V;
}

impl<V, S> UrlMapExt<V> for HashMap<UrlBuf, V, S>
where
	S: BuildHasher,
{
	fn get_or_insert_default<U>(&mut self, url: U) -> &mut V
	where
		U: AsUrl,
		V: Default,
	{
		self.get_or_insert_with(url, |_| Default::default())
	}

	fn get_or_insert_with<U, F>(&mut self, url: U, default: F) -> &mut V
	where
		U: AsUrl,
		F: FnOnce(Url<'_>) -> V,
	{
		let url = url.as_url();
		match self.entry_ref(&url) {
			EntryRef::Occupied(oe) => oe.into_mut(),
			EntryRef::Vacant(ve) => ve.insert_with_key(url.into(), default(url)),
		}
	}
}

impl<V, S> UrlMapExt<V> for IndexMap<UrlBuf, V, S>
where
	S: BuildHasher,
{
	fn get_or_insert_default<U>(&mut self, url: U) -> &mut V
	where
		U: AsUrl,
		V: Default,
	{
		self.get_or_insert_with(url, |_| Default::default())
	}

	fn get_or_insert_with<U, F>(&mut self, url: U, default: F) -> &mut V
	where
		U: AsUrl,
		F: FnOnce(Url<'_>) -> V,
	{
		let url = url.as_url();
		self.raw_entry_mut_v1().from_key(&url).or_insert_with(|| (url.into(), default(url))).1
	}
}

// --- UrlCovMapExt
pub trait UrlCovMapExt<V> {
	fn get_or_insert_default(&mut self, url: UrlCov<'_>) -> &mut V
	where
		V: Default;
}

impl<V, S> UrlCovMapExt<V> for HashMap<UrlBufCov, V, S>
where
	S: BuildHasher,
{
	fn get_or_insert_default(&mut self, url: UrlCov<'_>) -> &mut V
	where
		V: Default,
	{
		match self.entry_ref(&url) {
			EntryRef::Occupied(oe) => oe.into_mut(),
			EntryRef::Vacant(ve) => ve.insert_with_key(url.into(), Default::default()),
		}
	}
}