use crate::{
    ContextSelectionSet, GQLInputValue, GQLOutputValue, GQLType, QueryError, Result, Value,
};
use std::borrow::Cow;

impl<T: GQLType> GQLType for Vec<T> {
    fn type_name() -> Cow<'static, str> {
        Cow::Owned(format!("[{}]!", T::type_name()))
    }
}

impl<T: GQLInputValue> GQLInputValue for Vec<T> {
    fn parse(value: Value) -> Result<Self> {
        match value {
            Value::List(values) => {
                let mut result = Vec::new();
                for value in values {
                    result.push(GQLInputValue::parse(value)?);
                }
                Ok(result)
            }
            _ => {
                return Err(QueryError::ExpectedType {
                    expect: Self::type_name(),
                    actual: value,
                }
                .into())
            }
        }
    }

    fn parse_from_json(value: serde_json::Value) -> Result<Self> {
        match value {
            serde_json::Value::Array(values) => {
                let mut result = Vec::new();
                for value in values {
                    result.push(GQLInputValue::parse_from_json(value)?);
                }
                Ok(result)
            }
            _ => {
                return Err(QueryError::ExpectedJsonType {
                    expect: Self::type_name(),
                    actual: value,
                }
                .into())
            }
        }
    }
}

#[async_trait::async_trait]
impl<T: GQLOutputValue + Send + Sync> GQLOutputValue for Vec<T> {
    async fn resolve(&self, ctx: &ContextSelectionSet<'_>) -> Result<serde_json::Value> {
        let mut res = Vec::new();
        for item in self {
            res.push(item.resolve(ctx).await?);
        }
        Ok(res.into())
    }
}

impl<T: GQLType> GQLType for &[T] {
    fn type_name() -> Cow<'static, str> {
        Cow::Owned(format!("[{}]!", T::type_name()))
    }
}

#[async_trait::async_trait]
impl<T: GQLOutputValue + Send + Sync> GQLOutputValue for &[T] {
    async fn resolve(&self, ctx: &ContextSelectionSet<'_>) -> Result<serde_json::Value> {
        let mut res = Vec::new();
        for item in self.iter() {
            res.push(item.resolve(ctx).await?);
        }
        Ok(res.into())
    }
}