Adds a default value to `.remove()` on `dict` and `array` (#2346)

3 files changed, 27 insertions, 9 deletions

diff --git a/crates/typst/src/eval/array.rs b/crates/typst/src/eval/array.rs
index 92642702..bee3fae7 100644
--- a/crates/typst/src/eval/array.rs
+++ b/crates/typst/src/eval/array.rs
@@ -247,9 +247,14 @@ impl Array {
         /// The index at which to remove the item. If negative, indexes from
         /// the back.
         index: i64,
+        /// A default value to return if the index is out of bounds.
+        #[named]
+        default: Option<Value>,
     ) -> StrResult<Value> {
-        let i = self.locate(index, false)?;
-        Ok(self.0.remove(i))
+        self.locate_opt(index, false)
+            .map(|i| self.0.remove(i))
+            .or(default)
+            .ok_or_else(|| out_of_bounds_no_default(index, self.len()))
     }
 
     /// Extracts a subslice of the array. Fails with an error if the start or
diff --git a/crates/typst/src/eval/dict.rs b/crates/typst/src/eval/dict.rs
index c5da7c15..b77686fe 100644
--- a/crates/typst/src/eval/dict.rs
+++ b/crates/typst/src/eval/dict.rs
@@ -169,11 +169,17 @@ impl Dict {
 
     /// Removes a pair from the dictionary by key and return the value.
     #[func]
-    pub fn remove(&mut self, key: Str) -> StrResult<Value> {
-        match Arc::make_mut(&mut self.0).shift_remove(&key) {
-            Some(value) => Ok(value),
-            None => Err(missing_key(&key)),
-        }
+    pub fn remove(
+        &mut self,
+        key: Str,
+        /// A default value to return if the key does not exist.
+        #[named]
+        default: Option<Value>,
+    ) -> StrResult<Value> {
+        Arc::make_mut(&mut self.0)
+            .shift_remove(&key)
+            .or(default)
+            .ok_or_else(|| missing_key(&key))
     }
 
     /// Returns the keys of the dictionary as an array in insertion order.
diff --git a/crates/typst/src/eval/methods.rs b/crates/typst/src/eval/methods.rs
index b8d71c76..843549e6 100644
--- a/crates/typst/src/eval/methods.rs
+++ b/crates/typst/src/eval/methods.rs
@@ -51,13 +51,20 @@ pub fn call_mut(
             "insert" => {
                 array.insert(args.expect("index")?, args.expect("value")?).at(span)?
             }
-            "remove" => output = array.remove(args.expect("index")?).at(span)?,
+            "remove" => {
+                output = array
+                    .remove(args.expect("index")?, args.named("default")?)
+                    .at(span)?
+            }
             _ => return missing(),
         },
 
         Value::Dict(dict) => match method {
             "insert" => dict.insert(args.expect::<Str>("key")?, args.expect("value")?),
-            "remove" => output = dict.remove(args.expect::<Str>("key")?).at(span)?,
+            "remove" => {
+                output =
+                    dict.remove(args.expect("key")?, args.named("default")?).at(span)?
+            }
             _ => return missing(),
         },