108 lines
3.2 KiB
Rust
108 lines
3.2 KiB
Rust
use std::collections::HashSet;
|
|
|
|
use crate::{
|
|
model::{__Directive, __Type},
|
|
registry, Object,
|
|
};
|
|
|
|
pub struct __Schema<'a> {
|
|
registry: &'a registry::Registry,
|
|
visible_types: &'a HashSet<&'a str>,
|
|
}
|
|
|
|
impl<'a> __Schema<'a> {
|
|
pub fn new(registry: &'a registry::Registry, visible_types: &'a HashSet<&'a str>) -> Self {
|
|
Self {
|
|
registry,
|
|
visible_types,
|
|
}
|
|
}
|
|
}
|
|
|
|
/// A GraphQL Schema defines the capabilities of a GraphQL server. It exposes
|
|
/// all available types and directives on the server, as well as the entry
|
|
/// points for query, mutation, and subscription operations.
|
|
#[Object(internal, name = "__Schema")]
|
|
impl<'a> __Schema<'a> {
|
|
/// A list of all types supported by this server.
|
|
async fn types(&self) -> Vec<__Type<'a>> {
|
|
let mut types: Vec<_> = self
|
|
.registry
|
|
.types
|
|
.values()
|
|
.filter_map(|ty| {
|
|
if self.visible_types.contains(ty.name()) {
|
|
Some((
|
|
ty.name(),
|
|
__Type::new_simple(self.registry, self.visible_types, ty),
|
|
))
|
|
} else {
|
|
None
|
|
}
|
|
})
|
|
.collect();
|
|
types.sort_by(|a, b| a.0.cmp(b.0));
|
|
types.into_iter().map(|(_, ty)| ty).collect()
|
|
}
|
|
|
|
/// The type that query operations will be rooted at.
|
|
#[inline]
|
|
async fn query_type(&self) -> __Type<'a> {
|
|
__Type::new_simple(
|
|
self.registry,
|
|
self.visible_types,
|
|
&self.registry.types[&self.registry.query_type],
|
|
)
|
|
}
|
|
|
|
/// If this server supports mutation, the type that mutation operations will
|
|
/// be rooted at.
|
|
#[inline]
|
|
async fn mutation_type(&self) -> Option<__Type<'a>> {
|
|
self.registry.mutation_type.as_ref().and_then(|ty| {
|
|
if self.visible_types.contains(ty.as_str()) {
|
|
Some(__Type::new_simple(
|
|
self.registry,
|
|
self.visible_types,
|
|
&self.registry.types[ty],
|
|
))
|
|
} else {
|
|
None
|
|
}
|
|
})
|
|
}
|
|
|
|
/// If this server support subscription, the type that subscription
|
|
/// operations will be rooted at.
|
|
#[inline]
|
|
async fn subscription_type(&self) -> Option<__Type<'a>> {
|
|
self.registry.subscription_type.as_ref().and_then(|ty| {
|
|
if self.visible_types.contains(ty.as_str()) {
|
|
Some(__Type::new_simple(
|
|
self.registry,
|
|
self.visible_types,
|
|
&self.registry.types[ty],
|
|
))
|
|
} else {
|
|
None
|
|
}
|
|
})
|
|
}
|
|
|
|
/// A list of all directives supported by this server.
|
|
async fn directives(&self) -> Vec<__Directive<'a>> {
|
|
let mut directives: Vec<_> = self
|
|
.registry
|
|
.directives
|
|
.values()
|
|
.map(|directive| __Directive {
|
|
registry: self.registry,
|
|
visible_types: self.visible_types,
|
|
directive,
|
|
})
|
|
.collect();
|
|
directives.sort_by(|a, b| a.directive.name.cmp(b.directive.name));
|
|
directives
|
|
}
|
|
}
|