greg/mentat
1
0
Fork 0
Code Issues 294 Pull requests Projects Releases 9 Packages Wiki Activity

Pre: Handle SrcVar.

Browse source
This commit is contained in:
Nick Alexander 2017-04-05 15:30:22 -07:00
parent 62fda71fbc
commit 6544ca1594
1 changed files with 33 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

47
query/src/lib.rs
View file

@ -147,7 +147,11 @@ impl FromValue<SrcVar> for SrcVar {
impl SrcVar { impl SrcVar {
pub fn from_symbol(sym: &PlainSymbol) -> Option<SrcVar> { pub fn from_symbol(sym: &PlainSymbol) -> Option<SrcVar> {
if sym.is_src_symbol() { if sym.is_src_symbol() {
Some(SrcVar::NamedSrc(sym.plain_name().to_string())) if sym.0 == "$" {
Some(SrcVar::DefaultSrc)
} else {
Some(SrcVar::NamedSrc(sym.plain_name().to_string()))
}
} else { } else {
None None
} }
@ -185,19 +189,34 @@ pub enum FnArg {
impl FromValue<FnArg> for FnArg { impl FromValue<FnArg> for FnArg {
fn from_value(v: edn::ValueAndSpan) -> Option<FnArg> { fn from_value(v: edn::ValueAndSpan) -> Option<FnArg> {
// TODO: support SrcVars. use edn::SpannedValue::*;
Variable::from_value(v.clone()) // TODO: don't clone! match v.inner {
.and_then(|v| Some(FnArg::Variable(v))) Integer(x) =>
.or_else(|| { Some(FnArg::EntidOrInteger(x)),
println!("from_value {}", v.inner); PlainSymbol(ref x) if x.is_src_symbol() =>
match v.inner { SrcVar::from_symbol(x).map(FnArg::SrcVar),
edn::SpannedValue::Integer(i) => Some(FnArg::EntidOrInteger(i)), PlainSymbol(ref x) if x.is_var_symbol() =>
edn::SpannedValue::Boolean(b) => Some(FnArg::Constant(NonIntegerConstant::Boolean(b))), Variable::from_symbol(x).map(FnArg::Variable),
edn::SpannedValue::BigInteger(x) => Some(FnArg::Constant(NonIntegerConstant::BigInteger(x))), PlainSymbol(_) => None,
edn::SpannedValue::Float(f) => Some(FnArg::Constant(NonIntegerConstant::Float(f))), NamespacedKeyword(ref x) =>
edn::SpannedValue::Text(ref s) => Some(FnArg::Constant(NonIntegerConstant::Text(Rc::new(s.clone())))), Some(FnArg::Ident(x.clone())),
_ => unimplemented!(), Boolean(x) =>
}}) Some(FnArg::Constant(NonIntegerConstant::Boolean(x))),
Float(x) =>
Some(FnArg::Constant(NonIntegerConstant::Float(x))),
BigInteger(ref x) =>
Some(FnArg::Constant(NonIntegerConstant::BigInteger(x.clone()))),
Text(ref x) =>
// TODO: intern strings. #398.
Some(FnArg::Constant(NonIntegerConstant::Text(Rc::new(x.clone())))),
Nil |
NamespacedSymbol(_) |
Keyword(_) |
Vector(_) |
List(_) |
Set(_) |
Map(_) => None,
}
} }
} }