63 lines
1.9 KiB
Rust
63 lines
1.9 KiB
Rust
|
use std::borrow::Cow;
|
||
|
|
||
|
use crate::parser::types::Field;
|
||
|
use crate::resolver_utils::resolve_list;
|
||
|
use crate::{
|
||
|
registry, ContextSelectionSet, InputType, InputValueError, InputValueResult, OutputType,
|
||
|
Positioned, ServerResult, Type, Value,
|
||
|
};
|
||
|
|
||
|
impl<T: Type, const N: usize> Type for [T; N] {
|
||
|
fn type_name() -> Cow<'static, str> {
|
||
|
Cow::Owned(format!("[{}]", T::qualified_type_name()))
|
||
|
}
|
||
|
|
||
|
fn qualified_type_name() -> String {
|
||
|
format!("[{}]!", T::qualified_type_name())
|
||
|
}
|
||
|
|
||
|
fn create_type_info(registry: &mut registry::Registry) -> String {
|
||
|
T::create_type_info(registry);
|
||
|
Self::qualified_type_name()
|
||
|
}
|
||
|
}
|
||
|
|
||
|
impl<T: InputType, const N: usize> InputType for [T; N] {
|
||
|
fn parse(value: Option<Value>) -> InputValueResult<Self> {
|
||
|
if let Some(Value::List(values)) = value {
|
||
|
let items: Vec<T> = values
|
||
|
.into_iter()
|
||
|
.map(|value| InputType::parse(Some(value)))
|
||
|
.collect::<Result<_, _>>()
|
||
|
.map_err(InputValueError::propagate)?;
|
||
|
let len = items.len();
|
||
|
items.try_into().map_err(|_| {
|
||
|
InputValueError::custom(format!(
|
||
|
"Expected input type \"[{}; {}]\", found [{}; {}].",
|
||
|
T::type_name(),
|
||
|
N,
|
||
|
T::type_name(),
|
||
|
len
|
||
|
))
|
||
|
})
|
||
|
} else {
|
||
|
Err(InputValueError::expected_type(value.unwrap_or_default()))
|
||
|
}
|
||
|
}
|
||
|
|
||
|
fn to_value(&self) -> Value {
|
||
|
Value::List(self.iter().map(InputType::to_value).collect())
|
||
|
}
|
||
|
}
|
||
|
|
||
|
#[async_trait::async_trait]
|
||
|
impl<T: OutputType, const N: usize> OutputType for [T; N] {
|
||
|
async fn resolve(
|
||
|
&self,
|
||
|
ctx: &ContextSelectionSet<'_>,
|
||
|
field: &Positioned<Field>,
|
||
|
) -> ServerResult<Value> {
|
||
|
resolve_list(ctx, field, self.iter(), Some(self.len())).await
|
||
|
}
|
||
|
}
|