feat: add ts support

History.md

@@ -1,3 +1,8 @@
+3.1.0-beta.1 / 2019-04-17 (by Carrotzpc)
+==================
+
+  * add ts support
+
 3.0.0 / 2019-04-14 (by Carrotzpc)
 ==================
 

README.md

@@ -147,11 +147,11 @@ exports.middleware = [ 'graphql' ];
 
 ## 参考文章
 
-- [graphql官网](http://facebook.github.io/graphql)
+- [graphql 官网](http://facebook.github.io/graphql)
 
-- [如何在egg中使用graphql](https://zhuanlan.zhihu.com/p/30604868)
+- [如何在 egg 中使用 graphql](https://zhuanlan.zhihu.com/p/30604868)
 
-- [项目例子：结合sequelize](https://github.com/freebyron/egg-graphql-boilerplate)
+- [项目例子：结合 sequelize](https://github.com/freebyron/egg-graphql-boilerplate)
 
 ## 协议
 

agent.js

@@ -1,7 +1,5 @@
 'use strict';
 
 module.exports = agent => {
-  require('./lib/load_schema')(agent);
-  require('./lib/load_connector')(agent);
+  require('./lib/loader')(agent);
 };
-

app.js

@@ -1,7 +1,5 @@
 'use strict';
 
 module.exports = app => {
-  require('./lib/load_schema')(app);
-  require('./lib/load_connector')(app);
+  require('./lib/loader')(app);
 };
-

app/extend/context.js

@@ -8,7 +8,6 @@ module.exports = {
    * connector instance
    * @member Context#connector
    */
-
   get connector() {
     /* istanbul ignore else */
     if (!this[SYMBOL_CONNECTOR]) {
@@ -25,7 +24,6 @@ module.exports = {
    * graphql instance access
    * @member Context#graphql
    */
-
   get graphql() {
     return this.service.graphql;
   },

index.d.ts

@@ -0,0 +1,15 @@
+
+declare module 'egg' {
+  export interface IConnector extends PlainObject {}
+
+  // extend context
+  interface Context {
+    connector: IConnector;
+  }
+
+  // extend your config
+  interface EggAppConfig {
+    graphql: any;
+  }
+
+}

lib/loader.js

@@ -0,0 +1,116 @@
+'use strict';
+
+const { join, dirname } = require('path');
+const { merge, isFunction } = require('lodash');
+const is = require('is-type-of');
+const {
+  makeExecutableSchema,
+  SchemaDirectiveVisitor,
+} = require('graphql-tools');
+
+const SYMBOL_SCHEMA = Symbol('Application#schema');
+const SYMBOL_CONNECTOR_CLASS = Symbol('Application#connectorClass');
+
+module.exports = app => {
+  const directiveResolvers = {};
+  const schemaDirectives = {};
+  const resolvers = {};
+  const typeDefs = [];
+
+  class GraphqlLoader {
+    constructor(app) {
+      this.app = app;
+    }
+
+    load() {
+      const connectorClasses = new Map();
+      this.loadGraphql(connectorClasses);
+      this.loadTypeDefs();
+      /**
+       * create a GraphQL.js GraphQLSchema instance
+       */
+      Object.defineProperties(this.app, {
+        schema: {
+          get() {
+            if (!this[SYMBOL_SCHEMA]) {
+              this[SYMBOL_SCHEMA] = makeExecutableSchema({
+                typeDefs,
+                resolvers,
+                directiveResolvers,
+                schemaDirectives,
+              });
+            }
+            return this[SYMBOL_SCHEMA];
+          },
+        },
+        connectorClass: {
+          get() {
+            if (!this[SYMBOL_CONNECTOR_CLASS]) {
+              this[SYMBOL_CONNECTOR_CLASS] = connectorClasses;
+            }
+            return this[SYMBOL_CONNECTOR_CLASS];
+          },
+        },
+      });
+    }
+    // 加载 graphql
+    loadGraphql(connectorClasses) {
+      const loader = this.app.loader;
+      loader.timing.start('Loader Graphql');
+      const opt = {
+        caseStyle: 'lower',
+        directory: join(this.app.baseDir, 'app/graphql'),
+        target: {},
+        initializer: (obj, opt) => {
+          const pathName = opt.pathName.split('.').pop();
+          // 加载 resolver
+          if (pathName === 'resolver') {
+            if (isFunction(obj)) {
+              obj = obj(this.app);
+            }
+            merge(resolvers, obj);
+          }
+          // load schemaDirective
+          if (is.class(obj)) {
+            const proto = Object.getPrototypeOf(obj);
+            if (proto === SchemaDirectiveVisitor) {
+              const name = opt.pathName.split('.').pop();
+              schemaDirectives[name] = obj;
+            }
+          }
+          if (pathName === 'schemaDirective') {
+            merge(schemaDirectives, obj);
+          }
+          // load directiveResolver
+          if (pathName === 'directive') {
+            merge(directiveResolvers, obj);
+          }
+          // load connector
+          if (pathName === 'connector') {
+            // 获取文件目录名
+            const type = dirname(opt.path)
+              .split(/\/|\\/)
+              .pop();
+            connectorClasses.set(type, obj);
+          }
+        },
+      };
+      new this.app.loader.FileLoader(opt).load();
+      loader.timing.end('Loader Graphql');
+    }
+    // 加载 typeDefs
+    loadTypeDefs() {
+      const opt = {
+        directory: join(this.app.baseDir, 'app/graphql'),
+        match: '**/*.graphql',
+        target: {},
+        initializer: obj => {
+          typeDefs.push(obj.toString('utf8'));
+        },
+      };
+      new this.app.loader.FileLoader(opt).load();
+    }
+  }
+
+  new GraphqlLoader(app).load();
+};

package.json

@@ -1,6 +1,6 @@
 {
   "name": "@switchdog/egg-graphql",
-  "version": "3.0.0",
+  "version": "3.1.0-beta.1",
   "description": "egg graphql plugin",
   "eggPlugin": {
     "name": "graphql"
@@ -49,7 +49,8 @@
     "agent.js",
     "config",
     "app",
-    "lib"
+    "lib",
+    "index.d.ts"
   ],
   "ci": {
     "version": "8, 9"