yume/juniper
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add support for skipping fields in GraphQL objects (#224)

Browse source 
Fields can now be skipped with the `#[graphql(skip)]` annotation. Note this
doesn't really make sense for GraphQLInputObjects so this isn't supported there.

Fixes https://github.com/graphql-rust/juniper/issues/220.
This commit is contained in:
Christian Legnitto 2018-08-27 15:09:42 -07:00 committed by GitHub
parent 62d015cf86
commit 22c955599a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 112 additions and 33 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
changelog/master.md
View file

@ -45,6 +45,10 @@
[#219](https://github.com/graphql-rust/juniper/pull/219) [#219](https://github.com/graphql-rust/juniper/pull/219)
* When using the `GraphQLObject` custom derive, fields now be omitted by annotating the field with `#[graphql(skip)]`.
[#220](https://github.com/graphql-rust/juniper/issues/220)
* Due to newer dependencies, the oldest Rust version supported is now 1.22.0 * Due to newer dependencies, the oldest Rust version supported is now 1.22.0
[#231](https://github.com/graphql-rust/juniper/pull/231) [#231](https://github.com/graphql-rust/juniper/pull/231)

14
juniper_codegen/src/derive_enum.rs
View file

@ -25,9 +25,9 @@ impl EnumAttrs {
res.description = get_doc_comment(&input.attrs); res.description = get_doc_comment(&input.attrs);
// Check attributes for name and description. // Check attributes for name and description.
if let Some(items) = get_graphl_attr(&input.attrs) { if let Some(items) = get_graphql_attr(&input.attrs) {
for item in items { for item in items {
if let Some(val) = keyed_item_value(&item, "name", true) { if let Some(AttributeValue::String(val)) = keyed_item_value(&item, "name", AttributeValidation::String) {
if is_valid_name(&*val) { if is_valid_name(&*val) {
res.name = Some(val); res.name = Some(val);
continue; continue;
@ -38,7 +38,7 @@ impl EnumAttrs {
); );
} }
} }
if let Some(val) = keyed_item_value(&item, "description", true) { if let Some(AttributeValue::String(val)) = keyed_item_value(&item, "description", AttributeValidation::String) {
res.description = Some(val); res.description = Some(val);
continue; continue;
} }
@ -76,9 +76,9 @@ impl EnumVariantAttrs {
res.description = get_doc_comment(&variant.attrs); res.description = get_doc_comment(&variant.attrs);
// Check attributes for name and description. // Check attributes for name and description.
if let Some(items) = get_graphl_attr(&variant.attrs) { if let Some(items) = get_graphql_attr(&variant.attrs) {
for item in items { for item in items {
if let Some(val) = keyed_item_value(&item, "name", true) { if let Some(AttributeValue::String(val)) = keyed_item_value(&item, "name", AttributeValidation::String) {
if is_valid_name(&*val) { if is_valid_name(&*val) {
res.name = Some(val); res.name = Some(val);
continue; continue;
@ -89,11 +89,11 @@ impl EnumVariantAttrs {
); );
} }
} }
if let Some(val) = keyed_item_value(&item, "description", true) { if let Some(AttributeValue::String(val)) = keyed_item_value(&item, "description", AttributeValidation::String) {
res.description = Some(val); res.description = Some(val);
continue; continue;
} }
if let Some(val) = keyed_item_value(&item, "deprecated", true) { if let Some(AttributeValue::String(val)) = keyed_item_value(&item, "deprecated", AttributeValidation::String) {
res.deprecation = Some(val); res.deprecation = Some(val);
continue; continue;
} }

14
juniper_codegen/src/derive_input_object.rs
View file

@ -21,9 +21,9 @@ impl ObjAttrs {
res.description = get_doc_comment(&input.attrs); res.description = get_doc_comment(&input.attrs);
// Check attributes for name and description. // Check attributes for name and description.
if let Some(items) = get_graphl_attr(&input.attrs) { if let Some(items) = get_graphql_attr(&input.attrs) {
for item in items { for item in items {
if let Some(val) = keyed_item_value(&item, "name", true) { if let Some(AttributeValue::String(val)) = keyed_item_value(&item, "name", AttributeValidation::String) {
if is_valid_name(&*val) { if is_valid_name(&*val) {
res.name = Some(val); res.name = Some(val);
continue; continue;
@ -34,7 +34,7 @@ impl ObjAttrs {
); );
} }
} }
if let Some(val) = keyed_item_value(&item, "description", true) { if let Some(AttributeValue::String(val)) = keyed_item_value(&item, "description", AttributeValidation::String) {
res.description = Some(val); res.description = Some(val);
continue; continue;
} }
@ -73,9 +73,9 @@ impl ObjFieldAttrs {
res.description = get_doc_comment(&variant.attrs); res.description = get_doc_comment(&variant.attrs);
// Check attributes for name and description. // Check attributes for name and description.
if let Some(items) = get_graphl_attr(&variant.attrs) { if let Some(items) = get_graphql_attr(&variant.attrs) {
for item in items { for item in items {
if let Some(val) = keyed_item_value(&item, "name", true) { if let Some(AttributeValue::String(val)) = keyed_item_value(&item, "name", AttributeValidation::String) {
if is_valid_name(&*val) { if is_valid_name(&*val) {
res.name = Some(val); res.name = Some(val);
continue; continue;
@ -86,11 +86,11 @@ impl ObjFieldAttrs {
); );
} }
} }
if let Some(val) = keyed_item_value(&item, "description", true) { if let Some(AttributeValue::String(val)) = keyed_item_value(&item, "description", AttributeValidation::String) {
res.description = Some(val); res.description = Some(val);
continue; continue;
} }
if let Some(val) = keyed_item_value(&item, "default", true) { if let Some(AttributeValue::String(val)) = keyed_item_value(&item, "default", AttributeValidation::Any) {
res.default_expr = Some(val); res.default_expr = Some(val);
continue; continue;
} }

30
juniper_codegen/src/derive_object.rs
View file

@ -18,9 +18,9 @@ impl ObjAttrs {
res.description = get_doc_comment(&input.attrs); res.description = get_doc_comment(&input.attrs);
// Check attributes for name and description. // Check attributes for name and description.
if let Some(items) = get_graphl_attr(&input.attrs) { if let Some(items) = get_graphql_attr(&input.attrs) {
for item in items { for item in items {
if let Some(val) = keyed_item_value(&item, "name", true) { if let Some(AttributeValue::String(val)) = keyed_item_value(&item, "name", AttributeValidation::String) {
if is_valid_name(&*val) { if is_valid_name(&*val) {
res.name = Some(val); res.name = Some(val);
continue; continue;
@ -31,12 +31,12 @@ impl ObjAttrs {
); );
} }
} }
if let Some(val) = keyed_item_value(&item, "description", true) { if let Some(AttributeValue::String(val)) = keyed_item_value(&item, "description", AttributeValidation::String) {
res.description = Some(val); res.description = Some(val);
continue; continue;
} }
panic!(format!( panic!(format!(
"Unknown attribute for #[derive(GraphQLObject)]: {:?}", "Unknown object attribute for #[derive(GraphQLObject)]: {:?}",
item item
)); ));
} }
@ -50,6 +50,7 @@ struct ObjFieldAttrs {
name: Option<String>, name: Option<String>,
description: Option<String>, description: Option<String>,
deprecation: Option<String>, deprecation: Option<String>,
skip: bool,
} }
impl ObjFieldAttrs { impl ObjFieldAttrs {
@ -59,10 +60,10 @@ impl ObjFieldAttrs {
// Check doc comments for description. // Check doc comments for description.
res.description = get_doc_comment(&variant.attrs); res.description = get_doc_comment(&variant.attrs);
// Check attributes for name and description. // Check attributes.
if let Some(items) = get_graphl_attr(&variant.attrs) { if let Some(items) = get_graphql_attr(&variant.attrs) {
for item in items { for item in items {
if let Some(val) = keyed_item_value(&item, "name", true) { if let Some(AttributeValue::String(val)) = keyed_item_value(&item, "name", AttributeValidation::String) {
if is_valid_name(&*val) { if is_valid_name(&*val) {
res.name = Some(val); res.name = Some(val);
continue; continue;
@ -73,16 +74,20 @@ impl ObjFieldAttrs {
); );
} }
} }
if let Some(val) = keyed_item_value(&item, "description", true) { if let Some(AttributeValue::String(val)) = keyed_item_value(&item, "description", AttributeValidation::String) {
res.description = Some(val); res.description = Some(val);
continue; continue;
} }
if let Some(val) = keyed_item_value(&item, "deprecation", true) { if let Some(AttributeValue::String(val)) = keyed_item_value(&item, "deprecation", AttributeValidation::String) {
res.deprecation = Some(val); res.deprecation = Some(val);
continue; continue;
} }
if let Some(_) = keyed_item_value(&item, "skip", AttributeValidation::Bare) {
res.skip = true;
continue;
}
panic!(format!( panic!(format!(
"Unknown attribute for #[derive(GraphQLObject)]: {:?}", "Unknown field attribute for #[derive(GraphQLObject)]: {:?}",
item item
)); ));
} }
@ -122,6 +127,11 @@ pub fn impl_object(ast: &syn::DeriveInput) -> TokenStream {
let field_attrs = ObjFieldAttrs::from_input(field); let field_attrs = ObjFieldAttrs::from_input(field);
let field_ident = field.ident.as_ref().unwrap(); let field_ident = field.ident.as_ref().unwrap();
// Check if we should skip this field.
if field_attrs.skip {
continue;
}
// Build value. // Build value.
let name = match field_attrs.name { let name = match field_attrs.name {
Some(ref name) => { Some(ref name) => {

42
juniper_codegen/src/util.rs
View file

@ -1,6 +1,17 @@
use regex::Regex; use regex::Regex;
use syn::{Attribute, Lit, Meta, MetaNameValue, NestedMeta}; use syn::{Attribute, Lit, Meta, MetaNameValue, NestedMeta};
pub enum AttributeValidation {
Any,
Bare,
String,
}
pub enum AttributeValue {
Bare,
String(String),
}
// Gets doc comment. // Gets doc comment.
pub fn get_doc_comment(attrs: &Vec<Attribute>) -> Option<String> { pub fn get_doc_comment(attrs: &Vec<Attribute>) -> Option<String> {
if let Some(items) = get_doc_attr(attrs) { if let Some(items) = get_doc_attr(attrs) {
@ -59,7 +70,7 @@ fn get_doc_attr(attrs: &Vec<Attribute>) -> Option<Vec<MetaNameValue>> {
} }
// Get the nested items of a a #[graphql(...)] attribute. // Get the nested items of a a #[graphql(...)] attribute.
pub fn get_graphl_attr(attrs: &Vec<Attribute>) -> Option<Vec<NestedMeta>> { pub fn get_graphql_attr(attrs: &Vec<Attribute>) -> Option<Vec<NestedMeta>> {
for attr in attrs { for attr in attrs {
match attr.interpret_meta() { match attr.interpret_meta() {
Some(Meta::List(ref list)) if list.ident == "graphql" => { Some(Meta::List(ref list)) if list.ident == "graphql" => {
@ -71,21 +82,36 @@ pub fn get_graphl_attr(attrs: &Vec<Attribute>) -> Option<Vec<NestedMeta>> {
None None
} }
pub fn keyed_item_value(item: &NestedMeta, name: &str, must_be_string: bool) -> Option<String> { pub fn keyed_item_value(item: &NestedMeta, name: &str, validation: AttributeValidation) -> Option<AttributeValue> {
match item { match item {
// Attributes in the form of `#[graphql(name = "value")]`.
&NestedMeta::Meta(Meta::NameValue(ref nameval)) if nameval.ident == name => { &NestedMeta::Meta(Meta::NameValue(ref nameval)) if nameval.ident == name => {
match &nameval.lit { match &nameval.lit {
&Lit::Str(ref strlit) => Some(strlit.value()), // We have a string attribute value.
_ => if must_be_string { &Lit::Str(ref strlit) => match validation {
AttributeValidation::Bare => {
panic!(format!(
"Invalid format for attribute \"{:?}\": expected a bare attribute without a value",
item
));
},
_ => Some(AttributeValue::String(strlit.value())),
},
_ => None,
}
},
// Attributes in the form of `#[graphql(name)]`.
&NestedMeta::Meta(Meta::Word(ref ident)) if ident.to_string() == name => {
match validation {
AttributeValidation::String => {
panic!(format!( panic!(format!(
"Invalid format for attribute \"{:?}\": expected a string", "Invalid format for attribute \"{:?}\": expected a string value",
item item
)); ));
} else {
None
}, },
_ => Some(AttributeValue::Bare),
} }
} },
_ => None, _ => None,
} }
} }

39
juniper_tests/src/codegen/derive_object.rs
View file

@ -48,6 +48,13 @@ struct OverrideDocComment {
regular_field: bool, regular_field: bool,
} }
#[derive(GraphQLObject, Debug, PartialEq)]
struct SkippedFieldObj {
regular_field: bool,
#[graphql(skip)]
skipped: i32,
}
graphql_object!(Query: () |&self| { graphql_object!(Query: () |&self| {
field obj() -> Obj { field obj() -> Obj {
Obj{ Obj{
@ -82,6 +89,13 @@ graphql_object!(Query: () |&self| {
regular_field: true, regular_field: true,
} }
} }
field skipped_field_obj() -> SkippedFieldObj {
SkippedFieldObj{
regular_field: false,
skipped: 42,
}
}
}); });
#[test] #[test]
@ -171,6 +185,31 @@ fn test_derived_object() {
); );
} }
#[test]
#[should_panic]
fn test_cannot_query_skipped_field() {
let doc = r#"
{
skippedFieldObj {
skippedField
}
}"#;
let schema = RootNode::new(Query, EmptyMutation::<()>::new());
execute(doc, None, &schema, &Variables::new(), &()).unwrap();
}
#[test]
fn test_skipped_field_siblings_unaffected() {
let doc = r#"
{
skippedFieldObj {
regularField
}
}"#;
let schema = RootNode::new(Query, EmptyMutation::<()>::new());
execute(doc, None, &schema, &Variables::new(), &()).unwrap();
}
#[test] #[test]
fn test_derived_object_nested() { fn test_derived_object_nested() {
let doc = r#" let doc = r#"