mroot/yazi
1
0
Fork 0
mirror of https://github.com/sxyazi/yazi.git synced 2026-07-25 08:41:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

feat: add is_absolute, has_root, starts_with, ends_with, strip_prefix to Url Userdata (#1605)

Browse source
This commit is contained in:
三咲雅 · Misaki Masa 2024-09-04 19:17:48 +08:00 • committed by GitHub
parent ff04e2d21f
commit 322f5fc230
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 27 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
yazi-plugin/src/url/url.rs
View file

@ -13,6 +13,8 @@ impl Url {
reg.add_field_method_get("is_regular", |_, me| Ok(me.is_regular())); reg.add_field_method_get("is_regular", |_, me| Ok(me.is_regular()));
reg.add_field_method_get("is_search", |_, me| Ok(me.is_search())); reg.add_field_method_get("is_search", |_, me| Ok(me.is_search()));
reg.add_field_method_get("is_archive", |_, me| Ok(me.is_archive())); reg.add_field_method_get("is_archive", |_, me| Ok(me.is_archive()));
reg.add_field_method_get("is_absolute", |_, me| Ok(me.is_absolute()));
reg.add_field_method_get("has_root", |_, me| Ok(me.has_root()));
reg.add_method("name", |lua, me, ()| { reg.add_method("name", |lua, me, ()| {
me.file_name().map(|s| lua.create_string(s.as_encoded_bytes())).transpose() me.file_name().map(|s| lua.create_string(s.as_encoded_bytes())).transpose()
@ -24,18 +26,37 @@ impl Url {
me.extension().map(|s| lua.create_string(s.as_encoded_bytes())).transpose() me.extension().map(|s| lua.create_string(s.as_encoded_bytes())).transpose()
}); });
reg.add_method("join", |lua, me, other: Value| { reg.add_method("join", |lua, me, other: Value| {
Ok(match other { Self::cast(lua, match other {
Value::String(s) => Self::cast(lua, me.join(s.to_str()?)), Value::String(s) => me.join(s.to_str()?),
Value::UserData(ud) => { Value::UserData(ud) => me.join(&*ud.borrow::<yazi_shared::fs::Url>()?),
let url = ud.borrow::<yazi_shared::fs::Url>()?;
Self::cast(lua, me.join(&*url))
}
_ => Err("must be a string or a Url".into_lua_err())?, _ => Err("must be a string or a Url".into_lua_err())?,
}) })
}); });
reg.add_method("parent", |lua, me, ()| { reg.add_method("parent", |lua, me, ()| {
me.parent_url().map(|u| Self::cast(lua, u)).transpose() me.parent_url().map(|u| Self::cast(lua, u)).transpose()
}); });
reg.add_method("starts_with", |_, me, base: Value| {
Ok(match base {
Value::String(s) => me.starts_with(s.to_str()?),
Value::UserData(ud) => me.starts_with(&*ud.borrow::<yazi_shared::fs::Url>()?),
_ => Err("must be a string or a Url".into_lua_err())?,
})
});
reg.add_method("ends_with", |_, me, child: Value| {
Ok(match child {
Value::String(s) => me.ends_with(s.to_str()?),
Value::UserData(ud) => me.ends_with(&*ud.borrow::<yazi_shared::fs::Url>()?),
_ => Err("must be a string or a Url".into_lua_err())?,
})
});
reg.add_method("strip_prefix", |lua, me, base: Value| {
let path = match base {
Value::String(s) => me.strip_prefix(s.to_str()?),
Value::UserData(ud) => me.strip_prefix(&*ud.borrow::<yazi_shared::fs::Url>()?),
_ => Err("must be a string or a Url".into_lua_err())?,
};
path.map(|p| Self::cast(lua, yazi_shared::fs::Url::from(p))).transpose()
});
reg.add_meta_method(MetaMethod::Eq, |_, me, other: UrlRef| Ok(me == &*other)); reg.add_meta_method(MetaMethod::Eq, |_, me, other: UrlRef| Ok(me == &*other));
reg.add_meta_method(MetaMethod::ToString, |lua, me, ()| { reg.add_meta_method(MetaMethod::ToString, |lua, me, ()| {