Merge pull request #181 from basile-henry/text-replace

Implement Text/replace

2 files changed, 51 insertions, 2 deletions

diff --git a/dhall/src/builtins.rs b/dhall/src/builtins.rs
index 16d656f..e80bf6b 100644
--- a/dhall/src/builtins.rs
+++ b/dhall/src/builtins.rs
@@ -3,7 +3,8 @@ use std::convert::TryInto;
 
 use crate::operations::{BinOp, OpKind};
 use crate::semantics::{
-    skip_resolve_expr, typecheck, Hir, HirKind, Nir, NirKind, NzEnv, VarEnv,
+    nze, skip_resolve_expr, typecheck, Hir, HirKind, Nir, NirKind, NzEnv,
+    VarEnv,
 };
 use crate::syntax::Const::Type;
 use crate::syntax::{
@@ -43,6 +44,7 @@ pub enum Builtin {
     ListIndexed,
     ListReverse,
     TextShow,
+    TextReplace,
 }
 
 impl Builtin {
@@ -78,6 +80,7 @@ impl Builtin {
             "List/indexed" => Some(ListIndexed),
             "List/reverse" => Some(ListReverse),
             "Text/show" => Some(TextShow),
+            "Text/replace" => Some(TextReplace),
             _ => None,
         }
     }
@@ -211,7 +214,12 @@ pub fn type_of_builtin(b: Builtin) -> Hir {
 
         DoubleShow => make_type!(Double -> Text),
         TextShow => make_type!(Text -> Text),
-
+        TextReplace => make_type!(
+            forall (needle: Text) ->
+            forall (replacement: Text) ->
+            forall (haystack: Text) ->
+            Text
+        ),
         ListBuild => make_type!(
             forall (a: Type) ->
             (forall (list: Type) ->
@@ -401,6 +409,44 @@ fn apply_builtin(b: Builtin, args: Vec<Nir>, env: NzEnv) -> NirKind {
             }
             _ => Ret::DoneAsIs,
         },
+        (Builtin::TextReplace, [needle, replacement, haystack]) => {
+            // Helper to match a Nir as a text literal
+            fn nir_to_string(n: &Nir) -> Option<String> {
+                match &*n.kind() {
+                    TextLit(n_lit) => n_lit.as_text(),
+                    _ => None,
+                }
+            }
+
+            // The needle and the haystack need to be fully
+            // evaluated as Text otherwise no progress can be made
+            match (nir_to_string(needle), nir_to_string(haystack)) {
+                (Some(n), Some(h)) => {
+                    // When the needle is empty the haystack is returned untouched
+                    if n.is_empty() {
+                        Ret::Nir(haystack.clone())
+                    // Fast case when replacement is fully evaluated
+                    } else if let Some(r) = nir_to_string(replacement) {
+                        Ret::Nir(Nir::from_text(h.replace(&n, &r)))
+                    } else {
+                        use itertools::Itertools;
+
+                        let parts = h.split(&n).map(|s| {
+                            InterpolatedTextContents::Text(s.to_string())
+                        });
+                        let replacement =
+                            InterpolatedTextContents::Expr(replacement.clone());
+
+                        Ret::Nir(Nir::from_kind(NirKind::TextLit(
+                            nze::nir::TextLit::new(
+                                parts.intersperse(replacement),
+                            ),
+                        )))
+                    }
+                }
+                _ => Ret::DoneAsIs,
+            }
+        }
         (Builtin::ListLength, [_, l]) => match &*l.kind() {
             EmptyListLit(_) => Ret::NirKind(Num(Natural(0))),
             NEListLit(xs) => Ret::NirKind(Num(Natural(xs.len()))),
@@ -560,6 +606,7 @@ impl std::fmt::Display for Builtin {
             ListIndexed => "List/indexed",
             ListReverse => "List/reverse",
             TextShow => "Text/show",
+            TextReplace => "Text/replace",
         })
     }
 }
diff --git a/dhall/src/syntax/text/dhall.abnf b/dhall/src/syntax/text/dhall.abnf
index 2d487f5..e913341 100644
--- a/dhall/src/syntax/text/dhall.abnf
+++ b/dhall/src/syntax/text/dhall.abnf
@@ -412,6 +412,7 @@ builtin =
     / List-indexed

     / List-reverse

     / Text-show

+    / Text-replace

     / Bool

     / True

     / False

@@ -464,6 +465,7 @@ List-last         = %x4c.69.73.74.2f.6c.61.73.74
 List-indexed      = %x4c.69.73.74.2f.69.6e.64.65.78.65.64

 List-reverse      = %x4c.69.73.74.2f.72.65.76.65.72.73.65

 Text-show         = %x54.65.78.74.2f.73.68.6f.77

+Text-replace      = %x54.65.78.74.2f.72.65.70.6c.61.63.65

 

 ; Operators

 combine       = %x2227 / "/\"