From ea7f61dc1b235357249d5eb10bc94d147225a5b5 Mon Sep 17 00:00:00 2001
From: Tpt <thomaspt@hotmail.fr>
Date: Mon, 8 Oct 2018 21:42:30 +0200
Subject: [PATCH] Adds some cast evaluation

---
 lib/src/sparql/eval.rs | 43 +++++++++++++++++++++++-------------------
 1 file changed, 24 insertions(+), 19 deletions(-)

diff --git a/lib/src/sparql/eval.rs b/lib/src/sparql/eval.rs
index e243912e..a59daf96 100644
--- a/lib/src/sparql/eval.rs
+++ b/lib/src/sparql/eval.rs
@@ -212,25 +212,9 @@ impl<S: EncodedQuadsStore> SimpleEvaluator<S> {
             PlanExpression::UnaryNot(e) => self
                 .to_bool(self.eval_expression(e, tuple)?)
                 .map(|v| (!v).into()),
-            PlanExpression::Str(e) => match self.eval_expression(e, tuple)? {
-                EncodedTerm::NamedNode { iri_id } => {
-                    Some(EncodedTerm::SimpleLiteral { value_id: iri_id })
-                }
-                EncodedTerm::SimpleLiteral { value_id } => {
-                    Some(EncodedTerm::SimpleLiteral { value_id })
-                }
-                EncodedTerm::LangStringLiteral { value_id, .. } => {
-                    Some(EncodedTerm::SimpleLiteral { value_id })
-                }
-                EncodedTerm::TypedLiteral { value_id, .. } => {
-                    Some(EncodedTerm::SimpleLiteral { value_id })
-                }
-                EncodedTerm::StringLiteral { value_id } => {
-                    Some(EncodedTerm::SimpleLiteral { value_id })
-                }
-                //TODO EncodedTerm::BooleanLiteral(v),
-                _ => None,
-            },
+            PlanExpression::Str(e) => Some(EncodedTerm::SimpleLiteral {
+                value_id: self.to_string_id(self.eval_expression(e, tuple)?)?,
+            }),
             PlanExpression::Lang(e) => match self.eval_expression(e, tuple)? {
                 EncodedTerm::LangStringLiteral { language_id, .. } => {
                     Some(EncodedTerm::SimpleLiteral {
@@ -272,6 +256,12 @@ impl<S: EncodedQuadsStore> SimpleEvaluator<S> {
             PlanExpression::IsLiteral(e) => {
                 Some(self.eval_expression(e, tuple)?.is_literal().into())
             }
+            PlanExpression::BooleanCast(e) => {
+                Some(self.to_bool(self.eval_expression(e, tuple)?)?.into())
+            }
+            PlanExpression::StringCast(e) => Some(EncodedTerm::StringLiteral {
+                value_id: self.to_string_id(self.eval_expression(e, tuple)?)?,
+            }),
             e => unimplemented!(),
         }
     }
@@ -285,6 +275,21 @@ impl<S: EncodedQuadsStore> SimpleEvaluator<S> {
         }
     }
 
+    fn to_string_id(&self, term: EncodedTerm) -> Option<u64> {
+        match term {
+            EncodedTerm::NamedNode { iri_id } => Some(iri_id),
+            EncodedTerm::SimpleLiteral { value_id } => Some(value_id),
+            EncodedTerm::LangStringLiteral { value_id, .. } => Some(value_id),
+            EncodedTerm::TypedLiteral { value_id, .. } => Some(value_id),
+            EncodedTerm::StringLiteral { value_id } => Some(value_id),
+            EncodedTerm::BooleanLiteral(value) => self
+                .store
+                .insert_bytes(if value { b"true" } else { b"false" })
+                .ok(),
+            _ => None,
+        }
+    }
+
     fn decode_bindings(
         &self,
         iter: EncodedTuplesIterator,