feature: support ts schema

packages/graphqlgen/package.json

@@ -66,6 +66,7 @@
     "@types/yargs": "12.0.8",
     "benchmark": "2.1.4",
     "flow-bin": "0.86.0",
+    "graphql-tag": "^2.10.1",
     "jest": "23.6.0",
     "ts-jest": "23.10.5",
     "ts-node": "8.0.2",

packages/graphqlgen/src/parse.ts

@@ -2,6 +2,7 @@ import * as Ajv from 'ajv'
 import * as chalk from 'chalk'
 import * as fs from 'fs'
 import * as yaml from 'js-yaml'
+import { print } from 'graphql'
 import { importSchema } from 'graphql-import'
 
 import {
@@ -83,16 +84,28 @@ export function parseContext(
 }
 
 export function parseSchema(schemaPath: string): GraphQLTypes {
-  if (!fs.existsSync(schemaPath)) {
+  const [filePath, constName] = schemaPath.split(':')
+
+  if (!fs.existsSync(filePath)) {
     console.error(
-      chalk.default.red(`The schema file ${schemaPath} does not exist`),
+      chalk.default.red(`The schema file ${filePath} does not exist`),
     )
     process.exit(1)
   }
 
   let schema: string | undefined
   try {
-    schema = importSchema(schemaPath)
+    if (filePath.endsWith('.ts')) {
+      const loadedSchema = require(filePath)[constName || 'default']
+
+      if (typeof loadedSchema === 'string') {
+        schema = loadedSchema
+      } else {
+        schema = print(loadedSchema)
+      }
+    } else {
+      schema = importSchema(filePath)
+    }
   } catch (e) {
     console.error(
       chalk.default.red(`Error occurred while reading schema: ${e}`),

packages/graphqlgen/tests/validation/mocks/gqlSchema/model.ts

@@ -0,0 +1,3 @@
+interface PostModel {
+  id: string
+}

packages/graphqlgen/tests/validation/mocks/gqlSchema/schema.ts

@@ -0,0 +1,12 @@
+import gql from 'graphql-tag'
+
+export const typeDefs = gql`
+  type User {
+    id: String!
+    posts: [Post!]!
+  }
+
+  type Post {
+    id: String!
+  }
+`

packages/graphqlgen/tests/validation/mocks/gqlSchema/types.ts

@@ -0,0 +1,3 @@
+interface User {
+  id: string
+}

packages/graphqlgen/tests/validation/mocks/tsSchemaConst/model.ts

@@ -0,0 +1,3 @@
+interface PostModel {
+  id: string
+}

packages/graphqlgen/tests/validation/mocks/tsSchemaConst/schema.ts

@@ -0,0 +1,10 @@
+export const typeDefs = `
+  type User {
+    id: String!
+    posts: [Post!]!
+  }
+
+  type Post {
+    id: String!
+  }
+`

packages/graphqlgen/tests/validation/mocks/tsSchemaConst/types.ts

@@ -0,0 +1,3 @@
+interface User {
+  id: string
+}

packages/graphqlgen/tests/validation/mocks/tsSchemaDefault/model.ts

@@ -0,0 +1,3 @@
+interface PostModel {
+  id: string
+}

packages/graphqlgen/tests/validation/mocks/tsSchemaDefault/schema.ts

@@ -0,0 +1,10 @@
+export default `
+  type User {
+    id: String!
+    posts: [Post!]!
+  }
+
+  type Post {
+    id: String!
+  }
+`

packages/graphqlgen/tests/validation/mocks/tsSchemaDefault/types.ts

@@ -0,0 +1,3 @@
+interface User {
+  id: string
+}

packages/graphqlgen/tests/validation/validateModels.test.ts

@@ -74,4 +74,49 @@ describe('test validateModels()', () => {
       false,
     )
   })
+
+  test('import schema ts default', () => {
+    testValidateModels(
+      {
+        schema: relative('./mocks/tsSchemaDefault/schema.ts'),
+        models: {
+          files: [relative('./mocks/tsSchemaDefault/types.ts')],
+          override: {
+            Post: relative('./mocks/tsSchemaDefault/model.ts:PostModel'),
+          },
+        },
+      },
+      true,
+    )
+  })
+
+  test('import schema ts const', () => {
+    testValidateModels(
+      {
+        schema: relative('./mocks/tsSchemaConst/schema.ts:typeDefs'),
+        models: {
+          files: [relative('./mocks/tsSchemaConst/types.ts')],
+          override: {
+            Post: relative('./mocks/tsSchemaConst/model.ts:PostModel'),
+          },
+        },
+      },
+      true,
+    )
+  })
+
+  test('import schema gql default', () => {
+    testValidateModels(
+      {
+        schema: relative('./mocks/gqlSchema/schema.ts:typeDefs'),
+        models: {
+          files: [relative('./mocks/gqlSchema/types.ts')],
+          override: {
+            Post: relative('./mocks/gqlSchema/model.ts:PostModel'),
+          },
+        },
+      },
+      true,
+    )
+  })
 })