simplify declaraction of map fns
This commit is contained in:
ConnorSkees
parent
2eef3e9f6a
commit
2d094490ba
@ -1,141 +1,155 @@
|
|||||||
use super::GlobalFunctionMap;
|
use super::GlobalFunctionMap;
|
||||||
|
|
||||||
use super::Builtin;
|
use super::Builtin;
|
||||||
|
use crate::args::CallArgs;
|
||||||
use crate::common::{Brackets, ListSeparator};
|
use crate::common::{Brackets, ListSeparator};
|
||||||
|
use crate::error::SassResult;
|
||||||
|
use crate::scope::Scope;
|
||||||
|
use crate::selector::Selector;
|
||||||
use crate::value::{SassMap, Value};
|
use crate::value::{SassMap, Value};
|
||||||
|
|
||||||
pub(crate) fn register(f: &mut GlobalFunctionMap) {
|
pub(crate) fn register(f: &mut GlobalFunctionMap) {
|
||||||
f.insert(
|
fn map_get(mut args: CallArgs, scope: &Scope, super_selector: &Selector) -> SassResult<Value> {
|
||||||
"map-get",
|
args.max_args(2)?;
|
||||||
Builtin::new(|mut args, scope, super_selector| {
|
let key = arg!(args, scope, super_selector, 1, "key");
|
||||||
args.max_args(2)?;
|
let map = match arg!(args, scope, super_selector, 0, "map") {
|
||||||
let key = arg!(args, scope, super_selector, 1, "key");
|
Value::Map(m) => m,
|
||||||
let map = match arg!(args, scope, super_selector, 0, "map") {
|
Value::List(v, ..) if v.is_empty() => SassMap::new(),
|
||||||
Value::Map(m) => m,
|
v => {
|
||||||
Value::List(v, ..) if v.is_empty() => SassMap::new(),
|
return Err((
|
||||||
v => {
|
format!("$map: {} is not a map.", v.to_css_string(args.span())?),
|
||||||
return Err((
|
args.span(),
|
||||||
format!("$map: {} is not a map.", v.to_css_string(args.span())?),
|
)
|
||||||
args.span(),
|
.into())
|
||||||
)
|
|
||||||
.into())
|
|
||||||
}
|
|
||||||
};
|
|
||||||
Ok(map.get(&key, args.span())?.unwrap_or(Value::Null))
|
|
||||||
}),
|
|
||||||
);
|
|
||||||
f.insert(
|
|
||||||
"map-has-key",
|
|
||||||
Builtin::new(|mut args, scope, super_selector| {
|
|
||||||
args.max_args(2)?;
|
|
||||||
let key = arg!(args, scope, super_selector, 1, "key");
|
|
||||||
let map = match arg!(args, scope, super_selector, 0, "map") {
|
|
||||||
Value::Map(m) => m,
|
|
||||||
Value::List(v, ..) if v.is_empty() => SassMap::new(),
|
|
||||||
v => {
|
|
||||||
return Err((
|
|
||||||
format!("$map: {} is not a map.", v.to_css_string(args.span())?),
|
|
||||||
args.span(),
|
|
||||||
)
|
|
||||||
.into())
|
|
||||||
}
|
|
||||||
};
|
|
||||||
Ok(Value::bool(map.get(&key, args.span())?.is_some()))
|
|
||||||
}),
|
|
||||||
);
|
|
||||||
f.insert(
|
|
||||||
"map-keys",
|
|
||||||
Builtin::new(|mut args, scope, super_selector| {
|
|
||||||
args.max_args(1)?;
|
|
||||||
let map = match arg!(args, scope, super_selector, 0, "map") {
|
|
||||||
Value::Map(m) => m,
|
|
||||||
Value::List(v, ..) if v.is_empty() => SassMap::new(),
|
|
||||||
v => {
|
|
||||||
return Err((
|
|
||||||
format!("$map: {} is not a map.", v.to_css_string(args.span())?),
|
|
||||||
args.span(),
|
|
||||||
)
|
|
||||||
.into())
|
|
||||||
}
|
|
||||||
};
|
|
||||||
Ok(Value::List(
|
|
||||||
map.keys(),
|
|
||||||
ListSeparator::Comma,
|
|
||||||
Brackets::None,
|
|
||||||
))
|
|
||||||
}),
|
|
||||||
);
|
|
||||||
f.insert(
|
|
||||||
"map-values",
|
|
||||||
Builtin::new(|mut args, scope, super_selector| {
|
|
||||||
args.max_args(1)?;
|
|
||||||
let map = match arg!(args, scope, super_selector, 0, "map") {
|
|
||||||
Value::Map(m) => m,
|
|
||||||
Value::List(v, ..) if v.is_empty() => SassMap::new(),
|
|
||||||
v => {
|
|
||||||
return Err((
|
|
||||||
format!("$map: {} is not a map.", v.to_css_string(args.span())?),
|
|
||||||
args.span(),
|
|
||||||
)
|
|
||||||
.into())
|
|
||||||
}
|
|
||||||
};
|
|
||||||
Ok(Value::List(
|
|
||||||
map.values(),
|
|
||||||
ListSeparator::Comma,
|
|
||||||
Brackets::None,
|
|
||||||
))
|
|
||||||
}),
|
|
||||||
);
|
|
||||||
f.insert(
|
|
||||||
"map-merge",
|
|
||||||
Builtin::new(|mut args, scope, super_selector| {
|
|
||||||
args.max_args(2)?;
|
|
||||||
let mut map1 = match arg!(args, scope, super_selector, 0, "map1") {
|
|
||||||
Value::Map(m) => m,
|
|
||||||
Value::List(v, ..) if v.is_empty() => SassMap::new(),
|
|
||||||
v => {
|
|
||||||
return Err((
|
|
||||||
format!("$map1: {} is not a map.", v.to_css_string(args.span())?),
|
|
||||||
args.span(),
|
|
||||||
)
|
|
||||||
.into())
|
|
||||||
}
|
|
||||||
};
|
|
||||||
let map2 = match arg!(args, scope, super_selector, 1, "map2") {
|
|
||||||
Value::Map(m) => m,
|
|
||||||
Value::List(v, ..) if v.is_empty() => SassMap::new(),
|
|
||||||
v => {
|
|
||||||
return Err((
|
|
||||||
format!("$map2: {} is not a map.", v.to_css_string(args.span())?),
|
|
||||||
args.span(),
|
|
||||||
)
|
|
||||||
.into())
|
|
||||||
}
|
|
||||||
};
|
|
||||||
map1.merge(map2);
|
|
||||||
Ok(Value::Map(map1))
|
|
||||||
}),
|
|
||||||
);
|
|
||||||
f.insert(
|
|
||||||
"map-remove",
|
|
||||||
Builtin::new(|mut args, scope, super_selector| {
|
|
||||||
let mut map = match arg!(args, scope, super_selector, 0, "map") {
|
|
||||||
Value::Map(m) => m,
|
|
||||||
Value::List(v, ..) if v.is_empty() => SassMap::new(),
|
|
||||||
v => {
|
|
||||||
return Err((
|
|
||||||
format!("$map: {} is not a map.", v.to_css_string(args.span())?),
|
|
||||||
args.span(),
|
|
||||||
)
|
|
||||||
.into())
|
|
||||||
}
|
|
||||||
};
|
|
||||||
let keys = args.get_variadic(scope, super_selector)?;
|
|
||||||
for key in keys {
|
|
||||||
map.remove(&key);
|
|
||||||
}
|
}
|
||||||
Ok(Value::Map(map))
|
};
|
||||||
}),
|
Ok(map.get(&key, args.span())?.unwrap_or(Value::Null))
|
||||||
);
|
}
|
||||||
|
|
||||||
|
fn map_has_key(
|
||||||
|
mut args: CallArgs,
|
||||||
|
scope: &Scope,
|
||||||
|
super_selector: &Selector,
|
||||||
|
) -> SassResult<Value> {
|
||||||
|
args.max_args(2)?;
|
||||||
|
let key = arg!(args, scope, super_selector, 1, "key");
|
||||||
|
let map = match arg!(args, scope, super_selector, 0, "map") {
|
||||||
|
Value::Map(m) => m,
|
||||||
|
Value::List(v, ..) if v.is_empty() => SassMap::new(),
|
||||||
|
v => {
|
||||||
|
return Err((
|
||||||
|
format!("$map: {} is not a map.", v.to_css_string(args.span())?),
|
||||||
|
args.span(),
|
||||||
|
)
|
||||||
|
.into())
|
||||||
|
}
|
||||||
|
};
|
||||||
|
Ok(Value::bool(map.get(&key, args.span())?.is_some()))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn map_keys(mut args: CallArgs, scope: &Scope, super_selector: &Selector) -> SassResult<Value> {
|
||||||
|
args.max_args(1)?;
|
||||||
|
let map = match arg!(args, scope, super_selector, 0, "map") {
|
||||||
|
Value::Map(m) => m,
|
||||||
|
Value::List(v, ..) if v.is_empty() => SassMap::new(),
|
||||||
|
v => {
|
||||||
|
return Err((
|
||||||
|
format!("$map: {} is not a map.", v.to_css_string(args.span())?),
|
||||||
|
args.span(),
|
||||||
|
)
|
||||||
|
.into())
|
||||||
|
}
|
||||||
|
};
|
||||||
|
Ok(Value::List(
|
||||||
|
map.keys(),
|
||||||
|
ListSeparator::Comma,
|
||||||
|
Brackets::None,
|
||||||
|
))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn map_values(
|
||||||
|
mut args: CallArgs,
|
||||||
|
scope: &Scope,
|
||||||
|
super_selector: &Selector,
|
||||||
|
) -> SassResult<Value> {
|
||||||
|
args.max_args(1)?;
|
||||||
|
let map = match arg!(args, scope, super_selector, 0, "map") {
|
||||||
|
Value::Map(m) => m,
|
||||||
|
Value::List(v, ..) if v.is_empty() => SassMap::new(),
|
||||||
|
v => {
|
||||||
|
return Err((
|
||||||
|
format!("$map: {} is not a map.", v.to_css_string(args.span())?),
|
||||||
|
args.span(),
|
||||||
|
)
|
||||||
|
.into())
|
||||||
|
}
|
||||||
|
};
|
||||||
|
Ok(Value::List(
|
||||||
|
map.values(),
|
||||||
|
ListSeparator::Comma,
|
||||||
|
Brackets::None,
|
||||||
|
))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn map_merge(
|
||||||
|
mut args: CallArgs,
|
||||||
|
scope: &Scope,
|
||||||
|
super_selector: &Selector,
|
||||||
|
) -> SassResult<Value> {
|
||||||
|
args.max_args(2)?;
|
||||||
|
let mut map1 = match arg!(args, scope, super_selector, 0, "map1") {
|
||||||
|
Value::Map(m) => m,
|
||||||
|
Value::List(v, ..) if v.is_empty() => SassMap::new(),
|
||||||
|
v => {
|
||||||
|
return Err((
|
||||||
|
format!("$map1: {} is not a map.", v.to_css_string(args.span())?),
|
||||||
|
args.span(),
|
||||||
|
)
|
||||||
|
.into())
|
||||||
|
}
|
||||||
|
};
|
||||||
|
let map2 = match arg!(args, scope, super_selector, 1, "map2") {
|
||||||
|
Value::Map(m) => m,
|
||||||
|
Value::List(v, ..) if v.is_empty() => SassMap::new(),
|
||||||
|
v => {
|
||||||
|
return Err((
|
||||||
|
format!("$map2: {} is not a map.", v.to_css_string(args.span())?),
|
||||||
|
args.span(),
|
||||||
|
)
|
||||||
|
.into())
|
||||||
|
}
|
||||||
|
};
|
||||||
|
map1.merge(map2);
|
||||||
|
Ok(Value::Map(map1))
|
||||||
|
}
|
||||||
|
|
||||||
|
fn map_remove(
|
||||||
|
mut args: CallArgs,
|
||||||
|
scope: &Scope,
|
||||||
|
super_selector: &Selector,
|
||||||
|
) -> SassResult<Value> {
|
||||||
|
let mut map = match arg!(args, scope, super_selector, 0, "map") {
|
||||||
|
Value::Map(m) => m,
|
||||||
|
Value::List(v, ..) if v.is_empty() => SassMap::new(),
|
||||||
|
v => {
|
||||||
|
return Err((
|
||||||
|
format!("$map: {} is not a map.", v.to_css_string(args.span())?),
|
||||||
|
args.span(),
|
||||||
|
)
|
||||||
|
.into())
|
||||||
|
}
|
||||||
|
};
|
||||||
|
let keys = args.get_variadic(scope, super_selector)?;
|
||||||
|
for key in keys {
|
||||||
|
map.remove(&key);
|
||||||
|
}
|
||||||
|
Ok(Value::Map(map))
|
||||||
|
}
|
||||||
|
|
||||||
|
f.insert("map-get", Builtin::new(map_get));
|
||||||
|
f.insert("map-has-key", Builtin::new(map_has_key));
|
||||||
|
f.insert("map-keys", Builtin::new(map_keys));
|
||||||
|
f.insert("map-values", Builtin::new(map_values));
|
||||||
|
f.insert("map-merge", Builtin::new(map_merge));
|
||||||
|
f.insert("map-remove", Builtin::new(map_remove));
|
||||||
}
|
}
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user