86 lines
1.9 KiB
Rust
86 lines
1.9 KiB
Rust
use crate::sql::table::{table, Table};
|
|
use nom::branch::alt;
|
|
use nom::bytes::complete::tag;
|
|
use nom::combinator::map;
|
|
use nom::multi::many1;
|
|
use nom::IResult;
|
|
use serde::{Deserialize, Serialize};
|
|
use std::fmt;
|
|
|
|
#[derive(Clone, Debug, Eq, PartialEq, Serialize, Deserialize)]
|
|
pub enum Kind {
|
|
Any,
|
|
Array,
|
|
Bool,
|
|
Datetime,
|
|
Number,
|
|
Object,
|
|
String,
|
|
Geometry(String),
|
|
Record(Vec<Table>),
|
|
}
|
|
|
|
impl Default for Kind {
|
|
fn default() -> Kind {
|
|
Kind::Any
|
|
}
|
|
}
|
|
|
|
impl fmt::Display for Kind {
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
match self {
|
|
Kind::Any => write!(f, "any"),
|
|
Kind::Array => write!(f, "array"),
|
|
Kind::Bool => write!(f, "bool"),
|
|
Kind::Datetime => write!(f, "datetime"),
|
|
Kind::Number => write!(f, "number"),
|
|
Kind::Object => write!(f, "object"),
|
|
Kind::String => write!(f, "string"),
|
|
Kind::Geometry(v) => write!(f, "geometry({})", v),
|
|
Kind::Record(v) => write!(
|
|
f,
|
|
"record({})",
|
|
v.iter().map(|ref v| format!("{}", v)).collect::<Vec<_>>().join(", ")
|
|
),
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn kind(i: &str) -> IResult<&str, Kind> {
|
|
alt((
|
|
map(tag("any"), |_| Kind::Any),
|
|
map(tag("array"), |_| Kind::Array),
|
|
map(tag("bool"), |_| Kind::Bool),
|
|
map(tag("datetime"), |_| Kind::Datetime),
|
|
map(tag("number"), |_| Kind::Number),
|
|
map(tag("object"), |_| Kind::Object),
|
|
map(tag("string"), |_| Kind::String),
|
|
map(geometry, |v| Kind::Geometry(v)),
|
|
map(record, |v| Kind::Record(v)),
|
|
))(i)
|
|
}
|
|
|
|
fn geometry(i: &str) -> IResult<&str, String> {
|
|
let (i, _) = tag("geometry")(i)?;
|
|
let (i, _) = tag("(")(i)?;
|
|
let (i, v) = alt((
|
|
tag("feature"),
|
|
tag("point"),
|
|
tag("line"),
|
|
tag("polygon"),
|
|
tag("multipoint"),
|
|
tag("multiline"),
|
|
tag("multipolygon"),
|
|
tag("collection"),
|
|
))(i)?;
|
|
let (i, _) = tag(")")(i)?;
|
|
Ok((i, String::from(v)))
|
|
}
|
|
|
|
fn record(i: &str) -> IResult<&str, Vec<Table>> {
|
|
let (i, _) = tag("record")(i)?;
|
|
let (i, _) = tag("(")(i)?;
|
|
let (i, v) = many1(table)(i)?;
|
|
let (i, _) = tag(")")(i)?;
|
|
Ok((i, v))
|
|
}
|