feat(schema): support array of primitives

RisingStack · Jul 26, 2015 · 9e6a439 · 9e6a439
1 parent edca7e0
commit 9e6a439
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 6 deletions.
diff --git a/example/app.js b/example/app.js
@@ -16,6 +16,8 @@ function *run() {
       age
       createdAt
       nums
+      bools
+      strings
       removed
       friends {
         name
@@ -29,7 +31,6 @@ function *run() {
   //     name
   //     age
   //     createdAt
-  //     nums
   //     removed
   //     friends {
   //       name

diff --git a/example/user.js b/example/user.js
@@ -11,7 +11,20 @@ var UserSchema = new mongoose.Schema({
     ref: 'User'
   }],
   nums: [Number],
-  removed: Boolean
+  bools: [Boolean],
+  strings: [String],
+  removed: Boolean,
+
+  // TODO: support objects
+  body: {
+    eye: String,
+    foot: Number
+  },
+
+  // TODO: support sub-documents
+  pets: [{
+    name: String
+  }]
 });
 
 var User = mongoose.model('User', UserSchema);

diff --git a/src/schema.js b/src/schema.js
@@ -98,10 +98,12 @@ function getSchema (models) {
 
     // Array
     else if (path.instance === 'Array') {
-      var type = path.caster.options.ref;
+      var type;
 
       // Array of refs
       if (path.caster.instance === 'ObjectID') {
+        type = path.caster.options.ref;
+
         return {
           type: new GraphQLList(types[type]),
           resolve: (modelInstance, params, source, fieldASTs) => {
@@ -118,9 +120,19 @@ function getSchema (models) {
       // Array of primitives
       // FIXME: cannot handle array
       } else {
-        return {
-          // type: new GraphQLList(getField(path.caster))
-        };
+        if (path.caster.instance === 'Number') {
+          return {
+            type: new GraphQLList(GraphQLInt)
+          };
+        } else if (path.caster.instance === 'String') {
+          return {
+            type: new GraphQLList(GraphQLString)
+          };
+        } else if (path.caster.instance === 'Boolean') {
+          return {
+            type: new GraphQLList(GraphQLBoolean)
+          };
+        }
       }
     }